97 lines
2.5 KiB
Go
97 lines
2.5 KiB
Go
package cmd
|
|
|
|
import (
|
|
//"io"
|
|
"fmt"
|
|
"strings"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gin-contrib/cors"
|
|
)
|
|
|
|
func printJails(jails []string, fields []string) ([]byte, error) {
|
|
var res []byte
|
|
|
|
res = append(res, []byte("[")...)
|
|
|
|
// Build a filtered jail list
|
|
var filteredJails []Jail
|
|
if (len(jails) >= 1 && len(jails[0]) > 0) {
|
|
for _, j := range gJails {
|
|
if isStringInArray(jails, j.Name) {
|
|
filteredJails = append(filteredJails, j)
|
|
}
|
|
}
|
|
} else {
|
|
filteredJails = gJails
|
|
}
|
|
|
|
for i, j := range filteredJails {
|
|
out, err := j.PrintJSON(fields)
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
res = append(res, out...)
|
|
if i < len(filteredJails) - 1 {
|
|
res = append(res, []byte(",")...)
|
|
}
|
|
}
|
|
res = append(res, []byte("]")...)
|
|
return res, nil
|
|
}
|
|
|
|
func startApi(address string, port int) {
|
|
// Initialize gJails.
|
|
// FIXME: How to update this at running time without doubling the same jails?
|
|
// core will do this for us in add and delete functions
|
|
ListJails([]string{""}, false)
|
|
|
|
r := gin.Default()
|
|
|
|
// Cross Origin Request Support
|
|
config := cors.DefaultConfig()
|
|
// DANGER !
|
|
config.AllowOrigins = []string{"*"}
|
|
r.Use(cors.New(config))
|
|
|
|
r.GET("/jail/list", func(c *gin.Context) {
|
|
jailstr := c.Query("jail")
|
|
fields := c.Query("fields")
|
|
if len(fields) == 0 {
|
|
// TODO : Put this default value in (user?) configuration
|
|
fields = "Name,Running,Config.Release,Config.Ip4_addr"
|
|
}
|
|
jsout, err := printJails(strings.Split(jailstr, ","), strings.Split(fields, ","))
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"status": 1, "error": err.Error()})
|
|
} else {
|
|
c.Data(http.StatusOK, "application/json", jsout)
|
|
}
|
|
})
|
|
|
|
r.POST("/jail/start/:jail", func(c *gin.Context) {
|
|
jailstr := c.Params.ByName("jail")
|
|
// TODO : Capture or redirect output so we can know if jail was started or not
|
|
StartJail(strings.Split(jailstr, ","))
|
|
c.JSON(http.StatusOK, gin.H{"status": 0, "error": ""})
|
|
})
|
|
r.POST("/jail/stop/:jail", func(c *gin.Context) {
|
|
jailstr := c.Params.ByName("jail")
|
|
// TODO : Capture or redirect output so we can know if jail was started or not
|
|
StopJail(strings.Split(jailstr, ","))
|
|
c.JSON(http.StatusOK, gin.H{"status": 0, "error": ""})
|
|
})
|
|
|
|
r.GET("/jail/properties", func(c *gin.Context) {
|
|
props, err := ListJailsProps("json")
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"status": 1, "error": err.Error()})
|
|
} else {
|
|
c.Data(http.StatusOK, "application/json", []byte(props))
|
|
}
|
|
})
|
|
|
|
r.Run(fmt.Sprintf("%s:%d", address, port))
|
|
}
|