oc-lib/models/utils/abstracts.go

108 lines
3.5 KiB
Go
Raw Normal View History

2024-07-18 11:51:12 +02:00
package utils
import (
2024-07-18 13:35:14 +02:00
"cloud.o-forge.io/core/oc-lib/dbs/mongo"
"cloud.o-forge.io/core/oc-lib/logs"
2024-07-18 18:14:12 +02:00
"github.com/go-playground/validator/v10"
2024-07-19 10:54:58 +02:00
"github.com/google/uuid"
2024-07-18 11:51:12 +02:00
"github.com/rs/zerolog"
)
2024-07-19 10:54:58 +02:00
var validate = validator.New(validator.WithRequiredStructEnabled())
type AbstractObject struct {
2024-07-22 16:39:26 +02:00
UUID string `json:"id,omitempty" bson:"id,omitempty" validate:"required"`
Name string `json:"name,omitempty" bson:"name,omitempty" validate:"required"`
2024-07-19 10:54:58 +02:00
}
2024-07-19 13:27:34 +02:00
func (ao *AbstractObject) GetID() string {
return ao.UUID
}
2024-07-23 11:22:50 +02:00
func (ao *AbstractObject) GetName() string {
return ao.Name
}
2024-07-19 10:54:58 +02:00
func (r *AbstractObject) GenerateID() {
r.UUID = uuid.New().String()
}
2024-07-18 18:14:12 +02:00
2024-07-18 11:51:12 +02:00
type AbstractAccessor struct {
Logger zerolog.Logger
2024-07-18 16:46:54 +02:00
Type string
2024-07-18 11:51:12 +02:00
}
2024-07-18 16:46:54 +02:00
func (dma *AbstractAccessor) GetType() string {
return dma.Type
}
2024-07-19 10:54:58 +02:00
func (dma *AbstractAccessor) SetLogger(t DataType) {
2024-07-18 17:55:27 +02:00
dma.Logger = logs.CreateLogger(t.String(), "")
2024-07-18 16:46:54 +02:00
dma.Type = t.String()
2024-07-18 11:51:12 +02:00
}
2024-07-19 10:54:58 +02:00
2024-07-19 11:27:58 +02:00
func (wfa *AbstractAccessor) GenericStoreOne(data DBObject, accessor Accessor) (DBObject, int, error) {
2024-07-19 11:00:15 +02:00
data.GenerateID()
2024-07-19 10:54:58 +02:00
err := validate.Struct(data)
if err != nil {
2024-07-19 11:27:58 +02:00
return nil, 422, err
2024-07-19 10:54:58 +02:00
}
2024-07-22 14:02:48 +02:00
id, code, err := mongo.MONGOService.StoreOne(data, data.GetID(), wfa.GetType())
2024-07-19 10:54:58 +02:00
if err != nil {
wfa.Logger.Error().Msg("Could not store " + data.GetName() + " to db. Error: " + err.Error())
2024-07-19 11:27:58 +02:00
return nil, code, err
2024-07-19 10:54:58 +02:00
}
return accessor.LoadOne(id)
}
2024-07-19 11:27:58 +02:00
func (dma *AbstractAccessor) GenericDeleteOne(id string, accessor Accessor) (DBObject, int, error) {
res, code, err := accessor.LoadOne(id)
2024-07-18 15:35:30 +02:00
if err != nil {
dma.Logger.Error().Msg("Could not retrieve " + id + " to db. Error: " + err.Error())
2024-07-19 11:27:58 +02:00
return nil, code, err
2024-07-18 15:35:30 +02:00
}
2024-07-19 11:27:58 +02:00
_, code, err = mongo.MONGOService.DeleteOne(id, accessor.GetType())
2024-07-18 11:51:12 +02:00
if err != nil {
dma.Logger.Error().Msg("Could not delete " + id + " to db. Error: " + err.Error())
2024-07-19 11:27:58 +02:00
return nil, code, err
2024-07-18 11:51:12 +02:00
}
2024-07-19 11:27:58 +02:00
return res, 200, nil
2024-07-18 11:51:12 +02:00
}
2024-07-24 14:49:04 +02:00
func (dma *AbstractAccessor) GenericUpdateOne(set DBObject, id string, accessor Accessor, new DBObject) (DBObject, int, error) {
2024-07-23 08:40:24 +02:00
r, c, err := accessor.LoadOne(id)
if err != nil {
return nil, c, err
}
2024-07-24 14:49:04 +02:00
change := set.Serialize()
loaded := r.Serialize()
for k, v := range change {
loaded[k] = v
}
id, code, err := mongo.MONGOService.UpdateOne(new.Deserialize(loaded), id, accessor.GetType())
2024-07-18 11:51:12 +02:00
if err != nil {
dma.Logger.Error().Msg("Could not update " + id + " to db. Error: " + err.Error())
2024-07-19 11:27:58 +02:00
return nil, code, err
2024-07-18 11:51:12 +02:00
}
return accessor.LoadOne(id)
}
2024-07-26 10:36:23 +02:00
type AbstractResource struct {
AbstractObject
ShortDescription string `json:"short_description,omitempty" bson:"short_description,omitempty" validate:"required"`
Description string `json:"description,omitempty" bson:"description,omitempty"`
Logo string `json:"logo,omitempty" bson:"logo,omitempty" validate:"required"`
Owner string `json:"owner,omitempty" bson:"owner,omitempty" validate:"required"`
OwnerLogo string `json:"owner_logo,omitempty" bson:"owner_logo,omitempty"`
SourceUrl string `json:"source_url,omitempty" bson:"source_url,omitempty" validate:"required"`
Proxy *ResourceProxy `json:"proxy,omitempty" bson:"proxy,omitempty"`
}
type ResourceProxy struct {
Host string `json:"host,omitempty" bson:"host,omitempty"`
Port int `json:"port,omitempty" bson:"port,omitempty"`
Command string `json:"command,omitempty" bson:"command,omitempty"`
Args []string `json:"args,omitempty" bson:"args,omitempty"`
EnvArgs map[string]interface{} `json:"env_args,omitempty" bson:"env_args,omitempty"`
}