add ssh supports(unfinished)

1 files changed, 24 insertions, 2 deletions

diff --git a/utils/conf.go b/utils/conf.go
index 0b1a3990..75bd2462 100644
--- a/utils/conf.go
+++ b/utils/conf.go
@@ -7,17 +7,39 @@ package utils
 import (
 	"fmt"
 	"os"
+	"os/exec"
+	"path"
+	"path/filepath"
 
 	"github.com/Unknwon/goconfig"
 )
 
 var Cfg *goconfig.ConfigFile
 
+func exeDir() (string, error) {
+	file, err := exec.LookPath(os.Args[0])
+	if err != nil {
+		return "", err
+	}
+	p, err := filepath.Abs(file)
+	if err != nil {
+		return "", err
+	}
+	return path.Dir(p), nil
+}
+
 func init() {
 	var err error
-	Cfg, err = goconfig.LoadConfigFile("conf/app.ini")
+	workDir, err := exeDir()
+	if err != nil {
+		fmt.Printf("Fail to get work directory: %s\n", err)
+		os.Exit(2)
+	}
+
+	cfgPath := filepath.Join(workDir, "conf", "app.ini")
+	Cfg, err = goconfig.LoadConfigFile(cfgPath)
 	if err != nil {
-		fmt.Println("Cannot load config file 'app.ini'")
+		fmt.Printf("Cannot load config file '%s'\n", cfgPath)
 		os.Exit(2)
 	}
 	Cfg.BlockMode = false