forked from x/icebergs
121 lines
3.0 KiB
Go
121 lines
3.0 KiB
Go
package nfs
|
|
|
|
import (
|
|
"archive/tar"
|
|
"compress/gzip"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
|
|
ice "shylinux.com/x/icebergs"
|
|
"shylinux.com/x/icebergs/base/mdb"
|
|
kit "shylinux.com/x/toolkits"
|
|
)
|
|
|
|
func _tar_list(m *ice.Message, p string, cb func(*tar.Header, *tar.Reader, int)) *ice.Message {
|
|
const (
|
|
GZ = "gz"
|
|
)
|
|
var r io.Reader
|
|
if f, e := os.Open(p); m.Warn(e, ice.ErrNotValid, p) {
|
|
return m
|
|
} else {
|
|
defer f.Close()
|
|
r = f
|
|
}
|
|
for {
|
|
switch kit.Ext(p) {
|
|
case GZ:
|
|
if f, e := gzip.NewReader(r); m.Warn(e, ice.ErrNotValid, p) {
|
|
return m
|
|
} else {
|
|
defer f.Close()
|
|
r = f
|
|
}
|
|
p = kit.TrimExt(p, GZ)
|
|
case TAR:
|
|
i := 0
|
|
for r := tar.NewReader(r); ; i++ {
|
|
h, e := r.Next()
|
|
if m.Warn(e) {
|
|
break
|
|
}
|
|
cb(h, r, i)
|
|
}
|
|
m.StatusTimeCount(mdb.TOTAL, i)
|
|
return m
|
|
default:
|
|
return m
|
|
}
|
|
}
|
|
return m
|
|
}
|
|
|
|
const TAR = "tar"
|
|
|
|
func init() {
|
|
Index.MergeCommands(ice.Commands{
|
|
TAR: {Name: "tar path file auto page", Help: "打包", Actions: ice.MergeActions(ice.Actions{
|
|
mdb.NEXT: {Hand: func(m *ice.Message, arg ...string) { mdb.PrevPage(m, arg[0], kit.Slice(arg, 1)...) }},
|
|
mdb.PREV: {Hand: func(m *ice.Message, arg ...string) { mdb.NextPageLimit(m, arg[0], kit.Slice(arg, 1)...) }},
|
|
mdb.EXPORT: {Hand: func(m *ice.Message, arg ...string) {
|
|
list, size := map[string]bool{}, int64(0)
|
|
_tar_list(m, m.Option(PATH), func(h *tar.Header, r *tar.Reader, i int) {
|
|
if h.Name == m.Option(FILE) || m.Option(FILE) == "" {
|
|
p := path.Join(path.Dir(m.Option(PATH)), h.Name)
|
|
if strings.HasSuffix(h.Name, ice.PS) {
|
|
MkdirAll(m, p)
|
|
return
|
|
}
|
|
if !list[path.Dir(p)] {
|
|
list[path.Dir(p)] = true
|
|
MkdirAll(m, path.Dir(p))
|
|
}
|
|
if f, p, e := CreateFile(m, p); !m.Warn(e) {
|
|
defer f.Close()
|
|
if m.Option(FILE) != "" {
|
|
defer m.Cmdy(DIR, p, "time,path,size")
|
|
defer m.Cmdy(CAT, p)
|
|
}
|
|
if n, e := io.Copy(f, r); !m.Warn(e) {
|
|
size += n
|
|
m.Logs(mdb.EXPORT, LINE, i, SIZE, kit.FmtSize(size), FILE, p, SIZE, kit.FmtSize(n))
|
|
os.Chmod(p, os.FileMode(h.Mode))
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}},
|
|
}, mdb.PageListAction()), Hand: func(m *ice.Message, arg ...string) {
|
|
if len(arg) == 0 || strings.HasSuffix(arg[0], ice.PS) {
|
|
m.Cmdy(DIR, arg)
|
|
return
|
|
}
|
|
page, size := mdb.OptionPages(m, kit.Slice(arg, 2)...)
|
|
_tar_list(m, arg[0], func(h *tar.Header, r *tar.Reader, i int) {
|
|
if len(kit.Slice(arg, 0, 2)) > 1 {
|
|
if h.Name != arg[1] {
|
|
return
|
|
}
|
|
m.Echo(string(ReadAll(m, r)[:]))
|
|
}
|
|
if i >= (page-1)*size && i < page*size {
|
|
m.Push(mdb.TIME, h.ModTime.Format(ice.MOD_TIME)).Push(FILE, h.Name).Push(SIZE, kit.FmtSize(h.Size))
|
|
}
|
|
}).PushAction(mdb.EXPORT)
|
|
}},
|
|
})
|
|
}
|
|
func ReadAll(m *ice.Message, r io.Reader) []byte {
|
|
if buf, e := ioutil.ReadAll(r); m.Warn(e) {
|
|
return buf
|
|
} else {
|
|
return buf
|
|
}
|
|
}
|
|
func TarExport(m *ice.Message, path string, file ...string) {
|
|
m.Cmd(TAR, mdb.EXPORT, ice.Maps{PATH: path, FILE: kit.Select("", file, 0)})
|
|
}
|