89 lines
1.8 KiB
Go
89 lines
1.8 KiB
Go
package controller
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/drone/drone/model"
|
|
"github.com/drone/drone/router/middleware/context"
|
|
"github.com/drone/drone/router/middleware/session"
|
|
"github.com/drone/drone/shared/token"
|
|
"github.com/drone/drone/store"
|
|
)
|
|
|
|
func GetNodes(c *gin.Context) {
|
|
nodes, err := store.GetNodeList(c)
|
|
if err != nil {
|
|
c.String(400, err.Error())
|
|
} else {
|
|
c.JSON(200, nodes)
|
|
}
|
|
}
|
|
|
|
func ShowNodes(c *gin.Context) {
|
|
user := session.User(c)
|
|
nodes, _ := store.GetNodeList(c)
|
|
token, _ := token.New(token.CsrfToken, user.Login).Sign(user.Hash)
|
|
c.HTML(http.StatusOK, "nodes.html", gin.H{"User": user, "Nodes": nodes, "Csrf": token})
|
|
}
|
|
|
|
func GetNode(c *gin.Context) {
|
|
|
|
}
|
|
|
|
func PostNode(c *gin.Context) {
|
|
engine := context.Engine(c)
|
|
|
|
in := struct {
|
|
Addr string `json:"address"`
|
|
Arch string `json:"architecture"`
|
|
Cert string `json:"cert"`
|
|
Key string `json:"key"`
|
|
CA string `json:"ca"`
|
|
}{}
|
|
err := c.Bind(&in)
|
|
if err != nil {
|
|
c.AbortWithStatus(http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
node := &model.Node{}
|
|
node.Addr = in.Addr
|
|
node.Cert = in.Cert
|
|
node.Key = in.Key
|
|
node.CA = in.CA
|
|
node.Arch = "linux_amd64"
|
|
|
|
err = engine.Allocate(node)
|
|
if err != nil {
|
|
c.String(http.StatusBadRequest, err.Error())
|
|
return
|
|
}
|
|
|
|
err = store.CreateNode(c, node)
|
|
if err != nil {
|
|
c.AbortWithStatus(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
c.IndentedJSON(http.StatusOK, node)
|
|
}
|
|
|
|
func DeleteNode(c *gin.Context) {
|
|
engine := context.Engine(c)
|
|
|
|
id, _ := strconv.Atoi(c.Param("node"))
|
|
node, err := store.GetNode(c, int64(id))
|
|
if err != nil {
|
|
c.AbortWithStatus(http.StatusNotFound)
|
|
return
|
|
}
|
|
err = store.DeleteNode(c, node)
|
|
if err != nil {
|
|
c.AbortWithStatus(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
engine.Deallocate(node)
|
|
}
|