| 0,0 → 1,327 |
|---|
| /** |
| * EnvVarUpdate.nsh |
| * : Environmental Variables: append, prepend, and remove entries |
| * |
| * WARNING: If you use StrFunc.nsh header then include it before this file |
| * with all required definitions. This is to avoid conflicts |
| * |
| * Usage: |
| * ${EnvVarUpdate} "ResultVar" "EnvVarName" "Action" "RegLoc" "PathString" |
| * |
| * Credits: |
| * Version 1.0 |
| * * Cal Turney (turnec2) |
| * * Amir Szekely (KiCHiK) and e-circ for developing the forerunners of this |
| * function: AddToPath, un.RemoveFromPath, AddToEnvVar, un.RemoveFromEnvVar, |
| * WriteEnvStr, and un.DeleteEnvStr |
| * * Diego Pedroso (deguix) for StrTok |
| * * Kevin English (kenglish_hi) for StrContains |
| * * Hendri Adriaens (Smile2Me), Diego Pedroso (deguix), and Dan Fuhry |
| * (dandaman32) for StrReplace |
| * |
| * Version 1.1 (compatibility with StrFunc.nsh) |
| * * techtonik |
| * |
| * http://nsis.sourceforge.net/Environmental_Variables:_append%2C_prepend%2C_and_remove_entries |
| * |
| */ |
| |
| |
| !ifndef ENVVARUPDATE_FUNCTION |
| !define ENVVARUPDATE_FUNCTION |
| !verbose push |
| !verbose 3 |
| !include "LogicLib.nsh" |
| !include "WinMessages.NSH" |
| !include "StrFunc.nsh" |
| |
| ; ---- Fix for conflict if StrFunc.nsh is already includes in main file ----------------------- |
| !macro _IncludeStrFunction StrFuncName |
| !ifndef ${StrFuncName}_INCLUDED |
| ${${StrFuncName}} |
| !endif |
| !ifndef Un${StrFuncName}_INCLUDED |
| ${Un${StrFuncName}} |
| !endif |
| !define un.${StrFuncName} "${Un${StrFuncName}}" |
| !macroend |
| |
| !insertmacro _IncludeStrFunction StrTok |
| !insertmacro _IncludeStrFunction StrStr |
| !insertmacro _IncludeStrFunction StrRep |
| |
| ; ---------------------------------- Macro Definitions ---------------------------------------- |
| !macro _EnvVarUpdateConstructor ResultVar EnvVarName Action Regloc PathString |
| Push "${EnvVarName}" |
| Push "${Action}" |
| Push "${RegLoc}" |
| Push "${PathString}" |
| Call EnvVarUpdate |
| Pop "${ResultVar}" |
| !macroend |
| !define EnvVarUpdate '!insertmacro "_EnvVarUpdateConstructor"' |
| |
| !macro _unEnvVarUpdateConstructor ResultVar EnvVarName Action Regloc PathString |
| Push "${EnvVarName}" |
| Push "${Action}" |
| Push "${RegLoc}" |
| Push "${PathString}" |
| Call un.EnvVarUpdate |
| Pop "${ResultVar}" |
| !macroend |
| !define un.EnvVarUpdate '!insertmacro "_unEnvVarUpdateConstructor"' |
| ; ---------------------------------- Macro Definitions end------------------------------------- |
| |
| ;----------------------------------- EnvVarUpdate start---------------------------------------- |
| !define hklm_all_users 'HKLM "SYSTEM\CurrentControlSet\Control\Session Manager\Environment"' |
| !define hkcu_current_user 'HKCU "Environment"' |
| |
| !macro EnvVarUpdate UN |
| |
| Function ${UN}EnvVarUpdate |
| |
| Push $0 |
| Exch 4 |
| Exch $1 |
| Exch 3 |
| Exch $2 |
| Exch 2 |
| Exch $3 |
| Exch |
| Exch $4 |
| Push $5 |
| Push $6 |
| Push $7 |
| Push $8 |
| Push $9 |
| Push $R0 |
| |
| /* After this point: |
| ------------------------- |
| $0 = ResultVar (returned) |
| $1 = EnvVarName (input) |
| $2 = Action (input) |
| $3 = RegLoc (input) |
| $4 = PathString (input) |
| $5 = Orig EnvVar (read from registry) |
| $6 = Len of $0 (temp) |
| $7 = tempstr1 (temp) |
| $8 = Entry counter (temp) |
| $9 = tempstr2 (temp) |
| $R0 = tempChar (temp) */ |
| |
| ; Step 1: Read contents of EnvVarName from RegLoc |
| ; |
| ; Check for empty EnvVarName |
| ${If} $1 == "" |
| SetErrors |
| DetailPrint "ERROR: EnvVarName is blank" |
| Goto EnvVarUpdate_Restore_Vars |
| ${EndIf} |
| |
| ; Check for valid Action |
| ${If} $2 != "A" |
| ${AndIf} $2 != "P" |
| ${AndIf} $2 != "R" |
| SetErrors |
| DetailPrint "ERROR: Invalid Action - must be A, P, or R" |
| Goto EnvVarUpdate_Restore_Vars |
| ${EndIf} |
| |
| ${If} $3 == HKLM |
| ReadRegStr $5 ${hklm_all_users} $1 ; Get EnvVarName from all users into $5 |
| ${ElseIf} $3 == HKCU |
| ReadRegStr $5 ${hkcu_current_user} $1 ; Read EnvVarName from current user into $5 |
| ${Else} |
| SetErrors |
| DetailPrint 'ERROR: Action is [$3] but must be "HKLM" or HKCU"' |
| Goto EnvVarUpdate_Restore_Vars |
| ${EndIf} |
| |
| ; Check for empty PathString |
| ${If} $4 == "" |
| SetErrors |
| DetailPrint "ERROR: PathString is blank" |
| Goto EnvVarUpdate_Restore_Vars |
| ${EndIf} |
| |
| ; Make sure we've got some work to do |
| ${If} $5 == "" |
| ${AndIf} $2 == "R" |
| SetErrors |
| DetailPrint "$1 is empty - Nothing to remove" |
| Goto EnvVarUpdate_Restore_Vars |
| ${EndIf} |
| |
| ; Step 2: Scrub EnvVar |
| ; |
| StrCpy $0 $5 ; Copy the contents to $0 |
| ; Remove spaces around semicolons (NOTE: spaces before the 1st entry or |
| ; after the last one are not removed here but instead in Step 3) |
| ${If} $0 != "" ; If EnvVar is not empty ... |
| ${Do} |
| ${${UN}StrStr} $7 $0 " ;" |
| ${If} $7 == "" |
| ${ExitDo} |
| ${EndIf} |
| ${${UN}StrRep} $0 $0 " ;" ";" ; Remove '<space>;' |
| ${Loop} |
| ${Do} |
| ${${UN}StrStr} $7 $0 "; " |
| ${If} $7 == "" |
| ${ExitDo} |
| ${EndIf} |
| ${${UN}StrRep} $0 $0 "; " ";" ; Remove ';<space>' |
| ${Loop} |
| ${Do} |
| ${${UN}StrStr} $7 $0 ";;" |
| ${If} $7 == "" |
| ${ExitDo} |
| ${EndIf} |
| ${${UN}StrRep} $0 $0 ";;" ";" |
| ${Loop} |
| |
| ; Remove a leading or trailing semicolon from EnvVar |
| StrCpy $7 $0 1 0 |
| ${If} $7 == ";" |
| StrCpy $0 $0 "" 1 ; Change ';<EnvVar>' to '<EnvVar>' |
| ${EndIf} |
| StrLen $6 $0 |
| IntOp $6 $6 - 1 |
| StrCpy $7 $0 1 $6 |
| ${If} $7 == ";" |
| StrCpy $0 $0 $6 ; Change ';<EnvVar>' to '<EnvVar>' |
| ${EndIf} |
| ; DetailPrint "Scrubbed $1: [$0]" ; Uncomment to debug |
| ${EndIf} |
| |
| /* Step 3. Remove all instances of the target path/string (even if "A" or "P") |
| $6 = bool flag (1 = found and removed PathString) |
| $7 = a string (e.g. path) delimited by semicolon(s) |
| $8 = entry counter starting at 0 |
| $9 = copy of $0 |
| $R0 = tempChar */ |
| |
| ${If} $5 != "" ; If EnvVar is not empty ... |
| StrCpy $9 $0 |
| StrCpy $0 "" |
| StrCpy $8 0 |
| StrCpy $6 0 |
| |
| ${Do} |
| ${${UN}StrTok} $7 $9 ";" $8 "0" ; $7 = next entry, $8 = entry counter |
| |
| ${If} $7 == "" ; If we've run out of entries, |
| ${ExitDo} ; were done |
| ${EndIf} ; |
| |
| ; Remove leading and trailing spaces from this entry (critical step for Action=Remove) |
| ${Do} |
| StrCpy $R0 $7 1 |
| ${If} $R0 != " " |
| ${ExitDo} |
| ${EndIf} |
| StrCpy $7 $7 "" 1 ; Remove leading space |
| ${Loop} |
| ${Do} |
| StrCpy $R0 $7 1 -1 |
| ${If} $R0 != " " |
| ${ExitDo} |
| ${EndIf} |
| StrCpy $7 $7 -1 ; Remove trailing space |
| ${Loop} |
| ${If} $7 == $4 ; If string matches, remove it by not appending it |
| StrCpy $6 1 ; Set 'found' flag |
| ${ElseIf} $7 != $4 ; If string does NOT match |
| ${AndIf} $0 == "" ; and the 1st string being added to $0, |
| StrCpy $0 $7 ; copy it to $0 without a prepended semicolon |
| ${ElseIf} $7 != $4 ; If string does NOT match |
| ${AndIf} $0 != "" ; and this is NOT the 1st string to be added to $0, |
| StrCpy $0 $0;$7 ; append path to $0 with a prepended semicolon |
| ${EndIf} ; |
| |
| IntOp $8 $8 + 1 ; Bump counter |
| ${Loop} ; Check for duplicates until we run out of paths |
| ${EndIf} |
| |
| ; Step 4: Perform the requested Action |
| ; |
| ${If} $2 != "R" ; If Append or Prepend |
| ${If} $6 == 1 ; And if we found the target |
| DetailPrint "Target is already present in $1. It will be removed and" |
| ${EndIf} |
| ${If} $0 == "" ; If EnvVar is (now) empty |
| StrCpy $0 $4 ; just copy PathString to EnvVar |
| ${If} $6 == 0 ; If found flag is either 0 |
| ${OrIf} $6 == "" ; or blank (if EnvVarName is empty) |
| DetailPrint "$1 was empty and has been updated with the target" |
| ${EndIf} |
| ${ElseIf} $2 == "A" ; If Append (and EnvVar is not empty), |
| StrCpy $0 $0;$4 ; append PathString |
| ${If} $6 == 1 |
| DetailPrint "appended to $1" |
| ${Else} |
| DetailPrint "Target was appended to $1" |
| ${EndIf} |
| ${Else} ; If Prepend (and EnvVar is not empty), |
| StrCpy $0 $4;$0 ; prepend PathString |
| ${If} $6 == 1 |
| DetailPrint "prepended to $1" |
| ${Else} |
| DetailPrint "Target was prepended to $1" |
| ${EndIf} |
| ${EndIf} |
| ${Else} ; If Action = Remove |
| ${If} $6 == 1 ; and we found the target |
| DetailPrint "Target was found and removed from $1" |
| ${Else} |
| DetailPrint "Target was NOT found in $1 (nothing to remove)" |
| ${EndIf} |
| ${If} $0 == "" |
| DetailPrint "$1 is now empty" |
| ${EndIf} |
| ${EndIf} |
| |
| ; Step 5: Update the registry at RegLoc with the updated EnvVar and announce the change |
| ; |
| ClearErrors |
| ${If} $3 == HKLM |
| WriteRegExpandStr ${hklm_all_users} $1 $0 ; Write it in all users section |
| ${ElseIf} $3 == HKCU |
| WriteRegExpandStr ${hkcu_current_user} $1 $0 ; Write it to current user section |
| ${EndIf} |
| |
| IfErrors 0 +4 |
| MessageBox MB_OK|MB_ICONEXCLAMATION "Could not write updated $1 to $3" |
| DetailPrint "Could not write updated $1 to $3" |
| Goto EnvVarUpdate_Restore_Vars |
| |
| ; "Export" our change |
| SendMessage ${HWND_BROADCAST} ${WM_WININICHANGE} 0 "STR:Environment" /TIMEOUT=5000 |
| |
| EnvVarUpdate_Restore_Vars: |
| ; |
| ; Restore the user's variables and return ResultVar |
| Pop $R0 |
| Pop $9 |
| Pop $8 |
| Pop $7 |
| Pop $6 |
| Pop $5 |
| Pop $4 |
| Pop $3 |
| Pop $2 |
| Pop $1 |
| Push $0 ; Push my $0 (ResultVar) |
| Exch |
| Pop $0 ; Restore his $0 |
| |
| FunctionEnd |
| |
| !macroend ; EnvVarUpdate UN |
| !insertmacro EnvVarUpdate "" |
| !insertmacro EnvVarUpdate "un." |
| ;----------------------------------- EnvVarUpdate end---------------------------------------- |
| |
| !verbose pop |
| !endif |