mirror of
https://github.com/SphericalKat/medium.rip.git
synced 2025-12-18 07:25:58 +00:00
78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"crypto/tls"
|
|
"sync"
|
|
|
|
"github.com/medium.rip/api/routes"
|
|
"github.com/medium.rip/internal/config"
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/fiber/v2/middleware/filesystem"
|
|
"github.com/gofiber/fiber/v2/middleware/proxy"
|
|
"github.com/gofiber/template/html/v2"
|
|
)
|
|
|
|
func RegisterRoutes(ctx context.Context, wg *sync.WaitGroup, engine *html.Engine, fs http.FileSystem) {
|
|
// more Fiber options at https://docs.gofiber.io/api/fiber/
|
|
app := fiber.New(fiber.Config{
|
|
StreamRequestBody: true,
|
|
ServerHeader: "Katbox",
|
|
AppName: "Katbox",
|
|
DisableStartupMessage: false,
|
|
Views: engine,
|
|
Network: "tcp",
|
|
})
|
|
|
|
// static file server
|
|
app.Use("/assets", filesystem.New(filesystem.Config{
|
|
Root: fs,
|
|
Browse: config.Conf.Env == "dev",
|
|
}))
|
|
|
|
if config.Conf.Proxy != "" {
|
|
proxy.WithTlsConfig(&tls.Config{ InsecureSkipVerify: true,})
|
|
app.Use(proxy.Balancer(proxy.Config{
|
|
Servers: []string{ config.Conf.Proxy, },
|
|
ModifyRequest: func(c *fiber.Ctx) error {
|
|
//c.Request().Header.Add("X-Real-IP", c.IP())
|
|
//return nil
|
|
req_code, req_body, req_errs := c.String()
|
|
log.Printf("REQUEST: %s", req_body)
|
|
return nil
|
|
},
|
|
ModifyResponse: func(c *fiber.Ctx) error {
|
|
//c.Response().Header.Del(fiber.HeaderServer)
|
|
//return nil
|
|
res_code, res_body, res_errs := c.String()
|
|
log.Printf("RESPONSE: %s | %s | %s ", res_code, res_body, res_errs)
|
|
return nil
|
|
},
|
|
}))
|
|
log.Printf("Using proxy: %s", config.Conf.Proxy)
|
|
}
|
|
|
|
routes.RegisterRoutes(app)
|
|
|
|
go func(app *fiber.App) {
|
|
log.Printf("Starting http server at: http://localhost:%s", config.Conf.Port)
|
|
if err := app.Listen(fmt.Sprintf(":%s", config.Conf.Port)); err != nil {
|
|
log.Fatalf("Unable to start http server: %s", err)
|
|
}
|
|
}(app)
|
|
|
|
// listen for context cancellation
|
|
<-ctx.Done()
|
|
|
|
// shut down http server
|
|
log.Info("Gracefully shutting down http server...")
|
|
if err := app.Shutdown(); err != nil {
|
|
log.Warn("Server shutdown Failed: ", err)
|
|
}
|
|
wg.Done()
|
|
}
|