Update the env proto
Update the env proto
This commit is contained in:
		@@ -2,12 +2,10 @@ package v1
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"errors"
 | 
			
		||||
 | 
			
		||||
	"git.badhouseplants.net/softplayer/softplayer-backend/internal/controllers"
 | 
			
		||||
	proto "git.badhouseplants.net/softplayer/softplayer-go-proto/pkg/environments"
 | 
			
		||||
	"github.com/golang/protobuf/ptypes/empty"
 | 
			
		||||
	"google.golang.org/grpc/metadata"
 | 
			
		||||
	ctrl "sigs.k8s.io/controller-runtime"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
@@ -22,31 +20,41 @@ type EnvironmentsServer struct {
 | 
			
		||||
	controller ctrl.Manager
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (e *EnvironmentsServer) Create(ctx context.Context, in *proto.EnvironmentData) (*proto.EnvironmentFull, error) {
 | 
			
		||||
	md, ok := metadata.FromIncomingContext(ctx)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, errors.New("metadata is not provided")
 | 
			
		||||
	}
 | 
			
		||||
	token, ok := md["token"]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, errors.New("token is not sent via metadata")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	uuid, ok := md["uuid"]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, errors.New("used id is not sent via metadata")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
// Create an environment
 | 
			
		||||
func (e *EnvironmentsServer) Create(ctx context.Context, in *proto.CreateOptions) (*proto.EnvironmentFull, error) {
 | 
			
		||||
	data := &controllers.EnvironemntData{
 | 
			
		||||
		Name:       in.GetName(),
 | 
			
		||||
		Provider:   in.GetProvider().String(),
 | 
			
		||||
		Kubernetes: in.GetKubernetes().String(),
 | 
			
		||||
		Name:       in.GetName().GetName(),
 | 
			
		||||
		Provider:   in.GetData().GetProvider().String(),
 | 
			
		||||
		Kubernetes: in.GetData().GetKubernetes().String(),
 | 
			
		||||
	}
 | 
			
		||||
	environment := &controllers.Environemnt{
 | 
			
		||||
		UserID:     uuid[0],
 | 
			
		||||
		UserID:     in.GetOwnerId().GetUuid(),
 | 
			
		||||
		Controller: e.controller,
 | 
			
		||||
		Data:       data,
 | 
			
		||||
		Token:      token[0],
 | 
			
		||||
		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.GetName().GetName(),
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	environment := &controllers.Environemnt{
 | 
			
		||||
		UserID:     in.GetOwnerId().GetUuid(),
 | 
			
		||||
		Controller: e.controller,
 | 
			
		||||
		Data:       data,
 | 
			
		||||
		Token:      in.GetToken().GetToken(),
 | 
			
		||||
	}
 | 
			
		||||
	err := environment.Create(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
@@ -55,62 +63,16 @@ func (e *EnvironmentsServer) Create(ctx context.Context, in *proto.EnvironmentDa
 | 
			
		||||
	return nil, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (e *EnvironmentsServer) Update(ctx context.Context, in *proto.EnvironmentFull) (*proto.EnvironmentFull, error) {
 | 
			
		||||
	return nil, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (e *EnvironmentsServer) Delete(ctx context.Context, in *proto.EnvironmentFull) (*empty.Empty, error) {
 | 
			
		||||
	md, ok := metadata.FromIncomingContext(ctx)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, errors.New("metadata is not provided")
 | 
			
		||||
	}
 | 
			
		||||
	token, ok := md["token"]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, errors.New("token is not sent via metadata")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	uuid, ok := md["uuid"]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, errors.New("used id is not sent via metadata")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
func (e *EnvironmentsServer) Get(ctx context.Context, in *proto.GetOptions) (*proto.EnvironmentFull, error) {
 | 
			
		||||
	data := &controllers.EnvironemntData{
 | 
			
		||||
		Name:       in.Data.GetName(),
 | 
			
		||||
		Provider:   in.Data.GetProvider().String(),
 | 
			
		||||
		Kubernetes: in.Data.GetKubernetes().String(),
 | 
			
		||||
		Name: in.GetName().GetName(),
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	environment := &controllers.Environemnt{
 | 
			
		||||
		UserID:     uuid[0],
 | 
			
		||||
		UserID:     in.GetOwnerId().GetUuid(),
 | 
			
		||||
		Controller: e.controller,
 | 
			
		||||
		Data:       data,
 | 
			
		||||
		Token:      token[0],
 | 
			
		||||
	}
 | 
			
		||||
	err := environment.Create(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return nil, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (e *EnvironmentsServer) Get(ctx context.Context, in *proto.EnvironmentId) (*proto.EnvironmentFull, error) {
 | 
			
		||||
	md, ok := metadata.FromIncomingContext(ctx)
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, errors.New("metadata is not provided")
 | 
			
		||||
	}
 | 
			
		||||
	token, ok := md["token"]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, errors.New("token is not sent via metadata")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	uuid, ok := md["uuid"]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return nil, errors.New("used id is not sent via metadata")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	environment := &controllers.Environemnt{
 | 
			
		||||
		UserID:     uuid[0],
 | 
			
		||||
		Controller: e.controller,
 | 
			
		||||
		Token:      token[0],
 | 
			
		||||
		Token:      in.GetToken().GetToken(),
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if err := environment.Get(ctx); err != nil {
 | 
			
		||||
@@ -118,35 +80,19 @@ func (e *EnvironmentsServer) Get(ctx context.Context, in *proto.EnvironmentId) (
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return &proto.EnvironmentFull{
 | 
			
		||||
		Id: &proto.EnvironmentId{
 | 
			
		||||
			Id: "test",
 | 
			
		||||
		},
 | 
			
		||||
		Data: &proto.EnvironmentData{
 | 
			
		||||
		Name: &proto.EnvironmentName{
 | 
			
		||||
			Name: environment.Data.Name,
 | 
			
		||||
		},
 | 
			
		||||
	}, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (e *EnvironmentsServer) List(in *empty.Empty, stream proto.Environments_ListServer) error {
 | 
			
		||||
	md, ok := metadata.FromIncomingContext(stream.Context())
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return errors.New("metadata is not provided")
 | 
			
		||||
	}
 | 
			
		||||
	token, ok := md["token"]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return errors.New("token is not sent via metadata")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	uuid, ok := md["uuid"]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		return errors.New("used id is not sent via metadata")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
func (e *EnvironmentsServer) List(in *proto.ListOptions, stream proto.Environments_ListServer) error {
 | 
			
		||||
	environment := &controllers.Environemnt{
 | 
			
		||||
		UserID:     uuid[0],
 | 
			
		||||
		UserID:     in.GetOwnerId().GetUuid(),
 | 
			
		||||
		Controller: e.controller,
 | 
			
		||||
		Token:      token[0],
 | 
			
		||||
		Token:      in.GetToken().GetToken(),
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	envs, err := environment.ListEnvs(stream.Context())
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
@@ -154,8 +100,7 @@ func (e *EnvironmentsServer) List(in *empty.Empty, stream proto.Environments_Lis
 | 
			
		||||
 | 
			
		||||
	for _, env := range envs {
 | 
			
		||||
		if err := stream.Send(&proto.EnvironmentFull{
 | 
			
		||||
			Id: &proto.EnvironmentId{Id: "test"},
 | 
			
		||||
			Data: &proto.EnvironmentData{
 | 
			
		||||
			Name: &proto.EnvironmentName{
 | 
			
		||||
				Name: env,
 | 
			
		||||
			},
 | 
			
		||||
		}); err != nil {
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user