Add the get method

2024-04-30 09:47:00 +02:00
parent bbef71c512
commit 9f2489fc55
6 changed files with 110 additions and 32 deletions
--- a/internal/providers/common.go
+++ b/internal/providers/common.go
@ -15,11 +15,11 @@ type Providers interface {
 	RawServerLocation(string) (string, error)
 }

-func GetProvider(provider proto.Provider) (Providers, error) {
+func GetProvider(provider string) (Providers, error) {
 	switch provider {
-	case proto.Provider_PROVIDER_HETZNER:
+	case proto.Provider_PROVIDER_HETZNER.String(), "hetzner":
 		return &Hetzner{}, nil
 	default:
-		return nil, fmt.Errorf("unknown provider: %s", provider.String())
+		return nil, fmt.Errorf("unknown provider: %s", provider)
 	}
 }