mirror of
https://github.com/harelba/q.git
synced 2024-10-04 06:48:13 +03:00
441 lines
9.7 KiB
Plaintext
441 lines
9.7 KiB
Plaintext
!ifndef _AddToPath_nsh
|
|
!define _AddToPath_nsh
|
|
|
|
!verbose 3
|
|
!include "WinMessages.NSH"
|
|
!verbose 4
|
|
|
|
!ifndef WriteEnvStr_RegKey
|
|
!ifdef ALL_USERS
|
|
!define WriteEnvStr_RegKey \
|
|
'HKLM "SYSTEM\CurrentControlSet\Control\Session Manager\Environment"'
|
|
!else
|
|
!define WriteEnvStr_RegKey 'HKCU "Environment"'
|
|
!endif
|
|
!endif
|
|
|
|
; AddToPath - Adds the given dir to the search path.
|
|
; Input - head of the stack
|
|
; Note - Win9x systems requires reboot
|
|
|
|
Function AddToPath
|
|
Exch $0
|
|
Push $1
|
|
Push $2
|
|
Push $3
|
|
|
|
# don't add if the path doesn't exist
|
|
IfFileExists "$0\*.*" "" AddToPath_done
|
|
|
|
ReadEnvStr $1 PATH
|
|
Push "$1;"
|
|
Push "$0;"
|
|
Call StrStr
|
|
Pop $2
|
|
StrCmp $2 "" "" AddToPath_done
|
|
Push "$1;"
|
|
Push "$0\;"
|
|
Call StrStr
|
|
Pop $2
|
|
StrCmp $2 "" "" AddToPath_done
|
|
GetFullPathName /SHORT $3 $0
|
|
Push "$1;"
|
|
Push "$3;"
|
|
Call StrStr
|
|
Pop $2
|
|
StrCmp $2 "" "" AddToPath_done
|
|
Push "$1;"
|
|
Push "$3\;"
|
|
Call StrStr
|
|
Pop $2
|
|
StrCmp $2 "" "" AddToPath_done
|
|
|
|
Call IsNT
|
|
Pop $1
|
|
StrCmp $1 1 AddToPath_NT
|
|
; Not on NT
|
|
StrCpy $1 $WINDIR 2
|
|
FileOpen $1 "$1\autoexec.bat" a
|
|
FileSeek $1 -1 END
|
|
FileReadByte $1 $2
|
|
IntCmp $2 26 0 +2 +2 # DOS EOF
|
|
FileSeek $1 -1 END # write over EOF
|
|
FileWrite $1 "$\r$\nSET PATH=%PATH%;$3$\r$\n"
|
|
FileClose $1
|
|
SetRebootFlag true
|
|
Goto AddToPath_done
|
|
|
|
AddToPath_NT:
|
|
ReadRegStr $1 ${WriteEnvStr_RegKey} "PATH"
|
|
StrCmp $1 "" AddToPath_NTdoIt
|
|
Push $1
|
|
Call Trim
|
|
Pop $1
|
|
StrCpy $0 "$1;$0"
|
|
AddToPath_NTdoIt:
|
|
WriteRegExpandStr ${WriteEnvStr_RegKey} "PATH" $0
|
|
SendMessage ${HWND_BROADCAST} ${WM_WININICHANGE} 0 "STR:Environment" /TIMEOUT=5000
|
|
|
|
AddToPath_done:
|
|
Pop $3
|
|
Pop $2
|
|
Pop $1
|
|
Pop $0
|
|
FunctionEnd
|
|
|
|
; RemoveFromPath - Remove a given dir from the path
|
|
; Input: head of the stack
|
|
|
|
Function un.RemoveFromPath
|
|
Exch $0
|
|
Push $1
|
|
Push $2
|
|
Push $3
|
|
Push $4
|
|
Push $5
|
|
Push $6
|
|
|
|
IntFmt $6 "%c" 26 # DOS EOF
|
|
|
|
Call un.IsNT
|
|
Pop $1
|
|
StrCmp $1 1 unRemoveFromPath_NT
|
|
; Not on NT
|
|
StrCpy $1 $WINDIR 2
|
|
FileOpen $1 "$1\autoexec.bat" r
|
|
GetTempFileName $4
|
|
FileOpen $2 $4 w
|
|
GetFullPathName /SHORT $0 $0
|
|
StrCpy $0 "SET PATH=%PATH%;$0"
|
|
Goto unRemoveFromPath_dosLoop
|
|
|
|
unRemoveFromPath_dosLoop:
|
|
FileRead $1 $3
|
|
StrCpy $5 $3 1 -1 # read last char
|
|
StrCmp $5 $6 0 +2 # if DOS EOF
|
|
StrCpy $3 $3 -1 # remove DOS EOF so we can compare
|
|
StrCmp $3 "$0$\r$\n" unRemoveFromPath_dosLoopRemoveLine
|
|
StrCmp $3 "$0$\n" unRemoveFromPath_dosLoopRemoveLine
|
|
StrCmp $3 "$0" unRemoveFromPath_dosLoopRemoveLine
|
|
StrCmp $3 "" unRemoveFromPath_dosLoopEnd
|
|
FileWrite $2 $3
|
|
Goto unRemoveFromPath_dosLoop
|
|
unRemoveFromPath_dosLoopRemoveLine:
|
|
SetRebootFlag true
|
|
Goto unRemoveFromPath_dosLoop
|
|
|
|
unRemoveFromPath_dosLoopEnd:
|
|
FileClose $2
|
|
FileClose $1
|
|
StrCpy $1 $WINDIR 2
|
|
Delete "$1\autoexec.bat"
|
|
CopyFiles /SILENT $4 "$1\autoexec.bat"
|
|
Delete $4
|
|
Goto unRemoveFromPath_done
|
|
|
|
unRemoveFromPath_NT:
|
|
ReadRegStr $1 ${WriteEnvStr_RegKey} "PATH"
|
|
StrCpy $5 $1 1 -1 # copy last char
|
|
StrCmp $5 ";" +2 # if last char != ;
|
|
StrCpy $1 "$1;" # append ;
|
|
Push $1
|
|
Push "$0;"
|
|
Call un.StrStr ; Find `$0;` in $1
|
|
Pop $2 ; pos of our dir
|
|
StrCmp $2 "" unRemoveFromPath_done
|
|
; else, it is in path
|
|
# $0 - path to add
|
|
# $1 - path var
|
|
StrLen $3 "$0;"
|
|
StrLen $4 $2
|
|
StrCpy $5 $1 -$4 # $5 is now the part before the path to remove
|
|
StrCpy $6 $2 "" $3 # $6 is now the part after the path to remove
|
|
StrCpy $3 $5$6
|
|
|
|
StrCpy $5 $3 1 -1 # copy last char
|
|
StrCmp $5 ";" 0 +2 # if last char == ;
|
|
StrCpy $3 $3 -1 # remove last char
|
|
|
|
WriteRegExpandStr ${WriteEnvStr_RegKey} "PATH" $3
|
|
SendMessage ${HWND_BROADCAST} ${WM_WININICHANGE} 0 "STR:Environment" /TIMEOUT=5000
|
|
|
|
unRemoveFromPath_done:
|
|
Pop $6
|
|
Pop $5
|
|
Pop $4
|
|
Pop $3
|
|
Pop $2
|
|
Pop $1
|
|
Pop $0
|
|
FunctionEnd
|
|
|
|
|
|
|
|
; AddToEnvVar - Adds the given value to the given environment var
|
|
; Input - head of the stack $0 environement variable $1=value to add
|
|
; Note - Win9x systems requires reboot
|
|
|
|
Function AddToEnvVar
|
|
|
|
Exch $1 ; $1 has environment variable value
|
|
Exch
|
|
Exch $0 ; $0 has environment variable name
|
|
|
|
DetailPrint "Adding $1 to $0"
|
|
Push $2
|
|
Push $3
|
|
Push $4
|
|
|
|
|
|
ReadEnvStr $2 $0
|
|
Push "$2;"
|
|
Push "$1;"
|
|
Call StrStr
|
|
Pop $3
|
|
StrCmp $3 "" "" AddToEnvVar_done
|
|
|
|
Push "$2;"
|
|
Push "$1\;"
|
|
Call StrStr
|
|
Pop $3
|
|
StrCmp $3 "" "" AddToEnvVar_done
|
|
|
|
|
|
Call IsNT
|
|
Pop $2
|
|
StrCmp $2 1 AddToEnvVar_NT
|
|
; Not on NT
|
|
StrCpy $2 $WINDIR 2
|
|
FileOpen $2 "$2\autoexec.bat" a
|
|
FileSeek $2 -1 END
|
|
FileReadByte $2 $3
|
|
IntCmp $3 26 0 +2 +2 # DOS EOF
|
|
FileSeek $2 -1 END # write over EOF
|
|
FileWrite $2 "$\r$\nSET $0=%$0%;$4$\r$\n"
|
|
FileClose $2
|
|
SetRebootFlag true
|
|
Goto AddToEnvVar_done
|
|
|
|
AddToEnvVar_NT:
|
|
ReadRegStr $2 ${WriteEnvStr_RegKey} $0
|
|
StrCpy $3 $2 1 -1 # copy last char
|
|
StrCmp $3 ";" 0 +2 # if last char == ;
|
|
StrCpy $2 $2 -1 # remove last char
|
|
StrCmp $2 "" AddToEnvVar_NTdoIt
|
|
StrCpy $1 "$2;$1"
|
|
AddToEnvVar_NTdoIt:
|
|
WriteRegExpandStr ${WriteEnvStr_RegKey} $0 $1
|
|
SendMessage ${HWND_BROADCAST} ${WM_WININICHANGE} 0 "STR:Environment" /TIMEOUT=5000
|
|
|
|
AddToEnvVar_done:
|
|
Pop $4
|
|
Pop $3
|
|
Pop $2
|
|
Pop $0
|
|
Pop $1
|
|
|
|
FunctionEnd
|
|
|
|
; RemoveFromEnvVar - Remove a given value from a environment var
|
|
; Input: head of the stack
|
|
|
|
Function un.RemoveFromEnvVar
|
|
|
|
Exch $1 ; $1 has environment variable value
|
|
Exch
|
|
Exch $0 ; $0 has environment variable name
|
|
|
|
DetailPrint "Removing $1 from $0"
|
|
Push $2
|
|
Push $3
|
|
Push $4
|
|
Push $5
|
|
Push $6
|
|
Push $7
|
|
|
|
IntFmt $7 "%c" 26 # DOS EOF
|
|
|
|
Call un.IsNT
|
|
Pop $2
|
|
StrCmp $2 1 unRemoveFromEnvVar_NT
|
|
; Not on NT
|
|
StrCpy $2 $WINDIR 2
|
|
FileOpen $2 "$2\autoexec.bat" r
|
|
GetTempFileName $5
|
|
FileOpen $3 $5 w
|
|
GetFullPathName /SHORT $1 $1
|
|
StrCpy $1 "SET $0=%$0%;$1"
|
|
Goto unRemoveFromEnvVar_dosLoop
|
|
|
|
unRemoveFromEnvVar_dosLoop:
|
|
FileRead $2 $4
|
|
StrCpy $6 $4 1 -1 # read last char
|
|
StrCmp $6 $7 0 +2 # if DOS EOF
|
|
StrCpy $4 $4 -1 # remove DOS EOF so we can compare
|
|
StrCmp $4 "$1$\r$\n" unRemoveFromEnvVar_dosLoopRemoveLine
|
|
StrCmp $4 "$1$\n" unRemoveFromEnvVar_dosLoopRemoveLine
|
|
StrCmp $4 "$1" unRemoveFromEnvVar_dosLoopRemoveLine
|
|
StrCmp $4 "" unRemoveFromEnvVar_dosLoopEnd
|
|
FileWrite $3 $4
|
|
Goto unRemoveFromEnvVar_dosLoop
|
|
unRemoveFromEnvVar_dosLoopRemoveLine:
|
|
SetRebootFlag true
|
|
Goto unRemoveFromEnvVar_dosLoop
|
|
|
|
unRemoveFromEnvVar_dosLoopEnd:
|
|
FileClose $3
|
|
FileClose $2
|
|
StrCpy $2 $WINDIR 2
|
|
Delete "$2\autoexec.bat"
|
|
CopyFiles /SILENT $5 "$2\autoexec.bat"
|
|
Delete $5
|
|
Goto unRemoveFromEnvVar_done
|
|
|
|
unRemoveFromEnvVar_NT:
|
|
ReadRegStr $2 ${WriteEnvStr_RegKey} $0
|
|
StrCpy $6 $2 1 -1 # copy last char
|
|
StrCmp $6 ";" +2 # if last char != ;
|
|
StrCpy $2 "$2;" # append ;
|
|
Push $2
|
|
Push "$1;"
|
|
Call un.StrStr ; Find `$1;` in $2
|
|
Pop $3 ; pos of our dir
|
|
StrCmp $3 "" unRemoveFromEnvVar_done
|
|
; else, it is in path
|
|
# $1 - path to add
|
|
# $2 - path var
|
|
StrLen $4 "$1;"
|
|
StrLen $5 $3
|
|
StrCpy $6 $2 -$5 # $6 is now the part before the path to remove
|
|
StrCpy $7 $3 "" $4 # $7 is now the part after the path to remove
|
|
StrCpy $4 $6$7
|
|
|
|
StrCpy $6 $4 1 -1 # copy last char
|
|
StrCmp $6 ";" 0 +2 # if last char == ;
|
|
StrCpy $4 $4 -1 # remove last char
|
|
|
|
WriteRegExpandStr ${WriteEnvStr_RegKey} $0 $4
|
|
|
|
; delete reg value if null
|
|
StrCmp $4 "" 0 +2 # if null delete reg
|
|
DeleteRegValue ${WriteEnvStr_RegKey} $0
|
|
|
|
SendMessage ${HWND_BROADCAST} ${WM_WININICHANGE} 0 "STR:Environment" /TIMEOUT=5000
|
|
|
|
unRemoveFromEnvVar_done:
|
|
Pop $7
|
|
Pop $6
|
|
Pop $5
|
|
Pop $4
|
|
Pop $3
|
|
Pop $2
|
|
Pop $1
|
|
Pop $0
|
|
FunctionEnd
|
|
|
|
|
|
|
|
|
|
!ifndef IsNT_KiCHiK
|
|
!define IsNT_KiCHiK
|
|
|
|
###########################################
|
|
# Utility Functions #
|
|
###########################################
|
|
|
|
; IsNT
|
|
; no input
|
|
; output, top of the stack = 1 if NT or 0 if not
|
|
;
|
|
; Usage:
|
|
; Call IsNT
|
|
; Pop $R0
|
|
; ($R0 at this point is 1 or 0)
|
|
|
|
!macro IsNT un
|
|
Function ${un}IsNT
|
|
Push $0
|
|
ReadRegStr $0 HKLM "SOFTWARE\Microsoft\Windows NT\CurrentVersion" CurrentVersion
|
|
StrCmp $0 "" 0 IsNT_yes
|
|
; we are not NT.
|
|
Pop $0
|
|
Push 0
|
|
Return
|
|
|
|
IsNT_yes:
|
|
; NT!!!
|
|
Pop $0
|
|
Push 1
|
|
FunctionEnd
|
|
!macroend
|
|
!insertmacro IsNT ""
|
|
!insertmacro IsNT "un."
|
|
|
|
!endif ; IsNT_KiCHiK
|
|
|
|
; StrStr
|
|
; input, top of stack = string to search for
|
|
; top of stack-1 = string to search in
|
|
; output, top of stack (replaces with the portion of the string remaining)
|
|
; modifies no other variables.
|
|
;
|
|
; Usage:
|
|
; Push "this is a long ass string"
|
|
; Push "ass"
|
|
; Call StrStr
|
|
; Pop $R0
|
|
; ($R0 at this point is "ass string")
|
|
|
|
!macro StrStr un
|
|
Function ${un}StrStr
|
|
Exch $R1 ; st=haystack,old$R1, $R1=needle
|
|
Exch ; st=old$R1,haystack
|
|
Exch $R2 ; st=old$R1,old$R2, $R2=haystack
|
|
Push $R3
|
|
Push $R4
|
|
Push $R5
|
|
StrLen $R3 $R1
|
|
StrCpy $R4 0
|
|
; $R1=needle
|
|
; $R2=haystack
|
|
; $R3=len(needle)
|
|
; $R4=cnt
|
|
; $R5=tmp
|
|
loop:
|
|
StrCpy $R5 $R2 $R3 $R4
|
|
StrCmp $R5 $R1 done
|
|
StrCmp $R5 "" done
|
|
IntOp $R4 $R4 + 1
|
|
Goto loop
|
|
done:
|
|
StrCpy $R1 $R2 "" $R4
|
|
Pop $R5
|
|
Pop $R4
|
|
Pop $R3
|
|
Pop $R2
|
|
Exch $R1
|
|
FunctionEnd
|
|
!macroend
|
|
!insertmacro StrStr ""
|
|
!insertmacro StrStr "un."
|
|
|
|
Function Trim ; Added by Pelaca
|
|
Exch $R1
|
|
Push $R2
|
|
Loop:
|
|
StrCpy $R2 "$R1" 1 -1
|
|
StrCmp "$R2" " " RTrim
|
|
StrCmp "$R2" "$\n" RTrim
|
|
StrCmp "$R2" "$\r" RTrim
|
|
StrCmp "$R2" ";" RTrim
|
|
GoTo Done
|
|
RTrim:
|
|
StrCpy $R1 "$R1" -1
|
|
Goto Loop
|
|
Done:
|
|
Pop $R2
|
|
Exch $R1
|
|
FunctionEnd
|
|
|
|
!endif ; _AddToPath_nsh
|