2020-05-26 09:30:20 +03:00
|
|
|
import os
|
|
|
|
import v.pref
|
2021-02-18 19:11:09 +03:00
|
|
|
import v.util
|
2020-05-26 09:30:20 +03:00
|
|
|
|
2020-06-19 13:54:56 +03:00
|
|
|
$if windows {
|
|
|
|
$if tinyc {
|
|
|
|
#flag -lAdvapi32
|
|
|
|
#flag -lUser32
|
|
|
|
}
|
|
|
|
}
|
2020-10-15 16:17:52 +03:00
|
|
|
fn main() {
|
2021-02-18 19:11:09 +03:00
|
|
|
C.atexit(cleanup_vtmp_folder)
|
2020-10-22 17:28:58 +03:00
|
|
|
vexe := os.real_path(pref.vexe_path())
|
2020-05-26 09:30:20 +03:00
|
|
|
$if windows {
|
2020-07-16 19:33:26 +03:00
|
|
|
setup_symlink_windows(vexe)
|
2020-05-26 13:34:01 +03:00
|
|
|
} $else {
|
2021-02-18 19:11:09 +03:00
|
|
|
setup_symlink_unix(vexe)
|
2020-05-26 09:30:20 +03:00
|
|
|
}
|
2020-05-26 13:34:01 +03:00
|
|
|
}
|
|
|
|
|
2021-02-18 19:11:09 +03:00
|
|
|
fn cleanup_vtmp_folder() {
|
|
|
|
os.rmdir_all(util.get_vtmp_folder()) or { }
|
|
|
|
}
|
|
|
|
|
|
|
|
fn setup_symlink_unix(vexe string) {
|
2021-02-22 14:08:41 +03:00
|
|
|
mut link_path := '/data/data/com.termux/files/usr/bin/v'
|
|
|
|
if os.system("uname -o | grep -q '[A/a]ndroid'") == 1 {
|
|
|
|
link_dir := '/usr/local/bin'
|
|
|
|
if !os.exists(link_dir) {
|
2021-03-01 02:18:14 +03:00
|
|
|
os.mkdir_all(link_dir) or { panic(err) }
|
2021-02-22 14:08:41 +03:00
|
|
|
}
|
|
|
|
link_path = link_dir + '/v'
|
2020-10-16 10:55:55 +03:00
|
|
|
}
|
2021-03-08 21:52:13 +03:00
|
|
|
ret := os.execute_or_panic('ln -sf $vexe $link_path')
|
2020-05-26 09:30:20 +03:00
|
|
|
if ret.exit_code == 0 {
|
|
|
|
println('Symlink "$link_path" has been created')
|
|
|
|
} else {
|
2020-10-22 17:28:58 +03:00
|
|
|
eprintln('Failed to create symlink "$link_path". Try again with sudo.')
|
2020-05-26 09:30:20 +03:00
|
|
|
}
|
|
|
|
}
|
2020-05-26 13:34:01 +03:00
|
|
|
|
2020-10-15 16:17:52 +03:00
|
|
|
fn setup_symlink_windows(vexe string) {
|
2020-06-08 10:19:31 +03:00
|
|
|
$if windows {
|
2020-07-16 19:33:26 +03:00
|
|
|
// Create a symlink in a new local folder (.\.bin\.v.exe)
|
|
|
|
// Puts `v` in %PATH% without polluting it with anything else (like make.bat).
|
|
|
|
// This will make `v` available on cmd.exe, PowerShell, and MinGW(MSYS)/WSL/Cygwin
|
2020-10-15 16:17:52 +03:00
|
|
|
vdir := os.real_path(os.dir(vexe))
|
|
|
|
vsymlinkdir := os.join_path(vdir, '.bin')
|
2020-07-16 19:33:26 +03:00
|
|
|
mut vsymlink := os.join_path(vsymlinkdir, 'v.exe')
|
2021-02-27 10:39:36 +03:00
|
|
|
// Remove old symlink first (v could have been moved, symlink rerun)
|
2020-07-16 19:33:26 +03:00
|
|
|
if !os.exists(vsymlinkdir) {
|
2021-03-01 02:18:14 +03:00
|
|
|
os.mkdir(vsymlinkdir) or { panic(err) }
|
2021-02-27 10:39:36 +03:00
|
|
|
} else {
|
|
|
|
if os.exists(vsymlink) {
|
2021-03-01 02:18:14 +03:00
|
|
|
os.rm(vsymlink) or { panic(err) }
|
2021-02-27 10:39:36 +03:00
|
|
|
} else {
|
|
|
|
vsymlink = os.join_path(vsymlinkdir, 'v.bat')
|
|
|
|
if os.exists(vsymlink) {
|
2021-03-01 02:18:14 +03:00
|
|
|
os.rm(vsymlink) or { panic(err) }
|
2021-02-27 10:39:36 +03:00
|
|
|
}
|
|
|
|
vsymlink = os.join_path(vsymlinkdir, 'v.exe')
|
|
|
|
}
|
2020-06-08 10:19:31 +03:00
|
|
|
}
|
2020-10-22 17:28:58 +03:00
|
|
|
// First, try to create a native symlink at .\.bin\v.exe
|
2020-07-16 19:33:26 +03:00
|
|
|
os.symlink(vsymlink, vexe) or {
|
|
|
|
// typically only fails if you're on a network drive (VirtualBox)
|
|
|
|
// do batch file creation instead
|
2020-10-22 17:28:58 +03:00
|
|
|
eprintln('Could not create a native symlink: $err')
|
2020-07-16 19:33:26 +03:00
|
|
|
eprintln('Creating a batch file instead...')
|
|
|
|
vsymlink = os.join_path(vsymlinkdir, 'v.bat')
|
|
|
|
if os.exists(vsymlink) {
|
2021-03-01 02:18:14 +03:00
|
|
|
os.rm(vsymlink) or { panic(err) }
|
2020-06-08 10:19:31 +03:00
|
|
|
}
|
2021-03-01 02:18:14 +03:00
|
|
|
os.write_file(vsymlink, '@echo off\n$vexe %*') or { panic(err) }
|
2020-10-22 17:28:58 +03:00
|
|
|
eprintln('$vsymlink file written.')
|
2020-07-16 19:33:26 +03:00
|
|
|
}
|
|
|
|
if !os.exists(vsymlink) {
|
|
|
|
warn_and_exit('Could not create $vsymlink')
|
2020-06-08 10:19:31 +03:00
|
|
|
}
|
2020-10-22 17:28:58 +03:00
|
|
|
println('Symlink $vsymlink to $vexe created.')
|
|
|
|
println('Checking system %PATH%...')
|
2020-10-15 16:17:52 +03:00
|
|
|
reg_sys_env_handle := get_reg_sys_env_handle() or {
|
2021-02-28 23:20:21 +03:00
|
|
|
warn_and_exit(err.msg)
|
2020-06-08 10:19:31 +03:00
|
|
|
return
|
|
|
|
}
|
2020-09-19 23:26:32 +03:00
|
|
|
// TODO: Fix defers inside ifs
|
|
|
|
// defer {
|
2020-10-15 16:17:52 +03:00
|
|
|
// C.RegCloseKey(reg_sys_env_handle)
|
2020-09-19 23:26:32 +03:00
|
|
|
// }
|
2020-07-04 20:52:25 +03:00
|
|
|
// if the above succeeded, and we cannot get the value, it may simply be empty
|
2021-01-26 17:43:10 +03:00
|
|
|
sys_env_path := get_reg_value(reg_sys_env_handle, 'Path') or { '' }
|
2020-06-08 10:19:31 +03:00
|
|
|
current_sys_paths := sys_env_path.split(os.path_delimiter).map(it.trim('/$os.path_separator'))
|
2020-10-15 16:17:52 +03:00
|
|
|
mut new_paths := [vsymlinkdir]
|
2020-06-08 10:19:31 +03:00
|
|
|
for p in current_sys_paths {
|
2020-10-22 17:28:58 +03:00
|
|
|
if p == '' {
|
|
|
|
continue
|
|
|
|
}
|
2020-06-08 10:19:31 +03:00
|
|
|
if p !in new_paths {
|
|
|
|
new_paths << p
|
|
|
|
}
|
|
|
|
}
|
|
|
|
new_sys_env_path := new_paths.join(';')
|
|
|
|
if new_sys_env_path == sys_env_path {
|
2020-10-22 17:28:58 +03:00
|
|
|
println('System %PATH% was already configured.')
|
2020-10-15 16:17:52 +03:00
|
|
|
} else {
|
2020-10-22 17:28:58 +03:00
|
|
|
println('System %PATH% was not configured.')
|
|
|
|
println('Adding symlink directory to system %PATH%...')
|
2020-06-08 10:19:31 +03:00
|
|
|
set_reg_value(reg_sys_env_handle, 'Path', new_sys_env_path) or {
|
2020-09-19 23:26:32 +03:00
|
|
|
C.RegCloseKey(reg_sys_env_handle)
|
2021-02-28 23:20:21 +03:00
|
|
|
warn_and_exit(err.msg)
|
2020-06-08 10:19:31 +03:00
|
|
|
}
|
2021-02-27 10:39:36 +03:00
|
|
|
println('Done.')
|
2020-06-08 10:19:31 +03:00
|
|
|
}
|
2020-10-22 17:28:58 +03:00
|
|
|
println('Notifying running processes to update their Environment...')
|
2020-06-08 10:19:31 +03:00
|
|
|
send_setting_change_msg('Environment') or {
|
2020-10-22 17:28:58 +03:00
|
|
|
eprintln(err)
|
2020-09-19 23:26:32 +03:00
|
|
|
C.RegCloseKey(reg_sys_env_handle)
|
2021-02-27 10:39:36 +03:00
|
|
|
warn_and_exit('You might need to run this again to have the `v` command in your %PATH%')
|
2020-06-08 10:19:31 +03:00
|
|
|
}
|
2020-09-19 23:26:32 +03:00
|
|
|
C.RegCloseKey(reg_sys_env_handle)
|
2021-02-27 10:39:36 +03:00
|
|
|
println('Done.')
|
|
|
|
println('Note: Restart your shell/IDE to load the new %PATH%.')
|
|
|
|
println('After restarting your shell/IDE, give `v version` a try in another directory!')
|
2020-05-26 13:34:01 +03:00
|
|
|
}
|
2020-06-08 10:19:31 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
fn warn_and_exit(err string) {
|
|
|
|
eprintln(err)
|
|
|
|
exit(1)
|
|
|
|
}
|
|
|
|
|
|
|
|
// get the system environment registry handle
|
|
|
|
fn get_reg_sys_env_handle() ?voidptr {
|
2020-07-16 19:33:26 +03:00
|
|
|
$if windows { // wrap for cross-compile compat
|
2020-06-08 10:19:31 +03:00
|
|
|
// open the registry key
|
2020-10-15 16:17:52 +03:00
|
|
|
reg_key_path := 'Environment'
|
|
|
|
reg_env_key := voidptr(0) // or HKEY (HANDLE)
|
2020-07-16 19:33:26 +03:00
|
|
|
if C.RegOpenKeyEx(os.hkey_current_user, reg_key_path.to_wide(), 0, 1 | 2, ®_env_key) != 0 {
|
2020-06-08 10:19:31 +03:00
|
|
|
return error('Could not open "$reg_key_path" in the registry')
|
|
|
|
}
|
|
|
|
return reg_env_key
|
2020-05-26 13:34:01 +03:00
|
|
|
}
|
2020-06-08 10:19:31 +03:00
|
|
|
return error('not on windows')
|
|
|
|
}
|
|
|
|
|
|
|
|
// get a value from a given $key
|
|
|
|
fn get_reg_value(reg_env_key voidptr, key string) ?string {
|
|
|
|
$if windows {
|
|
|
|
// query the value (shortcut the sizing step)
|
|
|
|
reg_value_size := 4095 // this is the max length (not for the registry, but for the system %PATH%)
|
2021-02-15 18:15:52 +03:00
|
|
|
mut reg_value := unsafe { &u16(malloc(reg_value_size)) }
|
2020-06-08 10:19:31 +03:00
|
|
|
if C.RegQueryValueEx(reg_env_key, key.to_wide(), 0, 0, reg_value, ®_value_size) != 0 {
|
2021-02-27 10:39:36 +03:00
|
|
|
return error('Unable to get registry value for "$key".')
|
2020-05-26 13:34:01 +03:00
|
|
|
}
|
2021-02-15 18:15:52 +03:00
|
|
|
return unsafe { string_from_wide(reg_value) }
|
2020-05-26 13:34:01 +03:00
|
|
|
}
|
2020-06-08 10:19:31 +03:00
|
|
|
return error('not on windows')
|
|
|
|
}
|
|
|
|
|
|
|
|
// sets the value for the given $key to the given $value
|
|
|
|
fn set_reg_value(reg_key voidptr, key string, value string) ?bool {
|
|
|
|
$if windows {
|
2021-02-27 10:39:36 +03:00
|
|
|
if C.RegSetValueEx(reg_key, key.to_wide(), 0, C.REG_EXPAND_SZ, value.to_wide(),
|
|
|
|
value.len * 2) != 0 {
|
|
|
|
return error('Unable to set registry value for "$key". %PATH% may be too long.')
|
2020-06-08 10:19:31 +03:00
|
|
|
}
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
return error('not on windows')
|
|
|
|
}
|
|
|
|
|
2020-07-16 19:33:26 +03:00
|
|
|
// Broadcasts a message to all listening windows (explorer.exe in particular)
|
2020-06-08 10:19:31 +03:00
|
|
|
// letting them know that the system environment has changed and should be reloaded
|
|
|
|
fn send_setting_change_msg(message_data string) ?bool {
|
|
|
|
$if windows {
|
2021-02-18 19:11:09 +03:00
|
|
|
if C.SendMessageTimeout(os.hwnd_broadcast, os.wm_settingchange, 0, message_data.to_wide(),
|
|
|
|
os.smto_abortifhung, 5000, 0) == 0 {
|
2020-06-08 10:19:31 +03:00
|
|
|
return error('Could not broadcast WM_SETTINGCHANGE')
|
|
|
|
}
|
|
|
|
return true
|
2020-05-26 13:34:01 +03:00
|
|
|
}
|
2020-06-08 10:19:31 +03:00
|
|
|
return error('not on windows')
|
2020-05-26 13:34:01 +03:00
|
|
|
}
|