add control for partner ship

controllers/peer.go

@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 
 	oclib "cloud.o-forge.io/core/oc-lib"
+	"cloud.o-forge.io/core/oc-lib/models/peer"
 	beego "github.com/beego/beego/v2/server/web"
 )
 
@@ -59,3 +60,42 @@ func (o *PeerController) Get() {
 	o.Data["json"] = oclib.LoadOne(oclib.LibDataEnum(oclib.PEER), id)
 	o.ServeJSON()
 }
+
+// @Title Partner
+// @Description add partner peer by peerid
+// @Param	id		path 	string	true		"the peer id you want to partner"
+// @Success 200 {peer} models.peer
+// @router /:id/partner [post]
+func (o *PeerController) Partner() {
+	id := o.Ctx.Input.Param(":id")
+	o.Data["json"] = oclib.UpdateOne(oclib.LibDataEnum(oclib.PEER), map[string]interface{}{
+		"state": peer.PARTNER,
+	}, id)
+	o.ServeJSON()
+}
+
+// @Title Blacklist
+// @Description add blacklist peer by peerid
+// @Param	id		path 	string	true		"the peer id you want to blacklist"
+// @Success 200 {peer} models.peer
+// @router /:id/blacklist [post]
+func (o *PeerController) Blacklist() {
+	id := o.Ctx.Input.Param(":id")
+	o.Data["json"] = oclib.UpdateOne(oclib.LibDataEnum(oclib.PEER), map[string]interface{}{
+		"state": peer.BLACKLIST,
+	}, id)
+	o.ServeJSON()
+}
+
+// @Title DeleteState
+// @Description delete state peer by peerid
+// @Param	id		path 	string	true		"the peer id you want to delete state"
+// @Success 200 {peer} models.peer
+// @router /:id/undo_state [post]
+func (o *PeerController) DeleteState() {
+	id := o.Ctx.Input.Param(":id")
+	o.Data["json"] = oclib.UpdateOne(oclib.LibDataEnum(oclib.PEER), map[string]interface{}{
+		"state": peer.NONE,
+	}, id)
+	o.ServeJSON()
+}