From be0ba9ea88aff8a658d0495d36accf944b74888d Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Wed, 19 Feb 2014 17:50:53 +0800
Subject: add ssh supports(unfinished)

---
 utils/conf.go | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

(limited to 'utils')

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
-- 
cgit v1.2.3