softplayer-backend/api/v1/environments.go

package v1

import (
	"context"
	"fmt"

	"git.badhouseplants.net/softplayer/softplayer-backend/internal/controllers"
	"git.badhouseplants.net/softplayer/softplayer-backend/internal/providers"
	proto "git.badhouseplants.net/softplayer/softplayer-go-proto/pkg/environments"
	"github.com/golang/protobuf/ptypes/empty"
	ctrl "sigs.k8s.io/controller-runtime"
)

func NewapiGrpcImpl(controller ctrl.Manager) *EnvironmentsServer {
	return &EnvironmentsServer{
		controller: controller,
	}
}

type EnvironmentsServer struct {
	proto.UnimplementedEnvironmentsServer
	controller ctrl.Manager
}

// Create an environment
func (e *EnvironmentsServer) Create(ctx context.Context, in *proto.CreateOptions) (*proto.EnvironmentFull, error) {
	provider, err := providers.GetProvider(in.GetSpec().GetProvider())
	if err != nil {
		return nil, err
	}

	var location string
	switch in.GetSpec().GetProvider() {
	case proto.Provider_PROVIDER_HETZNER:
		location, err = provider.GetServerLocation(in.GetSpec().GetHetznerOptions().GetServerLocation().String())
		if err != nil {
			return nil, err
		}
	default:
		return nil, fmt.Errorf("unknown provider: %s", in.GetSpec().GetProvider())
	}

	serverType, err := provider.GetServerType(in.Spec.HetznerOptions.ServerType)
	if err != nil {
		return nil, err
	}

	data := &controllers.EnvironemntData{
		Name:        in.GetMetadata().GetName(),
		Description: in.GetMetadata().GetDescription(),
		Provider:    in.GetSpec().GetProvider(),
		Kubernetes:  in.GetSpec().GetKubernetes().String(),
		Location:    location,
		ServerType:  serverType,
	}

	environment := &controllers.Environemnt{
		UserID:     in.GetOwnerId().GetUuid(),
		Controller: e.controller,
		Data:       data,
		Token:      in.GetToken().GetToken(),
	}

	err = environment.Create(ctx)
	if err != nil {
		return nil, err
	}
	return nil, nil
}

func (e *EnvironmentsServer) Update(ctx context.Context, in *proto.UpdateOptions) (*proto.EnvironmentFull, error) {
	return nil, nil
}

func (e *EnvironmentsServer) Delete(ctx context.Context, in *proto.DeleteOptions) (*empty.Empty, error) {
	data := &controllers.EnvironemntData{
		Name: in.GetMetadata().GetName(),
	}

	environment := &controllers.Environemnt{
		UserID:     in.GetOwnerId().GetUuid(),
		Controller: e.controller,
		Data:       data,
		Token:      in.GetToken().GetToken(),
	}
	err := environment.Create(ctx)
	if err != nil {
		return nil, err
	}
	return nil, nil
}

func (e *EnvironmentsServer) Get(ctx context.Context, in *proto.GetOptions) (*proto.EnvironmentFull, error) {
	data := &controllers.EnvironemntData{
		Name: in.GetMetadata().GetName(),
	}

	environment := &controllers.Environemnt{
		UserID:     in.GetOwnerId().GetUuid(),
		Controller: e.controller,
		Data:       data,
		Token:      in.GetToken().GetToken(),
	}

	if err := environment.Get(ctx); err != nil {
		return nil, err
	}

	return &proto.EnvironmentFull{
		Metadata: &proto.EnvironmentMetadata{
			Name: environment.Data.Name,
		},
	}, nil
}

func (e *EnvironmentsServer) List(in *proto.ListOptions, stream proto.Environments_ListServer) error {
	environment := &controllers.Environemnt{
		UserID:     in.GetOwnerId().GetUuid(),
		Controller: e.controller,
		Token:      in.GetToken().GetToken(),
	}

	envs, err := environment.ListEnvs(stream.Context())
	if err != nil {
		return err
	}

	for _, env := range envs {
		if err := stream.Send(&proto.EnvironmentFull{
			Metadata: &proto.EnvironmentMetadata{
				Name: env,
			},
		}); err != nil {
			return err
		}
	}
	return nil
}