package code

import (
	"encoding/base64"
	"fmt"
	"io"
	"io/ioutil"
	"path"
	"strings"

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

func _binpack_file(m *ice.Message, w io.Writer, arg ...string) { // file name
	if f, e := nfs.OpenFile(m, arg[0]); e == nil {
		defer f.Close()
		if b, e := ioutil.ReadAll(f); e == nil && len(b) > 0 {
			fmt.Fprintf(w, "        \"%s\": \"%s\",\n", kit.Select(arg[0], arg, 1), base64.StdEncoding.EncodeToString(b))
			return
		}
	}
	fmt.Fprintf(w, "        // \"%s\": \"%s\",\n", kit.Select(arg[0], arg, 1), "")
}
func _binpack_dir(m *ice.Message, w io.Writer, dir string) {
	m.Option(nfs.DIR_ROOT, dir)
	m.Option(nfs.DIR_DEEP, true)
	m.Option(nfs.DIR_TYPE, nfs.CAT)

	m.Cmd(nfs.DIR, nfs.PWD).Sort(nfs.PATH).Tables(func(value ice.Maps) {
		switch path.Base(value[nfs.PATH]) {
		case ice.GO_MOD, ice.GO_SUM, "binpack.go", "version.go":
			return
		}
		switch strings.Split(value[nfs.PATH], ice.PS)[0] {
		case ice.BIN, ice.VAR, "website", "polaris":
			return
		}
		_binpack_file(m, w, path.Join(dir, value[nfs.PATH]))
	})
	fmt.Fprintln(w)
}

func _binpack_can(m *ice.Message, w io.Writer, dir string) {
	m.Option(nfs.DIR_ROOT, dir)
	m.Option(nfs.DIR_DEEP, true)
	m.Option(nfs.DIR_TYPE, nfs.CAT)

	for _, k := range []string{ice.FAVICON, ice.PROTO_JS, ice.FRAME_JS} {
		_binpack_file(m, w, path.Join(dir, k), path.Join(ice.USR_VOLCANOS, k))
	}
	for _, k := range []string{LIB, PAGE, PANEL, PLUGIN, "publish/client/nodejs/"} {
		m.Cmd(nfs.DIR, k).Sort(nfs.PATH).Tables(func(value ice.Maps) {
			_binpack_file(m, w, path.Join(dir, value[nfs.PATH]), path.Join(ice.USR_VOLCANOS, value[nfs.PATH]))
		})
	}
	fmt.Fprintln(w)
}
func _binpack_ctx(m *ice.Message, w io.Writer) {
	_binpack_dir(m, w, ice.SRC)
}
func _binpack_all(m *ice.Message) {
	if w, p, e := nfs.CreateFile(m, ice.SRC_BINPACK_GO); m.Assert(e) {
		defer w.Close()
		defer m.Echo(p)

		fmt.Fprintln(w, `package main

import (
	"encoding/base64"
	ice "shylinux.com/x/icebergs"
	"shylinux.com/x/icebergs/base/nfs"
)

func init() {
`)
		defer fmt.Fprintln(w, `}`)

		defer fmt.Fprintln(w, `
	for k, v := range pack {
		if b, e := base64.StdEncoding.DecodeString(v); e == nil {
			nfs.PackFile.WriteFile(k, b)
		}
	}
`)

		fmt.Fprintln(w, `	pack := ice.Maps{`)
		defer fmt.Fprintln(w, `	}`)

		if nfs.ExistsFile(m, ice.USR_VOLCANOS) && nfs.ExistsFile(m, ice.USR_INTSHELL) && m.Option(ice.MSG_USERPOD) == "" {
			_binpack_can(m, w, ice.USR_VOLCANOS)
			_binpack_dir(m, w, ice.USR_INTSHELL)
		}
		_binpack_ctx(m, w)

		_binpack_file(m, w, ice.ETC_MISS_SH)
		_binpack_file(m, w, ice.ETC_INIT_SHY)
		_binpack_file(m, w, ice.ETC_EXIT_SHY)
		fmt.Fprintln(w)

		_binpack_file(m, w, ice.LICENSE)
		_binpack_file(m, w, ice.MAKEFILE)
		_binpack_file(m, w, ice.README_MD)
		fmt.Fprintln(w)

		mdb.HashSelects(m).Sort(nfs.PATH).Tables(func(value ice.Maps) {
			if s, e := nfs.StatFile(m, value[nfs.PATH]); e == nil {
				if s.IsDir() {
					_binpack_dir(m, w, value[nfs.PATH])
				} else {
					_binpack_file(m, w, value[nfs.PATH])
				}
			}
		})
	}
}

const BINPACK = "binpack"

func init() {
	Index.MergeCommands(ice.Commands{
		BINPACK: {Name: "binpack path auto create insert", Help: "打包", Actions: ice.MergeActions(ice.Actions{
			mdb.CREATE: {Name: "create", Help: "创建", Hand: func(m *ice.Message, arg ...string) {
				_binpack_all(m)
			}},
			mdb.INSERT: {Name: "insert path", Help: "添加", Hand: func(m *ice.Message, arg ...string) {
				mdb.HashCreate(m, nfs.PATH, m.Option(nfs.PATH))
			}},
		}, mdb.HashAction(mdb.SHORT, nfs.PATH)), Hand: func(m *ice.Message, arg ...string) {
			mdb.HashSelect(m)
		}},
	})
}