重构项目
This commit is contained in:
ExplodingDragon
149
cmd/server/config.go
Normal file
149
cmd/server/config.go
Normal file
@@ -0,0 +1,149 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
_ "embed"
|
||||
"net/http"
|
||||
"os"
|
||||
"text/template"
|
||||
"time"
|
||||
|
||||
"github.com/alecthomas/units"
|
||||
"github.com/pkg/errors"
|
||||
"go.uber.org/zap"
|
||||
"gopkg.d7z.net/gitea-pages/pkg/utils"
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
//go:embed errors.html.tmpl
|
||||
var defaultErrPage string
|
||||
|
||||
type Config struct {
|
||||
Bind string `yaml:"bind"` // HTTP 绑定
|
||||
Domain string `yaml:"domain"` // 基础域名
|
||||
|
||||
Database ConfigDatabase `yaml:"database"` // 配置
|
||||
|
||||
Auth ConfigAuth `yaml:"auth"` // 后端认证配置
|
||||
|
||||
Cache ConfigCache `yaml:"cache"` // 缓存配置
|
||||
|
||||
Page ConfigPage `yaml:"page"` // 页面配置
|
||||
|
||||
Render ConfigRender `yaml:"render"` // 渲染配置
|
||||
Proxy ConfigProxy `yaml:"proxy"` // 反向代理配置
|
||||
|
||||
StaticDir string `yaml:"static"` // 静态资源提供路径
|
||||
|
||||
pageErrNotFound, pageErrUnknown *template.Template
|
||||
}
|
||||
|
||||
func (c *Config) ErrorHandler(w http.ResponseWriter, r *http.Request, err error) {
|
||||
if errors.Is(err, os.ErrNotExist) {
|
||||
w.WriteHeader(http.StatusNotFound)
|
||||
if err = c.pageErrNotFound.Execute(w, utils.NewTemplateInject(r, map[string]any{
|
||||
"UUID": r.Header.Get("Session-ID"),
|
||||
"Error": err,
|
||||
"Path": r.URL.Path,
|
||||
"Code": 404,
|
||||
})); err != nil {
|
||||
zap.L().Error("failed to render error page", zap.Error(err))
|
||||
}
|
||||
} else {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
if err = c.pageErrUnknown.Execute(w, utils.NewTemplateInject(r, map[string]any{
|
||||
"UUID": r.Header.Get("Session-ID"),
|
||||
"Error": err,
|
||||
"Path": r.URL.Path,
|
||||
"Code": 500,
|
||||
})); err != nil {
|
||||
zap.L().Error("failed to render error page", zap.Error(err))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
type ConfigAuth struct {
|
||||
// 服务器地址
|
||||
Server string `yaml:"server"`
|
||||
// 会话 Id
|
||||
Token string `yaml:"token"`
|
||||
}
|
||||
|
||||
type ConfigPage struct {
|
||||
DefaultBranch string `yaml:"default_branch"`
|
||||
ErrNotFoundPage string `yaml:"404"`
|
||||
ErrUnknownPage string `yaml:"500"`
|
||||
}
|
||||
|
||||
type ConfigDatabase struct {
|
||||
URL string `yaml:"url"`
|
||||
}
|
||||
|
||||
type ConfigProxy struct {
|
||||
Enable bool `yaml:"enable"` // 是否允许反向代理
|
||||
}
|
||||
|
||||
type ConfigRender struct {
|
||||
Enable bool `yaml:"enable"` // 是否开启渲染器
|
||||
}
|
||||
|
||||
type ConfigCache struct {
|
||||
Meta string `yaml:"meta"` // 元数据缓存
|
||||
MetaTTL time.Duration `yaml:"meta_ttl"` // 缓存时间
|
||||
|
||||
Blob string `yaml:"blob"` // 缓存归档位置
|
||||
BlobTTL time.Duration `yaml:"blob_ttl"` // 缓存归档位置
|
||||
BlobLimit units.Base2Bytes `yaml:"blob_limit"` // 单个文件最大大小
|
||||
}
|
||||
|
||||
func LoadConfig(path string) (*Config, error) {
|
||||
f, err := os.Open(path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer f.Close()
|
||||
var c Config
|
||||
decoder := yaml.NewDecoder(f)
|
||||
err = decoder.Decode(&c)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if c.Domain == "" {
|
||||
return nil, errors.New("domain is required")
|
||||
}
|
||||
if c.Database.URL == "" {
|
||||
return nil, errors.New("c is required")
|
||||
}
|
||||
if c.StaticDir != "" {
|
||||
stat, err := os.Stat(c.StaticDir)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "static dir not exists")
|
||||
}
|
||||
if !stat.IsDir() {
|
||||
return nil, errors.New("static dir is not a directory")
|
||||
}
|
||||
}
|
||||
if c.Page.DefaultBranch == "" {
|
||||
c.Page.DefaultBranch = "gh-pages"
|
||||
}
|
||||
defaultErr := utils.MustTemplate(defaultErrPage)
|
||||
if c.Page.ErrUnknownPage != "" {
|
||||
data, err := os.ReadFile(c.Page.ErrUnknownPage)
|
||||
if err != nil {
|
||||
return nil, errors.Wrapf(err, "failed to read file %s", string(data))
|
||||
}
|
||||
c.pageErrUnknown = utils.MustTemplate(string(data))
|
||||
} else {
|
||||
c.pageErrUnknown = defaultErr
|
||||
}
|
||||
if c.Page.ErrNotFoundPage != "" {
|
||||
data, err := os.ReadFile(c.Page.ErrNotFoundPage)
|
||||
if err != nil {
|
||||
return nil, errors.Wrapf(err, "failed to read file %s", c.Page.ErrNotFoundPage)
|
||||
}
|
||||
c.pageErrNotFound = utils.MustTemplate(string(data))
|
||||
} else {
|
||||
c.pageErrNotFound = defaultErr
|
||||
}
|
||||
return &c, nil
|
||||
}
|
||||
17
cmd/server/errors.html.tmpl
Normal file
17
cmd/server/errors.html.tmpl
Normal file
@@ -0,0 +1,17 @@
|
||||
<!DOCTYPE html>
|
||||
<html>
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport"
|
||||
content="width=device-width, user-scalable=no, initial-scale=1.0, maximum-scale=1.0, minimum-scale=1.0">
|
||||
<meta http-equiv="X-UA-Compatible" content="ie=edge">
|
||||
{{ if eq .Code 404 }}<title>404 Not Found</title>{{ else }}<title>500 Unknown Error</title>{{ end }}
|
||||
</head>
|
||||
<Body>
|
||||
<div style="text-align: center;">
|
||||
{{ if eq .Code 404 }}<h1>404 Not Found</h1>{{ else }}<h1>500 Unknown Error</h1>{{ end }}
|
||||
</div>
|
||||
<hr>
|
||||
<div style="text-align: center;">Gitea Pages({{.Request.Host}})/{{ .UUID }}</div>
|
||||
</Body>
|
||||
</html>
|
||||
102
cmd/server/main.go
Normal file
102
cmd/server/main.go
Normal file
@@ -0,0 +1,102 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
|
||||
"go.uber.org/zap"
|
||||
"gopkg.d7z.net/gitea-pages/pkg"
|
||||
"gopkg.d7z.net/gitea-pages/pkg/providers"
|
||||
"gopkg.d7z.net/middleware/cache"
|
||||
"gopkg.d7z.net/middleware/kv"
|
||||
)
|
||||
|
||||
var (
|
||||
configPath = "config-local.yaml"
|
||||
debug = false
|
||||
)
|
||||
|
||||
func init() {
|
||||
flag.StringVar(&configPath, "conf", configPath, "config file path")
|
||||
flag.BoolVar(&debug, "debug", debug, "debug mode")
|
||||
flag.Parse()
|
||||
}
|
||||
|
||||
func main() {
|
||||
call := logInject()
|
||||
defer call()
|
||||
config, err := LoadConfig(configPath)
|
||||
if err != nil {
|
||||
log.Fatalf("fail to load config file: %v", err)
|
||||
}
|
||||
|
||||
gitea, err := providers.NewGitea(config.Auth.Server, config.Auth.Token)
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
cacheMeta, err := kv.NewKVFromURL(config.Cache.Meta)
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
defer cacheMeta.Close()
|
||||
cacheBlob, err := cache.NewCacheFromURL(config.Cache.Blob)
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
defer cacheBlob.Close()
|
||||
backend := providers.NewProviderCache(gitea, cacheMeta, config.Cache.MetaTTL,
|
||||
cacheBlob, uint64(config.Cache.BlobLimit),
|
||||
)
|
||||
defer backend.Close()
|
||||
db, err := kv.NewKVFromURL(config.Database.URL)
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
defer db.Close()
|
||||
pageServer := pkg.NewPageServer(
|
||||
http.DefaultClient,
|
||||
backend,
|
||||
config.Domain,
|
||||
config.Page.DefaultBranch,
|
||||
db,
|
||||
cacheMeta,
|
||||
config.Cache.MetaTTL,
|
||||
config.ErrorHandler,
|
||||
)
|
||||
ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM, syscall.SIGQUIT)
|
||||
defer stop()
|
||||
|
||||
svc := http.Server{Addr: config.Bind, Handler: pageServer}
|
||||
go func() {
|
||||
<-ctx.Done()
|
||||
zap.L().Debug("shutdown gracefully")
|
||||
_ = svc.Close()
|
||||
}()
|
||||
_ = svc.ListenAndServe()
|
||||
}
|
||||
|
||||
func logInject() func() {
|
||||
atom := zap.NewAtomicLevel()
|
||||
if debug {
|
||||
atom.SetLevel(zap.DebugLevel)
|
||||
} else {
|
||||
atom.SetLevel(zap.InfoLevel)
|
||||
}
|
||||
cfg := zap.NewProductionConfig()
|
||||
cfg.Level = atom
|
||||
|
||||
logger, _ := cfg.Build()
|
||||
zap.ReplaceGlobals(logger)
|
||||
zap.L().Debug("debug enabled")
|
||||
return func() {
|
||||
if err := logger.Sync(); err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user