oc-scheduler/controllers/booking.go

package controllers

import (
	"fmt"
	"net/http"
	"strconv"
	"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"
	"cloud.o-forge.io/core/oc-lib/models/peer"
	beego "github.com/beego/beego/v2/server/web"
	"github.com/gorilla/websocket"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

// Operations about workspace
type BookingController struct {
	beego.Controller
}

var BookingExample booking.Booking

// bookingDestFilter builds a dest_peer_id filter for booking queries.
// Default: filter by self (master's own bookings).
// With nanoID: verify the ID belongs to a NANO peer, then filter by it.
// Returns nil when nanoID is provided but does not resolve to a NANO peer
// (caller should return HTTP 404 or empty result).
func bookingDestFilter(nanoID string) *dbs.Filters {
	targetID := ""
	if nanoID != "" {
		d := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.PEER), nil).Search(&dbs.Filters{
			And: map[string][]dbs.Filter{
				"id":       {{Operator: dbs.EQUAL.String(), Value: nanoID}},
				"relation": {{Operator: dbs.EQUAL.String(), Value: peer.NANO}},
			},
		}, "", false, 0, 1)
		if len(d.Data) == 0 {
			return nil
		}
		targetID = nanoID
	} else {
		if self, err := oclib.GetMySelf(); err == nil && self != nil {
			targetID = self.GetID()
		}
	}
	if targetID == "" {
		return nil
	}
	return &dbs.Filters{
		And: map[string][]dbs.Filter{
			"dest_peer_id": {{Operator: dbs.EQUAL.String(), Value: targetID}},
		},
	}
}

// @Title Search
// @Description search bookings
// @Param	start_date		path 	string	true		"the word search you want to get"
// @Param	end_date		path 	string	true		"the word search you want to get"
// @Param	is_draft		query 	string	false		"draft wished"
// @Param	nano_id		query 	string	false		"nano peer UUID — if set, return bookings for that nano instead of self"
// @Param  offset       query  string  false
// @Param  limit       query  string  false
// @Success 200 {workspace} models.workspace
// @router /search/:start_date/:end_date [get]
func (o *BookingController) Search() {
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	offset, _ := strconv.Atoi(o.Ctx.Input.Query("offset"))
	limit, _ := strconv.Atoi(o.Ctx.Input.Query("limit"))
	start_date, _ := time.ParseInLocation("2006-01-02", o.Ctx.Input.Param(":start_date"), time.UTC)
	end_date, _ := time.ParseInLocation("2006-01-02", o.Ctx.Input.Param(":end_date"), time.UTC)
	isDraft := o.Ctx.Input.Query("is_draft")
	nanoID := o.Ctx.Input.Query("nano_id")
	sd := primitive.NewDateTimeFromTime(start_date)
	ed := primitive.NewDateTimeFromTime(end_date)
	fmt.Println("SEARCH START END", start_date, end_date)
	df := bookingDestFilter(nanoID)
	if nanoID != "" && df == nil {
		o.Ctx.Output.SetStatus(http.StatusNotFound)
		o.Data["json"] = map[string]string{"error": "nano peer not found: " + nanoID}
		o.ServeJSON()
		return
	}
	f := dbs.Filters{
		And: map[string][]dbs.Filter{
			"expected_start_date": {{Operator: "gte", Value: sd}, {Operator: "lte", Value: ed}},
		},
	}
	if df != nil {
		for k, v := range df.And {
			f.And[k] = v
		}
	}
	o.Data["json"] = oclib.NewRequest(oclib.LibDataEnum(oclib.BOOKING), user, peerID, groups, nil).Search(&f, "", isDraft == "true", int64(offset), int64(limit))
	o.ServeJSON()
}

// @Title GetAll
// @Description find booking by id
// @Param	is_draft		query 	string	false		"draft wished"
// @Param	nano_id		query 	string	false		"nano peer UUID — if set, return bookings for that nano instead of self"
// @Param  offset       query  string  false
// @Param  limit       query  string  false
// @Success 200 {booking} models.booking
// @router / [get]
func (o *BookingController) GetAll() {
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	offset, _ := strconv.Atoi(o.Ctx.Input.Query("offset"))
	limit, _ := strconv.Atoi(o.Ctx.Input.Query("limit"))
	isDraft := o.Ctx.Input.Query("is_draft")
	nanoID := o.Ctx.Input.Query("nano_id")
	df := bookingDestFilter(nanoID)
	if nanoID != "" && df == nil {
		o.Ctx.Output.SetStatus(http.StatusNotFound)
		o.Data["json"] = map[string]string{"error": "nano peer not found: " + nanoID}
		o.ServeJSON()
		return
	}
	if df != nil {
		o.Data["json"] = oclib.NewRequest(oclib.LibDataEnum(oclib.BOOKING), user, peerID, groups, nil).Search(df, "", isDraft == "true", int64(offset), int64(limit))
	} else {
		o.Data["json"] = oclib.NewRequest(oclib.LibDataEnum(oclib.BOOKING), user, peerID, groups, nil).LoadAll(isDraft == "true", int64(offset), int64(limit))
	}
	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() {
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	id := o.Ctx.Input.Param(":id")
	o.Data["json"] = oclib.NewRequest(oclib.LibDataEnum(oclib.BOOKING), user, peerID, groups, nil).LoadOne(id)
	o.ServeJSON()
}

var upgrader = websocket.Upgrader{
	CheckOrigin: func(r *http.Request) bool { return true }, // allow all origins
}