diff options
Diffstat (limited to 'go/internal/config/config.go')
| -rw-r--r-- | go/internal/config/config.go | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/go/internal/config/config.go b/go/internal/config/config.go index 53de6f2..36a25c4 100644 --- a/go/internal/config/config.go +++ b/go/internal/config/config.go @@ -36,6 +36,7 @@ type Config struct { Secret string `yaml:"secret"` HttpSettings HttpSettingsConfig `yaml:"http_settings"` HttpClient *HttpClient + IPAddr string } func New() (*Config, error) { @@ -52,7 +53,10 @@ func NewFromDir(dir string) (*Config, error) { } func newFromFile(filename string) (*Config, error) { - cfg := &Config{RootDir: path.Dir(filename)} + cfg := &Config{ + RootDir: path.Dir(filename), + IPAddr: getIPAddr(), + } configBytes, err := ioutil.ReadFile(filename) if err != nil { @@ -123,7 +127,7 @@ func parseSecret(cfg *Config) error { return nil } -func (c *Config) IpAddr() string { +func getIPAddr() string { address := os.Getenv("SSH_CONNECTION") if address != "" { return strings.Fields(address)[0] |