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