better ux for the domain cli arg

1 files changed, 6 insertions, 3 deletions

diff --git a/nccli.go b/nccli.go
index 4485f94..d907362 100644
--- a/nccli.go
+++ b/nccli.go
@@ -69,7 +69,7 @@ func getEnv(key string) (string, error) {
 // Load ENV vars into the apiUrl var.
 // Returns nil if config loaded successfully, error otherwise.
 func loadConfig() (string, error) {
-	keys := [5]string{"API_USER", "API_KEY", "ACCESS_IP", "DOMAIN_MAIN", "DOMAIN_END"}
+	keys := [4]string{"API_USER", "API_KEY", "ACCESS_IP", "DOMAIN"}
 	for _, key := range keys {
 		_, err := getEnv(key)
 		if err != nil {
@@ -79,8 +79,11 @@ func loadConfig() (string, error) {
 	user, _ := getEnv("API_USER")
 	apiKey, _ := getEnv("API_KEY")
 	accessIP, _ := getEnv("ACCESS_IP")
-	domainMain, _ := getEnv("DOMAIN_MAIN")
-	domainEnd, _ := getEnv("DOMAIN_END")
+
+	domain, _ := getEnv("DOMAIN")
+	domainSlices := strings.Split(domain, ".")
+	domainMain := domainSlices[0]
+	domainEnd := domainSlices[1]
 
 	apiUrl := "https://api.namecheap.com/xml.response?ApiUser=" + user +
 		"&ApiKey=" + apiKey +