project.go

package transformers

import (
	"github.com/flyteorg/flyteadmin/pkg/repositories/models"
	"github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/admin"
	"github.com/golang/protobuf/proto"
)

type CreateProjectModelInput struct {
	Identifier  string
	Name        string
	Description string
}

func CreateProjectModel(project *admin.Project) models.Project {
	stateInt := int32(project.State)
	projectBytes, err := proto.Marshal(project)
	if err != nil {
		return models.Project{}
	}
	return models.Project{
		Identifier:  project.Id,
		Name:        project.Name,
		Description: project.Description,
		Labels:      projectBytes,
		State:       &stateInt,
	}
}

func FromProjectModel(projectModel models.Project, domains []*admin.Domain) admin.Project {
	projectDeserialized := &admin.Project{}
	err := proto.Unmarshal(projectModel.Labels, projectDeserialized)
	if err != nil {
		return admin.Project{}
	}
	project := admin.Project{
		Id:          projectModel.Identifier,
		Name:        projectModel.Name,
		Description: projectModel.Description,
		Labels:      projectDeserialized.Labels,
		State:       admin.Project_ProjectState(*projectModel.State),
	}
	project.Domains = domains
	return project
}

func FromProjectModels(projectModels []models.Project, domains []*admin.Domain) []*admin.Project {
	projects := make([]*admin.Project, len(projectModels))
	for index, projectModel := range projectModels {
		project := FromProjectModel(projectModel, domains)
		projects[index] = &project
	}
	return projects
}