Fixed wrong gitignore
parent
8c8680e5e4
commit
399a4d3adf
@ -0,0 +1,293 @@
|
|||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"git.phc.dm.unipi.it/phc/storage/config"
|
||||||
|
"git.phc.dm.unipi.it/phc/storage/database"
|
||||||
|
"git.phc.dm.unipi.it/phc/storage/utils"
|
||||||
|
"github.com/gofiber/fiber/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (r *Server) isAdminMiddleware(c *fiber.Ctx) error {
|
||||||
|
if _, found := r.adminSessions[c.Cookies("sid")]; !found {
|
||||||
|
return fmt.Errorf("invalid session token")
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.Next()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *Server) Api(api fiber.Router) {
|
||||||
|
isAPIKeyMiddleware := func(c *fiber.Ctx) error {
|
||||||
|
if _, isAdmin := r.adminSessions[c.Cookies("sid")]; !isAdmin { // if admin continue
|
||||||
|
token := c.Cookies("token")
|
||||||
|
if token == "" {
|
||||||
|
return fmt.Errorf("no api token")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := r.Database.CheckAPIKey(token); err != nil { // otherwise also check api token
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return c.Next()
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
// TODO: Change to /server-info (also in frontend)
|
||||||
|
// Setup "/api/monitor" routes
|
||||||
|
//
|
||||||
|
monitorRoute := api.Group("/monitor")
|
||||||
|
monitorRoute.Use(r.isAdminMiddleware)
|
||||||
|
r.ApiMonitor(monitorRoute)
|
||||||
|
|
||||||
|
api.Post("/login", func(c *fiber.Ctx) error {
|
||||||
|
var form struct {
|
||||||
|
Password string `form:"password"`
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := c.BodyParser(&form); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if form.Password != config.AdminPassword {
|
||||||
|
return c.JSON("invalid credentials")
|
||||||
|
}
|
||||||
|
|
||||||
|
token := utils.GenerateRandomString(32)
|
||||||
|
r.adminSessions[token] = struct{}{}
|
||||||
|
|
||||||
|
c.Cookie(&fiber.Cookie{
|
||||||
|
Name: "sid",
|
||||||
|
Value: token,
|
||||||
|
Path: "/",
|
||||||
|
Expires: time.Now().Add(3 * 24 * time.Hour),
|
||||||
|
})
|
||||||
|
|
||||||
|
return c.Redirect("/")
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Get("/status", func(c *fiber.Ctx) error {
|
||||||
|
return c.JSON("ok")
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Get("/current-user",
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
if _, found := r.adminSessions[c.Cookies("sid")]; !found {
|
||||||
|
return c.JSON("anonymous")
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON("admin")
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Get("/dashboard-state",
|
||||||
|
r.isAdminMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
state, err := r.Database.GetDashboardState()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON(state)
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Post("/dashboard-state",
|
||||||
|
r.isAdminMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
|
||||||
|
var state database.DashboardState
|
||||||
|
|
||||||
|
if err := c.BodyParser(&state); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := r.Database.SetDashboardState(state); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON("ok")
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Get("/buckets",
|
||||||
|
r.isAdminMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
buckets, err := r.Database.AllBuckets()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON(buckets)
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Post("/buckets",
|
||||||
|
r.isAdminMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
var req struct {
|
||||||
|
Bucket string `json:"bucket"`
|
||||||
|
Path string `json:"path"`
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Printf("%v", string(c.Body()))
|
||||||
|
|
||||||
|
if err := c.BodyParser(&req); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
settings := &database.JsonBucketSettings{}
|
||||||
|
if req.Path != "" {
|
||||||
|
settings.Path = req.Path
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := r.Database.CreateBucket(req.Bucket, settings); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON("ok")
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Get("/buckets/:bucket",
|
||||||
|
isAPIKeyMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
bucket := c.Params("bucket")
|
||||||
|
|
||||||
|
objects, err := r.Database.AllBucketObjects(bucket)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON(objects)
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Get("/buckets/:bucket/settings",
|
||||||
|
r.isAdminMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
bucket := c.Params("bucket")
|
||||||
|
|
||||||
|
settings, err := r.Database.GetBucketSettings(bucket)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON(settings)
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Post("/buckets/:bucket/settings",
|
||||||
|
r.isAdminMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
bucket := c.Params("bucket")
|
||||||
|
|
||||||
|
var settings *database.JsonBucketSettings
|
||||||
|
|
||||||
|
if err := c.BodyParser(&settings); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := r.Database.SetBucketSettings(bucket, settings); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON("ok")
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Post("/buckets/:bucket",
|
||||||
|
isAPIKeyMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
bucket := c.Params("bucket")
|
||||||
|
|
||||||
|
ff, err := c.FormFile("file")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
mf, err := ff.Open()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
id, err := r.Database.CreateBucketObject(bucket, mf)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON(fiber.Map{
|
||||||
|
"bucket": bucket,
|
||||||
|
"id": id,
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Get("/buckets/:bucket/:id",
|
||||||
|
isAPIKeyMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
bucket := c.Params("bucket")
|
||||||
|
id := c.Params("id")
|
||||||
|
|
||||||
|
buf := &bytes.Buffer{}
|
||||||
|
|
||||||
|
if err := r.Database.GetBucketObject(bucket, id, buf); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.SendStream(buf)
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Delete("/buckets/:bucket/:id",
|
||||||
|
isAPIKeyMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
bucket := c.Params("bucket")
|
||||||
|
id := c.Params("id")
|
||||||
|
|
||||||
|
if err := r.Database.DeleteBucketObject(bucket, id); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON("ok")
|
||||||
|
})
|
||||||
|
|
||||||
|
//
|
||||||
|
// API Keys
|
||||||
|
//
|
||||||
|
|
||||||
|
api.Get("/api-keys",
|
||||||
|
r.isAdminMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
apiKeys, err := r.Database.AllAPIKeys()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON(apiKeys)
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Post("/api-keys",
|
||||||
|
r.isAdminMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
key, err := r.Database.CreateAPIKey()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON(key)
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Delete("/api-keys/:key",
|
||||||
|
r.isAdminMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
if err := r.Database.RemoveAPIKey(c.Params("key")); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON("ok")
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Get("/api-keys/:key",
|
||||||
|
r.isAdminMiddleware,
|
||||||
|
func(c *fiber.Ctx) error {
|
||||||
|
if err := r.Database.CheckAPIKey(c.Params("key")); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON("valid")
|
||||||
|
})
|
||||||
|
}
|
@ -0,0 +1,29 @@
|
|||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/gofiber/fiber/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (r *Server) ApiMonitor(api fiber.Router) {
|
||||||
|
// Respond to requests like
|
||||||
|
// - "/api/monitor/status?script=SCRIPT_NAME" where SCRIPT_NAME is the name of a file inside "./scripts"
|
||||||
|
api.Get("/status", func(c *fiber.Ctx) error {
|
||||||
|
if qScript := c.Query("script"); qScript != "" {
|
||||||
|
output, err := r.Monitor.GetOutput(qScript)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON(output)
|
||||||
|
}
|
||||||
|
|
||||||
|
return fmt.Errorf("no script, device or entity provided")
|
||||||
|
})
|
||||||
|
|
||||||
|
api.Get("/invalidate-cache", func(c *fiber.Ctx) error {
|
||||||
|
r.Monitor.InvalidateCache()
|
||||||
|
return c.JSON("ok")
|
||||||
|
})
|
||||||
|
}
|
@ -0,0 +1,22 @@
|
|||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"git.phc.dm.unipi.it/phc/storage/database"
|
||||||
|
"git.phc.dm.unipi.it/phc/storage/serverinfo"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Server struct {
|
||||||
|
adminSessions map[string]struct{}
|
||||||
|
|
||||||
|
Database database.Database
|
||||||
|
Monitor *serverinfo.Service
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(db database.Database, m *serverinfo.Service) *Server {
|
||||||
|
return &Server{
|
||||||
|
adminSessions: map[string]struct{}{},
|
||||||
|
|
||||||
|
Database: db,
|
||||||
|
Monitor: m,
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue