优化本地调试

Author: ExplodingDragon

cmd/local/main.go

@@ -2,20 +2,24 @@ package main
 
 import (
 	"context"
+	"encoding/json"
 	"flag"
 	"fmt"
 	"io"
 	"net/http"
 	"os"
+	"path/filepath"
 	"time"
 
 	"github.com/pkg/errors"
 	"go.uber.org/zap"
 	"gopkg.d7z.net/gitea-pages/pkg"
+	"gopkg.d7z.net/gitea-pages/pkg/core"
 	"gopkg.d7z.net/gitea-pages/pkg/providers"
 	"gopkg.d7z.net/middleware/cache"
 	"gopkg.d7z.net/middleware/kv"
 	"gopkg.d7z.net/middleware/subscribe"
+	"gopkg.in/yaml.v3"
 )
 
 var (
@@ -53,6 +57,18 @@ func main() {
 	provider := providers.NewLocalProvider(map[string][]string{
 		org: {repo},
 	}, path)
+
+	file, _ := os.ReadFile(filepath.Join(path, ".pages.yaml"))
+	if file != nil {
+		var info core.PageConfig
+		err := yaml.Unmarshal(file, &info)
+		if err != nil {
+			zap.L().Fatal("parse yaml", zap.Error(err))
+		}
+		info.Alias = []string{}
+		marshal, _ := json.Marshal(info)
+		provider.AddOverlay(".pages.yaml", marshal)
+	}
 	memory, err := kv.NewMemory("")
 	if err != nil {
 		zap.L().Fatal("failed to init memory provider", zap.Error(err))
@@ -66,7 +82,11 @@ func main() {
 			} else if err != nil {
 				http.Error(w, err.Error(), http.StatusInternalServerError)
 			}
-		}, make(map[string]map[string]any))
+		}, map[string]map[string]any{
+			"redirect": {
+				"scheme": "http",
+			},
+		})
 	if err != nil {
 		zap.L().Fatal("failed to init page", zap.Error(err))
 	}

pkg/providers/local.go

@@ -12,26 +12,33 @@ import (
 	"path/filepath"
 	"slices"
 	"strconv"
+	"strings"
 	"time"
 
 	"gopkg.d7z.net/gitea-pages/pkg/core"
 )
 
 type LocalProvider struct {
-	graph map[string][]string
-	path  string
+	graph   map[string][]string
+	path    string
+	overlay map[string][]byte
 }
 
 func NewLocalProvider(
 	graph map[string][]string,
 	path string,
 ) *LocalProvider {
 	return &LocalProvider{
-		graph: graph,
-		path:  path,
+		graph:   graph,
+		path:    path,
+		overlay: map[string][]byte{},
 	}
 }
 
+func (l *LocalProvider) AddOverlay(path string, overlay []byte) {
+	l.overlay[strings.Trim(path, "/")] = overlay
+}
+
 func (l *LocalProvider) Close() error {
 	return nil
 }
@@ -65,20 +72,28 @@ func (l *LocalProvider) Branches(_ context.Context, owner, repo string) (map[str
 }
 
 func (l *LocalProvider) Open(_ context.Context, _, _, _, path string, _ http.Header) (*http.Response, error) {
-	open, err := os.Open(filepath.Join(l.path, path))
-	if err != nil {
-		return nil, errors.Join(err, os.ErrNotExist)
-	}
-	defer open.Close()
-	all, err := io.ReadAll(open)
-	if err != nil {
-		return nil, errors.Join(err, os.ErrNotExist)
-	}
+	var all []byte
 	recorder := httptest.NewRecorder()
+	if data, ok := l.overlay[strings.Trim(path, "/")]; ok {
+		all = data
+		recorder.Header().Add("Content-Length", strconv.FormatInt(int64(len(data)), 10))
+	} else {
+		open, err := os.Open(filepath.Join(l.path, path))
+		if err != nil {
+			return nil, errors.Join(err, os.ErrNotExist)
+		}
+		defer open.Close()
+		all, err = io.ReadAll(open)
+		if err != nil {
+			return nil, errors.Join(err, os.ErrNotExist)
+		}
+		stat, _ := open.Stat()
+		recorder.Header().Add("Content-Length", strconv.FormatInt(stat.Size(), 10))
+		recorder.Header().Add("Last-Modified", stat.ModTime().Format(http.TimeFormat))
+	}
+
 	recorder.Body = bytes.NewBuffer(all)
 	recorder.Header().Add("Content-Type", mime.TypeByExtension(filepath.Ext(path)))
-	stat, _ := open.Stat()
-	recorder.Header().Add("Content-Length", strconv.FormatInt(stat.Size(), 10))
-	recorder.Header().Add("Last-Modified", stat.ModTime().Format(http.TimeFormat))
+
 	return recorder.Result(), nil
 }