rimgo/main.go

67 lines
1.9 KiB
Go
Raw Normal View History

2022-01-17 20:23:04 +00:00
package main
import (
2022-05-31 01:33:44 +00:00
"fmt"
2022-01-17 20:23:04 +00:00
"net/http"
2022-01-17 22:34:05 +00:00
"codeberg.org/video-prize-ranch/rimgo/pages"
"codeberg.org/video-prize-ranch/rimgo/static"
2022-04-22 15:55:53 +00:00
"codeberg.org/video-prize-ranch/rimgo/utils"
2022-01-17 22:34:05 +00:00
"codeberg.org/video-prize-ranch/rimgo/views"
2022-01-17 20:23:04 +00:00
"github.com/gofiber/fiber/v2"
"github.com/gofiber/fiber/v2/middleware/filesystem"
2022-04-22 15:38:09 +00:00
"github.com/gofiber/fiber/v2/middleware/recover"
2022-01-17 20:23:04 +00:00
"github.com/gofiber/template/handlebars"
)
func main() {
2022-04-22 15:55:53 +00:00
utils.LoadConfig()
2022-01-17 22:34:05 +00:00
2022-01-17 20:23:04 +00:00
engine := handlebars.NewFileSystem(http.FS(views.GetFiles()), ".hbs")
app := fiber.New(fiber.Config{
2022-01-17 22:34:05 +00:00
Views: engine,
2022-04-22 15:55:53 +00:00
Prefork: utils.Config["fiberPrefork"].(bool),
2022-01-17 22:34:05 +00:00
UnescapePath: true,
2022-01-17 20:23:04 +00:00
StreamRequestBody: true,
})
2022-05-31 01:33:44 +00:00
app.Use(recover.New(recover.Config{
EnableStackTrace: true,
StackTraceHandler: func (c *fiber.Ctx, e interface{}) {
fmt.Println(e)
},
}))
2022-01-17 20:23:04 +00:00
app.Use("/static", filesystem.New(filesystem.Config{
2022-06-18 16:21:22 +00:00
Next: func(c *fiber.Ctx) bool {
c.Response().Header.Add("Cache-Control", "public,max-age=2592000")
return false
},
2022-01-17 20:23:04 +00:00
Root: http.FS(static.GetFiles()),
}))
2022-01-21 23:35:12 +00:00
app.Get("/robots.txt", func(c *fiber.Ctx) error {
file, _ := static.GetFiles().ReadFile("robots.txt")
_, err := c.Write(file)
return err
})
app.Get("/favicon.ico", func(c *fiber.Ctx) error {
file, _ := static.GetFiles().ReadFile("favicon/favicon.ico")
_, err := c.Write(file)
return err
})
2022-01-21 23:35:12 +00:00
app.Get("/", pages.HandleFrontpage)
app.Get("/:postID.gifv", pages.HandleEmbed)
2022-01-17 22:11:33 +00:00
app.Get("/:baseName.:extension", pages.HandleMedia)
app.Get("/:postID", pages.HandlePost)
2022-05-23 15:10:50 +00:00
app.Get("/a/:postID", pages.HandlePost)
2022-02-18 21:56:56 +00:00
app.Get("/t/:tag", pages.HandleTag)
2022-01-28 01:41:10 +00:00
app.Get("/user/:userID", pages.HandleUser)
app.Get("/r/:sub/:postID", pages.HandlePost)
2022-01-28 01:41:10 +00:00
app.Get("/user/:userID/cover", pages.HandleUserCover)
app.Get("/user/:userID/avatar", pages.HandleUserAvatar)
2022-05-23 15:10:50 +00:00
app.Get("/gallery/:postID", pages.HandlePost)
2022-01-17 20:23:04 +00:00
2022-06-12 16:12:36 +00:00
app.Listen(utils.Config["addr"].(string) + ":" + utils.Config["port"].(string))
2022-01-17 22:34:05 +00:00
}