补充实现细节

pkg/server.go

@@ -1,7 +1,12 @@
 package pkg
 
 import (
+	"errors"
+	"io"
+	"mime"
 	"net/http"
+	"os"
+	"path/filepath"
 	"time"
 
 	"code.d7z.net/d7z-project/gitea-pages/pkg/core"
@@ -10,7 +15,9 @@ import (
 )
 
 type ServerOptions struct {
-	Domain string
+	Domain        string
+	DefaultBranch string
+
 	Config utils.Config
 	Cache  utils.Cache
 
@@ -22,29 +29,72 @@ type ServerOptions struct {
 func DefaultOptions(domain string) ServerOptions {
 	configMemory, _ := utils.NewConfigMemory("")
 	return ServerOptions{
-		Domain:       domain,
-		Config:       configMemory,
-		Cache:        utils.NewCacheMemory(1024*1024*10, int(memory.FreeMemory()/3*2)),
-		MaxCacheSize: 1024 * 1024 * 10,
-		HttpClient:   http.DefaultClient,
+		Domain:        domain,
+		DefaultBranch: "gh-pages",
+		Config:        configMemory,
+		Cache:         utils.NewCacheMemory(1024*1024*10, int(memory.FreeMemory()/3*2)),
+		MaxCacheSize:  1024 * 1024 * 10,
+		HttpClient:    http.DefaultClient,
 	}
 }
 
 type Server struct {
-	meta    *core.ServerMeta
+	meta    *core.PageDomain
 	options *ServerOptions
+	reader  *core.CacheBackendBlobReader
 }
 
 func NewPageServer(backend core.Backend, options ServerOptions) *Server {
 	backend = core.NewCacheBackend(backend, options.Config, time.Minute)
+	svcMeta := core.NewServerMeta(options.HttpClient, backend, options.Config, time.Minute)
+	pageMeta := core.NewPageDomain(svcMeta, options.Domain, options.DefaultBranch)
+	reader := core.NewCacheBackendBlobReader(options.HttpClient, backend, options.Cache, options.MaxCacheSize)
 	return &Server{
-		meta:    core.NewServerMeta(options.HttpClient, backend, options.Config, time.Minute),
+		meta:    pageMeta,
 		options: &options,
+		reader:  reader,
 	}
 }
 
 func (s *Server) ServeHTTP(writer http.ResponseWriter, request *http.Request) {
+	meta, err := s.meta.ParseDomainMeta(request.Method, request.RequestURI, request.URL.Query().Get("branch"))
+	if err != nil {
+		if errors.Is(err, os.ErrNotExist) {
+			s.writeError(writer, err)
+		} else {
+			s.writeNotfoundError(writer, request.RequestURI)
+		}
+		return
+	}
+	result, err := s.reader.Open(meta.Owner, meta.Repo, meta.CommitID, meta.Path)
+	if err != nil {
+		if errors.Is(err, os.ErrNotExist) {
+			s.writeError(writer, err)
+		} else {
+			s.writeNotfoundError(writer, request.RequestURI)
+		}
+		return
+	}
+	fileName := filepath.Base(meta.Path)
+	if reader, ok := result.(*utils.CacheContent); ok {
+		http.ServeContent(writer, request, fileName, reader.LastModified, reader)
+		_ = reader.Close()
+		return
+	} else {
+		writer.Header().Set("Content-Type", mime.TypeByExtension(meta.Path))
+		writer.WriteHeader(http.StatusOK)
+		_, _ = io.Copy(writer, reader)
+		_ = reader.Close()
+		return
+	}
+}
 
+func (s *Server) writeError(writer http.ResponseWriter, err error) {
+	http.Error(writer, err.Error(), http.StatusInternalServerError)
+}
+
+func (s *Server) writeNotfoundError(writer http.ResponseWriter, path string) {
+	http.Error(writer, "page not found.", http.StatusNotFound)
 }
 
 func (s *Server) Close() error {