135 lines
2.5 KiB
Go
135 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"archive/zip"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
type Store interface {
|
|
ReadFile(filename string) ([]byte, error)
|
|
ReadDir(dirname string) ([]os.FileInfo, error)
|
|
Stat(name string) (os.FileInfo, error)
|
|
}
|
|
|
|
type fileStore struct {
|
|
base string
|
|
}
|
|
|
|
func (f *fileStore) ReadFile(filename string) ([]byte, error) {
|
|
path := filepath.Join(f.base, filename)
|
|
return ioutil.ReadFile(path)
|
|
}
|
|
|
|
func (f *fileStore) ReadDir(dirname string) ([]os.FileInfo, error) {
|
|
path := filepath.Join(f.base, dirname)
|
|
return ioutil.ReadDir(path)
|
|
}
|
|
|
|
func (f *fileStore) Stat(name string) (os.FileInfo, error) {
|
|
path := filepath.Join(f.base, name)
|
|
return os.Stat(path)
|
|
}
|
|
|
|
type zipStore struct {
|
|
archive *zip.ReadCloser
|
|
files map[string]*zip.File
|
|
dirs map[string][]*zip.File
|
|
}
|
|
|
|
func openZipStore(file string) (*zipStore, error) {
|
|
r, err := zip.OpenReader(file)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
zs := &zipStore{r, make(map[string]*zip.File), make(map[string][]*zip.File)}
|
|
|
|
log.Printf("zip %q loading files...", file)
|
|
for _, file := range r.File {
|
|
zs.files[file.Name] = file
|
|
log.Printf("%v...", file.Name)
|
|
}
|
|
|
|
log.Printf("zip %q loading directories...", file)
|
|
for _, dir := range r.File {
|
|
if !strings.HasSuffix(dir.Name, "/") {
|
|
continue
|
|
}
|
|
|
|
var fi []*zip.File
|
|
for _, file := range zs.files {
|
|
name := file.Name
|
|
if name == dir.Name {
|
|
continue
|
|
}
|
|
|
|
if strings.HasSuffix(name, "/") {
|
|
name = name[:len(name)-1]
|
|
}
|
|
|
|
d, _ := filepath.Split(name)
|
|
|
|
if d != dir.Name {
|
|
continue
|
|
}
|
|
|
|
fi = append(fi, file)
|
|
|
|
if strings.HasPrefix(file.Name, dir.Name) {
|
|
log.Printf("%v in dir %v", file.Name, dir.Name)
|
|
}
|
|
}
|
|
|
|
zs.dirs[dir.Name] = fi
|
|
log.Printf("%v...", dir.Name)
|
|
}
|
|
|
|
return zs, nil
|
|
}
|
|
|
|
func (z *zipStore) ReadFile(filename string) ([]byte, error) {
|
|
log.Printf("read of %v", filename)
|
|
zf, ok := z.files[filename]
|
|
if !ok {
|
|
return nil, os.ErrNotExist
|
|
}
|
|
|
|
zr, err := zf.Open()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer zr.Close()
|
|
zc, err := ioutil.ReadAll(zr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return zc, nil
|
|
}
|
|
|
|
func (z *zipStore) ReadDir(dirname string) ([]os.FileInfo, error) {
|
|
var fi []os.FileInfo
|
|
|
|
dir, ok := z.dirs[dirname]
|
|
if !ok {
|
|
return nil, os.ErrNotExist
|
|
}
|
|
|
|
for _, d := range dir {
|
|
fi = append(fi, d.FileInfo())
|
|
}
|
|
|
|
return fi, nil
|
|
}
|
|
|
|
func (z *zipStore) Stat(name string) (os.FileInfo, error) {
|
|
log.Printf("stat %v", name)
|
|
if f, ok := z.files[name]; ok {
|
|
return f.FileInfo(), nil
|
|
}
|
|
return nil, os.ErrNotExist
|
|
}
|