mirror of
https://github.com/woodchen-ink/webp_server_go.git
synced 2025-07-18 05:32:02 +08:00
a7b5992662
* recover middleware * simplify Atoi * metadata data prototype * InterestingAttention * resize itself * Bump version to 0.9.4 Added some comments Removed String() for Extraparams * Add metadata test * Fix CI * Remove unnecessary tests * Update file count * use t.Run to get test case --------- Co-authored-by: n0vad3v <n0vad3v@riseup.net>
98 lines
2.3 KiB
Go
98 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"runtime"
|
|
"webp_server_go/config"
|
|
"webp_server_go/encoder"
|
|
"webp_server_go/handler"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/fiber/v2/middleware/etag"
|
|
"github.com/gofiber/fiber/v2/middleware/logger"
|
|
"github.com/gofiber/fiber/v2/middleware/recover"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var app = fiber.New(fiber.Config{
|
|
ServerHeader: "Webp Server Go",
|
|
AppName: "Webp Server Go",
|
|
DisableStartupMessage: true,
|
|
ProxyHeader: "X-Real-IP",
|
|
})
|
|
|
|
func setupLogger() {
|
|
log.SetOutput(os.Stdout)
|
|
log.SetReportCaller(true)
|
|
formatter := &log.TextFormatter{
|
|
EnvironmentOverrideColors: true,
|
|
FullTimestamp: true,
|
|
TimestampFormat: config.TimeDateFormat,
|
|
CallerPrettyfier: func(f *runtime.Frame) (string, string) {
|
|
return fmt.Sprintf("[%d:%s]", f.Line, f.Function), ""
|
|
},
|
|
}
|
|
log.SetFormatter(formatter)
|
|
log.SetLevel(log.InfoLevel)
|
|
|
|
// fiber logger format
|
|
app.Use(logger.New(logger.Config{
|
|
Format: config.FiberLogFormat,
|
|
TimeFormat: config.TimeDateFormat,
|
|
}))
|
|
app.Use(recover.New(recover.Config{}))
|
|
log.Infoln("fiber ready.")
|
|
}
|
|
|
|
func init() {
|
|
// main init is the last one to be called
|
|
flag.Parse()
|
|
config.LoadConfig()
|
|
setupLogger()
|
|
}
|
|
|
|
func main() {
|
|
// Our banner
|
|
banner := fmt.Sprintf(`
|
|
▌ ▌ ▌ ▛▀▖ ▞▀▖ ▞▀▖
|
|
▌▖▌▞▀▖▛▀▖▙▄▘ ▚▄ ▞▀▖▙▀▖▌ ▌▞▀▖▙▀▖ ▌▄▖▞▀▖
|
|
▙▚▌▛▀ ▌ ▌▌ ▖ ▌▛▀ ▌ ▐▐ ▛▀ ▌ ▌ ▌▌ ▌
|
|
▘ ▘▝▀▘▀▀ ▘ ▝▀ ▝▀▘▘ ▘ ▝▀▘▘ ▝▀ ▝▀
|
|
|
|
WebP Server Go - v%s
|
|
Develop by WebP Server team. https://github.com/webp-sh`, config.Version)
|
|
|
|
// process cli params
|
|
if config.DumpConfig {
|
|
fmt.Println(config.SampleConfig)
|
|
os.Exit(0)
|
|
}
|
|
if config.DumpSystemd {
|
|
fmt.Println(config.SampleSystemd)
|
|
os.Exit(0)
|
|
}
|
|
if config.ShowVersion {
|
|
fmt.Printf("\n %c[1;32m%s%c[0m\n\n", 0x1B, banner+"", 0x1B)
|
|
os.Exit(0)
|
|
}
|
|
|
|
if config.Prefetch {
|
|
go encoder.PrefetchImages()
|
|
}
|
|
|
|
app.Use(etag.New(etag.Config{
|
|
Weak: true,
|
|
}))
|
|
|
|
listenAddress := config.Config.Host + ":" + config.Config.Port
|
|
app.Get("/*", handler.Convert)
|
|
|
|
fmt.Printf("\n %c[1;32m%s%c[0m\n\n", 0x1B, banner, 0x1B)
|
|
fmt.Println("WebP Server Go is Running on http://" + listenAddress)
|
|
|
|
_ = app.Listen(listenAddress)
|
|
|
|
}
|