Clean up the BambooHR module's API credentials loading

bamboohr/client.go

@@ -4,6 +4,8 @@ import (
 	"encoding/xml"
 	"fmt"
 	"os"
+
+	"github.com/senorprogrammer/wtf/wtf"
 )
 
 // A Client represents the data required to connect to the BambooHR API
@@ -16,11 +18,13 @@ type Client struct {
 // NewClient creates and returns a new BambooHR client
 func NewClient(url string) *Client {
 	client := Client{
-		apiBase:   url,
-		apiKey:    os.Getenv("WTF_BAMBOO_HR_TOKEN"),
-		subdomain: os.Getenv("WTF_BAMBOO_HR_SUBDOMAIN"),
+		apiBase: url,
+		//apiKey:    os.Getenv("WTF_BAMBOO_HR_TOKEN"),
+		//subdomain: os.Getenv("WTF_BAMBOO_HR_SUBDOMAIN"),
 	}
 
+	client.loadAPICredentials()
+
 	return &client
 }
 
@@ -60,3 +64,15 @@ func (client *Client) away(startDate, endDate string) (cal Calendar, err error)
 
 	return
 }
+
+func (client *Client) loadAPICredentials() {
+	client.apiKey = wtf.Config.UString(
+		"wtf.mods.bamboohr.apiKey",
+		os.Getenv("WTF_BAMBOO_HR_TOKEN"),
+	)
+
+	client.subdomain = wtf.Config.UString(
+		"wtf.mods.bamboohr.subdomain",
+		os.Getenv("WTF_BAMBOO_HR_SUBDOMAIN"),
+	)
+}