icebergs/core/code/xterm.go

package code

import (
	"encoding/base64"
	"fmt"
	"io"
	"os"
	"os/exec"
	"path"
	"strings"

	pty "shylinux.com/x/creackpty"
	ice "shylinux.com/x/icebergs"
	"shylinux.com/x/icebergs/base/aaa"
	"shylinux.com/x/icebergs/base/cli"
	"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"
	kit "shylinux.com/x/toolkits"
)

func _xterm_socket(m *ice.Message, h, t string) {
	defer mdb.RLock(m)()
	m.Option(ice.MSG_DAEMON, m.Conf("", kit.Keys(mdb.HASH, h, mdb.META, mdb.TEXT)))
	m.Option(mdb.TEXT, t)
}
func _xterm_get(m *ice.Message, h string, must bool) *os.File {
	t := mdb.HashSelectField(m, m.Option(mdb.HASH, h), mdb.TYPE)
	if f, ok := mdb.HashTarget(m, h, func() ice.Any {
		if !must {
			return nil
		}

		ls := kit.Split(kit.Select("sh", t))
		m.Logs(cli.DAEMON, ice.CMD, ls)
		cmd := exec.Command(cli.SystemFind(m, ls[0]), ls[1:]...)
		m.Logs(cli.DAEMON, ice.CMD, cmd.Path)
		cmd.Env = append(os.Environ(), "TERM=xterm")

		tty, err := pty.Start(cmd)
		m.Assert(err)

		m.Go(func() {
			mdb.HashSelectUpdate(m, h, func(value ice.Map) {
				value["_cmd"] = nfs.NewCloser(func() error { return cmd.Process.Kill() })
			})
			buf := make([]byte, ice.MOD_BUFS)
			for {
				if n, e := tty.Read(buf); !m.Warn(e) {
					_xterm_socket(m, h, base64.StdEncoding.EncodeToString(buf[:n]))
					web.PushNoticeGrow(m, "data")
				} else {
					break
				}
			}
			web.PushNoticeGrow(m, "exit")
		})
		return tty
	}).(*os.File); m.Warn(!ok, ice.ErrNotValid, f) {
		mdb.HashSelectUpdate(m, h, func(value ice.Map) { delete(value, mdb.TARGET) })
		return nil
	} else {
		return f
	}
}

const XTERM = "xterm"

func init() {
	Index.MergeCommands(ice.Commands{
		XTERM: {Name: "xterm hash refresh", Help: "终端", Actions: ice.MergeActions(ice.Actions{
			ice.CTX_EXIT: {Hand: func(m *ice.Message, arg ...string) {
				mdb.HashSelectValue(m, func(value ice.Map) {
					if c, ok := value["_cmd"].(io.Closer); ok {
						c.Close()
					}
				})
				m.Conf("", mdb.HASH, "")
			}},
			mdb.INPUTS: {Name: "inputs", Help: "补全", Hand: func(m *ice.Message, arg ...string) {
				switch mdb.HashInputs(m, arg); arg[0] {
				case mdb.TYPE:
					m.Push(arg[0], "ice.bin source stdio", "node", "python", "bash", "sh")
				case mdb.NAME:
					m.Push(arg[0], path.Base(m.Option(mdb.TYPE)))
				}
			}},
			mdb.CREATE: {Name: "create type name", Help: "创建", Hand: func(m *ice.Message, arg ...string) {
				mdb.HashCreate(m, arg, mdb.TEXT, m.Option(ice.MSG_DAEMON))
			}},
			mdb.REMOVE: {Name: "remove", Help: "删除", Hand: func(m *ice.Message, arg ...string) {
				mdb.HashSelectDetail(m, m.Option(mdb.HASH), func(value ice.Map) {
					if c, ok := value["_cmd"].(io.Closer); ok {
						c.Close()
					}
				})
				mdb.HashRemove(m)
			}},
			"install": {Name: "install", Help: "安装", Hand: func(m *ice.Message, arg ...string) {
				if m.Option(mdb.HASH) != "" {
					u := web.OptionUserWeb(m)
					host := strings.Split(u.Host, ice.DF)[0]
					if host == tcp.LOCALHOST {
						host = m.Cmd(tcp.HOST).Append(aaa.IP)
					}
					m.Option("httphost", fmt.Sprintf("%s://%s:%s", u.Scheme, host, kit.Select(kit.Select("443", "80", u.Scheme == ice.HTTP), strings.Split(u.Host, ice.DF), 1)))
					cmd := kit.Renders(`export ctx_dev={{.Option "httphost"}}{{.Option "ctx_env"}}; ctx_temp=$(mktemp); wget -O $ctx_temp -q $ctx_dev; source $ctx_temp app username {{.Option "user.name"}}`, m)
					_xterm_get(m, m.Option(mdb.HASH), true).Write([]byte(cmd + "\n"))
				}
				m.ProcessHold()
			}},
			"rename": {Name: "rename", Help: "重命名", Hand: func(m *ice.Message, arg ...string) {
				mdb.HashModify(m, arg)
			}},
			"resize": {Name: "resize", Help: "大小", Hand: func(m *ice.Message, arg ...string) {
				pty.Setsize(_xterm_get(m, m.Option(mdb.HASH), true), &pty.Winsize{Rows: uint16(kit.Int(m.Option("rows"))), Cols: uint16(kit.Int(m.Option("cols")))})
			}},
			"select": {Name: "select", Help: "连接", Hand: func(m *ice.Message, arg ...string) {
				mdb.HashModify(m, mdb.TEXT, m.Option(ice.MSG_DAEMON))
				_xterm_get(m, m.Option(mdb.HASH), true)
				m.Cmd("", "input", arg)
			}},
			"input": {Name: "input", Help: "输入", Hand: func(m *ice.Message, arg ...string) {
				if b, e := base64.StdEncoding.DecodeString(strings.Join(arg, "")); !m.Warn(e) {
					_xterm_get(m, m.Option(mdb.HASH), true).Write(b)
					mdb.HashModify(m, mdb.TIME, m.Time())
				}
			}},
		}, mdb.HashAction(mdb.FIELD, "time,hash,type,name,text"), ctx.CmdAction()), Hand: func(m *ice.Message, arg ...string) {
			ctx.DisplayLocal(mdb.HashSelect(m, kit.Slice(arg, 0, 1)...), "")
		}},
	})
}

func ProcessXterm(m *ice.Message, args []string, arg ...string) {
	if len(arg) > 2 && arg[0] == ice.RUN && arg[1] == ctx.ACTION && arg[2] == mdb.CREATE {
		arg = append(arg, args...)
	}
	ctx.ProcessField(m, "web.code.xterm", nil, arg...)
}