Client + Server: Set the current institute / custom server on adding

1 files changed, 17 insertions, 0 deletions

diff --git a/internal/server/secureinternet.go b/internal/server/secureinternet.go
index 776bb72..27d48a5 100644
--- a/internal/server/secureinternet.go
+++ b/internal/server/secureinternet.go
@@ -31,6 +31,23 @@ func (servers *Servers) GetSecureInternetHomeServer() (*SecureInternetHomeServer
 	return &servers.SecureInternetHomeServer, nil
 }
 
+func (servers *Servers) SetSecureInternet(server Server) error {
+	errorMessage := "failed setting secure internet server"
+	base, baseErr := server.GetBase()
+	if baseErr != nil {
+		return &types.WrappedErrorMessage{Message: errorMessage, Err: baseErr}
+	}
+
+	if base.Type != "secure_internet" {
+		return &types.WrappedErrorMessage{Message: errorMessage, Err: errors.New("Not a secure internet server")}
+	}
+
+	// The location should already be configured
+	// TODO: check for location?
+	servers.IsType = SecureInternetServerType
+	return nil
+}
+
 func (servers *Servers) RemoveSecureInternet() {
 	// Empty out the struct
 	servers.SecureInternetHomeServer = SecureInternetHomeServer{}