icebergs/base/ssh/ssh.go

package ssh

import (
	ice "github.com/shylinux/icebergs"
	"github.com/shylinux/icebergs/base/tcp"
	kit "github.com/shylinux/toolkits"
)

const SSH = "ssh"

var Index = &ice.Context{Name: SSH, Help: "终端模块", Commands: map[string]*ice.Command{
	ice.CTX_INIT: {Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
		m.Load()
	}},
	ice.CTX_EXIT: {Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
		if _, ok := m.Target().Server().(*Frame); ok {
			m.Done(true)
		}
		m.Richs(SERVICE, "", kit.MDB_FOREACH, func(key string, value map[string]interface{}) {
			kit.Value(value, "meta.status", tcp.CLOSE)
		})
		m.Richs(CHANNEL, "", kit.MDB_FOREACH, func(key string, value map[string]interface{}) {
			kit.Value(value, "meta.status", tcp.CLOSE)
		})
		m.Richs(SESSION, "", kit.MDB_FOREACH, func(key string, value map[string]interface{}) {
			kit.Value(value, "meta.status", tcp.CLOSE)
		})
		m.Richs(CONNECT, "", kit.MDB_FOREACH, func(key string, value map[string]interface{}) {
			kit.Value(value, "status", tcp.CLOSE)
		})

		m.Richs(SOURCE, "", STDIO, func(key string, value map[string]interface{}) {
			m.Conf(SOURCE, kit.Keys(kit.MDB_HASH), "")
			m.Conf(SOURCE, kit.Keys(kit.MDB_HASH, key), value)
		})
		m.Save()
	}},
}}

func init() {
	ice.Index.Register(Index, &Frame{},
		CONNECT, SESSION, SERVICE, CHANNEL,
		SOURCE, TARGET, PROMPT, RETURN,
	)
}