package processing

import (
	"encoding/json"

	resources "cloud.o-forge.io/core/oc-lib/models/resources"
	"cloud.o-forge.io/core/oc-lib/models/utils"
)

type ProcessingResource struct {
	resources.AbstractResource
	Container   string              `bson:"container,omitempty" json:"container,omitempty"`   // We could create a specific class for container, that could check if the name exists/is available
	Repository  string              `bson:"repository,omitempty" json:"repository,omitempty"` // Indicate where to find the container image	=> Could add a struct handling authentication to the repo
	Command     string              `bson:"command,omitempty" json:"command,omitempty"`
	Arguments   []string            `bson:"arguments,omitempty" json:"arguments,omitempty"`
	Environment []map[string]string `bson:"environment,omitempty" json:"environment,omitempty"` // a key/value struct is what ressembles the most a NAME=VALUE struct

	ExecutionRequirements ExecutionRequirementsModel `json:"execution_requirements,omitempty"`

	Price   uint   `bson:"price,omitempty" json:"price,omitempty"`
	License string `bson:"license,omitempty" json:"license,omitempty"`
}

type ExecutionRequirementsModel struct {
	CPUs uint `bson:"cp_us,omitempty" json:"cp_us,omitempty"`
	GPUs uint `bson:"gp_us,omitempty" json:"gp_us,omitempty"`
	RAM  uint `bson:"ram,omitempty" json:"ram,omitempty"`

	Parallel     bool   `bson:"parallel,omitempty" json:"parallel,omitempty"`
	ScalingModel uint   `bson:"scaling_model,omitempty" json:"scaling_model,omitempty"`
	DiskIO       string `bson:"disk_io,omitempty" json:"disk_io,omitempty"`
}

func (dma *ProcessingResource) Deserialize(j map[string]interface{}) utils.DBObject {
	b, err := json.Marshal(j)
	if err != nil {
		return nil
	}
	json.Unmarshal(b, dma)
	return dma
}

func (dma *ProcessingResource) Serialize() map[string]interface{} {
	var m map[string]interface{}
	b, err := json.Marshal(dma)
	if err != nil {
		return nil
	}
	json.Unmarshal(b, dma)
	return m
}

func (d *ProcessingResource) GetAccessor() utils.Accessor {
	data := &ProcessingMongoAccessor{}
	data.SetLogger(utils.PROCESSING_RESOURCE)
	return data
}