84 lines
1.9 KiB
Go
84 lines
1.9 KiB
Go
package controllers
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"mime/multipart"
|
|
"net/http"
|
|
|
|
"git.myrkvi.com/myrkvi/bin/global"
|
|
"git.myrkvi.com/myrkvi/bin/models"
|
|
"git.myrkvi.com/myrkvi/bin/utils"
|
|
"git.myrkvi.com/myrkvi/bin/views/components"
|
|
"git.myrkvi.com/myrkvi/bin/views/pages"
|
|
"github.com/alecthomas/chroma/lexers"
|
|
"github.com/labstack/echo/v4"
|
|
)
|
|
|
|
func GetNewHandler(c echo.Context) error {
|
|
values := c.QueryParams()
|
|
wantsText := values.Has("text")
|
|
return utils.RenderComponents(c, http.StatusOK, pages.NewFull(wantsText))
|
|
}
|
|
|
|
func PostNewHandler(c echo.Context) error {
|
|
file, err := c.FormFile("file")
|
|
text := c.FormValue("text")
|
|
|
|
name := c.FormValue("name")
|
|
description := c.FormValue("description")
|
|
lang := c.FormValue("lang")
|
|
ip := c.RealIP()
|
|
|
|
if (file == nil || err != nil) && text == "" {
|
|
return utils.RenderErrorToast(c, "file or text must be provided")
|
|
}
|
|
|
|
code, adminCode, err := utils.GenerateCodes()
|
|
if err != nil {
|
|
return utils.RenderErrorToast(c, "server-side error occurred")
|
|
}
|
|
|
|
if file != nil {
|
|
text, err = getTextFromFile(file)
|
|
if name == "" {
|
|
name = file.Filename
|
|
}
|
|
if err != nil {
|
|
return utils.RenderErrorToast(c, "server-side error occurred")
|
|
}
|
|
}
|
|
// Determine language from file extension if not set.
|
|
if lang == "" {
|
|
lexer := lexers.Match(name)
|
|
if lexer != nil {
|
|
lang = lexer.Config().Name
|
|
}
|
|
}
|
|
|
|
createdFile, err := models.CreateNewBin(global.DB, text, name, description, lang, code, adminCode, ip)
|
|
if err != nil {
|
|
return utils.RenderErrorToast(c, "server-side error occurred")
|
|
}
|
|
|
|
utils.SetHeader(c, "HX-Push", fmt.Sprintf("/b/%s", code))
|
|
return utils.RenderComponents(c, http.StatusOK,
|
|
pages.BinPartial(createdFile),
|
|
components.SetTitle(file.Filename),
|
|
)
|
|
}
|
|
|
|
func getTextFromFile(fileHeader *multipart.FileHeader) (text string, err error) {
|
|
f, err := fileHeader.Open()
|
|
if err != nil {
|
|
return
|
|
}
|
|
b, err := io.ReadAll(f)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
text = string(b)
|
|
return
|
|
}
|