#include-once

; corz registry functions v0.2
; (c) cor + corz.org 2006->tomorrow!

#cs

    miscelleneous registry comands

    @extended values..

        0 = directory (key)
        1 = REG_SZ
        2 = REG_EXPAND_SZ
        3 = REG_BINARY
        4 = REG_DWORD
        7 = REG_MULTI_SZ


    function list..

        MakeLongRegPaths()
        GetConceptCommands()
        KeyIsEmpty()
        ReturnAllSubkeys()
        ReturnAllValueNames()
        ReturnAllPairs()
        SetExplorerContextMenu()

#ce


; switch short (i.e "HKCU") registry paths, for "long" (i.e. "HKEY_CURRENT_USER") paths
; (doesn't AutoIt do this automatically yet?)..
func MakeLongRegPaths($reg_key)
    $reg_key = StringReplace($reg_key, "HKCR", "HKEY_CLASSES_ROOT")
    $reg_key = StringReplace($reg_key, "HKCU", "HKEY_CURRENT_USER")
    $reg_key = StringReplace($reg_key, "HKU", "HKEY_USERS")
    $reg_key = StringReplace($reg_key, "HKLM", "HKEY_LOCAL_MACHINE")
    $reg_key = StringReplace($reg_key, "HKCC", "HKEY_CURRENT_CONFIG")
    return $reg_key
endfunc


;
; GetConceptCommands()
;
; return names of the concept (right-click) commands
; of a particular file type as an "AutoIt array", that is,
; an array where the first value is the number of real
; values in the array (integer).
;
;    $array = GetConceptCommands("txtfile")
;
func GetConceptCommands($file_type)

    $file_type = "HKEY_CLASSES_ROOT\" & $file_type & "\Shell"
    $dirs = ""
    $y = 1
    while $y
        $var = RegEnumKey($file_type, $y)
        if @error <> 0 then exitloop
        if @extended = 0 then ; directory
            $dirs &= $var & "|"
        endif
        $y += 1
    wend
    if $dirs <> "" then
    $dirs = StringTrimRight($dirs, 1)
    return StringSplit($dirs, "|")
    endif
endfunc


; returns true is a given registry key is empty..
;
; requires ReturnAllSubkeys() & ReturnAllValueNames()
;
func KeyIsEmpty($key)
    $reg_test_array_k = ReturnAllSubkeys($key)
    $reg_test_array_v = ReturnAllValueNames($key)
    if $reg_test_array_k[1] = "" and  $reg_test_array_v[1] = "" then
        return 1
    endif
endfunc


; ReturnAllSubkeys()
; returns all the subkeys of a registry key as an "AutoIt array" of key names..
; eg. $array = ReturnAllSubkeys("HKCU\SOFTWARE\corz")
;
; requires MakeLongRegPaths($reg_key)
;
func ReturnAllSubkeys($key)
    $key = MakeLongRegPaths($key)
    $keys = ""
    $x = 1
    while $x
        $var = RegEnumKey($key, $x)
        if @error <> 0 then exitloop
        if $var <> "" then $keys &= $var & "|"
        $x += 1
    wend
    $keys = StringTrimRight($keys, 1)
    return StringSplit($keys, "|")
endfunc



; ReturnAllValueNames()
; returns all the names of a registry key as an "AutoIt array" of names..
; eg. $array = ReturnAllValueNames("HKCU\SOFTWARE\corz")
;
; requires MakeLongRegPaths($reg_key)
;
func ReturnAllValueNames($key)
    $key = MakeLongRegPaths($key)
    $names = ""
    $x = 1
    while $x
        $var = RegEnumVal($key, $x)
        if @error <> 0 then exitloop
        if $var <> "" then $names &= $var & "|"
        $x += 1
    wend
    ;if $names <> "" then
    $names = StringTrimRight($names, 1)
    return StringSplit($names, "|")
    ;endif
endfunc



;
; ReturnAllPairs()
;
; returns all the name/value pairs from a registry key as a
; 2-dimensional "AutoIt array" of name/value pairs, starting at
; array[1][0]. array[0][0] contains the total number of pairs
; returned.  Empty registry values can be skipped by passing the
; optional second parameter as true.
;
;    ReturnAllPairs("HKCU\SOFTWARE\corz")
;
#cs
 ReturnAllPairs example:
;
; display programs that run at startup for all users..

#include "corz_registry.au3"

    $str = ""
    $foo = ReturnAllPairs("HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\Run")
    for $i = 1 to $foo[0][0]
        $str &= "key: " & $foo[$i][0] & "    value: " & $foo[$i][1] & "    " & @LF
    next
    MsgBox (0, "These programs run at startup..", $str)
    exit

; or the whole lot..

    dim $startup_places[4] = _
    [    "HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Run",            _
        "HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\Run",            _
        "HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\RunOnce",        _
        "HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\RunOnceEx"    ]

    for $place in $startup_places
        $foo = ReturnAllPairs($place)
        for $i = 1 to $foo[0][0]
            $str &= $foo[$i][0] & " :    " & $foo[$i][1] & "    " & @LF
        next
    next
    MsgBox (0, "These programs run at startup..", $str)

#ce
func ReturnAllPairs($key, $skip_empties=false)

    $x = 1
    $y = 1
    local $pairs[1][2]

    while $x
        $var = RegEnumVal($key, $x)
        if @error <> 0 then exitloop
        $this_value = RegRead($key, $var)
        if (not $skip_empties) or ($skip_empties and $this_value <> "") then
            redim $pairs[$y+1][2]
            $pairs[$y][0] = $var
            $pairs[$y][1] = $this_value
            $y += 1
        endif
        $x += 1
    wend

    $pairs[0][0] = $y - 1
    return $pairs

endfunc




;
; SetExplorerContextMenu
;
; This sets an Explorer concept item for an application. It can be either the default
; command for a particular filetype, or else simply another command on the context menu.
;
; This function relies on my $my_name convention, where $my_name is a global variable
; containing a string which is the name of the application, used to create subkeys, etc.
;
; Uses KeyIsEmpty()    - Also in this collection.
;
; Parameters..
;
;        1. $ext            File Type extension to work with (minus the dot)
;        2. $switch        $ON or $OFF, $GUI_CHECKED & $GUI_UNCHECKED, 1 or 4, whatever
;                        This switch enables or disables the explorer menu item, depending.
;        3. $default        True if you wish to set the Explorer default command
;        4. $menu_item    The string of text that will appear in the Explorer context menu
;                        Defaults to $my_name if left blank
;        5. $fallback    If no file type has been set for this file extension, you can
;                        specify one here. Will only be used if the file extension
;                        doesn't already have a type. It will not steal an extension.
;        6. $switches    Any switches you would like to place in the command-line
;                        (they are placed between <app> & <path>, with a space added at either side)
;
;
func SetExplorerContextMenu($ext, $switch=4, $default=false, $menu_item="", $fallback="my-unique-key", $switches="")

    if not $ext then return false
    local $key = $my_name
    if $default then $key = "open"    ; become the default app for this file type
    if not $menu_item then $menu_item = $my_name

    local $reg1, $reg2, $reg3
    local $file_type = RegRead("HKEY_CLASSES_ROOT\." & $ext, "")                                ; "soundrec"
    if not $file_type then $file_type = $fallback
    local $shell = "HKEY_CLASSES_ROOT\" & $file_type & "\shell"                                    ;    HKEY_CLASSES_ROOT\soundrec\shell
    local $shellkey = $shell & "\" & $key                                                        ;    HKEY_CLASSES_ROOT\soundrec\shell\(MyAPP || open)
    local $BAX_key = "HKEY_CLASSES_ROOT\" & $file_type & "\" & $my_name & "_Backup\" & $key        ;    HKEY_CLASSES_ROOT\soundrec\MyAPP_Backup\(MyAPP || open)

    if @compiled then
        $me_app = @ScriptFullPath
    else
        $me_app = ".\" & $my_name & ".exe"
    endif

    ; user has this enabled.
    if $switch = 1 then

        ; check if it's still enabled..
        $regexist = RegRead($shellkey & "\command", "")
        if StringInStr($regexist, RemoveExtension(BaseName(@ScriptName))) then return true

        ; backup current key..
        local $bax1, $bax2, $bax3, $old_val, $old_key    ;    default shell command: "open"
        ; we don't simply set this to the \my_name command, because they could be two different commands.
        ; this is more flexible

        ; replacing default shell command..
        if $default then
            $old_key = RegRead($shell, "")                                            ;    "open"    or "02.other_command", etc.
            $bax1 = RegWrite($BAX_key, "", "REG_SZ", $old_key)
            $old_val = RegRead($shell & "\" & $old_key, "")                            ;    "DO THIS.."
            $bax2 = RegWrite($BAX_key, "menu", "REG_SZ", $old_val)
            $old_val = RegRead($shell & "\" &  $old_key & "\command", "")            ;    C:\Program Files\other-apps\some-app.exe "%1"
            $bax3 = RegWrite($BAX_key, "cmd", "REG_SZ", $old_val)                    ;    "cmd" because "command" is reserved!
            if not $bax1 and $bax2 and $bax3 then return false
        endif

        ; write new key..
        $reg1 = RegWrite($shellkey, "", "REG_SZ", $menu_item)
        $reg2 = RegWrite($shellkey & "\command", "", "REG_SZ", '"' & $me_app & '" ' & $switches & ' "%1"')
        if $default then RegWrite($shell, "", "REG_SZ", "open")

        if $reg1 and $reg2 then return true


    else ; Switch OFF..

        $regexist = RegRead($shellkey & "\command", "")
        if not StringInstr($regexist, $me_app) then return true

        ; okay, reg is in place, let's restore backup, if available, or delete keys..
        local $old_key1, $old_key2, $old_key3

        ; default command..
        if $default then
            $old_key1 = RegRead($BAX_key, "")
            $old_key2 = RegRead($BAX_key, "menu")
            $old_key3 = RegRead($BAX_key, "cmd")

            if $old_key1 and $old_key2 and $old_key3 then

                $reg1 = RegWrite($shell, "", "REG_SZ", $old_key1)
                $reg2 = RegWrite($shell & "\" & $old_key1, "", "REG_SZ", $old_key2)
                $reg3 = RegWrite($shell & "\" & $old_key1 & "\command", "", "REG_SZ", $old_key3)
                if $reg1 and $reg2 and $reg3 then
                    RegDelete($BAX_key)
                    if KeyIsEmpty("HKEY_CLASSES_ROOT\" & $file_type & "\" & $my_name & "_Backup") then
                        RegDelete("HKEY_CLASSES_ROOT\" & $file_type & "\" & $my_name & "_Backup")
                    endif
                    if $old_key1 <> "open" then RegDelete($shellkey)
                    return true
                endif
            endif
        else
        ; regular menu item, simply delete it..
            $restored = RegDelete($shellkey)
            if $restored then return true
        endif
    endif

    return false
endfunc


#cs

    Changes:

        0.2

            Added..

                SetExplorerContextMenu()

            which I cooked up for LoopDropZ & MangleeZee. All yours!


        0.1

            A small selection of useful registry functions..

                MakeLongRegPaths()
                GetConceptCommands()
                KeyIsEmpty()
                ReturnAllSubkeys()
                ReturnAllValueNames()
                ReturnAllPairs()


#ce