338 lines
10 KiB
Go
338 lines
10 KiB
Go
|
package controllers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"cloud.o-forge.io/core/oc-catalog/models"
|
||
|
|
"github.com/beego/beego/v2/core/logs"
|
||
|
|
beego "github.com/beego/beego/v2/server/web"
|
||
|
|
"github.com/vk496/cron"
|
||
|
|
)
|
||
|
|
|
||
|
|
type WorkflowController struct {
|
||
|
|
beego.Controller
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title Create a new workflow
|
||
|
|
// @Description Create a name for the new workflow
|
||
|
|
// @Param workflowName query string true "Name of the workflow"
|
||
|
|
// @Success 200 {string} Workflow created succeful
|
||
|
|
// @Failure 403 Authentication issue
|
||
|
|
// @Failure 400 {string} Other error
|
||
|
|
// // @Security jwtAPIToken
|
||
|
|
// @router / [post]
|
||
|
|
func (u *WorkflowController) CreateWorkflow(workflowName string) {
|
||
|
|
// token := u.Ctx.Input.GetData("jwtAPIToken").(string)
|
||
|
|
|
||
|
|
// //TODO: Implement as swagger security definition (api key?)
|
||
|
|
// username, err := GetUsernameFromToken(token)
|
||
|
|
// if err != nil {
|
||
|
|
// u.Data["json"] = "No valid token"
|
||
|
|
// u.Ctx.Output.Status = 403
|
||
|
|
// u.ServeJSON()
|
||
|
|
// return
|
||
|
|
// }
|
||
|
|
|
||
|
|
username := "asd"
|
||
|
|
|
||
|
|
if err := models.CreateWorkflow(username, workflowName); err != nil {
|
||
|
|
u.Data["json"] = err.Error()
|
||
|
|
u.Ctx.Output.Status = 400
|
||
|
|
u.ServeJSON()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title List workflows
|
||
|
|
// @Description List available workflows
|
||
|
|
// @Success 200 []string List of workflows
|
||
|
|
// @Failure 403 Authentication issue
|
||
|
|
// @Failure 400 {string} Other error
|
||
|
|
// // @Security jwtAPIToken
|
||
|
|
// @router / [get]
|
||
|
|
func (u *WorkflowController) ListWorkflows() {
|
||
|
|
// token := u.Ctx.Input.GetData("jwtAPIToken").(string)
|
||
|
|
|
||
|
|
username := "asd"
|
||
|
|
|
||
|
|
w := models.GetWorkspace(username)
|
||
|
|
|
||
|
|
if w == nil {
|
||
|
|
// No username
|
||
|
|
u.Data["json"] = "Workspace doesn't exist"
|
||
|
|
u.Ctx.Output.Status = 400
|
||
|
|
u.ServeJSON()
|
||
|
|
}
|
||
|
|
|
||
|
|
u.Data["json"] = w.GetWorkflows()
|
||
|
|
u.ServeJSON()
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title Get Workflow
|
||
|
|
// @Description Get a workflow by name
|
||
|
|
// @Param workflowName path string true "Workflow Name"
|
||
|
|
// @Success 200 {object} models.Workflow
|
||
|
|
// @Failure 403 Authentication issue
|
||
|
|
// @Failure 400 {string} Other error
|
||
|
|
// // @Security jwtAPIToken
|
||
|
|
// @router /:workflowName [get]
|
||
|
|
func (u *WorkflowController) GetWorkflow(workflowName string) {
|
||
|
|
// token := u.Ctx.Input.GetData("jwtAPIToken").(string)
|
||
|
|
|
||
|
|
username := "asd"
|
||
|
|
|
||
|
|
proj, err := models.GetWorkflow(username, workflowName)
|
||
|
|
if err != nil {
|
||
|
|
u.SetData(err.Error())
|
||
|
|
u.Ctx.Output.Status = 400
|
||
|
|
u.ServeJSON()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
u.Data["json"] = proj
|
||
|
|
u.ServeJSON()
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title Add new object to a Workflow
|
||
|
|
// @Description Create a Rtype object from already added resources to the workspace
|
||
|
|
// @Param workflowName path string true "workflow Name"
|
||
|
|
// @Param rID query string true "rID of already existing item in Workspace"
|
||
|
|
// @Success 200 {string} ID of the new object (rObjID)
|
||
|
|
// @Failure 403 Authentication issue
|
||
|
|
// @Failure 400 {string} Other error
|
||
|
|
// // @Security jwtAPIToken
|
||
|
|
// @router /:workflowName/add [post]
|
||
|
|
func (u *WorkflowController) AddElementWorkflow(workflowName, rID string) {
|
||
|
|
username := "asd"
|
||
|
|
|
||
|
|
rObjID, err := models.CreateObjectInWorkflow(username, workflowName, rID)
|
||
|
|
if err != nil {
|
||
|
|
logs.Debug(err.Error())
|
||
|
|
u.Data["json"] = err.Error()
|
||
|
|
u.Ctx.Output.Status = 400
|
||
|
|
u.ServeJSON()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
u.Data["json"] = rObjID
|
||
|
|
u.ServeJSON()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title Parse mxGraph
|
||
|
|
// @Description If we use this aproach to transofrm mxgraph representation in our representation, we should not use other API calls for modify the project structure or we'll have inconsistencies.
|
||
|
|
// @Param workflowName path string true "Workflow Name"
|
||
|
|
// @Param xmlData body string true "Xml representation of the workflow"
|
||
|
|
// @Success 200 The xmlgraph consumed correctly
|
||
|
|
// @Success 201 The xmlgraph consumed with issues
|
||
|
|
// @Failure 403 Authentication issue
|
||
|
|
// @Failure 400 {string} Other error
|
||
|
|
// @router /:workflowName/mxGraphParser [post]
|
||
|
|
func (u *WorkflowController) MxGraphParser(workflowName, xmlData string) {
|
||
|
|
username := "asd"
|
||
|
|
err, mxissues := models.ParseMxGraph(username, workflowName, xmlData)
|
||
|
|
if err != nil {
|
||
|
|
logs.Debug(err.Error())
|
||
|
|
u.CustomAbort(400, err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(mxissues) > 0 {
|
||
|
|
strErrors := make([]string, len(mxissues))
|
||
|
|
|
||
|
|
for i, err := range mxissues {
|
||
|
|
strErrors[i] = err.Error()
|
||
|
|
}
|
||
|
|
|
||
|
|
u.Data["json"] = strErrors
|
||
|
|
u.Ctx.Output.Status = 201
|
||
|
|
u.ServeJSON()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title Get mxGraph last status
|
||
|
|
// @Description Obtain the last mxgraph XML status from the workflow
|
||
|
|
// @Param workflowName path string true "Workflow Name"
|
||
|
|
// @Success 200 The xmlgraph
|
||
|
|
// @Success 201 Empty workflow
|
||
|
|
// @Failure 403 Authentication issue
|
||
|
|
// @Failure 400 {string} Other error
|
||
|
|
// @router /:workflowName/mxGraphParser [get]
|
||
|
|
func (u *WorkflowController) MxGraphParserConsume(workflowName string) {
|
||
|
|
username := "asd"
|
||
|
|
xmlData, err := models.GetMxGraph(username, workflowName)
|
||
|
|
if err != nil {
|
||
|
|
logs.Debug(err.Error())
|
||
|
|
u.Data["json"] = err.Error()
|
||
|
|
u.Ctx.Output.Status = 400
|
||
|
|
u.ServeJSON()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if xmlData == nil {
|
||
|
|
u.Ctx.Output.Status = 201
|
||
|
|
} else {
|
||
|
|
u.Ctx.Output.Status = 200
|
||
|
|
u.Ctx.Output.Body([]byte(*xmlData))
|
||
|
|
u.ServeXML()
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title Create a realtionship between two Robjects
|
||
|
|
// @Description Create a Rtype object from already added resources to the workspace
|
||
|
|
// @Param workflowName path string true "Workflow Name"
|
||
|
|
// @Param rObjIDsource query string true "Robject source. Usually Data"
|
||
|
|
// @Param isInput query bool true "If the operation is for input (true) linkage or output (false)"
|
||
|
|
// @Param rObjIDtarger query string true "Robject where will be written the association"
|
||
|
|
// @Success 200 {string} ID of the new object (rObjID)
|
||
|
|
// @Failure 403 Authentication issue
|
||
|
|
// @Failure 400 {string} Other error
|
||
|
|
// // @Security jwtAPIToken
|
||
|
|
// @router /:workflowName/link [post]
|
||
|
|
func (u *WorkflowController) LinkElementsWorkflow(workflowName, rObjIDsource, rObjIDtarger string, isInput bool) {
|
||
|
|
username := "asd"
|
||
|
|
|
||
|
|
err := models.LinkObjectsInWorkspace(username, workflowName, rObjIDsource, isInput, rObjIDtarger)
|
||
|
|
if err != nil {
|
||
|
|
logs.Debug(err.Error())
|
||
|
|
u.Data["json"] = err.Error()
|
||
|
|
u.Ctx.Output.Status = 400
|
||
|
|
u.ServeJSON()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title Get Schedule
|
||
|
|
// @Description Obtain the desired schedule of this workflow
|
||
|
|
// @Param workflowName path string true "Workflow Name"
|
||
|
|
// @Success 200 {object} models.ScheduleTime
|
||
|
|
// @Failure 403 Authentication issue
|
||
|
|
// @Failure 400 Workflow doesn't exist
|
||
|
|
// @Failure 401 Other error
|
||
|
|
// @router /:workflowName/schedule [get]
|
||
|
|
func (u *WorkflowController) GetWorkflowSchedule(workflowName string) {
|
||
|
|
username := "asd"
|
||
|
|
sched, err := models.GetWorkflowSchedule(username, workflowName)
|
||
|
|
|
||
|
|
// Some error
|
||
|
|
if err != nil {
|
||
|
|
u.CustomAbort(401, err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
// No workflow
|
||
|
|
if sched == nil {
|
||
|
|
u.Ctx.Output.Status = 400
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
u.Ctx.Output.Status = 200
|
||
|
|
u.Data["json"] = sched
|
||
|
|
u.ServeJSON()
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title Set Schedule
|
||
|
|
// @Description Set desired schedule by the user. No other effects a part of saving the user input
|
||
|
|
// @Param workflowName path string true "Workflow Name"
|
||
|
|
// @Param isService query bool true "True: Service, False: Task"
|
||
|
|
// @Param startDate query time.Time true "RFC3339 time for startDate"
|
||
|
|
// @Param stopDate query time.Time true "RFC3339 time for stopDate"
|
||
|
|
// @Param events query string false "List of events separated by comma"
|
||
|
|
// @Param cronString query string false "Cron string"
|
||
|
|
// @Param duration query uint false "Duration in seconds"
|
||
|
|
// @Success 200 {object} models.ScheduleInfo
|
||
|
|
// @Failure 403 Authentication issue
|
||
|
|
// @Failure 400 Workflow doesn't exist
|
||
|
|
// @Failure 401 Other error
|
||
|
|
// @Failure 402 Bad user input
|
||
|
|
// @router /:workflowName/schedule [put]
|
||
|
|
func (u *WorkflowController) SetWorkflowSchedule(workflowName, cronString, events string, isService bool, startDate, stopDate time.Time, duration uint) {
|
||
|
|
username := "asd"
|
||
|
|
|
||
|
|
// Check Dates
|
||
|
|
if startDate.After(stopDate) || startDate.Equal(stopDate) {
|
||
|
|
u.CustomAbort(402, "startDate must be before stopDate")
|
||
|
|
}
|
||
|
|
|
||
|
|
if startDate.Before(time.Now().UTC()) {
|
||
|
|
u.CustomAbort(402, "Current server time ("+time.Now().UTC().String()+") is after the startDate ("+startDate.String()+")")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Tasks must have cron and duration
|
||
|
|
if !isService {
|
||
|
|
if cronString == "" {
|
||
|
|
u.CustomAbort(402, "Tasks cronString must not be empty")
|
||
|
|
}
|
||
|
|
|
||
|
|
if duration == 0 {
|
||
|
|
u.CustomAbort(402, "Tasks duration musn't be 0")
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err := cron.Parse(cronString)
|
||
|
|
// Check cron
|
||
|
|
if err != nil {
|
||
|
|
u.CustomAbort(402, "Bad cron message: "+err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
schedInfo, err := models.SetWorkflowSchedule(username, workflowName, cronString, events, isService, startDate, stopDate, duration)
|
||
|
|
|
||
|
|
// Some error
|
||
|
|
if err != nil {
|
||
|
|
u.CustomAbort(401, err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
// No workflow
|
||
|
|
if schedInfo == nil {
|
||
|
|
u.CustomAbort(400, "")
|
||
|
|
}
|
||
|
|
|
||
|
|
u.Ctx.Output.Status = 200
|
||
|
|
u.Data["json"] = schedInfo
|
||
|
|
u.ServeJSON()
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title Check Schedule
|
||
|
|
// @Description Check if we can schedule the project in other DCs. Must set a desired schedule first!
|
||
|
|
// @Param workflowName path string true "Workflow Name"
|
||
|
|
// @Success 200 {object} []models.DCstatus
|
||
|
|
// @Failure 403 Authentication issue
|
||
|
|
// @Failure 401 Other error
|
||
|
|
// @router /:workflowName/schedule/check [get]
|
||
|
|
func (u *WorkflowController) CheckWorkflowSchedule(workflowName string) {
|
||
|
|
username := "asd"
|
||
|
|
|
||
|
|
data, err := models.CheckAndBookWorkflowSchedule(username, workflowName, false)
|
||
|
|
if err != nil {
|
||
|
|
u.CustomAbort(401, err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
u.Ctx.Output.Status = 200
|
||
|
|
u.Data["json"] = data
|
||
|
|
u.ServeJSON()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// @Title Book Schedule
|
||
|
|
// @Description Book a schedule in all DCs of the workflow. Must set a desired schedule first!
|
||
|
|
// @Param workflowName path string true "Workflow Name"
|
||
|
|
// @Success 200 {object} []models.DCstatus
|
||
|
|
// @Failure 403 Authentication issue
|
||
|
|
// @Failure 401 Other error. Check output
|
||
|
|
// @router /:workflowName/schedule/book [post]
|
||
|
|
func (u *WorkflowController) BookWorkflowSchedule(workflowName string) {
|
||
|
|
username := "asd"
|
||
|
|
|
||
|
|
data, err := models.CheckAndBookWorkflowSchedule(username, workflowName, true)
|
||
|
|
if err != nil {
|
||
|
|
u.CustomAbort(401, err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
u.Ctx.Output.Status = 200
|
||
|
|
u.Data["json"] = data
|
||
|
|
u.ServeJSON()
|
||
|
|
return
|
||
|
|
}
|