'   Copyright 2017 bin jin
'
'   Licensed under the Apache License, Version 2.0 (the "License");
'   you may not use this file except in compliance with the License.
'   You may obtain a copy of the License at
'
'       http://www.apache.org/licenses/LICENSE-2.0
'
'   Unless required by applicable law or agreed to in writing, software
'   distributed under the License is distributed on an "AS IS" BASIS,
'   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
'   See the License for the specific language governing permissions and
'   limitations under the License.

' Framework
'     If the function names conform to the specifications:
'         External call function.
'         Error handling.
'         Display help information.
'         Print the functions list.
'
'     e.g.
'         ''' [brief_introduction] '[description_1] '[description_2] ...
'         Function [script_name_without_suffix]_[function_name]()
'             ...
'             [function_body]
'             '''
'             setErr "[error_description]" ' exit and display [error_description]
'             ...
'             setErr 1 ' return false status
'         End Function

''' Output version and exit
Function xlib_version()
    printLine "0.21.4.11"
ENd Function

''' Sleep some milliseconds 'Usage: xlib sleep [ms]
Function xlib_sleep(ms)
    ' Test ms
    If Not IsNumeric(ms) Then setErr "Args not a number"
    WScript.Sleep ms
End Function

''' Run some command at background 'Usage: xlib vbhide "[command]"
Function xlib_vbhide(command)
    CreateObject("WScript.Shell").Run command, vbhide
End Function

''' Download and save 'Usage: xlib get [url] [output_path]
Function xlib_get(url, output)
    Set htt = gXmlHttp()
    htt.Open "GET", url, 0
    htt.Send
    Set str = CreateObject("ADODB.Stream")
    str.Type = 1
    str.Open
    str.Write htt.ResponseBody
    str.SaveToFile output, 2
End Function

''' Download and print as text 'Usage: xlib getprint [url]
Function xlib_getprint(url)
    Set htt = gXmlHttp()
    htt.Open "GET", url, 0
    htt.Send
    printLine htt.ResponseText
End Function

' REF http://demon.tw/programming/vbs-unzip-file.html
''' Create zip file 'Usage: xlib zip [source_path] [zip_output_path]
Function xlib_zip(sourcePath, zipPath)
    Dim emptyZipFile, zipFile
    ' Create empty zip file
    Set emptyZipFile = fso.CreateTextFile(zipPath, true)
    emptyZipFile.Write "PK" & Chr(5) & Chr(6) & String(18, Chr(0))
    emptyZipFile.Close
    Set zipFile = CreateObject("Shell.Application").NameSpace(zipPath)
    zipFile.CopyHere sourcePath, 256
    Do
        WScript.Sleep 1000
    Loop Until zipFile.Items.Count > 0
End Function

' REF http://demon.tw/programming/vbs-unzip-file.html
''' Uncompress 'Usage: xlib unzip [zip_path] [output_path]
Function xlib_unZip(zipPath, targetPath)
    Dim shellApp, source, target
    If not fso.FolderExists(targetPath) Then
        fso.CreateFolder(targetPath)
    End If
    Set shellApp = CreateObject("Shell.Application")
    Set source = shellApp.NameSpace(zipPath).Items()
    Set target = shellApp.NameSpace(targetPath)
    target.CopyHere source, 256
End Function

''' Get clipboard data
Function xlib_gClip()
    printLine gClip()
End Function

''' Set clipboard data 'Usage: xlib sClip [String]
Function xlib_sClip(base64Strng)
    Dim fso, logFile, logPath, wshShell
    Set fso = CreateObject("Scripting.FileSystemObject")
    Set wshShell = CreateObject("WScript.Shell")
    If fso.FileExists(fso.GetSpecialFolder(0) & "\System32\clip.exe") Then
        ' use clip.exe
        logPath = fso.GetSpecialFolder(2) & "\" & guid & ".log"
        Set logFile = fso.CreateTextFile(logPath, true)
        logFile.Write base64Strng
        logFile.Close
        WshShell.Run "%ComSpec% /c ""%windir%\system32\clip.exe < """ & logPath, 0, true
        fso.deleteFile logPath
    Else
        ' use javascript
        wshShell.Environment("process").item("@") = base64Strng
        wshShell.Run "mshta ""javascript:clipboardData.setData('text', new ActiveXObject('WScript.Shell').Environment('process').item('@'));close();""", 0, true
    End If
End Function

''' Convert base64 to binary file from text 'Usage: xlib txt2bin [text_path] [output_path]
Function xlib_txt2bin(source, target)
    Dim content, text
    ' Read base64 text from file
    Set content = CreateObject("Scripting.FileSystemObject").opentextfile(source, 1)
    text = gBase64(content.readall)
    content.Close
    base64ToBin text, target
End Function

''' Convert binary file to base64 string 'Usage: xlib bin2Base64 [file_path]
Function xlib_bin2Base64(source)
    printLine bin2Base64(source)
End Function

''' Get guid
Function xlib_guuid()
    xlib_guuid = Left(CreateObject("Scriptlet.TypeLib").Guid, 38)
    printLine xlib_guuid
End Function

''' Get format time
Function xlib_gnow()
    Dim regEx
    Set regEx = New RegExp
    regEx.Pattern = "[\D]"
    regEx.Global = True
    xlib_gnow = regEx.Replace(Now & right(FormatNumber(timer * 100, 0), 2), "")
    printLine xlib_gnow
End Function

''' Convert ansi head to unicode head, or reconvert 'Usage: xlib ansi2unic
Function xlib_ansi2unic(path)
    Dim bin, stream
    Set stream = CreateObject("ADODB.Stream")
    stream.Type = 1
    stream.Open
    stream.LoadFromFile path
    Set bin = CreateObject("Microsoft.XMLDOM").CreateElement("binary")
    bin.DataType = "bin.hex"
    bin.NodeTypedValue = stream.Read
    stream.Close
    If 1 = gType(bin.Text) Then
        bin.Text = mid(bin.Text, 17, len(bin.Text))
    Else
        bin.Text = "fffe2026636c7326" & bin.Text
    End If
    stream.Open
    stream.Write bin.NodeTypedValue
    stream.saveToFile path, 2
    stream.Close
End Function

''' Trim *.inf text for drivers 'Usage: xlib inftrim [source_path] [target_path]
Function xlib_inftrim(source, target)
    Set fso = CreateObject("Scripting.FileSystemObject")
    Set fIn = fso.opentextfile(source)
    text = fIn.readall
    fIn.close
    text = replace(text, chr(9), "")
    text = replace(text, chr(32), "")
    ' for i = 1 to 10
    '     text = replace(text, chr(32) & chr(32), chr(32))
    ' next
    ' {59 ;} {44 ,}
    text = replace(text, chr(59), vbNewLine)
    text = replace(text, chr(44), vbNewLine)
    ' text = replace(text, chr(32) & vbNewLine, chr(13))
    ' create new text file
    Set fOut = fso.CreateTextFile(target, true)
    fOut.writeline text
    fOut.close
End Function

''' XML trans form Node by XSL 'Usage: xlib doxsl [xml_path] [xsl_path] [output_path]
Function xlib_doxsl(xml, xsl, output)
    Dim ver
    ' ver = 6 '
    ver = 3 ' support NT52
    Set xmlObj = CreateObject("MSXML2.DOMDocument." & ver & ".0")
    xmlObj.async = False
    xmlObj.validateOnParse = False
    xmlObj.load xml

    Set xslObj = CreateObject("MSXML2.DOMDocument." & ver & ".0")
    xslObj.async = False
    xslObj.validateOnParse = False
    xslObj.load xsl

    Set fso = CreateObject("Scripting.FileSystemObject").createTextFile(output, True)
    fso.Write(xmlObj.transformNode(xslObj))
    fso.Close()
End Function

''' Get target file system drive Letter 'Usage: xlib gfsd [NTFS/FAT32/EXFAT]
Function xlib_gfsd(tag)
    Dim drv
    For Each drv in CreateObject("Scripting.FileSystemObject").Drives
        If drv.IsReady Then
            If "tag" = drv.FileSystem Then
                printLine drv.DriveLetter & chr(58)
            End If
        End If
    Next
End Function

''' Create shortcut at desktop ''Usage: xlib lnkd [source] [target] ''       [target]: Desktop AllUsersDesktop C:\
Function xlib_lnkd(sourceFilePath, targetFolder)
    Set WshShell = WScript.CreateObject("WScript.Shell")
    Set Fso = CreateObject("Scripting.FileSystemObject")

    targetDir = WshShell.SpecialFolders(targetFolder)
    If targetDir = "" Then
        targetDir = targetFolder
    End If

    Set lnk = WshShell.CreateShortcut(targetDir & "\" & Fso.GetBaseName(sourceFilePath) & ".lnk")
    lnk.TargetPath = sourceFilePath
    lnk.Arguments = ""
    lnk.WorkingDirectory = Fso.GetFile(sourceFilePath).ParentFolder.Path
    lnk.WindowStyle = 1 ' Nomal Windows
    lnk.Hotkey = ""
    lnk.IconLocation = sourceFilePath & ", 0"
    lnk.Description = ""
    lnk.Save
End Function


''''''''''''''''
'   Template   '
''''''''''''''''

''' Tag date time each line 'Usage: [command] | log [strftime format]
Function xlib_log(format)
    format = Replace(format, "$F", "yyyy-MM-dd")
    format = Replace(format, "$T", "hh:mm:ss")
    format = Replace(format, "$Y", "yyyy")
    format = Replace(format, "$y", "yy")
    format = Replace(format, "$m", "MM")
    format = Replace(format, "$d", "dd")
    format = Replace(format, "$H", "hh")
    format = Replace(format, "$M", "mm")
    format = Replace(format, "$S", "ss")
    format = "{0:" & format & "}"

    Set Sb = CreateObject("System.Text.StringBuilder")
    Set StdIn = WScript.StdIn
    Set StdOut = WScript.StdOut
    Do While Not StdIn.AtEndOfStream
        Sb.AppendFormat_4 format, Array(now())
        StdOut.WriteLine Sb.ToString() & StdIn.ReadLine
        Sb.Length = 0
    Loop
End Function

' Function xlib_log(separator)
'     Set StdIn = WScript.StdIn
'     Do While Not StdIn.AtEndOfStream
'         line = StdIn.ReadLine
'         printLine Replace(FormatDateTime(Now()) , "/", "-") & separator & line
'     Loop
' End Function

''''''''''''''''''''''''
'   private function   '
''''''''''''''''''''''''

' Get HttpGet Object
Function gXmlHttp()
    ' To prevent antivirus false positives
    Set gXmlHttp = CreateObject("Microsoft" & Chr(46) & "XML" & "HT" & Chr(84) & "P")
End Function

' BKDRHash
''' BKDR Hash 'Usage: xlib hash [strng]
Function xlib_hash(key)
    Dim seed, hash, i
    seed = 131 ' 31 131 1313 13131 131313
    hash = 0
    For i = 1 To len(key)
        hash = hash * seed + Asc(Mid(key, i, 1))
    Next
    printLine hash Mod &H7FFFFFFF
End Function

' Get clip
Function gClip()
    Dim ieApp
    ' htmlfile, xmlfile, mhtmlfile
    gClip = CreateObject("htmlfile").parentWindow.clipboardData.getData("text")
    If len(gClip) = 0 Then
        gClip = GetObject("\", "htmlfile").parentWindow.clipboardData.getData("text")
        If len(gClip) = 0 Then
            Set ieApp = CreateObject("InternetExplorer.Application")
            ieApp.navigate "about:blank"
            ieApp.visible = False
            gClip = ieApp.document.parentwindow.clipboarddata.getdata("text")
        End If
    End If
End Function

' Format line to 64 column
Function to64Column(strng)
    Dim regEx
    Set regEx = New RegExp
    regEx.Pattern = "(.{64})"
    regEx.Global = True
    to64Column = regEx.Replace(Replace(strng, chr(10), ""), "$1" & chr(10)) & chr(10) & chr(10)
End Function

'Test text is base64 'for built-in function
Function iBase64(Strng)
    Dim regEx
    Set regEx = New RegExp
    regEx.Global = True
    regEx.Pattern = "^([0-9a-zA-Z/+]{64}\r?\n)+([0-9a-zA-Z/+]{1,63}[=]{0,3})?"
    iBase64 = regEx.Test(strng)
    Set regEx = Nothing
End Function

' Get base64 text from source text 'for built-in function
Function gBase64(Strng)
    Dim Matches, regEx
    Set regEx = New RegExp
    regEx.Pattern = "([0-9a-zA-Z/+]{64}\r?\n)+([0-9a-zA-Z/+]{1,63}[=]{0,3})?"
    Set Matches = regEx.Execute(Strng)
    If Matches.Count > 0 Then
        gBase64 = Matches(0)
    Else
        gBase64 = ""
    End If
End Function

' Convert binary file to base64 string
Function bin2Base64(path)
    Dim bin, stream
    Set stream = CreateObject("ADODB.Stream")
    stream.Type = 1
    stream.Open
    stream.LoadFromFile path
    Set bin = CreateObject("Microsoft.XMLDOM").CreateElement("binary")
    bin.DataType = "bin.base64"
    bin.NodeTypedValue = stream.Read
    stream.Close
    bin2Base64 = bin.Text
End Function

' Convert base64 string to binary file
Sub base64ToBin(base64Strng, path)
    Dim bin, stream
    Set bin = CreateObject("Microsoft.XMLDOM").CreateElement("binary")
    bin.DataType = "bin.base64"
    bin.Text = base64Strng
    Set stream = CreateObject("ADODB.Stream")
    stream.Type = 1
    stream.Open
    stream.Write bin.NodeTypedValue
    stream.saveToFile path, 2
    stream.Close
End Sub

' Run other vbscript
Sub vbs(path)
    Execute(CreateObject("Scripting.FileSystemObject").OpenTextFile(path).ReadAll)
End Sub

' Test file head 'Hex or Base64
Function gType(source)
    If "fffe2026636c7326" = (mid(source, 1, 16)) Then ' Unicode [Hex]
        gType = 1
    ElseIf "UEsDB" = (mid(source, 1, 5)) Then ' zip UEsDBBQAAAAIA [Base64]
        gType = 2
    ElseIf "H4sIA" = (mid(source, 1, 5)) Then ' Gz [Base64]
        gType = 3
    ElseIf "QlpoO" = (mid(source, 1, 5)) Then ' Bz2 QlpoOTFBWSZTW [Base64]
        gType = 4
    ElseIf "4d5a" = (mid(source, 1, 4)) Then ' dos exec [Hex]
        gType = 5
    ElseIf "7f45" = (mid(source, 1, 4)) Then ' unix exec [Hex]
        gType = 6
    Else
        gType = 0
    End If
End Function



' Function copy(path)
'     Dim input, output
'     Set input = CreateObject("ADODB.Stream")
'     input.Type = 1
'     input.Open
'     input.LoadFromFile path

'     Set output = CreateObject("ADODB.Stream")
'     output.Type = 1
'     output.Open

'     Do
'         output.Position = output.size
'         output.Write input.read(100000)
'         ' WScript.Sleep 5000
'         output.saveToFile path & ".bak", 2
'         output.Flush
'         ' output.Position = 0
'         ' output.SetEOS
'     Loop Until input.EOS
'     input.close
'     output.close
' End Function

' CreateObject("WScript.Shell").Run WScript.Arguments(0), 0

' Function getDp0
'     getDp0 = getThis.ParentFolder.Path
' End Function



'''''''''''''''''''''''''''''''''''''''''''''''''''
'                    Framework                    '
' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' '

' Make Error code Or Output error info
Sub setErr(Description)
    Err.Clear
    Err.Source = "Custom Error:"
    If IsNumeric(Description) Then
        Err.Raise Description
    Else
        Err.Description = Description
        Err.Raise 2
    End If
End Sub

' Test script run with cscript.exe
Function iCscript()
    If "\cscript.exe" = LCase(Right(WScript.FullName, 12)) Then
        iCscript = True
    Else
        iCscript = False
    End If
End Function

' Print strng, like WScript.Echo
Sub printLine(desc)
    If iCscript() Then
        WScript.StdOut.WriteLine desc
    Else
        MsgBox desc
    End If
End Sub

' Print strng
Sub errLine(desc)
    If iCscript() Then
        WScript.StdErr.WriteLine desc
    Else
        MsgBox desc, 48, "Error"
    End If
End Sub

' This script name
Function gScriptName()
    gScriptName = Mid(WScript.ScriptName, 1, InStrRev(WScript.ScriptName, Chr(46)) - 1)
End Function

' Print sort string
' System.Collections.ArrayList not support at Windows PE
Sub sortPrint(str)
    Dim arr
    arr = Split(str, vbNewLine)
    quickSort arr, Lbound(arr), Ubound(arr)
    printLine Join(arr, vbNewLine)
End Sub

' Quick sort
' REF http://www.cnblogs.com/falconshh/archive/2011/05/30/2063204.html
Sub quickSort(arr, low, high)
    Dim pivotPos
    If low < high Then
        pivotPos = partition(arr, low, high)
        quickSort arr, low, pivotPos - 1
        quickSort arr, pivotPos + 1, high
    End if
End Sub

' For quickSort sub
' REF http://www.cnblogs.com/falconshh/archive/2011/05/30/2063204.html
Function partition(arr, low, high)
    Dim i, j, pivot
    i = low
    j = high
    pivot = arr(low)
    While i < j
        While i < j And arr(j) >= pivot
            j = j - 1
        Wend
        arr(i) = arr(j)
        While i < j And arr(i) <= pivot
            i = i + 1
        Wend
        arr(j) = arr(i)
    Wend
    arr(i) = pivot
    partition = i
End Function

' Search this script function
Function gfuncAnno(method)
    Dim annotation, text, str, line, prefix, i
    prefix = gScriptName()
    ' This script body annotation
    Set text = CreateObject("Scripting.FileSystemObject").OpenTextFile(WScript.ScriptFullName)
    Do Until text.AtEndOfStream
        ' Trim line
        line = Trim(text.ReadLine)
        str = LCase(line)
        ' If case function name
        If "function " & prefix & Chr(95) & method = Mid(str, 1, Len(method) + 13) Then

            If Len(annotation) > 1 Then
                ' Get single function info
                If Len(method) > 0 Then
                    ' Get annotation replace ' to \n
                    For Each i In Split(annotation, Chr(39))
                        gfuncAnno = gfuncAnno & i & vbNewLine
                    Next
                    Exit Do
                End If
                ' Get first annotation split by '
                i = InStr(annotation, Chr(39)) - 2
                If i > 0 Then annotation = Left(annotation, i)
            End If

            ' Get function name
            str = Trim(Mid(str, 14, InStr(str, Chr(40)) - 14))
            ' Align annotation
            gfuncAnno = gfuncAnno & str & Space(15 - Len(str)) & annotation & vbNewLine
            ' Clear annotation
            annotation = ""
        ElseIf Chr(39) & Chr(39) & Chr(39) & Chr(32) = Mid(str, 1, 4) Then
            ' Get all annotations
            annotation = Trim(Mid(line, 4))
        End If
    Loop
    text.Close
End Function

' Replace - to StdIn.ReadAll
Function rArg(i)
    rArg = args(i)
    If "-" = rArg Then
        rArg = "WScript.StdIn.ReadAll"
    Else
        rArg = "args(" & i & ")"
    End If
End Function

' Test help
Function iHelp(i)
    Dim str
    str = LCase(args(i))
    If 0 = args.Count Then
        iHelp = True
        Exit Function
    ElseIf "-h" = str Or "--help" = str Then
        iHelp = True
    Else
        iHelp = False
    End If
End Function

Sub main()
    Dim arg, MethoParas, funcName, ScriptName, i
    ScriptName = gScriptName()

    ' ' Cache arguments
    ' Set args = WScript.Arguments
    ' ' Cache arguments count
    ' count = args.Count
    ' ' From secend arguments
    ' For i = 1 To count

    ' Next

    i = 0
    ' Assembling method name and parameter list
    ' WScript.Arguments not array
    For Each arg In WScript.Arguments
        i = i + 1
        If i = 1 Then
            ' Info list
            If LCase(arg) = "-h" Or Lcase(arg) = "--help" Then
                i = 0
                Exit For ' Will print all function introduction
            End If
            ' e.g. xlib_func(
            MethoParas = ScriptName & Chr(95) & arg & Chr(40)
            funcName = arg
        ElseIf i = 2 Then
            ' Info single
            If LCase(arg) = "-h" Or Lcase(arg) = "--help" Then
                ' Print target function annotation
                printLine gfuncAnno(funcName)
                i = -1
                Exit For
            End If
            ' e.g. xlib_func("arg1"
            MethoParas = MethoParas & Chr(34) & arg & Chr(34)
        Else
            ' e.g. xlib_func("arg1","arg2"
            MethoParas = MethoParas & Chr(44) & Chr(34) & arg & Chr(34)
        End If
    Next

    if i > 0 Then
        ' e.g. xlib_func(...)
        If Len(MethoParas) > 0 Then MethoParas = MethoParas & Chr(41)
        On Error Resume Next
        ' Run Function by Arguments
        eval MethoParas
        ' Catch Error
    ElseIf i = 0 Then
        arg = gfuncAnno("")
        ' Print all function introduction
        sortPrint Left(arg, Len(arg) - Len(vbNewLine))
    End If

    Select Case Err.Number
        Case 0
            ' Boolean: True
            WScript.Quit(0)
        Case 1
            ' setErr 1: return False to caller
            Err.Clear
            WScript.Quit(1)
        Case 2
            ' Custom error
            errLine "Error: " & Err.Description & Chr(32) _
            & Chr(40) & WScript.ScriptFullName & Chr(58) & ScriptName & Chr(95) & funcName & Chr(41)
        Case 13
            errLine "Error: No function found"
        Case 450
            errLine "Error: " & Err.Description
        Case Else
            errLine "Error Code: " & Err.Number & vbNewLine _
            & Err.Description & vbNewLine & "args: " & MethoParas
    End Select
    WScript.Quit(Err.Number)
    ' On Error GoTo 0
End Sub

' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' '
'                    Framework                    '
'''''''''''''''''''''''''''''''''''''''''''''''''''

main()

'' Err number in use
' 0, 5, 6, 7, 9,
' 10, 11, 13, 14, 17,
' 28, 35, 48,
' 51, 52, 53, 54, 55, 57, 58,
' 61, 62, 67, 68,
' 70, 71, 74, 75, 76,
' 91, 92, 94,
' 322,
' 424, 429, 430, 432, 438, 440, 445, 446, 447, 448, 449, 450, 451, 453, 455, 457, 458, 462, 481,
' 500, 501, 502, 503, 504, 505, 506, 507,
' 1001, 1002, 1003, 1005, 1006, 1007, 1010, 1011, 1012, 1013, 1014, 1015, 1016, 1017, 1018, 1019, 1020, 1021, 1022, 1023, 1024, 1025, 1026, 1027, 1028, 1029, 1030, 1031, 1032, 1033, 1034, 1037, 1038, 1039, 1040, 1041, 1042, 1043, 1044, 1045, 1046, 1047, 1048, 1049, 1050, 1051, 1052, 1053, 1054, 1055, 1056, 1057, 1058,
' 4096, 4097,
' 5016, 5017, 5018, 5019, 5020, 5021,
' 30000, 32766, 32767, 32768, 32769, 32770, 32811, 32812, 32813,
' 65536
''