''    USMT Environment Setup    ''
''   Created by Matt Shadbolt   ''

Option Explicit

'Declare Variables
Dim objFSO, objFolder, strDirectory
Dim objNetwork
Dim profilePath
Dim x
Dim filetxt, getname, path
Dim LocalAppDataPath, RoamingAppDataPath
Dim MyDocuments, MyDesktop, MyPictures

'Set Variables
x = 0
Set objFSO = CreateObject("Scripting.FileSystemObject")
Set objNetwork = CreateObject("WScript.Network")
profilePath = UserInput( "Is this for XP or WIN7?")

' Check if XP or Windows 7
If profilePath = "XP" Then
profilePath = "C:\Documents and Settings\"
LocalAppDataPath = "\Local Settings\Application Data\"
RoamingAppDataPath = "\Application Data\"
MyDocuments = "My Documents"
MyDesktop = "Desktop"
MyPictures = "My Documents\My Pictures"

ElseIf profilePath = "WIN7" Then
profilePath = "C:\Users\"
LocalAppDataPath = "\AppData\Local\"
RoamingAppDataPath = "\AppData\Roaming\"
MyDocuments = "Documents"
MyDesktop = "Desktop"
MyPictures = "Pictures"
Else 
MsgBox "You need to enter either XP or WIN7 (CaSe SeNsItIvE) "
Wscript.Quit
End If

' Desktop Folders
While x < 10
	strDirectory = profilePath & objNetwork.Username & "\" & MyDesktop & "\DesktopFolder" & x
	Set objFolder = objFSO.CreateFolder(strDirectory)
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "TextFile" & x & ".txt", True)
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "WordDoc" & x & ".doc", True)
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "Shortcut" & x & ".lnk", True)
	x = x + 1
Wend 
x = 0

' My Documents
While x < 10
	strDirectory = profilePath & objNetwork.Username & "\" & MyDocuments & "\DocumentsFolder" & x
	Set objFolder = objFSO.CreateFolder(strDirectory)
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "WordDoc" & x & ".doc", True)
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "WordDoc1" & x & ".doc", True)
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "WordDoc2" & x & ".doc", True)
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "WordDoc3" & x & ".doc", True)
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "WordDoc4" & x & ".doc", True)
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "WordDoc5" & x & ".doc", True)
	x = x + 1
Wend 
x = 0

' My Pictures
While x < 10
	strDirectory = profilePath & objNetwork.Username & "\" & MyPictures & "\PicturesFolder" & x
	Set objFolder = objFSO.CreateFolder(strDirectory)
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "Picture" & x & ".jpg", True)
	x = x + 1
Wend 
x = 0

' AppDatat\Local
strDirectory = profilePath & objNetwork.Username & LocalAppDataPath & "SomeAppData"
Set objFolder = objFSO.CreateFolder(strDirectory)
While x < 10
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "AppDataFile" & x & ".appData", True)
	x = x + 1
Wend 
x = 0

' AppData\Roaming
strDirectory = profilePath & objNetwork.Username & RoamingAppDataPath & "SomeAppData"
Set objFolder = objFSO.CreateFolder(strDirectory)
While x < 10
	Set filetxt = objFSO.CreateTextFile(strDirectory &"\" & "AppDataFile" & x & ".appData", True)
	x = x + 1
Wend 
x = 0

MsgBox "Success!"

''' Functions '''

'Input OS type
Function UserInput( myPrompt )
    If UCase( Right( WScript.FullName, 12 ) ) = "\CSCRIPT.EXE" Then
        WScript.StdOut.Write myPrompt & " "
        UserInput = WScript.StdIn.ReadLine
    Else
        UserInput = InputBox( myPrompt )
    End If
End Function