diff --git a/gitlab/widget.go b/gitlab/widget.go index 861e8ec3..3033e120 100644 --- a/gitlab/widget.go +++ b/gitlab/widget.go @@ -32,13 +32,8 @@ type Widget struct { } func NewWidget(app *tview.Application, pages *tview.Pages) *Widget { - apiKey := wtf.Config.UString( - "wtf.mods.gitlab.apiKey", - os.Getenv("WTF_GITLAB_TOKEN"), - ) - baseURL := wtf.Config.UString("wtf.mods.gitlab.domain") - gitlab := glb.NewClient(nil, apiKey) + gitlab := glb.NewClient(nil, apiKey()) if baseURL != "" { gitlab.SetBaseURL(baseURL) @@ -92,6 +87,13 @@ func (widget *Widget) Prev() { /* -------------------- Unexported Functions -------------------- */ +func apiKey() string { + return wtf.Config.UString( + "wtf.mods.gitlab.apiKey", + os.Getenv("WTF_GITLAB_TOKEN"), + ) +} + func (widget *Widget) buildProjectCollection(projectData map[string]interface{}) []*GitlabProject { gitlabProjects := []*GitlabProject{} diff --git a/logger/log.go b/logger/log.go index 69b5fceb..b654779d 100644 --- a/logger/log.go +++ b/logger/log.go @@ -36,7 +36,7 @@ func Log(msg string) { return } - f, err := os.OpenFile(logFilePath(), os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666) + f, err := os.OpenFile(logFilePath(), os.O_RDWR|os.O_CREATE|os.O_APPEND, 0600) if err != nil { log.Fatalf("error opening file: %v", err) } diff --git a/main.go b/main.go index b5da66cb..32bbe493 100644 --- a/main.go +++ b/main.go @@ -130,7 +130,10 @@ func refreshAllWidgets() { } func setTerm() { - os.Setenv("TERM", Config.UString("wtf.term", os.Getenv("TERM"))) + err := os.Setenv("TERM", Config.UString("wtf.term", os.Getenv("TERM"))) + if err != nil { + return + } } func watchForConfigChanges(app *tview.Application, configFilePath string, grid *tview.Grid, pages *tview.Pages) { diff --git a/travisci/client.go b/travisci/client.go index 6a88e5b3..dbd906fc 100644 --- a/travisci/client.go +++ b/travisci/client.go @@ -13,8 +13,6 @@ import ( "github.com/senorprogrammer/wtf/wtf" ) -const APIEnvToken = "WTF_TRAVIS_API_TOKEN" - var TRAVIS_HOSTS = map[bool]string{ false: "travis-ci.org", true: "travis-ci.com", @@ -75,7 +73,7 @@ func travisRequest(path string) (*http.Response, error) { func apiToken() string { return wtf.Config.UString( "wtf.mods.travisci.apiKey", - os.Getenv(APIEnvToken), + os.Getenv("WTF_TRAVIS_API_TOKEN"), ) }