lure-web/main.go

50 lines
1002 B
Go

package main
import (
"context"
"embed"
"net/http"
"os"
"github.com/uptrace/bunrouter"
"go.elara.ws/logger"
"go.elara.ws/logger/log"
"lure.sh/lure/pkg/loggerctx"
"lure.sh/lure/pkg/repos"
)
//go:embed static
var static embed.FS
func init() {
log.Logger = logger.NewPretty(os.Stderr)
}
func main() {
mux := bunrouter.New()
fileServer := http.FileServer(http.FS(static))
mux.Use(cache).GET("/static/*path", bunrouter.HTTPHandler(fileServer))
ctx := loggerctx.With(context.Background(), log.Logger)
err := repos.Pull(ctx, nil)
if err != nil {
log.Fatal("Error pulling repos").Err(err).Send()
}
registerBadge(mux)
registerSite(mux)
registerAPI(mux)
registerWebhook(ctx, mux)
http.ListenAndServe(":8080", mux)
}
func cache(next bunrouter.HandlerFunc) bunrouter.HandlerFunc {
return bunrouter.HandlerFunc(func(w http.ResponseWriter, req bunrouter.Request) error {
w.Header().Set("Cache-Control", "max-age=86400, stale-while-revalidate=86400")
return next(w, req)
})
}