oc-catalog/controllers/data.go

package controllers

import (
	"encoding/json"
	"oc-catalog/infrastructure"

	oclib "cloud.o-forge.io/core/oc-lib"
	"cloud.o-forge.io/core/oc-lib/tools"
	beego "github.com/beego/beego/v2/server/web"
)

// Operations about data
type DataController struct {
	beego.Controller
}

var data_collection = oclib.LibDataEnum(oclib.DATA_RESOURCE)
var data_dt = tools.DATA_RESOURCE

// @Title Update
// @Description create datas
// @Param	id		path 	string	true		"the data id you want to get"
// @Param	body		body 	models.data	true		"The data content"
// @Success 200 {data} models.data
// @router /:id [put]
func (o *DataController) Put() {
	// store and return Id or post with UUID
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	var res map[string]interface{}
	id := o.Ctx.Input.Param(":id")
	json.Unmarshal(o.Ctx.Input.CopyBody(10000), &res)
	data := oclib.NewRequest(data_collection, user, peerID, groups, nil).UpdateOne(res, id)
	if data.Err == "" {
		data, _ := json.Marshal(data.Data.Serialize(data.Data))
		infrastructure.EmitNATS(tools.PropalgationMessage{
			Action:   tools.PB_UPDATE,
			DataType: data_dt.EnumIndex(),
			User:     user,
			Payload:  data,
		})
	}
	o.Data["json"] = data
	o.ServeJSON()
}

// @Title Create
// @Description create data
// @Param	data		body 	json	true		"body for data content (Json format)"
// @Success 200 {data} models.data
// @router / [post]
func (o *DataController) Post() {
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	var res map[string]interface{}
	json.Unmarshal(o.Ctx.Input.CopyBody(10000), &res)
	data := oclib.NewRequest(data_collection, user, peerID, groups, nil).StoreOne(res)
	if data.Err == "" {
		data, _ := json.Marshal(data.Data.Serialize(data.Data))
		infrastructure.EmitNATS(tools.PropalgationMessage{
			Action:   tools.PB_CREATE,
			DataType: data_dt.EnumIndex(),
			User:     user,
			Payload:  data,
		})
	}
	o.Data["json"] = data
	o.ServeJSON()
}

// @Title GetAll
// @Description find data by id
// @Param	is_draft		query 	string	false		"draft wished"
// @Success 200 {data} models.data
// @router / [get]
func (o *DataController) GetAll() {
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	isDraft := o.Ctx.Input.Query("is_draft")
	o.Data["json"] = oclib.NewRequest(data_collection, user, peerID, groups, nil).LoadAll(isDraft == "true")
	o.ServeJSON()
}

// @Title Get
// @Description find data by key word
// @Param	search		path 	string	true		"the search you want to get"
// @Param	is_draft		query 	string	false		"draft wished"

// @Success 200 {data} models.data
// @router /search/:search [get]
func (o *DataController) Search() {
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	search := o.Ctx.Input.Param(":search")
	isDraft := o.Ctx.Input.Query("is_draft")
	o.Data["json"] = oclib.NewRequest(data_collection, user, peerID, groups, nil).Search(nil, search, isDraft == "true")
	o.ServeJSON()
}

// @Title Get
// @Description find data by id
// @Param	id		path 	string	true		"the id you want to get"
// @Success 200 {data} models.data
// @router /:id [get]
func (o *DataController) Get() {
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	id := o.Ctx.Input.Param(":id")
	o.Data["json"] = oclib.NewRequest(data_collection, user, peerID, groups, nil).LoadOne(id)
	o.ServeJSON()
}

// @Title Delete
// @Description delete the data
// @Param	id		path 	string	true		"The id you want to delete"
// @Success 200 {data} delete success!
// @router /:id [delete]
func (o *DataController) Delete() {
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	id := o.Ctx.Input.Param(":id")
	data := oclib.NewRequest(data_collection, user, peerID, groups, nil).DeleteOne(id)
	if data.Err == "" {
		data, _ := json.Marshal(data.Data.Serialize(data.Data))
		infrastructure.EmitNATS(tools.PropalgationMessage{
			Action:   tools.PB_DELETE,
			DataType: data_dt.EnumIndex(),
			User:     user,
			Payload:  data,
		})
	}
	o.Data["json"] = data
	o.ServeJSON()
}

// @Title Search Decentralized
// @Description find workflow by key word
// @Param	search		path 	string	true		"the search you want to get"
// @Param	is_draft		query 	string	false		"draft wished"
// @Success 200 {workflow} models.workflow
// @router /search/:search/decentralized/:type [get]
func (o *DataController) SearchDecentralized() {
	user, _, _ := oclib.ExtractTokenInfo(*o.Ctx.Request)
	search := o.Ctx.Input.Param(":search")
	t := o.Ctx.Input.Param(":type")
	b, err := json.Marshal(map[string]string{
		"search": search,
		"type":   t,
	})
	infrastructure.EmitNATS(tools.PropalgationMessage{
		Action:   tools.PB_SEARCH,
		DataType: data_dt.EnumIndex(),
		User:     user,
		Payload:  b,
	})
	if err != nil {
		o.Data["json"] = map[string]interface{}{
			"data":  nil,
			"code":  400,
			"error": err,
		}
		o.ServeJSON()
		return
	}
	Websocket(o.Ctx.Request.Context(), user, o.Ctx.ResponseWriter, o.Ctx.Request)
}