53 lines
996 B
Go
53 lines
996 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"crypto/rsa"
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
|
|
_ "embed"
|
|
|
|
"tinyskin/config"
|
|
"tinyskin/server"
|
|
"tinyskin/utils/sign"
|
|
|
|
"github.com/pelletier/go-toml/v2"
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
var configPath string
|
|
|
|
func init() {
|
|
flag.StringVar(&configPath, "c", "config.toml", "")
|
|
flag.Parse()
|
|
}
|
|
|
|
func main() {
|
|
ctx := context.Background()
|
|
|
|
if _, err := os.Stat(configPath); errors.Is(err, os.ErrNotExist) {
|
|
rsa2048 := lo.Must(rsa.GenerateKey(rand.Reader, 4096))
|
|
as := sign.NewAuthlibSignWithKey(rsa2048)
|
|
|
|
c := config.Default()
|
|
c.RsaPriKey = lo.Must(as.GetPriKey())
|
|
|
|
lo.Must0(os.WriteFile(configPath, lo.Must(toml.Marshal(c)), 0600))
|
|
fmt.Println("未找到配置文件，已写入模板配置文件")
|
|
}
|
|
|
|
b, err := os.ReadFile(configPath)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
var config config.Config
|
|
lo.Must0(toml.Unmarshal(b, &config))
|
|
s, cancel := lo.Must2(server.InitializeRoute(ctx, config))
|
|
defer cancel()
|
|
panic(s.ListenAndServe())
|
|
}
|