oc-datacenter/controllers/booking.go

150 lines
4.2 KiB
Go
Raw Normal View History

2024-08-13 11:17:09 +02:00
package controllers
import (
"encoding/json"
"errors"
"time"
oclib "cloud.o-forge.io/core/oc-lib"
"cloud.o-forge.io/core/oc-lib/dbs"
"cloud.o-forge.io/core/oc-lib/models/booking"
b "cloud.o-forge.io/core/oc-lib/models/booking"
"cloud.o-forge.io/core/oc-lib/models/workflow_execution"
beego "github.com/beego/beego/v2/server/web"
)
// Operations about workspace
type BookingController struct {
beego.Controller
}
// @Title Search
// @Description search booking
// @Param search path string true "the word search you want to get"
// @Success 200 {booking} models.booking
// @router /search/:search [get]
func (o *BookingController) 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.BOOKING))
o.ServeJSON()
}
// @Title GetAll
// @Description find booking by id
// @Success 200 {booking} models.booking
// @router / [get]
func (o *BookingController) GetAll() {
o.Data["json"] = oclib.LoadAll(oclib.LibDataEnum(oclib.BOOKING))
o.ServeJSON()
}
// @Title Get
// @Description find booking by id
// @Param id path string true "the id you want to get"
// @Success 200 {booking} models.booking
// @router /:id [get]
func (o *BookingController) Get() {
id := o.Ctx.Input.Param(":id")
o.Data["json"] = oclib.LoadOne(oclib.LibDataEnum(oclib.BOOKING), id)
o.ServeJSON()
}
// @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 *BookingController) 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 {
booking := &b.Booking{}
isAvailable, err2 := booking.CheckBooking(date, &date2)
code := 200
err := ""
if !isAvailable {
code = 409
err = "booking not available"
if err2 != nil {
err += " - " + err2.Error()
}
}
o.Data["json"] = map[string]interface{}{
"data": map[string]interface{}{
"is_available": isAvailable,
},
"code": code,
"error": err,
}
}
o.ServeJSON()
}
// @Title Post
// @Description create booking
// @Param booking body string true "the booking you want to post"
// @Success 200 {object} models.object
// @router / [post]
func (o *BookingController) Post() {
// TODO retrieve objects in body DON'T FORGET TO CHECK IF THE OBJECT IS VALID
var resp workflow_execution.WorkflowExecutions
json.Unmarshal(o.Ctx.Input.CopyBody(10000), &resp)
dc_id := resp.ResourceID
res := oclib.Search(&dbs.Filters{And: map[string][]dbs.Filter{
"datacenter_resource_id": {{Operator: dbs.EQUAL.String(), Value: dc_id}},
}}, "", oclib.LibDataEnum(oclib.BOOKING))
if res.Code != 200 {
o.Data["json"] = map[string]interface{}{
"data": nil,
"code": res.Code,
"error": res.Err,
}
o.ServeJSON()
return
}
for _, b := range res.Data {
oclib.DeleteOne(oclib.LibDataEnum(oclib.BOOKING), b.GetID())
}
books := make([]interface{}, 0)
errormsg := ""
for _, exec := range resp.Executions {
if ok, _ := (&booking.Booking{}).CheckBooking(*exec.ExecDate, exec.EndDate); !ok {
res.Err += " -> the booking from " + exec.ExecDate.String() + " is already taken."
o.Data["json"] = map[string]interface{}{
"data": nil,
"code": res.Code,
"error": res.Err + " -> the booking from " + exec.ExecDate.String() + " is already taken.",
}
o.ServeJSON()
return
}
new := &booking.Booking{
DatacenterResourceID: dc_id,
WorkflowExecution: *exec,
}
b := oclib.StoreOne(oclib.LibDataEnum(oclib.BOOKING), new.Serialize())
if b.Code == 200 {
books = append(books, b.Data)
} else {
errormsg += " -> " + b.Err
}
}
o.Data["json"] = map[string]interface{}{
"data": books,
"code": 200,
"error": errormsg,
}
o.ServeJSON()
}