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) string:
			list := []string{}
			bio := bufio.NewScanner(f)
			for i := 0; bio.Scan(); i++ {
				list = append(list, cb(bio.Text(), i))
			}
			m.Echo(strings.Join(list, "\n") + "\n")

		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.Info.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 (
	PATH = "path"
	SIZE = "size"
)
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])
			}},
		},
	})
}