55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"crypto/rsa"
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
|
|
_ "embed"
|
|
|
|
"github.com/samber/lo"
|
|
"github.com/xmdhs/authlib-skin/config"
|
|
"github.com/xmdhs/authlib-skin/server"
|
|
"github.com/xmdhs/authlib-skin/utils/sign"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
var configPath string
|
|
|
|
//go:embed config.yaml.template
|
|
var configTempLate []byte
|
|
|
|
func init() {
|
|
flag.StringVar(&configPath, "c", "config.yaml", "")
|
|
flag.Parse()
|
|
}
|
|
|
|
func main() {
|
|
ctx := context.Background()
|
|
b, err := os.ReadFile(configPath)
|
|
if err != nil {
|
|
if errors.Is(err, os.ErrNotExist) {
|
|
lo.Must0(os.WriteFile(configPath, configTempLate, 0600))
|
|
fmt.Println("未找到配置文件，已写入模板配置文件")
|
|
return
|
|
}
|
|
panic(err)
|
|
}
|
|
config := lo.Must(config.YamlDeCode(b))
|
|
|
|
if config.RsaPriKey == "" {
|
|
rsa2048 := lo.Must(rsa.GenerateKey(rand.Reader, 4096))
|
|
as := sign.NewAuthlibSignWithKey(rsa2048)
|
|
config.RsaPriKey = lo.Must(as.GetPriKey())
|
|
lo.Must0(os.WriteFile(configPath, lo.Must(yaml.Marshal(config)), 0600))
|
|
}
|
|
|
|
s, cancel := lo.Must2(server.InitializeRoute(ctx, config))
|
|
defer cancel()
|
|
panic(s.ListenAndServe())
|
|
}
|