Refactor providers data

internal/controllers/environments.go

@@ -9,7 +9,6 @@ import (
 	"github.com/joho/godotenv"
 
 	"git.badhouseplants.net/softplayer/softplayer-backend/internal/helpers/kube"
-	proto "git.badhouseplants.net/softplayer/softplayer-go-proto/pkg/environments"
 	corev1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/types"
@@ -28,7 +27,7 @@ type Environemnt struct {
 type EnvironemntData struct {
 	Name        string
 	Description string
-	Provider    proto.Provider
+	Provider    string
 	Kubernetes  string
 	Location    string
 	ServerType  string
@@ -58,7 +57,7 @@ func (env *Environemnt) isNsVerified(ctx context.Context) error {
 
 	val, ok := ns.GetLabels()["email-verified"]
 	if !ok || val == "false" {
-		return errors.New("User email is not verified, can't create an new env")
+		return errors.New("user email is not verified, can't create an new env")
 	}
 	return nil
 }

internal/kubernetes/common.go

@@ -0,0 +1,21 @@
+package kubernetes
+
+import (
+	"fmt"
+
+	proto "git.badhouseplants.net/softplayer/softplayer-go-proto/pkg/environments"
+)
+
+type Kubernetes interface {
+	GetKubernetesName() string
+	RawKubernetesName() string
+}
+
+func GetKubernetes(k8s proto.Kubernetes) (Kubernetes, error) {
+	switch k8s {
+	case proto.Kubernetes_KUBERNETES_K3S:
+		return &K3s{}, nil
+	default:
+		return nil, fmt.Errorf("unknown provider: %s", k8s.String())
+	}
+}

internal/kubernetes/k3s.go

@@ -0,0 +1,15 @@
+package kubernetes
+
+import (
+	proto "git.badhouseplants.net/softplayer/softplayer-go-proto/pkg/environments"
+)
+
+type K3s struct{}
+
+func (k *K3s) GetKubernetesName() string {
+	return "k3s"
+}
+
+func (k *K3s) RawKubernetesName() string {
+	return proto.Kubernetes_KUBERNETES_K3S.String()
+}

internal/providers/common.go

@@ -7,8 +7,12 @@ import (
 )
 
 type Providers interface {
-	GetServerType(proto.ServerType) (string, error)
+	GetProviderName() string
+	RawProviderName() string
+	GetServerType(string) (string, error)
 	GetServerLocation(string) (string, error)
+	RawServerType(string) (string, error)
+	RawServerLocation(string) (string, error)
 }
 
 func GetProvider(provider proto.Provider) (Providers, error) {

internal/providers/hetzner.go

@@ -1,47 +1,101 @@
 package providers
 
 import (
+	"errors"
 	"fmt"
+	"strings"
 
 	proto "git.badhouseplants.net/softplayer/softplayer-go-proto/pkg/environments"
 )
 
 type Hetzner struct{}
 
+// GetProviderName implements Providers.
+func (h *Hetzner) GetProviderName() string {
+	return "hetzner"
+}
+
+// RawProviderName implements Providers.
+func (h *Hetzner) RawProviderName() string {
+	proto.Provider_PROVIDER_HETZNER.String()
+}
+
+// RawServerLocation implements Providers.
+func (h *Hetzner) RawServerLocation(location string) (string, error) {
+	switch location {
+	case "ash":
+		return proto.Location_LOCATION_HETZNER_ASHBURN.String(), nil
+	case "hil":
+		return proto.Location_LOCATION_HETZNER_HILLSBORO.String(), nil
+	case "fsn1":
+		return proto.Location_LOCATION_HETZNER_FALKENSTEIN.String(), nil
+	case "nbg1":
+		return proto.Location_LOCATION_HETZNER_NUREMBERG.String(), nil
+	case "hel1":
+		return proto.Location_LOCATION_HETZNER_HELSINKI.String(), nil
+	default:
+		return "", fmt.Errorf("unknown location: %s", location)
+	}
+
+}
+
+// RawServerType implements Providers.
+func (h *Hetzner) RawServerType(kind string) (string, error) {
+	switch kind {
+	case "cpx21":
+		return proto.ServerType_SERVER_TYPE_STARTER.String(), nil
+	case "cpx31":
+		return proto.ServerType_SERVER_TYPE_REGULAR.String(), nil
+	case "cpx41":
+		return proto.ServerType_SERVER_TYPE_PLUS.String(), nil
+	case "cpx51":
+		return proto.ServerType_SERVER_TYPE_PRO.String(), nil
+	default:
+		err := fmt.Errorf("unknown server type: %s", kind)
+		return "", err
+	}
+}
+
 // GetServerLocation implements Providers.
 func (h *Hetzner) GetServerLocation(location string) (string, error) {
+	if !strings.HasPrefix(location, "HETZNER") {
+		return "", fmt.Errorf("location isn't supported by hetzner: %s", location)
+	}
 	switch location {
-	case proto.HetznerLocation_HETZNER_LOCATION_ASHBURN.String():
+	case proto.Location_LOCATION_HETZNER_ASHBURN.String():
 		return "ash", nil
-	case proto.HetznerLocation_HETZNER_LOCATION_HILLSBORO.String():
+	case proto.Location_LOCATION_HETZNER_HILLSBORO.String():
 		return "hil", nil
-	case proto.HetznerLocation_HETZNER_LOCATION_FALKENSTEIN.String():
+	case proto.Location_LOCATION_HETZNER_FALKENSTEIN.String():
 		return "fsn1", nil
-	case proto.HetznerLocation_HETZNER_LOCATION_NUREMBERG.String():
+	case proto.Location_LOCATION_HETZNER_NUREMBERG.String():
 		return "nbg1", nil
-	case proto.HetznerLocation_HETZNER_LOCATION_HELSINKI.String():
+	case proto.Location_LOCATION_HETZNER_HELSINKI.String():
 		return "hel1", nil
 	default:
 		return "", fmt.Errorf("unknown location: %s", location)
 	}
 }
 
-func (h *Hetzner) GetServerType(kind proto.ServerType) (serverType string, err error) {
+func (h *Hetzner) GetServerType(kind string) (serverType string, err error) {
 	switch kind {
-	case proto.ServerType_SERVER_TYPE_STARTER:
+	case proto.ServerType_SERVER_TYPE_STARTER.String():
 		serverType = "cpx21"
 		return
-	case proto.ServerType_SERVER_TYPE_REGULAR:
+	case proto.ServerType_SERVER_TYPE_REGULAR.String():
 		serverType = "cpx31"
 		return
-	case proto.ServerType_SERVER_TYPE_PLUS:
+	case proto.ServerType_SERVER_TYPE_PLUS.String():
 		serverType = "cpx41"
 		return
-	case proto.ServerType_SERVER_TYPE_PRO:
+	case proto.ServerType_SERVER_TYPE_PRO.String():
 		serverType = "cpx51"
 		return
+	case proto.ServerType_SERVER_TYPE_CUSTOM.String():
+		err = errors.New("custom server types are not supported yet")
+		return
 	default:
-		err = fmt.Errorf("unknown server type: %s", kind.String())
+		err = fmt.Errorf("unknown server type: %s", kind)
 		return
 	}
 }