forked from github-mirrorer/quake-kube
57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package content
|
|
|
|
import (
|
|
"hash/crc32"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
type File struct {
|
|
Name string `json:"name"`
|
|
Compressed int64 `json:"compressed"`
|
|
Checksum uint32 `json:"checksum"`
|
|
}
|
|
|
|
func getAssets(dir string) (files []*File, err error) {
|
|
err = walk(dir, func(path string, info os.FileInfo, err error) error {
|
|
data, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
n := crc32.ChecksumIEEE(data)
|
|
path = strings.TrimPrefix(path, dir+"/")
|
|
files = append(files, &File{path, info.Size(), n})
|
|
return nil
|
|
}, ".pk3", ".sh", ".run")
|
|
return
|
|
}
|
|
|
|
func hasExts(path string, exts ...string) bool {
|
|
for _, ext := range exts {
|
|
if strings.HasSuffix(path, ext) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func walk(root string, walkFn filepath.WalkFunc, exts ...string) error {
|
|
return filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
if os.IsPermission(err) {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
if !hasExts(path, exts...) {
|
|
return nil
|
|
}
|
|
if info.IsDir() {
|
|
return nil
|
|
}
|
|
return walkFn(path, info, err)
|
|
})
|
|
}
|