softplayer/softplayer-backend
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Activity

Refactor providers data

Browse Source
This commit is contained in:
Nikolai Rodionov
2024-04-29 20:03:37 +02:00
parent a8325fd202
commit 87e204ed46
8 changed files with 123 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
internal/controllers/environments.go
View File

@@ -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
}

21
internal/kubernetes/common.go Normal file
View File

@@ -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())
}
}

15
internal/kubernetes/k3s.go Normal file
View File

@@ -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()
}

6
internal/providers/common.go
View File

@@ -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) {

76
internal/providers/hetzner.go
View File

@@ -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
}
}