Whois error handling

alex27riva · Dec 2, 2024 · b9864fc · b9864fc
1 parent 69fcf17
commit b9864fc
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/cmd/whois.go b/cmd/whois.go
@@ -52,8 +52,14 @@ var whoisCmd = &cobra.Command{
 			os.Exit(1)
 		}
 
-		whoisData := apis.GetWhoisData(target)
-		displayData(*whoisData)
+		whoisData, err := apis.GetWhoisData(target)
+		if err != nil {
+			color.Red("An error has occured.")
+		} else {
+			displayData(*whoisData)
+
+		}
+
 	},
 }
 

diff --git a/internal/apis/whodat.go b/internal/apis/whodat.go
@@ -8,7 +8,6 @@ package apis
 
 import (
 	"fmt"
-	"log"
 	"soc-cli/internal/util"
 )
 
@@ -64,15 +63,12 @@ type DomainInfo struct {
 	Technical      Contact   `json:"technical"`
 }
 
-func GetWhoisData(domain string) *DomainInfo {
+func GetWhoisData(domain string) (*DomainInfo, error) {
 	apiUrl := fmt.Sprintf(whodatAPIURL, domain)
 
 	var whois DomainInfo
 
 	err := util.MakeGETRequest(apiUrl, nil, &whois)
-	if err != nil {
-		log.Fatalf("Error fetching whodat API: %v", err)
-	}
 
-	return &whois
+	return &whois, err
 }