home *** CD-ROM | disk | FTP | other *** search
/ Chip 2001 May / W2KPRK.iso / netmgmt.cab / persist.vbs < prev    next >
Text File  |  1999-11-04  |  10KB  |  409 lines
  1. '----------------------------------------------------------------------
  2. '
  3. ' Copyright (c) Microsoft Corporation 1998-1999
  4. ' All Rights Reserved
  5. '
  6. ' Abstract:
  7. '
  8. ' persist.vbs - script for saving and restoring printer configuration
  9. '
  10. ' Usage:
  11. ' persist [-rs?] [-b printer-name][-f file-name]
  12. ' 
  13. ' Examples:
  14. ' persist.vbs -s -b \\server\printer -f file.txt -all
  15. ' persist.vbs -r -b printer -f file.txt -sec
  16. '
  17. '----------------------------------------------------------------------
  18.  
  19. option explicit
  20.  
  21. '
  22. ' Debugging trace flags, to enable debug output trace message
  23. ' change gDebugFlag to true.
  24. '
  25. const kDebugTrace = 1
  26. const kDebugError = 2
  27. dim   gDebugFlag
  28.  
  29. gDebugFlag = false
  30.  
  31. '
  32. ' Messages to be displayed if the scripting host is not cscript
  33. '                            
  34. const kMessage1 = "Please run this script using CScript."  
  35. const kMessage2 = "This can be achieved by"
  36. const kMessage3 = "1. Using ""CScript script.vbs arguments"" or" 
  37. const kMessage4 = "2. Changing the default Windows Scripting Host to CScript"
  38. const kMessage5 = "   using ""CScript //H:CScript //S"" and running the script "
  39. const kMessage6 = "   ""script.vbs arguments""."
  40.  
  41. '
  42. ' Operation action values.
  43. '
  44. const kActionUnknown    = 0
  45. const kActionSave       = 1
  46. const kActionRestore    = 2
  47.  
  48. const kErrorSuccess     = 0
  49. const KErrorFailure     = 1
  50.  
  51. const kPrinterData      = 1
  52. const kPrinterInfo2     = 2
  53. const kPrinterInfo7     = 4
  54. const kPrinterSec       = 8
  55. const kUserDevmode      = 16
  56. const kPrinterDevmode   = 32
  57. const kColorProf        = 64
  58. const kForceName        = 128
  59. const kResolveName      = 256
  60. const kResolvePort      = 512
  61. const kResolveShare     = 1024
  62. const kDontGenerateShare= 2048
  63.  
  64. '
  65. ' kPrinterData + kPrinterInfo2 + kPrinterDevmode
  66. '
  67. const kMinimumSettings  = 35  
  68.  
  69. '
  70. ' kMinimumSettings + kPrinterInfo7 + kPrinterSec + 
  71. ' kUserDevmode + kColorProf
  72. '
  73. const kAllSettings      = 127 
  74.  
  75. main
  76.  
  77. '
  78. ' Main execution starts here
  79. '
  80. sub main
  81.  
  82.     dim iAction
  83.     dim iRetval
  84.     dim strPrinter
  85.     dim strFile 
  86.     dim iFlags
  87.     
  88.     '
  89.     ' Abort if the host is not cscript
  90.     '
  91.     if not IsHostCscript() then
  92.    
  93.         call wscript.echo(kMessage1 & vbCRLF & kMessage2 & vbCRLF & _
  94.                           kMessage3 & vbCRLF & kMessage4 & vbCRLF & _
  95.                           kMessage5 & vbCRLF & kMessage6 & vbCRLF)
  96.         
  97.         wscript.quit
  98.    
  99.     end if
  100.                
  101.     iRetval = ParseCommandLine(iAction, strPrinter, strFile, iFlags)
  102.  
  103.     if iRetval = kErrorSuccess then
  104.  
  105.         select case iAction
  106.  
  107.             case kActionSave
  108.                  iRetval = SavePrinter(strPrinter, strFile, iFlags)
  109.  
  110.             case kActionRestore
  111.                  iRetval = RestorePrinter(strPrinter, strFile, iFlags)
  112.  
  113.             case else
  114.                  Usage(True)
  115.                  exit sub
  116.  
  117.         end select
  118.  
  119.     end if
  120.  
  121. end sub
  122.  
  123. '
  124. ' Save printer configuration
  125. '
  126. function SavePrinter(strPrinter, strFile, iFlags)
  127.  
  128.     on error resume next
  129.     
  130.     DebugPrint kDebugTrace, "In SavePrinter"
  131.  
  132.     dim oMaster
  133.  
  134.     set oMaster = CreateObject("PrintMaster.PrintMaster.1")
  135.     
  136.     oMaster.PrinterPersistSave strPrinter, strFile, iFlags
  137.  
  138.     if Err.Number = kErrorSuccess then
  139.  
  140.         wscript.echo "Success saving the configuration of the printer """ & strPrinter & """ "
  141.         
  142.         SavePrinter = kErrorSuccess
  143.         
  144.     else
  145.     
  146.         wscript.echo "Unable to save the configuration of the printer """ & strPrinter _
  147.                      & """, error 0x" & Hex(Err.Number) & ". " & Err.Description
  148.     
  149.         SavePrinter = kErrorFailure    
  150.  
  151.     end if
  152.  
  153. end function
  154.  
  155. '
  156. ' Restore printer configuration
  157. '
  158. function RestorePrinter(strPrinter, strFile, iFlags)
  159.  
  160.     on error resume next    
  161.  
  162.     DebugPrint kDebugTrace, "In RestorePrinter"
  163.  
  164.     dim oMaster
  165.  
  166.     Set oMaster = CreateObject("PrintMaster.PrintMaster.1")
  167.     
  168.     oMaster.PrinterPersistRestore strPrinter, strFile, iFlags
  169.  
  170.     if Err.Number = kErrorSuccess then
  171.     
  172.         wscript.echo "Success restoring the configuration of the printer """ & strPrinter & """ "
  173.     
  174.         RestorePrinter = kErrorSuccess
  175.         
  176.     else
  177.     
  178.         wscript.echo "Unable to restore the configuration of the printer """ & strPrinter _
  179.                      & """, error: 0x" & Hex(Err.Number) & ". " & Err.Description
  180.     
  181.         RestorePrinter = kErrorFailure    
  182.  
  183.     end if
  184.  
  185. end function
  186.  
  187. '
  188. ' Debug display helper function
  189. '
  190. sub DebugPrint(uFlags, strString)
  191.  
  192.     if gDebugFlag = true then
  193.  
  194.         if uFlags = kDebugTrace then
  195.  
  196.             wscript.echo "Debug: " & strString
  197.  
  198.         end if
  199.  
  200.         if uFlags = kDebugError then
  201.  
  202.             if Err <> 0 then
  203.  
  204.                 wscript.echo "Debug: " & strString & " Failed with " & Hex(Err)
  205.  
  206.             end if
  207.  
  208.         end if
  209.  
  210.     end if
  211.  
  212. end sub
  213.  
  214. '
  215. ' Parse the command line into it's components
  216. '
  217. function ParseCommandLine(iAction, strPrinter, strFile, iFlags)
  218.  
  219.     DebugPrint kDebugTrace, "In the ParseCommandLine"
  220.  
  221.     dim oArgs
  222.     dim iIndex
  223.     
  224.     iFlags = 0
  225.  
  226.     iAction = kActionUnknown
  227.     iIndex  = 0
  228.  
  229.     set oArgs = wscript.Arguments
  230.  
  231.     while iIndex < oArgs.Count
  232.  
  233.         select case oArgs(iIndex)
  234.  
  235.             case "-r"
  236.                 iAction = kActionRestore
  237.  
  238.             case "-s"
  239.                 iAction = kActionSave
  240.  
  241.             case "-b"
  242.                 iIndex = iIndex + 1
  243.                 strPrinter = oArgs(iIndex)
  244.  
  245.             case "-f"
  246.                 iIndex = iIndex + 1
  247.                 strFile = oArgs(iIndex)
  248.             
  249.             case "-data"
  250.                 iFlags = iFlags + kPrinterData
  251.  
  252.             case "-2"
  253.                 iFlags = iFlags + kPrinterInfo2
  254.                 
  255.             case "-7"
  256.                 iFlags = iFlags + kPrinterInfo7
  257.  
  258.             case "-sec"
  259.                 iFlags = iFlags + kPrinterSec
  260.  
  261.             case "-udev"
  262.                 iFlags = iFlags + kUserDevmode
  263.                 
  264.             case "-pdev"
  265.                 iFlags = iFlags + kPrinterDevmode    
  266.  
  267.             case "-color"
  268.                 iFlags = iFlags + kColorProf
  269.  
  270.             case "-force"
  271.                 iFlags = iFlags + kForceName
  272.  
  273.             case "-resname"
  274.                 iFlags = iFlags + kResolveName
  275.  
  276.             case "-resport"
  277.                 iFlags = iFlags + kResolvePort
  278.  
  279.             case "-resshare"
  280.                 iFlags = iFlags + kResolveShare
  281.  
  282.             case "-noshare"
  283.                 iFlags = iFlags + kDontGenerateShare
  284.  
  285.             case "-min"
  286.                 iFlags = iFlags + kMinimumSettings
  287.  
  288.             case "-all"
  289.                 iFlags = iFlags + kAllSettings
  290.              
  291.             case "-?"
  292.                 Usage(true)
  293.                 exit function
  294.  
  295.             case else
  296.                 Usage(true)
  297.                 exit function
  298.  
  299.         end select
  300.  
  301.         iIndex = iIndex + 1
  302.  
  303.     wend
  304.     
  305.     if Err.Number = kErrorSuccess then
  306.  
  307.         ParseCommandLine = kErrorSuccess
  308.  
  309.     else
  310.  
  311.         wscript.echo "Unable to parse command line, error 0x" & _
  312.                      Hex(Err.Number) & ". " & Err.Description
  313.         
  314.         ParseCommandLine = KErrorFailure
  315.         
  316.     end if
  317.  
  318. end function
  319.  
  320. '
  321. ' Display command usage.
  322. '
  323. sub Usage(bExit)
  324.  
  325.     wscript.echo "Usage: persist [-rs?] [-b printer-name][-f file-name][-data]"
  326.     wscript.echo "               [-2][-7][-sec][-udev][-pdev][-color][-force]"
  327.     wscript.echo "               [-resname][-resport][-resshare][-noshare][-min][-all]" 
  328.     wscript.echo ""
  329.     wscript.echo "Arguments:"
  330.     wscript.echo "-r        - restore printer configuration"
  331.     wscript.echo "-s        - save printer configuration"
  332.     wscript.echo "-?        - display command usage"
  333.     wscript.echo "-b        - printer name"
  334.     wscript.echo "-data     - printer data"
  335.     wscript.echo "-2        - Printer Info 2"
  336.     wscript.echo "-7        - Printer Info 7"
  337.     wscript.echo "-sec      - security"
  338.     wscript.echo "-color    - color profile"
  339.     wscript.echo "-force    - force name"
  340.     wscript.echo "-udev     - user devmode"
  341.     wscript.echo "-pdev     - printer devmode"
  342.     wscript.echo "-resname  - resolve name"
  343.     wscript.echo "-resport  - resolve port"
  344.     wscript.echo "-resshare - resolve share"
  345.     wscript.echo "-noshare  - don't share the printer"
  346.     wscript.echo "-min      - minimum settings"
  347.     wscript.echo "-all      - all settings"
  348.     wscript.echo ""
  349.     wscript.echo "Examples:"
  350.     wscript.echo "persist.vbs -s -b \\server\printer -f file.txt -all"
  351.     wscript.echo "persist.vbs -r -b printer -f file.txt -sec"
  352.     
  353.     if bExit then
  354.  
  355.         wscript.quit(1)
  356.  
  357.     end if
  358.  
  359. end sub
  360.  
  361. '
  362. ' Determines which program is used to run this script. 
  363. ' Returns true if the script host is cscript.exe
  364. '
  365. function IsHostCscript()
  366.  
  367.     on error resume next
  368.     
  369.     dim strFullName 
  370.     dim strCommand 
  371.     dim i, j 
  372.     dim bReturn
  373.     
  374.     bReturn = false
  375.     
  376.     strFullName = WScript.FullName
  377.     
  378.     i = InStr(1, strFullName, ".exe", 1)
  379.     
  380.     if i <> 0 then
  381.         
  382.         j = InStrRev(strFullName, "\", i, 1)
  383.         
  384.         if j <> 0 then
  385.             
  386.             strCommand = Mid(strFullName, j+1, i-j-1)
  387.             
  388.             if LCase(strCommand) = "cscript" then
  389.             
  390.                 bReturn = true  
  391.             
  392.             end if    
  393.                 
  394.         end if
  395.         
  396.     end if
  397.     
  398.     if Err <> 0 then
  399.     
  400.         call wscript.echo("Error 0x" & hex(Err.Number) & " occurred. " & Err.Description _
  401.                           & ". " & vbCRLF & "The scripting host could not be determined.")       
  402.         
  403.     end if
  404.     
  405.     IsHostCscript = bReturn
  406.  
  407. end function
  408.  
  409.