收敛 Pages 部署包读取路径

This commit is contained in:
ryan
2026-06-21 12:04:54 +08:00
parent 13a375e042
commit 6ed3c0c81f
10 changed files with 274 additions and 159 deletions
+28 -112
View File
@@ -22,20 +22,17 @@ import (
"github.com/Rain-kl/Wavelet/internal/apps/upload"
"github.com/Rain-kl/Wavelet/internal/model"
"github.com/Rain-kl/Wavelet/internal/repository"
platformstorage "github.com/Rain-kl/Wavelet/internal/storage"
)
const (
pagesLegacyArtifactCandidateCapacity = 8
pagesLegacyArtifactRootCapacity = 4
pagesMaxDeploymentFiles = 1000
pagesMaxDeploymentBytes = 100 * 1024 * 1024
defaultPagesEntryFile = "index.html"
defaultPagesFallbackPath = "/index.html"
pagesDeploymentUploadType = "openflare_pages_deployment"
mimeTypeApplicationZip = "application/zip"
pagesMaxPathLength = 512
bytesPerKiB = 1024
pagesMaxDeploymentFiles = 1000
pagesMaxDeploymentBytes = 100 * 1024 * 1024
defaultPagesEntryFile = "index.html"
defaultPagesFallbackPath = "/index.html"
pagesDeploymentUploadType = "openflare_pages_deployment"
mimeTypeApplicationZip = "application/zip"
pagesMaxPathLength = 512
bytesPerKiB = 1024
)
var pagesSlugPattern = regexp.MustCompile(`^[a-z0-9][a-z0-9-]{0,126}[a-z0-9]$|^[a-z0-9]$`)
@@ -179,26 +176,34 @@ func persistPagesUploadTemp(fileHeader *multipart.FileHeader) (string, string, i
return temp.Name(), hex.EncodeToString(hash.Sum(nil)), written, nil
}
func pagesLegacyRelativeCandidates(project *model.PagesProject, deployment *model.PagesDeployment) []string {
if project == nil || deployment == nil {
return nil
}
slug := strings.TrimSpace(project.Slug)
checksum := strings.TrimSpace(deployment.Checksum)
if slug == "" || checksum == "" {
return nil
}
fileName := checksum + ".zip"
return []string{
filepath.Join("artifacts", slug, fileName),
filepath.Join("pages", "artifacts", slug, fileName),
filepath.Join("data", "pages", "artifacts", slug, fileName),
}
}
func ingestPagesDeploymentPackage(
ctx context.Context,
tempPath string,
localPath string,
checksum string,
size int64,
projectSlug string,
fileName string,
) (upload.IngestResult, error) {
file, err := os.Open(tempPath) //nolint:gosec // tempPath is a validated pages deployment staging file
if err != nil {
return upload.IngestResult{}, err
}
defer func() { _ = file.Close() }()
systemUser := repository.GetSystemUser(ctx)
accessMode := 0
return upload.Ingest(ctx, upload.IngestRequest{
return upload.IngestFromLocalPath(ctx, localPath, upload.IngestRequest{
UserID: systemUser.ID,
Reader: file,
Size: size,
FileName: fileName,
MimeType: mimeTypeApplicationZip,
Extension: "zip",
@@ -215,101 +220,12 @@ func ingestPagesDeploymentPackage(
})
}
func legacyArtifactCandidatePaths(ctx context.Context, project *model.PagesProject, deployment *model.PagesDeployment) []string {
if deployment == nil {
return nil
}
seen := make(map[string]struct{})
candidates := make([]string, 0, pagesLegacyArtifactCandidateCapacity)
add := func(raw string) {
value := strings.TrimSpace(raw)
if value == "" {
return
}
if _, ok := seen[value]; ok {
return
}
info, statErr := os.Stat(value)
if statErr != nil || info.IsDir() {
return
}
seen[value] = struct{}{}
candidates = append(candidates, value)
}
storedPath := strings.TrimSpace(deployment.ArtifactPath)
add(storedPath)
if storedPath != "" {
add(filepath.Clean(storedPath))
add(strings.ReplaceAll(storedPath, "/data/data/", "/data/"))
add(strings.ReplaceAll(filepath.Clean(storedPath), string(filepath.Separator)+string(filepath.Separator), string(filepath.Separator)))
}
slug := ""
if project != nil {
slug = strings.TrimSpace(project.Slug)
}
checksum := strings.TrimSpace(deployment.Checksum)
if slug != "" && checksum != "" {
add(filepath.Join("artifacts", slug, checksum+".zip"))
add(filepath.Join("pages", "artifacts", slug, checksum+".zip"))
add(filepath.Join("data", "pages", "artifacts", slug, checksum+".zip"))
}
roots := make([]string, 0, pagesLegacyArtifactRootCapacity)
if cfg, err := platformstorage.LoadConfig(ctx); err == nil {
root := strings.TrimSpace(cfg.Local.Root)
if root != "" {
roots = append(roots, root)
}
}
for _, root := range roots {
if storedPath != "" && !filepath.IsAbs(storedPath) {
add(filepath.Join(root, storedPath))
}
if slug != "" && checksum != "" {
add(filepath.Join(root, "artifacts", slug, checksum+".zip"))
add(filepath.Join(root, "pages", "artifacts", slug, checksum+".zip"))
add(filepath.Join(root, "data", "pages", "artifacts", slug, checksum+".zip"))
}
}
return candidates
}
func openLegacyDeploymentArtifact(ctx context.Context, project *model.PagesProject, deployment *model.PagesDeployment) (string, *os.File, os.FileInfo, error) {
for _, candidate := range legacyArtifactCandidatePaths(ctx, project, deployment) {
file, err := os.Open(candidate) //nolint:gosec // candidate is resolved from managed legacy artifact metadata
if err != nil {
continue
}
info, statErr := file.Stat()
if statErr != nil {
_ = file.Close()
continue
}
if info.IsDir() {
_ = file.Close()
continue
}
return candidate, file, info, nil
}
return "", nil, nil, os.ErrNotExist
}
func removeDeploymentArtifact(ctx context.Context, deployment *model.PagesDeployment) {
if deployment == nil {
return
}
if deployment.UploadID > 0 {
if _, err := upload.Remove(ctx, deployment.UploadID); err != nil {
return
}
return
}
if strings.TrimSpace(deployment.ArtifactPath) != "" {
_ = os.Remove(deployment.ArtifactPath)
_, _ = upload.Remove(ctx, deployment.UploadID)
}
}