package relay

import (
	"io"
	"os"
	"path"
	"strings"

	"shylinux.com/x/ice"
	"shylinux.com/x/icebergs/base/ctx"
	"shylinux.com/x/icebergs/base/mdb"
	"shylinux.com/x/icebergs/base/nfs"
	"shylinux.com/x/icebergs/base/tcp"
	"shylinux.com/x/icebergs/base/web"
	"shylinux.com/x/icebergs/misc/ssh"
	kit "shylinux.com/x/toolkits"
)

const (
	TRANS     = "trans"
	SSH_TRANS = "ssh.trans"
)

type trans struct {
	send string `name:"send from"`
	list string `name:"list machine path auto" help:"文件"`
}

func (s trans) Inputs(m *ice.Message, arg ...string) {
	switch arg[0] {
	case nfs.FROM:
		m.Cmdy(nfs.DIR, ice.USR_PUBLISH, nfs.PATH, nfs.SIZE, mdb.TIME)
	}
}
func (s trans) Upload(m *ice.Message, arg ...string) {
	ls := web.Upload(m.Message)
	m.Options(nfs.FROM, m.Cmd(web.CACHE, ls[0]).Append(nfs.FILE), nfs.FILE, ls[1])
	s.Send(m)
}
func (s trans) Send(m *ice.Message, arg ...string) {
	defer web.ToastProcess(m.Message)()
	nfs.Open(m.Message, m.Option(nfs.FROM), func(r io.Reader, info os.FileInfo) {
		s.open(m, func(fs *ssh.FileSystem) {
			nfs.Create(m.Message, path.Join(m.Option(nfs.PATH), m.OptionDefault(nfs.FILE, path.Base(m.Option(nfs.FROM)))), func(w io.Writer, p string) {
				m.Logs(tcp.SEND, nfs.TO, p, m.OptionSimple(tcp.HOST, nfs.FROM), nfs.SIZE, kit.FmtSize(info.Size()))
				web.GoToast(m.Message, "", func(toast func(name string, count, total int)) []string {
					last := 0
					nfs.CopyStream(m.Message, w, r, 81920, int(info.Size()), func(count, total int) {
						if size := count / 1024 / 1024; size != last {
							toast(p, count, total)
							last = size
						}
					})
					return nil
				})
			})
		})
	})
}
func (s trans) List(m *ice.Message, arg ...string) {
	if len(arg) == 0 {
		m.OptionFields("time,machine,text,username,host,port")
		m.Cmdy(SSH_RELAY).Set(ctx.ACTION).Option(ice.MSG_ACTION, "")
	} else {
		m.Action(s.Send, s.Upload)
		s.open(m, func(fs *ssh.FileSystem) {
			if len(arg) == 1 || strings.HasSuffix(arg[1], nfs.PS) {
				m.Cmdy(nfs.DIR, arg[1:]).PushAction(s.Trash)
			} else {
				m.Cmdy(nfs.CAT, arg[1])
			}
		}, arg...)
	}
}
func (s trans) Trash(m *ice.Message, arg ...string) {
	s.open(m, func(fs *ssh.FileSystem) { fs.Remove(m.Option(nfs.PATH)) })
}

func init() { ice.Cmd(SSH_TRANS, trans{}) }

func (s trans) open(m *ice.Message, cb func(*ssh.FileSystem), arg ...string) {
	ssh.Open(m.Options(m.Cmd(SSH_RELAY, kit.Select(m.Option(MACHINE), arg, 0)).AppendSimple()).Message, func(fs *ssh.FileSystem) {
		defer m.Options(ice.MSG_FILES, fs).Options(ice.MSG_FILES, nil)
		cb(fs)
	})
}