1package main 2 3import ( 4 "flag" 5 "fmt" 6 "path/filepath" 7 "syscall" 8 "unsafe" 9) 10 11var ( 12 shell32 = syscall.NewLazyDLL("shell32.dll") 13 kernel32 = syscall.NewLazyDLL("kernel32.dll") 14) 15 16var ( 17 proc_sh_get_folder_path = shell32.NewProc("SHGetFolderPathW") 18 proc_get_module_file_name = kernel32.NewProc("GetModuleFileNameW") 19) 20 21func create_sock_flag(name, desc string) *string { 22 return flag.String(name, "tcp", desc) 23} 24 25// Full path of the current executable 26func get_executable_filename() string { 27 b := make([]uint16, syscall.MAX_PATH) 28 ret, _, err := syscall.Syscall(proc_get_module_file_name.Addr(), 3, 29 0, uintptr(unsafe.Pointer(&b[0])), uintptr(len(b))) 30 if int(ret) == 0 { 31 panic(fmt.Sprintf("GetModuleFileNameW : err %d", int(err))) 32 } 33 return syscall.UTF16ToString(b) 34} 35 36const ( 37 csidl_appdata = 0x1a 38) 39 40func get_appdata_folder_path() string { 41 b := make([]uint16, syscall.MAX_PATH) 42 ret, _, err := syscall.Syscall6(proc_sh_get_folder_path.Addr(), 5, 43 0, csidl_appdata, 0, 0, uintptr(unsafe.Pointer(&b[0])), 0) 44 if int(ret) != 0 { 45 panic(fmt.Sprintf("SHGetFolderPathW : err %d", int(err))) 46 } 47 return syscall.UTF16ToString(b) 48} 49 50func config_dir() string { 51 return filepath.Join(get_appdata_folder_path(), "gocode") 52} 53 54func config_file() string { 55 return filepath.Join(get_appdata_folder_path(), "gocode", "config.json") 56} 57