63 lines
1.2 KiB
Go
63 lines
1.2 KiB
Go
|
package handlers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/gofiber/fiber/v3"
|
||
|
|
"lmika.dev/lmika/weiro/services/uploads"
|
||
|
|
)
|
||
|
|
|
||
|
|
type UploadsHandler struct {
|
||
|
|
UploadsService *uploads.Service
|
||
|
|
}
|
||
|
|
|
||
|
|
func (uh UploadsHandler) Index(c fiber.Ctx) error {
|
||
|
|
return c.Render("uploads/index", nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (uh UploadsHandler) New(c fiber.Ctx) error {
|
||
|
|
var req uploads.NewPendingRequest
|
||
|
|
|
||
|
|
if err := c.Bind().Body(&req); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
res, err := uh.UploadsService.NewPending(c.Context(), req)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return c.JSON(res)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (uh UploadsHandler) UploadPart(c fiber.Ctx) error {
|
||
|
|
guid := c.Params("guid")
|
||
|
|
if guid == "" {
|
||
|
|
return fiber.ErrBadRequest
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := uh.UploadsService.WriteToPending(c.Context(), guid, c.Body()); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return c.Status(fiber.StatusAccepted).JSON(fiber.Map{})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (uh UploadsHandler) UploadComplete(c fiber.Ctx) error {
|
||
|
|
guid := c.Params("guid")
|
||
|
|
if guid == "" {
|
||
|
|
return fiber.ErrBadRequest
|
||
|
|
}
|
||
|
|
|
||
|
|
var res struct {
|
||
|
|
Hash string `json:"hash"`
|
||
|
|
}
|
||
|
|
if err := c.Bind().Body(&res); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := uh.UploadsService.FinalizePending(c.Context(), guid, res.Hash); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return c.Status(fiber.StatusAccepted).JSON(fiber.Map{})
|
||
|
|
}
|