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 == "" || m.Warn(!kit.FileExists(from), ice.ErrNotFound, from) {
		return
	}
	os.Remove(name)
	MkdirAll(m, path.Dir(name))
	if e := os.Link(from, name); e != nil {
		if m.Warn(os.Symlink(from, name), ice.ErrFailure, from) {
			return
		}
	}
	m.Log_EXPORT(name, "from", 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])
		}},
	}})
}