byop-engine/models/component.go

package models

import (
	"encoding/json"
	"time"

	"gorm.io/gorm"
)

type Component struct {
	ID          string        `json:"id" gorm:"primaryKey"`
	Name        string        `json:"name" gorm:"not null"`
	Description string        `json:"description"`
	Type        ComponentType `json:"type" gorm:"index"` // frontend, backend, api, database, or microservice
	Language    string        `json:"language"`          // Programming language or framework
	Version     string        `json:"version"`           // Version number (e.g., 1.0.0)

	// Configuration details
	ConfigFile   string `json:"configFile" gorm:"type:text"`   // JSON configuration as a string
	EnvVariables string `json:"envVariables" gorm:"type:text"` // Environment variables as a string

	// Source code details
	Repository   string `json:"repository"`                   // Git repository URL
	Branch       string `json:"branch" gorm:"default:'main'"` // Git branch (default: main)
	BuildCommand string `json:"buildCommand"`                 // Command to build the app

	// Resource allocation - stored as JSON
	ResourcesJSON     string `json:"-" gorm:"column:resources;type:text"`
	ScaleSettingsJSON string `json:"-" gorm:"column:scale_settings;type:text"`

	// Virtual fields for ORM serialization/deserialization
	Resources     ResourceRequirements `json:"resources" gorm:"-"`
	ScaleSettings ScaleSettings        `json:"scaleSettings" gorm:"-"`

	CreatedAt time.Time      `json:"createdAt" gorm:"autoCreateTime"`
	UpdatedAt time.Time      `json:"updatedAt" gorm:"autoUpdateTime"`
	CreatedBy string         `json:"createdBy" gorm:"index"` // User ID who created the component
	DeletedAt gorm.DeletedAt `json:"-" gorm:"index"`         // Soft delete support

	// Relationships
	Deployments []DeployedComponent `json:"deployments" gorm:"foreignKey:ComponentID"` // Components deployed in deployments
}

type ResourceRequirements struct {
	CPU     string `json:"cpu"`     // e.g., "0.5"
	Memory  string `json:"memory"`  // e.g., "512Mi"
	Storage string `json:"storage"` // e.g., "1Gi"
}

type ScaleSettings struct {
	MinInstances int `json:"minInstances"` // Minimum number of instances
	MaxInstances int `json:"maxInstances"` // Maximum number of instances
	CPUThreshold int `json:"cpuThreshold"` // CPU threshold for scaling (percentage)
}

// ComponentType represents the type of component
type ComponentType string

const (
	Frontend     ComponentType = "frontend"
	Backend      ComponentType = "backend"
	API          ComponentType = "api"
	Database     ComponentType = "database"
	Microservice ComponentType = "microservice"
)

// BeforeSave serializes the embedded JSON fields
func (c *Component) BeforeSave(tx *gorm.DB) error {
	var err error

	// Marshal Resources to JSON
	resourcesJSON, err := json.Marshal(c.Resources)
	if err != nil {
		return err
	}
	c.ResourcesJSON = string(resourcesJSON)

	// Marshal ScaleSettings to JSON
	scaleSettingsJSON, err := json.Marshal(c.ScaleSettings)
	if err != nil {
		return err
	}
	c.ScaleSettingsJSON = string(scaleSettingsJSON)

	return nil
}

// AfterFind deserializes the JSON fields
func (c *Component) AfterFind(tx *gorm.DB) error {
	// Unmarshal Resources from JSON
	if c.ResourcesJSON != "" {
		if err := json.Unmarshal([]byte(c.ResourcesJSON), &c.Resources); err != nil {
			return err
		}
	}

	// Unmarshal ScaleSettings from JSON
	if c.ScaleSettingsJSON != "" {
		if err := json.Unmarshal([]byte(c.ScaleSettingsJSON), &c.ScaleSettings); err != nil {
			return err
		}
	}

	return nil
}