重构项目

cmd/local/main.go

@@ -1,8 +1,20 @@
 package main
 
 import (
+	"context"
 	"flag"
+	"fmt"
+	"io"
+	"net/http"
 	"os"
+	"time"
+
+	"github.com/pkg/errors"
+	"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 (
@@ -11,19 +23,68 @@ var (
 	repo   = org + "." + domain
 	path   = ""
 
-	port = 8080
+	port = ":8080"
 )
 
 func init() {
+	atom := zap.NewAtomicLevel()
+	atom.SetLevel(zap.DebugLevel)
+	cfg := zap.NewProductionConfig()
+	cfg.Level = atom
+	logger, _ := cfg.Build()
+	zap.ReplaceGlobals(logger)
 	dir, _ := os.Getwd()
 	path = dir
+	zap.L().Info("exec workdir", zap.String("path", path))
 	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.StringVar(&port, "port", port, "port")
 	flag.Parse()
 }
 
 func main() {
+	fmt.Printf("请访问 http://%s%s/", repo, port)
+	if stat, err := os.Stat(path); err != nil || !stat.IsDir() {
+		zap.L().Fatal("path is not a directory", zap.String("path", path))
+	}
+	provider := providers.NewLocalProvider(map[string][]string{
+		org: {repo},
+	}, path)
+	memory, err := kv.NewMemory("")
+	if err != nil {
+		zap.L().Fatal("failed to init memory provider", zap.Error(err))
+	}
+	server := pkg.NewPageServer(http.DefaultClient,
+		provider, domain, "gh-pages", memory, memory, 0, &nopCache{},
+		func(w http.ResponseWriter, r *http.Request, err error) {
+			if errors.Is(err, os.ErrNotExist) {
+				http.Error(w, "page not found.", http.StatusNotFound)
+			} else if err != nil {
+				http.Error(w, err.Error(), http.StatusInternalServerError)
+			}
+		})
+	err = http.ListenAndServe(port, server)
+	if err != nil && !errors.Is(err, http.ErrServerClosed) {
+		zap.L().Fatal("failed to start server", zap.Error(err))
+	}
+}
+
+type nopCache struct{}
+
+func (n *nopCache) Child(_ string) cache.Cache {
+	return n
+}
+
+func (n *nopCache) Put(_ context.Context, _ string, _ map[string]string, _ io.Reader, _ time.Duration) error {
+	return nil
+}
+
+func (n *nopCache) Get(_ context.Context, _ string) (*cache.Content, error) {
+	return nil, os.ErrNotExist
+}
+
+func (n *nopCache) Delete(_ context.Context, _ string) error {
+	return nil
 }

cmd/server/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"errors"
 	"flag"
 	"fmt"
 	"log"
@@ -36,7 +37,7 @@ func main() {
 		log.Fatalf("fail to load config file: %v", err)
 	}
 
-	gitea, err := providers.NewGitea(config.Auth.Server, config.Auth.Token)
+	gitea, err := providers.NewGitea(http.DefaultClient, config.Auth.Server, config.Auth.Token)
 	if err != nil {
 		log.Fatalln(err)
 	}
@@ -51,7 +52,7 @@ func main() {
 	}
 	defer cacheBlob.Close()
 	backend := providers.NewProviderCache(gitea, cacheMeta, config.Cache.MetaTTL,
-		cacheBlob, uint64(config.Cache.BlobLimit),
+		cacheBlob.Child("backend"), uint64(config.Cache.BlobLimit),
 	)
 	defer backend.Close()
 	db, err := kv.NewKVFromURL(config.Database.URL)
@@ -59,14 +60,19 @@ func main() {
 		log.Fatalln(err)
 	}
 	defer db.Close()
+	cdb, ok := db.(kv.RawKV).Raw().(kv.CursorPagedKV)
+	if !ok {
+		log.Fatalln(errors.New("database not support cursor"))
+	}
 	pageServer := pkg.NewPageServer(
 		http.DefaultClient,
 		backend,
 		config.Domain,
 		config.Page.DefaultBranch,
-		db,
+		cdb,
 		cacheMeta,
 		config.Cache.MetaTTL,
+		cacheBlob.Child("filter"),
 		config.ErrorHandler,
 	)
 	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM, syscall.SIGQUIT)