mirror of https://github.com/dirtbags/moth.git
68 lines
959 B
Go
68 lines
959 B
Go
|
package mothball
|
||
|
|
|
||
|
|
import (
|
||
|
|
"archive/zip"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"os"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Mothball struct {
|
||
|
|
zf *zip.ReadCloser
|
||
|
|
filename string
|
||
|
|
mtime time.Time
|
||
|
|
}
|
||
|
|
|
||
|
|
func Open(filename string) (*Mothball, error) {
|
||
|
|
var m Mothball
|
||
|
|
|
||
|
|
m.filename = filename
|
||
|
|
|
||
|
|
err := m.Refresh()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return &m, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Mothball) Close() (error) {
|
||
|
|
return m.zf.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Mothball) Refresh() (error) {
|
||
|
|
info, err := os.Stat(m.filename)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
mtime := info.ModTime()
|
||
|
|
|
||
|
|
if mtime == m.mtime {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
zf, err := zip.OpenReader(m.filename)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if m.zf != nil {
|
||
|
|
m.zf.Close()
|
||
|
|
}
|
||
|
|
m.zf = zf
|
||
|
|
m.mtime = mtime
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Mothball) Open(filename string) (io.ReadCloser, error) {
|
||
|
|
for _, f := range m.zf.File {
|
||
|
|
if filename == f.Name {
|
||
|
|
ret, err := f.Open()
|
||
|
|
return ret, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil, fmt.Errorf("File not found: %s in %s", filename, m.filename)
|
||
|
|
}
|