storage/server/api.go

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")
		})
}