forked from x/icebergs
105 lines
2.5 KiB
Go
105 lines
2.5 KiB
Go
package nfs
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
|
|
ice "github.com/shylinux/icebergs"
|
|
"github.com/shylinux/icebergs/base/aaa"
|
|
"github.com/shylinux/icebergs/base/mdb"
|
|
kit "github.com/shylinux/toolkits"
|
|
)
|
|
|
|
func _cat_ext(name string) string {
|
|
return strings.ToLower(kit.Select(path.Base(name), strings.TrimPrefix(path.Ext(name), ".")))
|
|
}
|
|
func _cat_right(m *ice.Message, name string) bool {
|
|
switch strings.Split(name, "/")[0] {
|
|
case kit.SSH_ETC, kit.SSH_VAR:
|
|
if m.Warn(m.Option(ice.MSG_USERROLE) == aaa.VOID, ice.ErrNotRight, "of", name) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
func _cat_show(m *ice.Message, name string) {
|
|
if !_cat_right(m, name) {
|
|
return // 没有权限
|
|
}
|
|
|
|
// 本地文件
|
|
if f, e := os.Open(path.Join(m.Option(DIR_ROOT), name)); e == nil {
|
|
defer f.Close()
|
|
|
|
switch cb := m.Optionv(kit.Keycb(CAT)).(type) {
|
|
case func(string, int):
|
|
bio := bufio.NewScanner(f)
|
|
for i := 0; bio.Scan(); i++ {
|
|
cb(bio.Text(), i)
|
|
}
|
|
|
|
default:
|
|
if s, e := f.Stat(); m.Assert(e) {
|
|
buf := make([]byte, s.Size())
|
|
if n, e := f.Read(buf); m.Assert(e) {
|
|
m.Log_IMPORT(kit.MDB_FILE, name, kit.MDB_SIZE, n)
|
|
m.Echo(string(buf[:n]))
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// 打包文件
|
|
if b, ok := ice.BinPack[name]; ok {
|
|
m.Logs("binpack", name, kit.MDB_SIZE, len(b))
|
|
m.Echo(string(b))
|
|
return
|
|
}
|
|
|
|
// 远程文件
|
|
switch cb := m.Optionv(kit.Keycb(CAT)).(type) {
|
|
case func(string, int):
|
|
bio := bufio.NewScanner(bytes.NewBufferString(m.Cmdx("web.spide", "dev", "raw", "GET", path.Join("/share/local/", name))))
|
|
for i := 0; bio.Scan(); i++ {
|
|
cb(bio.Text(), i)
|
|
}
|
|
default:
|
|
m.Cmdy("web.spide", "dev", "raw", "GET", path.Join("/share/local/", name))
|
|
}
|
|
}
|
|
|
|
const CAT = "cat"
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{
|
|
Configs: map[string]*ice.Config{
|
|
CAT: {Name: CAT, Help: "文件", Value: kit.Data(
|
|
kit.SSH_SOURCE, kit.Dict(
|
|
"sh", "true", "shy", "true", "py", "true",
|
|
"go", "true", "vim", "true", "js", "true",
|
|
"conf", "true", "json", "true",
|
|
"makefile", "true",
|
|
"yml", "true",
|
|
),
|
|
)},
|
|
},
|
|
Commands: map[string]*ice.Command{
|
|
CAT: {Name: "cat path auto", Help: "文件", Action: map[string]*ice.Action{
|
|
mdb.RENDER: {Name: "render type name text", Help: "渲染", Hand: func(m *ice.Message, arg ...string) {
|
|
_cat_show(m, path.Join(arg[2], arg[1]))
|
|
}},
|
|
}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) == 0 || strings.HasSuffix(arg[0], "/") {
|
|
m.Cmdy(DIR, arg)
|
|
return
|
|
}
|
|
_cat_show(m, arg[0])
|
|
}},
|
|
},
|
|
})
|
|
}
|