74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/rs/cors"
|
|
"github.com/uptrace/bunrouter"
|
|
"lure.sh/lure/pkg/loggerctx"
|
|
"lure.sh/lure/pkg/search"
|
|
"go.elara.ws/logger/log"
|
|
)
|
|
|
|
func registerAPI(mux *bunrouter.Router) {
|
|
g := mux.NewGroup(
|
|
"/api",
|
|
bunrouter.WithMiddleware(apiErrorHandler, corsMiddleware()),
|
|
)
|
|
|
|
g.GET("/*path", func(w http.ResponseWriter, req bunrouter.Request) error {
|
|
return HTTPError{404, "endpoint not found: " + req.URL.Path}
|
|
})
|
|
|
|
g.GET("/search", func(w http.ResponseWriter, req bunrouter.Request) error {
|
|
ctx := loggerctx.With(req.Context(), log.Logger)
|
|
pkgs, err := search.Search(ctx, searchOptions(req.Request))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return bunrouter.JSON(w, translatePkgs(getLanguages(req.Request), pkgs))
|
|
})
|
|
|
|
g.GET("/pkg/:repo/:pkg", func(w http.ResponseWriter, req bunrouter.Request) error {
|
|
ctx := loggerctx.With(req.Context(), log.Logger)
|
|
pkg, err := search.GetPkg(ctx, req.Param("repo"), req.Param("pkg"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return bunrouter.JSON(w, translatePkg(getLanguages(req.Request), pkg))
|
|
})
|
|
|
|
g.GET("/pkg/:repo/:pkg/script", func(w http.ResponseWriter, req bunrouter.Request) error {
|
|
ctx := loggerctx.With(req.Context(), log.Logger)
|
|
script, err := search.GetScript(ctx, req.Param("repo"), req.Param("pkg"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer script.Close()
|
|
|
|
if req.URL.Query().Get("highlight") == "true" {
|
|
data, err := highlight(script)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
w.Header().Set("Content-Type", "text/html")
|
|
_, err = io.WriteString(w, string(data))
|
|
return err
|
|
} else {
|
|
w.Header().Set("Content-Type", "text/x-shellscript")
|
|
_, err = io.Copy(w, script)
|
|
return err
|
|
}
|
|
})
|
|
}
|
|
|
|
func corsMiddleware() bunrouter.MiddlewareFunc {
|
|
ch := cors.AllowAll()
|
|
return func(next bunrouter.HandlerFunc) bunrouter.HandlerFunc {
|
|
return bunrouter.HTTPHandler(ch.Handler(next))
|
|
}
|
|
}
|