重构项目

2025-11-13 23:31:23 +08:00
parent 54bbef0205
commit 02df131beb
12 changed files with 248 additions and 68 deletions
--- a/cmd/local/main.go
+++ b/cmd/local/main.go
@@ -0,0 +1,29 @@
+package main
+
+import (
+	"flag"
+	"os"
+)
+
+var (
+	org    = "pub"
+	domain = "fbi.com"
+	repo   = org + "." + domain
+	path   = ""
+
+	port = 8080
+)
+
+func init() {
+	dir, _ := os.Getwd()
+	path = dir
+	flag.StringVar(&org, "org", org, "org")
+	flag.StringVar(&repo, "repo", repo, "repo")
+	flag.StringVar(&domain, "domain", domain, "domain")
+	flag.StringVar(&path, "path", path, "path")
+	flag.IntVar(&port, "port", port, "port")
+	flag.Parse()
+}
+
+func main() {
+}
--- a/cmd/server/config.go
+++ b/cmd/server/config.go
@@ -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
+}
--- a/cmd/server/errors.html.tmpl
+++ b/cmd/server/errors.html.tmpl
@@ -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>
--- a/cmd/server/main.go
+++ b/cmd/server/main.go
@@ -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)
+		}
+	}
+}