Merge remote-tracking branch 'origin/master' into feature/admiralty

2025-02-19 14:40:56 +01:00
parent 198c1e1a28 331c0835f1
commit 3c313171c3
19 changed files with 738 additions and 308 deletions
--- a/controllers/booking.go
+++ b/controllers/booking.go
@@ -4,6 +4,8 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"oc-datacenter/infrastructure"
+	"strings"
 	"time"

 	oclib "cloud.o-forge.io/core/oc-lib"
@@ -105,6 +107,40 @@ func (o *BookingController) Get() {
 	o.ServeJSON()
 }

+// @Title Update
+// @Description create computes
+// @Param	id		path 	string	true		"the compute id you want to get"
+// @Param	body		body 	models.compute	true		"The compute content"
+// @Success 200 {compute} models.compute
+// @router /:id [put]
+func (o *BookingController) Put() {
+	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
+	// store and return Id or post with UUID
+	var res map[string]interface{}
+	id := o.Ctx.Input.Param(":id")
+	book := oclib.NewRequest(oclib.LibDataEnum(oclib.BOOKING), user, peerID, groups, nil).LoadOne(id)
+	if book.Code != 200 {
+		o.Data["json"] = map[string]interface{}{
+			"data":  nil,
+			"code":  book.Code,
+			"error": book.Err,
+		}
+		o.ServeJSON()
+		return
+	}
+	booking := book.Data.(*b.Booking)
+	if time.Now().After(booking.ExpectedStartDate) {
+		o.Data["json"] = oclib.NewRequest(oclib.LibDataEnum(oclib.BOOKING), user, peerID, groups, nil).UpdateOne(res, id)
+	} else {
+		o.Data["json"] = map[string]interface{}{
+			"data":  nil,
+			"code":  409,
+			"error": "booking is not already started",
+		}
+	}
+	o.ServeJSON()
+}
+
 // @Title Check
 // @Description check booking
 // @Param	id	path 	string		"id of the datacenter"
@@ -156,14 +192,13 @@ func (o *BookingController) Check() {
 	o.ServeJSON()
 }

-// @Title Poststatic.
+// @Title Post.
 // @Description create booking
 // @Param	booking		body 	string	true		"the booking you want to post"
 // @Param	is_draft		query 	string	false		"draft wished"
 // @Success 200 {object} models.object
 // @router / [post]
 func (o *BookingController) Post() {
-	fmt.Println("POST")
 	/*
 	 * This function is used to create a booking.
 	 * It takes the following parameters:
@@ -195,11 +230,71 @@ func (o *BookingController) Post() {
 		oclib.NewRequest(oclib.LibDataEnum(oclib.BOOKING), user, peerID, groups, nil).DeleteOne(b.GetID())
 	}
 	b := oclib.NewRequest(oclib.LibDataEnum(oclib.BOOKING), user, peerID, groups, nil).StoreOne(resp.Serialize(&resp))
-	errormsg := ""
+	if b.Code != 200 {
+		o.Data["json"] = map[string]interface{}{
+			"data":  nil,
+			"code":  b.Code,
+			"error": b.Err,
+		}
+		o.ServeJSON()
+		return
+	}
+	if err := o.createNamespace(resp.ExecutionsID); err != nil {
+		if strings.Contains(err.Error(), "already exists") {
+			err = nil
+		} else {
+			o.Data["json"] = map[string]interface{}{
+				"data":  nil,
+				"code":  500,
+				"error": err.Error(),
+			}
+			o.ServeJSON()
+			return
+		}
+	}
 	o.Data["json"] = map[string]interface{}{
 		"data":  []interface{}{b},
 		"code":  200,
-		"error": errormsg,
+		"error": "",
 	}
 	o.ServeJSON()
 }
+
+func (o *BookingController) createNamespace(ns string) error {
+	/*
+	 * This function is used to create a namespace.
+	 * It takes the following parameters:
+	 * - ns: the namespace you want to create
+	 */
+	serv, err := infrastructure.NewService()
+	if err != nil {
+		return nil
+	}
+	err = serv.CreateNamespace(o.Ctx.Request.Context(), ns)
+	if err != nil {
+		return err
+	}
+	err = serv.CreateServiceAccount(o.Ctx.Request.Context(), ns)
+	if err != nil {
+		return err
+	}
+	role := "argo-role"
+	err = serv.CreateRole(o.Ctx.Request.Context(), ns, role,
+		[][]string{
+			{"coordination.k8s.io"},
+			{""},
+			{""}},
+		[][]string{
+			{"leases"},
+			{"secrets"},
+			{"pods"}},
+		[][]string{
+			{"get", "create", "update"},
+			{"get"},
+			{"patch"}})
+	if err != nil {
+		return err
+	}
+	fmt.Println("ROLLLLLE BIND")
+	return serv.CreateRoleBinding(o.Ctx.Request.Context(), ns, "argo-role-binding", role)
+}
--- a/controllers/datacenter.go
+++ b/controllers/datacenter.go
@@ -21,12 +21,12 @@ func (o *DatacenterController) GetAll() {
 	isDraft := o.Ctx.Input.Query("is_draft")
 	storages := oclib.NewRequest(oclib.LibDataEnum(oclib.STORAGE_RESOURCE), user, peerID, groups, nil).Search(&dbs.Filters{
 		Or: map[string][]dbs.Filter{
-			"abstractintanciatedresource.abstractresource.abstractobject.creator_id": {{Operator: dbs.EQUAL.String(), Value: peerID}},
+			"abstractinstanciatedresource.abstractresource.abstractobject.creator_id": {{Operator: dbs.EQUAL.String(), Value: peerID}},
 		},
 	}, "", isDraft == "true")
 	computes := oclib.NewRequest(oclib.LibDataEnum(oclib.COMPUTE_RESOURCE), user, peerID, groups, nil).Search(&dbs.Filters{
 		Or: map[string][]dbs.Filter{
-			"abstractintanciatedresource.abstractresource.abstractobject.creator_id": {{Operator: dbs.EQUAL.String(), Value: peerID}},
+			"abstractinstanciatedresource.abstractresource.abstractobject.creator_id": {{Operator: dbs.EQUAL.String(), Value: peerID}},
 		},
 	}, "", isDraft == "true")
 	storages.Data = append(storages.Data, computes.Data...)
@@ -49,15 +49,15 @@ func (o *DatacenterController) Get() {
 	id := o.Ctx.Input.Param(":id")
 	storages := oclib.NewRequest(oclib.LibDataEnum(oclib.STORAGE_RESOURCE), user, peerID, groups, nil).Search(&dbs.Filters{
 		Or: map[string][]dbs.Filter{
-			"abstractintanciatedresource.abstractresource.abstractobject.id":         {{Operator: dbs.EQUAL.String(), Value: id}},
-			"abstractintanciatedresource.abstractresource.abstractobject.creator_id": {{Operator: dbs.EQUAL.String(), Value: peerID}},
+			"abstractinstanciatedresource.abstractresource.abstractobject.id":         {{Operator: dbs.EQUAL.String(), Value: id}},
+			"abstractinstanciatedresource.abstractresource.abstractobject.creator_id": {{Operator: dbs.EQUAL.String(), Value: peerID}},
 		},
 	}, "", isDraft == "true")
 	if len(storages.Data) == 0 {
 		computes := oclib.NewRequest(oclib.LibDataEnum(oclib.COMPUTE_RESOURCE), user, peerID, groups, nil).Search(&dbs.Filters{
 			Or: map[string][]dbs.Filter{
-				"abstractintanciatedresource.abstractresource.abstractobject.id":         {{Operator: dbs.EQUAL.String(), Value: id}},
-				"abstractintanciatedresource.abstractresource.abstractobject.creator_id": {{Operator: dbs.EQUAL.String(), Value: peerID}},
+				"abstractinstanciatedresource.abstractresource.abstractobject.id":         {{Operator: dbs.EQUAL.String(), Value: id}},
+				"abstractinstanciatedresource.abstractresource.abstractobject.creator_id": {{Operator: dbs.EQUAL.String(), Value: peerID}},
 			},
 		}, "", isDraft == "true")
 		if len(computes.Data) == 0 {
--- a/controllers/session.go
+++ b/controllers/session.go
@@ -0,0 +1,53 @@
+package controllers
+
+import (
+	"fmt"
+	"oc-datacenter/infrastructure"
+	"strconv"
+
+	beego "github.com/beego/beego/v2/server/web"
+)
+
+type SessionController struct {
+	beego.Controller
+}
+
+// @Title GetToken
+// @Description find booking by id
+// @Param	id	path 	string		"id of the datacenter"
+// @Param	duration	path 	string		"duration of the token"
+// @Success 200 {booking} models.booking
+// @router /token/:id/:duration [get]
+func (o *SessionController) GetToken() {
+	id := o.Ctx.Input.Param(":id")
+	durationSTR := o.Ctx.Input.Param(":duration")
+
+	duration, err := strconv.Atoi(durationSTR)
+	if err != nil {
+		// change code to 400
+		o.Ctx.Output.SetStatus(400)
+		o.Data["json"] = map[string]string{"error": "duration invalid"}
+		o.ServeJSON()
+		return
+	}
+
+	serv, err := infrastructure.NewService()
+	if err != nil {
+		// change code to 500
+		o.Ctx.Output.SetStatus(500)
+		o.Data["json"] = map[string]string{"error": err.Error()}
+		o.ServeJSON()
+		return
+	}
+	fmt.Println("BLAPO", id, duration)
+	token, err := serv.GetToken(o.Ctx.Request.Context(), id, duration)
+	if err != nil {
+		// change code to 500
+		o.Ctx.Output.SetStatus(500)
+		o.Data["json"] = map[string]string{"error": err.Error()}
+		o.ServeJSON()
+		return
+	}
+	o.Data["json"] = token
+	o.ServeJSON()
+}