91 lines
2.6 KiB
Go
91 lines
2.6 KiB
Go
package yggdrasil
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"io"
|
|
"log/slog"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
"github.com/go-playground/validator/v10"
|
|
"github.com/julienschmidt/httprouter"
|
|
"github.com/samber/lo"
|
|
"github.com/xmdhs/authlib-skin/config"
|
|
"github.com/xmdhs/authlib-skin/model/yggdrasil"
|
|
yggdrasilM "github.com/xmdhs/authlib-skin/model/yggdrasil"
|
|
yggdrasilS "github.com/xmdhs/authlib-skin/service/yggdrasil"
|
|
"github.com/xmdhs/authlib-skin/utils"
|
|
)
|
|
|
|
type PubRsaKey string
|
|
|
|
type Yggdrasil struct {
|
|
logger *slog.Logger
|
|
validate *validator.Validate
|
|
yggdrasilService *yggdrasilS.Yggdrasil
|
|
config config.Config
|
|
pubkey PubRsaKey
|
|
}
|
|
|
|
func NewYggdrasil(logger *slog.Logger, validate *validator.Validate, yggdrasilService *yggdrasilS.Yggdrasil, config config.Config, pubkey PubRsaKey) *Yggdrasil {
|
|
return &Yggdrasil{
|
|
logger: logger,
|
|
validate: validate,
|
|
yggdrasilService: yggdrasilService,
|
|
config: config,
|
|
pubkey: pubkey,
|
|
}
|
|
}
|
|
|
|
func getAnyModel[K any](ctx context.Context, w http.ResponseWriter, r io.Reader, validate *validator.Validate, slog *slog.Logger) (K, bool) {
|
|
a, err := utils.DeCodeBody[K](r, validate)
|
|
if err != nil {
|
|
slog.DebugContext(ctx, err.Error())
|
|
handleYgError(ctx, w, yggdrasil.Error{ErrorMessage: err.Error()}, 400)
|
|
return a, false
|
|
}
|
|
return a, true
|
|
}
|
|
|
|
func (y *Yggdrasil) YggdrasilRoot() httprouter.Handle {
|
|
return func(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
|
|
var host string
|
|
if y.config.TextureBaseUrl != "" {
|
|
u := lo.Must(url.Parse(y.config.TextureBaseUrl))
|
|
host = u.Hostname()
|
|
} else {
|
|
host, _ = lo.TryOr[string](func() (string, error) {
|
|
h, _, err := net.SplitHostPort(r.Host)
|
|
return h, err
|
|
}, r.Host)
|
|
}
|
|
homepage, _ := url.JoinPath(y.config.WebBaseUrl, "/login")
|
|
register, _ := url.JoinPath(y.config.WebBaseUrl, "/register")
|
|
|
|
w.Write(lo.Must1(json.Marshal(yggdrasilM.Yggdrasil{
|
|
Meta: yggdrasilM.YggdrasilMeta{
|
|
ImplementationName: "authlib-skin",
|
|
ImplementationVersion: "0.0.1",
|
|
Links: yggdrasilM.YggdrasilMetaLinks{
|
|
Homepage: homepage,
|
|
Register: register,
|
|
},
|
|
ServerName: y.config.ServerName,
|
|
EnableProfileKey: true,
|
|
},
|
|
SignaturePublickey: string(y.pubkey),
|
|
SkinDomains: []string{host},
|
|
})))
|
|
|
|
}
|
|
}
|
|
|
|
func (y *Yggdrasil) TextureAssets() httprouter.Handle {
|
|
return func(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
|
|
w.Header().Set("Content-Type", "image/png")
|
|
http.StripPrefix("/texture/", http.FileServer(http.Dir(y.config.TexturePath))).ServeHTTP(w, r)
|
|
}
|
|
}
|