oc-workflow/controllers/workflow.go

195 lines
6.2 KiB
Go
Raw Normal View History

2024-07-11 11:40:11 +02:00
package controllers
import (
2024-07-24 10:25:08 +02:00
"encoding/json"
2024-08-12 16:58:24 +02:00
"errors"
"time"
2024-07-11 11:40:11 +02:00
2024-07-24 10:25:08 +02:00
oclib "cloud.o-forge.io/core/oc-lib"
2024-08-12 16:58:24 +02:00
w "cloud.o-forge.io/core/oc-lib/models/workflow"
tools "cloud.o-forge.io/core/oc-lib/tools"
2024-07-11 11:40:11 +02:00
beego "github.com/beego/beego/v2/server/web"
2024-08-13 08:35:01 +02:00
"github.com/nats-io/nats.go"
2024-07-11 11:40:11 +02:00
)
// Operations about workflow
type WorkflowController struct {
beego.Controller
}
2024-08-06 11:09:38 +02:00
// @Title Search
// @Description search workspace
// @Param search path string true "the word search you want to get"
// @Success 200 {workspace} models.workspace
// @router /search/:search [get]
func (o *WorkflowController) Search() {
// store and return Id or post with UUID
search := o.Ctx.Input.Param(":search")
o.Data["json"] = oclib.Search(nil, search, oclib.LibDataEnum(oclib.WORKFLOW))
o.ServeJSON()
}
2024-07-24 10:25:08 +02:00
// @Title Update
// @Description create workflows
// @Param id path string true "the workflowid you want to get"
// @Param body body models.workflow true "The workflow content"
// @Success 200 {object} models.workflow
// @router /:id [put]
func (o *WorkflowController) Put() {
// store and return Id or post with UUID
var res map[string]interface{}
id := o.Ctx.Input.Param(":id")
json.Unmarshal(o.Ctx.Input.CopyBody(10000), &res)
2024-08-13 11:04:27 +02:00
caller := tools.NewHTTPCaller(
map[string]map[tools.METHOD]string{
oclib.LibDataEnum(oclib.BOOKING).String(): {
tools.GET: oclib.GetPath(oclib.LibDataEnum(oclib.BOOKING)) + "/oc/booking/:id",
tools.DELETE: oclib.GetPath(oclib.LibDataEnum(oclib.BOOKING)) + "/oc/booking/:id",
tools.POST: oclib.GetPath(oclib.LibDataEnum(oclib.BOOKING)) + "/oc/booking",
tools.PUT: oclib.GetPath(oclib.LibDataEnum(oclib.BOOKING)) + "/oc/booking",
},
})
2024-08-13 08:35:01 +02:00
data := oclib.UpdateOne(oclib.LibDataEnum(oclib.WORKFLOW), res, id, caller)
data.Err += o.SetNATSPub(res, data)
o.Data["json"] = data
2024-07-24 10:25:08 +02:00
o.ServeJSON()
}
2024-08-13 08:35:01 +02:00
func (o *WorkflowController) SetNATSPub(res map[string]interface{}, data oclib.LibData) string {
if sched, ok := res["schedule"]; ok && sched != nil && data.Code == 200 {
workflow := data.ToWorkflow()
if workflow != nil && workflow.Schedule != nil {
// HERE NATS
nc, err := nats.Connect(nats.DefaultURL)
if err != nil {
return " -> Could not reach NATS server"
}
start := ""
if workflow.Schedule.Start != nil {
start = workflow.Schedule.Start.String()
}
end := ""
if workflow.Schedule.End != nil {
end = workflow.Schedule.End.String()
}
str_description, _ := json.Marshal(map[string]string{
"workflow_id": workflow.UUID,
"start_date": start,
"stop_date": end,
})
err = nc.Publish("workflowsUpdate", str_description)
time.Sleep(time.Second * 2)
if err != nil {
return " -> " + err.Error()
}
}
}
return ""
}
2024-07-11 11:40:11 +02:00
// @Title Create
// @Description create workflows
2024-07-24 10:25:08 +02:00
// @Param data body json true "body for data content (Json format)"
// @Success 200 {object} models.workflow
2024-07-11 11:40:11 +02:00
// @router / [post]
func (o *WorkflowController) Post() {
2024-07-24 10:25:08 +02:00
var res map[string]interface{}
json.Unmarshal(o.Ctx.Input.CopyBody(10000), &res)
2024-08-13 11:04:27 +02:00
caller := tools.NewHTTPCaller(map[string]map[tools.METHOD]string{
oclib.LibDataEnum(oclib.BOOKING).String(): {
tools.GET: oclib.GetPath(oclib.LibDataEnum(oclib.BOOKING)) + "/oc/booking/:id",
tools.DELETE: oclib.GetPath(oclib.LibDataEnum(oclib.BOOKING)) + "/oc/booking/:id",
tools.POST: oclib.GetPath(oclib.LibDataEnum(oclib.BOOKING)) + "/oc/booking",
tools.PUT: oclib.GetPath(oclib.LibDataEnum(oclib.BOOKING)) + "/oc/booking",
},
})
2024-08-13 08:35:01 +02:00
data := oclib.StoreOne(oclib.LibDataEnum(oclib.WORKFLOW), res, caller)
data.Err += o.SetNATSPub(res, data)
o.Data["json"] = data
2024-07-11 11:40:11 +02:00
o.ServeJSON()
}
2024-07-24 10:25:08 +02:00
// @Title Publish
// @Description create workflows
// @Param id path string true "the workflowid you want to get"
// @Success 200 {object} models.workflow
// @router /publish/:id [post]
func (o *WorkflowController) Publish() {
id := o.Ctx.Input.Param(":id")
data := oclib.LoadOne(oclib.LibDataEnum(oclib.WORKFLOW), id)
if data.Data != nil {
o.Data["json"] = oclib.CopyOne(oclib.LibDataEnum(oclib.WORKFLOW_RESOURCE), data.Data.Serialize())
} else {
o.Data["json"] = data
}
o.ServeJSON()
}
// @Title GetAll
2024-07-11 11:40:11 +02:00
// @Description find workflow by workflowid
// @Success 200 {workflow} models.workflow
2024-07-24 10:25:08 +02:00
// @router / [get]
func (o *WorkflowController) GetAll() {
o.Data["json"] = oclib.LoadAll(oclib.LibDataEnum(oclib.WORKFLOW))
2024-07-11 11:40:11 +02:00
o.ServeJSON()
}
2024-07-24 10:25:08 +02:00
// @Title Get
// @Description find workflow by workflowid
// @Param id path string true "the workflowid you want to get"
// @Success 200 {workflow} models.workflow
// @router /:id [get]
func (o *WorkflowController) Get() {
id := o.Ctx.Input.Param(":id")
o.Data["json"] = oclib.LoadOne(oclib.LibDataEnum(oclib.WORKFLOW), id)
2024-07-11 11:40:11 +02:00
o.ServeJSON()
}
// @Title Delete
// @Description delete the workflow
2024-07-24 10:25:08 +02:00
// @Param id path string true "The workflowId you want to delete"
2024-07-11 11:40:11 +02:00
// @Success 200 {string} delete success!
2024-07-24 10:25:08 +02:00
// @router /:id [delete]
2024-07-11 11:40:11 +02:00
func (o *WorkflowController) Delete() {
2024-07-24 10:25:08 +02:00
id := o.Ctx.Input.Param(":id")
o.Data["json"] = oclib.DeleteOne(oclib.LibDataEnum(oclib.WORKFLOW), id)
2024-07-11 11:40:11 +02:00
o.ServeJSON()
}
2024-08-12 16:58:24 +02:00
// @Title Check
// @Description check booking
// @Param start_date path string "the booking start date" format "2006-01-02T15:04:05"
// @Param end_date path string "the booking end date" format "2006-01-02T15:04:05"
// @Success 200 {object} models.object
// @router /check/:start_date/:end_date [get]
func (o *WorkflowController) Check() {
// store and return Id or post with UUID
date, err := time.Parse("2006-01-02T15:04:05", o.Ctx.Input.Param(":start_date"))
date2, err2 := time.Parse("2006-01-02T15:04:05", o.Ctx.Input.Param(":end_date"))
if err != nil || err2 != nil {
o.Data["json"] = map[string]interface{}{
"data": map[string]interface{}{
"is_available": false,
},
"code": 400,
"error": errors.New("invalid date format"),
}
} else {
workflow := &w.Workflow{}
workflow.Schedule = &w.WorkflowSchedule{Start: &date, End: &date2}
isAvailable, err := workflow.CheckBooking(oclib.GetPath(oclib.LibDataEnum(oclib.BOOKING)) + "/oc/booking/check/:start_date/:end_date")
code := 200
if !isAvailable {
code = 409
}
o.Data["json"] = map[string]interface{}{
"data": map[string]interface{}{
"is_available": isAvailable,
},
"code": code,
"error": err,
}
}
o.ServeJSON()
}