You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
1.5 KiB
76 lines
1.5 KiB
2 years ago
|
package bootstrap
|
||
|
|
|
||
|
|
import (
|
||
|
|
"archive/zip"
|
||
|
|
"crypto/sha256"
|
||
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/util"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"io"
|
||
|
|
"io/fs"
|
||
|
|
"sort"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewFS(zipContent string) fs.FS {
|
||
|
|
zipReader, err := zip.NewReader(strings.NewReader(zipContent), int64(len(zipContent)))
|
||
|
|
if err != nil {
|
||
|
|
util.Log().Panic("静态资源不是合法的zip文件: %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var files []file
|
||
|
|
err = fs.WalkDir(zipReader, ".", func(path string, d fs.DirEntry, err error) error {
|
||
|
|
if err != nil {
|
||
|
|
return errors.Errorf("无法获取[%s]的信息, %s, 跳过...", path, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if path == "." {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var f file
|
||
|
|
if d.IsDir() {
|
||
|
|
f.name = path + "/"
|
||
|
|
} else {
|
||
|
|
f.name = path
|
||
|
|
|
||
|
|
rc, err := zipReader.Open(path)
|
||
|
|
if err != nil {
|
||
|
|
return errors.Errorf("无法打开文件[%s], %s, 跳过...", path, err)
|
||
|
|
}
|
||
|
|
defer rc.Close()
|
||
|
|
|
||
|
|
data, err := io.ReadAll(rc)
|
||
|
|
if err != nil {
|
||
|
|
return errors.Errorf("无法读取文件[%s], %s, 跳过...", path, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
f.data = string(data)
|
||
|
|
|
||
|
|
hash := sha256.Sum256(data)
|
||
|
|
for i := range f.hash {
|
||
|
|
f.hash[i] = ^hash[i]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
files = append(files, f)
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
util.Log().Panic("初始化静态资源失败: %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
sort.Slice(files, func(i, j int) bool {
|
||
|
|
fi, fj := files[i], files[j]
|
||
|
|
di, ei, _ := split(fi.name)
|
||
|
|
dj, ej, _ := split(fj.name)
|
||
|
|
|
||
|
|
if di != dj {
|
||
|
|
return di < dj
|
||
|
|
}
|
||
|
|
return ei < ej
|
||
|
|
})
|
||
|
|
|
||
|
|
var embedFS FS
|
||
|
|
embedFS.files = &files
|
||
|
|
return embedFS
|
||
|
|
}
|