forked from x/ContextOS
312 lines
8.9 KiB
Go
312 lines
8.9 KiB
Go
package nfs // {{{
|
|
// }}}
|
|
import ( // {{{
|
|
"context"
|
|
|
|
"bufio"
|
|
"fmt"
|
|
"github.com/skip2/go-qrcode"
|
|
"io"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// }}}
|
|
|
|
type NFS struct {
|
|
in *os.File
|
|
out *os.File
|
|
buf []string
|
|
*ctx.Context
|
|
}
|
|
|
|
func (nfs *NFS) print(str string, arg ...interface{}) bool { // {{{
|
|
if nfs.out == nil {
|
|
return false
|
|
}
|
|
|
|
fmt.Fprintf(nfs.out, str, arg...)
|
|
return true
|
|
}
|
|
|
|
// }}}
|
|
|
|
func (nfs *NFS) Spawn(m *ctx.Message, c *ctx.Context, arg ...string) ctx.Server {
|
|
c.Caches = map[string]*ctx.Cache{
|
|
"pos": &ctx.Cache{Name: "读写位置", Value: "0", Help: "读写位置"},
|
|
"nline": &ctx.Cache{Name: "缓存命令行数", Value: "0", Help: "缓存命令行数"},
|
|
"return": &ctx.Cache{Name: "缓存命令行数", Value: "0", Help: "缓存命令行数"},
|
|
}
|
|
c.Configs = map[string]*ctx.Config{}
|
|
|
|
if len(arg) > 1 {
|
|
if info, e := os.Stat(arg[1]); e == nil {
|
|
c.Caches["name"] = &ctx.Cache{Name: "name", Value: info.Name(), Help: "文件名"}
|
|
c.Caches["mode"] = &ctx.Cache{Name: "mode", Value: info.Mode().String(), Help: "文件权限"}
|
|
c.Caches["size"] = &ctx.Cache{Name: "size", Value: fmt.Sprintf("%d", info.Size()), Help: "文件大小"}
|
|
c.Caches["time"] = &ctx.Cache{Name: "time", Value: info.ModTime().Format("15:03:04"), Help: "创建时间"}
|
|
}
|
|
}
|
|
|
|
s := new(NFS)
|
|
s.Context = c
|
|
return s
|
|
|
|
}
|
|
|
|
func (nfs *NFS) Begin(m *ctx.Message, arg ...string) ctx.Server { // {{{
|
|
nfs.Context.Master(nil)
|
|
if nfs.Context == Index {
|
|
Pulse = m
|
|
}
|
|
return nfs
|
|
}
|
|
|
|
// }}}
|
|
func (nfs *NFS) Start(m *ctx.Message, arg ...string) bool {
|
|
if out, ok := m.Data["out"]; ok {
|
|
nfs.out = out.(*os.File)
|
|
}
|
|
if in, ok := m.Data["in"]; ok {
|
|
nfs.in = in.(*os.File)
|
|
}
|
|
if len(arg) > 1 {
|
|
if m.Cap("stream", arg[1]); arg[0] == "open" {
|
|
return false
|
|
}
|
|
}
|
|
|
|
cli := m.Reply()
|
|
yac := m.Find(cli.Conf("yac"))
|
|
bio := bufio.NewScanner(nfs.in)
|
|
nfs.Context.Master(nil)
|
|
pos := 0
|
|
|
|
if buf, ok := m.Data["buf"]; ok {
|
|
nfs.buf = buf.([]string)
|
|
m.Capi("nline", len(nfs.buf))
|
|
goto out
|
|
}
|
|
|
|
if len(arg) > 2 {
|
|
nfs.print("%v\n", arg[2])
|
|
}
|
|
nfs.print("%s", cli.Conf("PS1"))
|
|
|
|
for rest, text := "", ""; pos < m.Capi("nline") || bio.Scan(); {
|
|
if pos == m.Capi("nline") {
|
|
if text = bio.Text(); len(text) > 0 && text[len(text)-1] == '\\' {
|
|
rest += text[:len(text)-1]
|
|
continue
|
|
}
|
|
|
|
if text, rest = rest+text, ""; len(text) == 0 && len(nfs.buf) > 0 && nfs.in == os.Stdin {
|
|
pos--
|
|
} else {
|
|
nfs.buf = append(nfs.buf, text)
|
|
m.Capi("nline", 1)
|
|
}
|
|
}
|
|
|
|
for ; pos < m.Capi("nline"); pos++ {
|
|
|
|
for text = nfs.buf[pos] + "\n"; text != ""; {
|
|
line := m.Spawn(yac.Target())
|
|
line.Optioni("pos", pos)
|
|
line.Put("option", "cli", cli.Target())
|
|
text = line.Cmd("parse", "line", "void", text).Get("result")
|
|
cli.Target(line.Data["cli"].(*ctx.Context))
|
|
if line.Has("return") {
|
|
goto out
|
|
}
|
|
if line.Has("back") {
|
|
pos = line.Appendi("back")
|
|
}
|
|
|
|
if result := strings.TrimRight(strings.Join(line.Meta["result"][1:len(line.Meta["result"])-1], ""), "\n"); len(result) > 0 {
|
|
nfs.print("%s", result+"\n")
|
|
}
|
|
}
|
|
}
|
|
|
|
nfs.print("%s", cli.Conf("PS1"))
|
|
}
|
|
|
|
out:
|
|
if len(arg) > 1 {
|
|
cli.Cmd("end")
|
|
} else {
|
|
m.Cap("status", "stop")
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (nfs *NFS) Close(m *ctx.Message, arg ...string) bool {
|
|
switch nfs.Context {
|
|
case m.Target():
|
|
if nfs.in != nil {
|
|
m.Log("info", nil, "%d close %s", Pulse.Capi("nfile", -1)+1, m.Cap("name"))
|
|
nfs.in.Close()
|
|
nfs.in = nil
|
|
}
|
|
case m.Source():
|
|
}
|
|
return true
|
|
}
|
|
|
|
var Pulse *ctx.Message
|
|
var Index = &ctx.Context{Name: "nfs", Help: "存储中心",
|
|
Caches: map[string]*ctx.Cache{
|
|
"nfile": &ctx.Cache{Name: "nfile", Value: "0", Help: "已经打开的文件数量"},
|
|
},
|
|
Configs: map[string]*ctx.Config{
|
|
"size": &ctx.Config{Name: "size", Value: "1024", Help: "读取文件的默认大小值"},
|
|
},
|
|
Commands: map[string]*ctx.Command{
|
|
"buffer": &ctx.Command{Name: "buffer [index string]", Help: "扫描文件, file: 文件名", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
|
|
if nfs, ok := m.Target().Server.(*NFS); m.Assert(ok) && nfs.buf != nil { // {{{
|
|
for i, v := range nfs.buf {
|
|
m.Echo("%d: %s\n", i, v)
|
|
}
|
|
} // }}}
|
|
}},
|
|
"copy": &ctx.Command{Name: "copy name [begin [end]]", Help: "复制文件, file: 文件名", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
|
|
if nfs, ok := m.Target().Server.(*NFS); m.Assert(ok) && len(nfs.buf) > 0 {
|
|
begin, end := 0, len(nfs.buf)
|
|
if len(arg) > 1 {
|
|
i, e := strconv.Atoi(arg[1])
|
|
m.Assert(e)
|
|
begin = i
|
|
}
|
|
if len(arg) > 2 {
|
|
i, e := strconv.Atoi(arg[2])
|
|
m.Assert(e)
|
|
end = i
|
|
}
|
|
m.Put("option", "buf", nfs.buf[begin:end])
|
|
m.Start(arg[0], "扫描文件", key)
|
|
}
|
|
}},
|
|
"scan": &ctx.Command{Name: "scan file", Help: "扫描文件, file: 文件名", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
|
|
if arg[0] == "stdio" { // {{{
|
|
m.Put("option", "in", os.Stdin).Put("option", "out", os.Stdout).Start("stdio", "扫描文件", m.Meta["detail"]...)
|
|
} else if f, e := os.Open(arg[0]); m.Assert(e) {
|
|
m.Put("option", "in", f).Start(fmt.Sprintf("file%d", Pulse.Capi("nfile", 1)), "扫描文件", m.Meta["detail"]...)
|
|
}
|
|
// }}}
|
|
}},
|
|
"print": &ctx.Command{Name: "print str", Help: "扫描文件, file: 文件名", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
|
|
if nfs, ok := m.Target().Server.(*NFS); m.Assert(ok) && nfs.out != nil { // {{{
|
|
fmt.Fprintf(nfs.out, "%s\n", arg[0])
|
|
}
|
|
// }}}
|
|
}},
|
|
"open": &ctx.Command{Name: "open file", Help: "打开文件, file: 文件名", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
|
|
if f, e := os.OpenFile(arg[0], os.O_RDWR|os.O_CREATE, os.ModePerm); m.Assert(e) { // {{{
|
|
m.Put("option", "in", f).Put("option", "out", f)
|
|
m.Start(fmt.Sprintf("file%d", Pulse.Capi("nfile", 1)), "打开文件", m.Meta["detail"]...)
|
|
m.Echo(m.Target().Name)
|
|
} // }}}
|
|
}},
|
|
"read": &ctx.Command{Name: "read [size [pos]]", Help: "读取文件, size: 读取大小, pos: 读取位置", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
|
|
if nfs, ok := m.Target().Server.(*NFS); m.Assert(ok) && nfs.in != nil { // {{{
|
|
var e error
|
|
n := m.Confi("size")
|
|
if len(arg) > 0 {
|
|
n, e = strconv.Atoi(arg[0])
|
|
m.Assert(e)
|
|
}
|
|
if len(arg) > 1 {
|
|
m.Cap("pos", arg[1])
|
|
}
|
|
|
|
buf := make([]byte, n)
|
|
if n, e = nfs.in.ReadAt(buf, int64(m.Capi("pos"))); e != io.EOF {
|
|
m.Assert(e)
|
|
}
|
|
m.Echo(string(buf))
|
|
|
|
if m.Capi("pos", n); m.Capi("pos") == m.Capi("size") {
|
|
m.Cap("pos", "0")
|
|
}
|
|
} // }}}
|
|
}},
|
|
"write": &ctx.Command{Name: "write string [pos]", Help: "写入文件, string: 写入内容, pos: 写入位置", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
|
|
if nfs, ok := m.Target().Server.(*NFS); m.Assert(ok) && nfs.out != nil { // {{{
|
|
|
|
if len(arg) > 1 {
|
|
m.Cap("pos", arg[1])
|
|
}
|
|
|
|
if len(arg[0]) == 0 {
|
|
m.Assert(nfs.out.Truncate(int64(m.Capi("pos"))))
|
|
m.Cap("size", m.Cap("pos"))
|
|
m.Cap("pos", "0")
|
|
} else {
|
|
n, e := nfs.out.WriteAt([]byte(arg[0]), int64(m.Capi("pos")))
|
|
if m.Assert(e) && m.Capi("pos", n) > m.Capi("size") {
|
|
m.Cap("size", m.Cap("pos"))
|
|
}
|
|
nfs.out.Sync()
|
|
}
|
|
|
|
m.Echo(m.Cap("pos"))
|
|
} // }}}
|
|
}},
|
|
"load": &ctx.Command{Name: "load file [size]", Help: "写入文件, string: 写入内容, pos: 写入位置", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
|
|
if f, e := os.Open(arg[0]); m.Assert(e) { // {{{
|
|
defer f.Close()
|
|
|
|
m.Meta = nil
|
|
size := 1024
|
|
if len(arg) > 1 {
|
|
if s, e := strconv.Atoi(arg[1]); m.Assert(e) {
|
|
size = s
|
|
}
|
|
}
|
|
buf := make([]byte, size)
|
|
|
|
if l, e := f.Read(buf); m.Assert(e) {
|
|
m.Log("info", nil, "read %d", l)
|
|
m.Echo(string(buf[:l]))
|
|
}
|
|
} // }}}
|
|
}},
|
|
"save": &ctx.Command{Name: "save file string...", Help: "写入文件, string: 写入内容, pos: 写入位置", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
|
|
if f, e := os.Create(arg[0]); m.Assert(e) { // {{{
|
|
defer f.Close()
|
|
|
|
fmt.Fprint(f, strings.Join(arg[1:], ""))
|
|
} // }}}
|
|
}},
|
|
"genqr": &ctx.Command{Name: "genqr [size] file string...", Help: "写入文件, string: 写入内容, pos: 写入位置", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
|
|
size := 256 // {{{
|
|
if len(arg) > 2 {
|
|
if s, e := strconv.Atoi(arg[0]); e == nil {
|
|
arg = arg[1:]
|
|
size = s
|
|
}
|
|
}
|
|
qrcode.WriteFile(strings.Join(arg[1:], ""), qrcode.Medium, size, arg[0]) // }}}
|
|
}},
|
|
},
|
|
Index: map[string]*ctx.Context{
|
|
"void": &ctx.Context{Name: "void",
|
|
Commands: map[string]*ctx.Command{
|
|
"scan": &ctx.Command{},
|
|
"open": &ctx.Command{},
|
|
"save": &ctx.Command{},
|
|
"load": &ctx.Command{},
|
|
"genqr": &ctx.Command{},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
nfs := &NFS{}
|
|
nfs.Context = Index
|
|
ctx.Index.Register(Index, nfs)
|
|
}
|