mirror of
https://shylinux.com/x/icebergs
synced 2025-04-26 01:24:05 +08:00
120 lines
3.0 KiB
Go
120 lines
3.0 KiB
Go
package nfs
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
|
|
ice "shylinux.com/x/icebergs"
|
|
kit "shylinux.com/x/toolkits"
|
|
)
|
|
|
|
func _defs_file(m *ice.Message, name string, text ...string) {
|
|
if kit.FileExists(path.Join(m.Option(DIR_ROOT), name)) {
|
|
return
|
|
}
|
|
for i, v := range text {
|
|
b, _ := kit.Render(v, m)
|
|
text[i] = string(b)
|
|
}
|
|
_save_file(m, name, text...)
|
|
}
|
|
func _save_file(m *ice.Message, name string, text ...string) {
|
|
if f, p, e := kit.Create(path.Join(m.Option(DIR_ROOT), name)); m.Assert(e) {
|
|
defer f.Close()
|
|
|
|
for _, v := range text {
|
|
if n, e := f.WriteString(v); m.Assert(e) {
|
|
m.Log_EXPORT(FILE, p, SIZE, n)
|
|
}
|
|
}
|
|
m.Echo(p)
|
|
}
|
|
}
|
|
func _push_file(m *ice.Message, name string, text ...string) {
|
|
p := path.Join(m.Option(DIR_ROOT), name)
|
|
if strings.Contains(p, ice.PS) {
|
|
MkdirAll(m, path.Dir(p))
|
|
}
|
|
|
|
if f, e := os.OpenFile(p, os.O_WRONLY|os.O_APPEND|os.O_CREATE, ice.MOD_FILE); m.Assert(e) {
|
|
defer f.Close()
|
|
|
|
for _, k := range text {
|
|
if n, e := f.WriteString(k); m.Assert(e) {
|
|
m.Log_EXPORT(FILE, p, SIZE, n)
|
|
}
|
|
}
|
|
m.Echo(p)
|
|
}
|
|
}
|
|
func _copy_file(m *ice.Message, name string, from ...string) {
|
|
if f, p, e := kit.Create(path.Join(m.Option(DIR_ROOT), name)); m.Assert(e) {
|
|
defer f.Close()
|
|
|
|
for _, v := range from {
|
|
if s, e := os.Open(v); !m.Warn(e, ice.ErrNotFound, name) {
|
|
defer s.Close()
|
|
|
|
if n, e := io.Copy(f, s); !m.Warn(e, ice.ErrNotFound, name) {
|
|
m.Log_IMPORT(FILE, v, SIZE, n)
|
|
m.Log_EXPORT(FILE, p, SIZE, n)
|
|
}
|
|
}
|
|
}
|
|
m.Echo(p)
|
|
}
|
|
}
|
|
func _link_file(m *ice.Message, name string, from string) {
|
|
if from == "" {
|
|
return
|
|
}
|
|
os.Remove(name)
|
|
MkdirAll(m, path.Dir(name))
|
|
if e := os.Link(from, name); e != nil {
|
|
m.Warn(os.Symlink(from, name), ice.ErrFailure, from)
|
|
}
|
|
m.Echo(name)
|
|
}
|
|
|
|
const (
|
|
CONTENT = "content"
|
|
)
|
|
const DEFS = "defs"
|
|
const SAVE = "save"
|
|
const PUSH = "push"
|
|
const COPY = "copy"
|
|
const LINK = "link"
|
|
|
|
func init() {
|
|
Index.Merge(&ice.Context{Commands: map[string]*ice.Command{
|
|
DEFS: {Name: "defs file text...", Help: "默认", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
_defs_file(m, arg[0], arg[1:]...)
|
|
}},
|
|
SAVE: {Name: "save file text...", Help: "保存", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) == 1 {
|
|
arg = append(arg, m.Option(CONTENT))
|
|
}
|
|
_save_file(m, arg[0], arg[1:]...)
|
|
}},
|
|
PUSH: {Name: "push file text...", Help: "追加", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) == 1 {
|
|
arg = append(arg, m.Option(CONTENT))
|
|
}
|
|
_push_file(m, arg[0], arg[1:]...)
|
|
}},
|
|
COPY: {Name: "copy file from...", Help: "复制", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
for _, file := range arg[1:] {
|
|
if kit.FileExists(file) {
|
|
_copy_file(m, arg[0], arg[1:]...)
|
|
return
|
|
}
|
|
}
|
|
}},
|
|
LINK: {Name: "link file from", Help: "链接", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
_link_file(m, arg[0], arg[1])
|
|
}},
|
|
}})
|
|
}
|