Client: Refactor out adding a Server from getting a config

1 files changed, 9 insertions, 1 deletions

diff --git a/internal/server/secureinternet.go b/internal/server/secureinternet.go
index 3981022..776bb72 100644
--- a/internal/server/secureinternet.go
+++ b/internal/server/secureinternet.go
@@ -1,11 +1,12 @@
 package server
 
 import (
+	"errors"
 	"fmt"
 
 	"github.com/eduvpn/eduvpn-common/internal/oauth"
-	"github.com/eduvpn/eduvpn-common/types"
 	"github.com/eduvpn/eduvpn-common/internal/util"
+	"github.com/eduvpn/eduvpn-common/types"
 )
 
 // A secure internet server which has its own OAuth tokens
@@ -23,6 +24,13 @@ type SecureInternetHomeServer struct {
 	CurrentLocation       string `json:"current_location"`
 }
 
+func (servers *Servers) GetSecureInternetHomeServer() (*SecureInternetHomeServer, error) {
+	if !servers.HasSecureLocation() {
+		return nil, errors.New("No secure internet home server")
+	}
+	return &servers.SecureInternetHomeServer, nil
+}
+
 func (servers *Servers) RemoveSecureInternet() {
 	// Empty out the struct
 	servers.SecureInternetHomeServer = SecureInternetHomeServer{}