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