package nfs

import (
	"archive/tar"
	"compress/gzip"
	"io"
	"os"
	"strings"

	ice "shylinux.com/x/icebergs"
	"shylinux.com/x/icebergs/base/mdb"
	kit "shylinux.com/x/toolkits"
)

const TAR = "tar"

func init() {
	Index.MergeCommands(ice.Commands{
		TAR: {Name: "tar file path auto", Help: "打包", Actions: ice.Actions{
			mdb.IMPORT: {Name: "import", Help: "导入", Hand: func(m *ice.Message, arg ...string) {
				if len(arg) == 1 {
					arg = append(arg, arg[0])
				}
				if !strings.HasSuffix(arg[0], ".tar.gz") {
					arg[0] += ".tar.gz"
				}
				m.Cmd("cli.system", "tar", "zcvf", arg)
				m.Echo(arg[0])
			}},
			mdb.EXPORT: {Name: "export", Help: "导出", Hand: func(m *ice.Message, arg ...string) {
				m.Cmd("cli.system", "tar", "xvf", arg)
				m.Echo(arg[0])
			}},
		}, Hand: func(m *ice.Message, arg ...string) {
			m.Option("cmd_dir", m.Option(DIR_ROOT))
			m.Debug("cmd_dir: %v", m.Option("cmd_dir"))
			m.Cmdy("cli.system", "tar", "zcvf", arg)
			return

			file, _, err := CreateFile(m, arg[0])
			m.Assert(err)
			defer file.Close()

			var out io.WriteCloser = file
			if strings.HasSuffix(arg[0], ".gz") {
				out = gzip.NewWriter(out)
				defer out.Close()
			}
			t := tar.NewWriter(out)
			defer t.Close()

			dir_root := m.Option(DIR_ROOT)
			var count, total int64
			for _, k := range arg[1:] {
				m.Option(DIR_TYPE, TYPE_CAT)
				m.Option(DIR_DEEP, ice.TRUE)
				m.Cmdy(DIR, k, func(f os.FileInfo, p string) {
					total += f.Size()

					header, err := tar.FileInfoHeader(f, p)
					if m.Warn(err != nil, err) {
						return
					}

					header.Name = strings.TrimPrefix(p, dir_root+ice.PS)
					if err = t.WriteHeader(header); m.Warn(err) {
						return
					}

					file, err := OpenFile(m, p)
					if m.Warn(err) {
						return
					}
					defer file.Close()

					if _, err = io.Copy(t, file); m.Warn(err) {
						return
					}

					count++
				})
			}
			m.StatusTimeCountTotal(kit.FmtSize(total))
		}},
	})
}