package main import ( "flag" "fmt" "net/http" "os" "time" "codeberg.org/rimgo/rimgo/pages" "codeberg.org/rimgo/rimgo/static" "codeberg.org/rimgo/rimgo/utils" "codeberg.org/rimgo/rimgo/views" "github.com/aymerick/raymond" "github.com/gofiber/fiber/v2" "github.com/gofiber/fiber/v2/middleware/cache" "github.com/gofiber/fiber/v2/middleware/filesystem" "github.com/gofiber/fiber/v2/middleware/recover" "github.com/gofiber/template/handlebars/v2" "github.com/joho/godotenv" ) func main() { envPath := flag.String("c", ".env", "Path to env file") godotenv.Load(*envPath) utils.LoadConfig() pages.InitializeApiClient() views := http.FS(views.GetFiles()) if os.Getenv("ENV") == "dev" { views = http.Dir("./views") } engine := handlebars.NewFileSystem(views, ".hbs") engine.AddFunc("noteq", func(a interface{}, b interface{}, options *raymond.Options) interface{} { if raymond.Str(a) != raymond.Str(b) { return options.Fn() } return "" }) app := fiber.New(fiber.Config{ Views: engine, Prefork: utils.Config.FiberPrefork, UnescapePath: true, StreamRequestBody: true, ErrorHandler: func(ctx *fiber.Ctx, err error) error { code := fiber.StatusInternalServerError if e, ok := err.(*fiber.Error); ok { code = e.Code } err = ctx.Status(code).Render("errors/error", fiber.Map{ "err": err, }) if err != nil { return ctx.Status(fiber.StatusInternalServerError).SendString("Internal Server Error") } return nil }, }) app.Use(recover.New(recover.Config{ EnableStackTrace: true, StackTraceHandler: func(c *fiber.Ctx, e interface{}) { fmt.Println(e) }, })) if os.Getenv("ENV") == "dev" { app.Use("/static", filesystem.New(filesystem.Config{ MaxAge: 2592000, Root: http.Dir("./static"), })) app.Get("/errors/429", func(c *fiber.Ctx) error { return c.Render("errors/429", nil) }) app.Get("/errors/404", func(c *fiber.Ctx) error { return c.Render("errors/404", nil) }) app.Get("/errors/error", func(c *fiber.Ctx) error { return c.Render("errors/error", fiber.Map{ "err": "Test error", }) }) } else { app.Use("/static", filesystem.New(filesystem.Config{ Root: http.FS(static.GetFiles()), })) app.Use(cache.New(cache.Config{ Expiration: 30 * time.Minute, MaxBytes: 25000000, KeyGenerator: func(c *fiber.Ctx) string { return c.OriginalURL() }, CacheControl: true, StoreResponseHeaders: true, })) } 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 }) app.Get("/", pages.HandleFrontpage) app.Get("/about", pages.HandleAbout) app.Get("/privacy", pages.HandlePrivacy) app.Get("/search", pages.HandleSearch) app.Get("/a/:postID", pages.HandlePost) app.Get("/a/:postID/embed", pages.HandleEmbed) app.Get("/t/:tag", pages.HandleTag) app.Get("/t/:tag/:postID", pages.HandlePost) app.Get("/user/:userID", pages.HandleUser) app.Get("/r/:sub/:postID", pages.HandlePost) app.Get("/user/:userID/cover", pages.HandleUserCover) app.Get("/user/:userID/avatar", pages.HandleUserAvatar) app.Get("/gallery/:postID", pages.HandlePost) app.Get("/gallery/:postID/embed", pages.HandleEmbed) app.Get("/:postID.gifv", pages.HandleGifv) app.Get("/:baseName.:extension", pages.HandleMedia) app.Get("/stack/:baseName.:extension", pages.HandleMedia) app.Get("/:postID", pages.HandlePost) app.Get("/:postID/embed", pages.HandleEmbed) app.Listen(utils.Config.Addr + ":" + utils.Config.Port) }