Set args = WScript.Arguments
Dim objFSO, objShell, objCmdExec

Set objFSO = CreateObject("Scripting.FileSystemObject")

If args.Count > 0 Then
    partitionScriptLocation = args(0)
Else
    Dim objFile
	
    scriptDirectory = objFSO.GetSpecialFolder(2)
	scriptFilename  = objFSO.GetTempName
	partitionScriptLocation = objFSO.BuildPath(scriptDirectory, scriptFilename)
	
	Set objFile = objFSO.CreateTextFile(partitionScriptLocation, False)
	objFile.WriteLine("select disk 1")
	objFile.WriteLine("clean")
	objFile.WriteLine("convert dynamic")
	objFile.WriteLine("select disk 0")
	objFile.WriteLine("convert dynamic")
	objFile.WriteLine("select volume 0")
	objFile.WriteLine("add disk 1")
	objFile.Close
End If

Set objShell = CreateObject("WScript.Shell")
Set objCmdExec = objShell.exec("diskpart /s " + partitionScriptLocation)