mozhi/serve/serve.go
2024-04-14 18:43:31 +05:30

135 lines
4.0 KiB
Go

package serve
import (
"html/template"
"log"
"net/http"
"os"
"runtime"
"strings"
"codeberg.org/aryak/libmozhi"
_ "codeberg.org/aryak/mozhi/docs"
"codeberg.org/aryak/mozhi/pages"
"codeberg.org/aryak/mozhi/public"
"codeberg.org/aryak/mozhi/utils"
"codeberg.org/aryak/mozhi/views"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/fiber/v2/middleware/compress"
"github.com/gofiber/fiber/v2/middleware/filesystem"
//"github.com/gofiber/fiber/v2/middleware/limiter"
// For debugging purposes
// "github.com/gofiber/fiber/v2/middleware/logger"
"github.com/gofiber/fiber/v2/middleware/recover"
"github.com/gofiber/swagger"
"github.com/gofiber/template/html"
_ "github.com/joho/godotenv/autoload"
)
// @title Mozhi API
// @version 1.0
// @description API for Mozhi, the alternative-frontend for many translation engines.
// @license.name AGPL 3.0
// @license.url https://www.gnu.org/licenses/agpl-3.0.txt
// @BasePath /api
func Serve(port string) {
views := http.FS(views.GetFiles())
engine := html.NewFileSystem(views, ".html")
engine.AddFunc(
"newlinetobr", func(s string) template.HTML {
return template.HTML(strings.ReplaceAll(strings.ReplaceAll(s, "\n", "<br>"), "\r", ""))
},
)
app := fiber.New(fiber.Config{
Views: engine,
Prefork: false,
AppName: "Mozhi",
// kind of screwed up way to fix rate limits
EnableTrustedProxyCheck: true,
TrustedProxies: []string{"0.0.0.0/0"},
ProxyHeader: fiber.HeaderXForwardedFor,
})
// For debugging purposes
// app.Use(logger.New(logger.Config{
// Format: "[${ip}]:${port} ${status} - ${method} ${path} ${queryParams}\n",
// }))
app.Use(compress.New(compress.Config{
Level: compress.LevelBestSpeed, // 1
}))
app.Use(recover.New())
//ratelimiter := limiter.New(limiter.Config{
// Max: 5,
// Expiration: 5 * time.Minute,
// LimitReached: func(c *fiber.Ctx) error {
// return c.Status(429).Render("ratelimit_gt", fiber.Map{
// "Title": "Rate limit exceeded",
// })
// },
//})
// add global headers
app.Use(func(c *fiber.Ctx) error {
c.Set("X-Frame-Options", "SAMEORIGIN")
c.Set("X-XSS-Protection", "1; mode=block")
c.Set("X-Content-Type-Options", "nosniff")
c.Set("Referrer-Policy", "no-referrer")
c.Set("Content-Security-Policy", "default-src 'self'; style-src 'self' 'unsafe-inline'; img-src 'self' data:; font-src 'self'; script-src 'self' 'unsafe-inline'; frame-ancestors 'self'; form-action 'self'; base-uri 'self'; connect-src 'self';")
c.Set("Strict-Transport-Security", "max-age=31536000; includeSubDomains; preload")
return c.Next()
})
api := app.Group("/api")
api.All("/translate", pages.HandleTranslate)
api.All("/image", pages.HandleImg)
api.Get("/source_languages", pages.HandleSourceLanguages)
api.Get("/target_languages", pages.HandleTargetLanguages)
api.Get("/engines", pages.HandleEngines)
api.Get("/tts", pages.HandleTTS)
api.Get("/version", func(c *fiber.Ctx) error {
return c.JSON(fiber.Map{
"fiberversion": fiber.Version,
"goversion": runtime.Version(),
})
})
api.Get("/swagger/*", swagger.HandlerDefault) // default
app.All("/", pages.HandleIndex)
app.All("/switchlanguages", func(c *fiber.Ctx) error {
engine := utils.Sanitize(utils.GetQueryOrFormValue(c, "engine"), "alpha")
from := utils.Sanitize(utils.GetQueryOrFormValue(c, "from"), "alpha")
to := utils.Sanitize(utils.GetQueryOrFormValue(c, "to"), "alpha")
text := utils.GetQueryOrFormValue(c, "text")
var swapText string
if engine != "all" && text != "" {
translation, tlerr := libmozhi.Translate(engine, to, from, text)
if tlerr == nil {
swapText = translation.OutputText
} else {
swapText = text
}
}
return c.Redirect("/?engine="+engine+"&from="+to+"&to="+from+"&text="+swapText+"&redirected=true", 301)
})
app.Get("/about", pages.HandleAbout)
app.Use("/", filesystem.New(filesystem.Config{
MaxAge: 2592000,
Root: http.FS(public.GetFiles()),
}))
val, ok := os.LookupEnv("MOZHI_PORT")
if !ok {
val = "3000"
}
if port != "" {
val = port
}
log.Fatal(app.Listen(":" + val))
}