Refactor: Secure internet into a different type but with interface

1 files changed, 9 insertions, 3 deletions

diff --git a/internal/wireguard.go b/internal/wireguard.go
index 7977dbc..318e0dc 100644
--- a/internal/wireguard.go
+++ b/internal/wireguard.go
@@ -30,8 +30,14 @@ func wireguardConfigAddKey(config string, key wgtypes.Key) string {
 	return interface_re.ReplaceAllString(config, to_replace)
 }
 
-func (server *Server) WireguardGetConfig() (string, error) {
-	profile_id := server.Profiles.Current
+func WireguardGetConfig(server Server) (string, error) {
+	base, baseErr := server.GetBase()
+
+	if baseErr != nil {
+		return "", &WireguardGetConfigError{Err: baseErr}
+	}
+
+	profile_id := base.Profiles.Current
 	wireguardKey, wireguardErr := wireguardGenerateKey()
 
 	if wireguardErr != nil {
@@ -39,7 +45,7 @@ func (server *Server) WireguardGetConfig() (string, error) {
 	}
 
 	wireguardPublicKey := wireguardKey.PublicKey().String()
-	configWireguard, _, configErr := server.APIConnectWireguard(profile_id, wireguardPublicKey)
+	configWireguard, _, configErr := APIConnectWireguard(server, profile_id, wireguardPublicKey)
 
 	if configErr != nil {
 		return "", &WireguardGetConfigError{Err: wireguardErr}