forked from x/icebergs
1672 lines
56 KiB
Go
1672 lines
56 KiB
Go
package web
|
|
|
|
import (
|
|
"github.com/gorilla/websocket"
|
|
"github.com/shylinux/icebergs"
|
|
"github.com/shylinux/toolkits"
|
|
"github.com/skip2/go-qrcode"
|
|
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"math/rand"
|
|
"mime/multipart"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"text/template"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
MSG_MAPS = 1
|
|
)
|
|
|
|
type Frame struct {
|
|
*http.Client
|
|
*http.Server
|
|
*http.ServeMux
|
|
m *ice.Message
|
|
send map[string]*ice.Message
|
|
}
|
|
|
|
func Cookie(msg *ice.Message, sessid string) string {
|
|
expire := time.Now().Add(kit.Duration(msg.Conf(ice.AAA_SESS, ice.Meta("expire"))))
|
|
msg.Log("cookie", "expire:%v sessid:%s", kit.Format(expire), sessid)
|
|
http.SetCookie(msg.W, &http.Cookie{Name: ice.WEB_SESS, Value: sessid, Path: "/", Expires: expire})
|
|
return sessid
|
|
}
|
|
func IsLocalIP(ip string) bool {
|
|
return ip == "::1" || strings.HasPrefix(ip, "127.")
|
|
}
|
|
|
|
func (web *Frame) Login(msg *ice.Message, w http.ResponseWriter, r *http.Request) bool {
|
|
if msg.Options(ice.WEB_SESS) {
|
|
// 会话认证
|
|
sub := msg.Cmd(ice.AAA_SESS, "check", msg.Option(ice.WEB_SESS))
|
|
msg.Info("role: %s user: %s", msg.Option(ice.MSG_USERROLE, sub.Append("userrole")),
|
|
msg.Option(ice.MSG_USERNAME, sub.Append("username")))
|
|
}
|
|
if strings.HasPrefix(msg.Option(ice.MSG_USERURL), "/space/") {
|
|
return true
|
|
}
|
|
|
|
if (!msg.Options(ice.MSG_SESSID) || !msg.Options(ice.MSG_USERNAME)) && IsLocalIP(msg.Option(ice.MSG_USERIP)) {
|
|
// 自动认证
|
|
msg.Option(ice.MSG_USERNAME, msg.Conf(ice.CLI_RUNTIME, "boot.username"))
|
|
msg.Option(ice.MSG_USERROLE, msg.Cmdx(ice.AAA_ROLE, "check", msg.Option(ice.MSG_USERNAME)))
|
|
msg.Option(ice.MSG_SESSID, msg.Rich(ice.AAA_SESS, nil, kit.Dict(
|
|
"username", msg.Option(ice.MSG_USERNAME), "userrole", msg.Option(ice.MSG_USERROLE),
|
|
)))
|
|
Cookie(msg, msg.Option(ice.MSG_SESSID))
|
|
msg.Info("user: %s role: %s sess: %s", msg.Option(ice.MSG_USERNAME), msg.Option(ice.MSG_USERROLE), msg.Option(ice.MSG_SESSID))
|
|
}
|
|
|
|
if s, ok := msg.Target().Commands[ice.WEB_LOGIN]; ok {
|
|
// 权限检查
|
|
msg.Target().Run(msg, s, ice.WEB_LOGIN, kit.Simple(msg.Optionv("cmds"))...)
|
|
}
|
|
return msg.Option(ice.MSG_USERURL) != ""
|
|
}
|
|
func (web *Frame) HandleWSS(m *ice.Message, safe bool, c *websocket.Conn, name string) bool {
|
|
for running := true; running; {
|
|
if t, b, e := c.ReadMessage(); m.Warn(e != nil, "space recv %d msg %v", t, e) {
|
|
break
|
|
} else {
|
|
switch t {
|
|
case MSG_MAPS:
|
|
// 接收报文
|
|
socket, msg := c, m.Spawn(b)
|
|
source := kit.Simple(msg.Optionv(ice.MSG_SOURCE), name)
|
|
target := kit.Simple(msg.Optionv(ice.MSG_TARGET))
|
|
msg.Info("recv %v %v->%v %v", t, source, target, msg.Format("meta"))
|
|
|
|
if len(target) == 0 {
|
|
// 本地执行
|
|
if msg.Optionv(ice.MSG_HANDLE, "true"); !msg.Warn(!safe, "no right") {
|
|
m.Option("_dev", name)
|
|
msg = msg.Cmd()
|
|
}
|
|
if source, target = []string{}, kit.Revert(source)[1:]; msg.Detail() == "exit" {
|
|
return true
|
|
}
|
|
|
|
} else if s, ok := msg.Confv(ice.WEB_SPACE, kit.Keys("hash", target[0], "socket")).(*websocket.Conn); ok {
|
|
// 转发报文
|
|
socket, source, target = s, source, target[1:]
|
|
msg.Info("space route")
|
|
|
|
} else if call, ok := web.send[msg.Option(ice.MSG_TARGET)]; len(target) == 1 && ok {
|
|
// 接收响应
|
|
delete(web.send, msg.Option(ice.MSG_TARGET))
|
|
msg.Info("space done")
|
|
call.Back(msg)
|
|
break
|
|
|
|
} else if msg.Warn(msg.Option("_handle") == "true", "space miss") {
|
|
// 回复失败
|
|
break
|
|
|
|
} else {
|
|
// 下发失败
|
|
msg.Warn(true, "space error")
|
|
source, target = []string{}, kit.Revert(source)[1:]
|
|
}
|
|
|
|
// 发送报文
|
|
msg.Optionv(ice.MSG_SOURCE, source)
|
|
msg.Optionv(ice.MSG_TARGET, target)
|
|
socket.WriteMessage(t, []byte(msg.Format("meta")))
|
|
msg.Info("send %v %v->%v %v", t, source, target, msg.Format("meta"))
|
|
msg.Log("cost", "%s: ", msg.Format("cost"))
|
|
}
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
func (web *Frame) HandleCGI(m *ice.Message, alias map[string]interface{}, which string) *template.Template {
|
|
cgi := template.FuncMap{}
|
|
|
|
tmpl := template.New(ice.WEB_TMPL)
|
|
cb := func(k string, p []string, v *ice.Command) {
|
|
cgi[k] = func(arg ...interface{}) (res interface{}) {
|
|
m.TryCatch(m.Spawn(), true, func(msg *ice.Message) {
|
|
msg.Target().Run(msg, v, k, kit.Simple(p, arg)...)
|
|
|
|
buffer := bytes.NewBuffer([]byte{})
|
|
m.Assert(tmpl.ExecuteTemplate(buffer, msg.Option(ice.WEB_TMPL), msg))
|
|
res = string(buffer.Bytes())
|
|
})
|
|
return
|
|
}
|
|
}
|
|
for k, v := range alias {
|
|
list := kit.Simple(v)
|
|
if v, ok := m.Target().Commands[list[0]]; ok {
|
|
cb(k, list[1:], v)
|
|
}
|
|
}
|
|
for k, v := range m.Target().Commands {
|
|
if strings.HasPrefix(k, "/") || strings.HasPrefix(k, "_") {
|
|
continue
|
|
}
|
|
cb(k, nil, v)
|
|
}
|
|
|
|
tmpl = tmpl.Funcs(cgi)
|
|
// tmpl = template.Must(tmpl.ParseGlob(path.Join(m.Conf(ice.WEB_SERVE, ice.Meta("template", "path")), "/*.tmpl")))
|
|
// tmpl = template.Must(tmpl.ParseGlob(path.Join(m.Conf(ice.WEB_SERVE, ice.Meta("template", "path")), m.Target().Name, "/*.tmpl")))
|
|
tmpl, e := tmpl.ParseFiles(which)
|
|
if e != nil {
|
|
}
|
|
// m.Confm(ice.WEB_SERVE, ice.Meta("template", "list"), func(index int, value string) { tmpl = template.Must(tmpl.Parse(value)) })
|
|
return tmpl
|
|
}
|
|
func (web *Frame) HandleCmd(m *ice.Message, key string, cmd *ice.Command) {
|
|
web.HandleFunc(key, func(w http.ResponseWriter, r *http.Request) {
|
|
m.TryCatch(m.Spawns(), true, func(msg *ice.Message) {
|
|
defer func() { msg.Cost("%s %v", r.URL.Path, msg.Optionv("cmds")) }()
|
|
|
|
// 解析请求
|
|
msg.Option(ice.MSG_USERUA, r.Header.Get("User-Agent"))
|
|
msg.Option(ice.MSG_USERIP, r.Header.Get(ice.MSG_USERIP))
|
|
msg.Option(ice.MSG_USERURL, r.URL.Path)
|
|
msg.Option(ice.MSG_USERNAME, "")
|
|
msg.Option(ice.WEB_SESS, "")
|
|
msg.R, msg.W = r, w
|
|
|
|
// 请求环境
|
|
for _, v := range r.Cookies() {
|
|
if v.Value != "" {
|
|
msg.Option(v.Name, v.Value)
|
|
}
|
|
}
|
|
|
|
// 请求数据
|
|
switch r.Header.Get("Content-Type") {
|
|
case "application/json":
|
|
var data interface{}
|
|
if e := json.NewDecoder(r.Body).Decode(&data); !msg.Warn(e != nil, "%s", e) {
|
|
msg.Optionv("content_data", data)
|
|
msg.Info("%s", kit.Formats(data))
|
|
}
|
|
|
|
switch d := data.(type) {
|
|
case map[string]interface{}:
|
|
for k, v := range d {
|
|
msg.Optionv(k, v)
|
|
}
|
|
}
|
|
default:
|
|
r.ParseMultipartForm(kit.Int64(kit.Select(r.Header.Get("Content-Length"), "4096")))
|
|
if r.ParseForm(); len(r.PostForm) > 0 {
|
|
for k, v := range r.PostForm {
|
|
msg.Info("%s: %v", k, v)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 请求参数
|
|
for k, v := range r.Form {
|
|
msg.Optionv(k, v)
|
|
}
|
|
|
|
if msg.Optionv("cmds") == nil {
|
|
msg.Optionv("cmds", strings.Split(msg.Option(ice.MSG_USERURL), "/")[2:])
|
|
}
|
|
|
|
// 执行命令
|
|
if web.Login(msg, w, r) && msg.Target().Run(msg, cmd, msg.Option(ice.MSG_USERURL), kit.Simple(msg.Optionv("cmds"))...) != nil {
|
|
// 输出响应
|
|
switch msg.Append("_output") {
|
|
case "void":
|
|
case "status":
|
|
msg.Info("status %s", msg.Result())
|
|
w.WriteHeader(kit.Int(kit.Select("200", msg.Result())))
|
|
|
|
case "file":
|
|
msg.Info("_output: %s %s", msg.Append("_output"), msg.Append("file"))
|
|
w.Header().Set("Content-Disposition", fmt.Sprintf("filename=%s", kit.Select(msg.Append("name"), msg.Append("story"))))
|
|
w.Header().Set("Content-Type", kit.Select("text/html", msg.Append("type")))
|
|
http.ServeFile(w, r, msg.Append("file"))
|
|
case "qrcode":
|
|
if qr, e := qrcode.New(msg.Result(), qrcode.Medium); m.Assert(e) {
|
|
w.Header().Set("Content-Type", "image/png")
|
|
m.Assert(qr.Write(256, w))
|
|
}
|
|
case "result":
|
|
w.Header().Set("Content-Type", kit.Select("text/html", msg.Append("type")))
|
|
fmt.Fprint(w, msg.Result())
|
|
default:
|
|
fmt.Fprint(w, msg.Formats("meta"))
|
|
}
|
|
} else {
|
|
switch msg.Append("_output") {
|
|
case "status":
|
|
msg.Info("status %s", msg.Result())
|
|
w.WriteHeader(kit.Int(kit.Select("200", msg.Result())))
|
|
}
|
|
}
|
|
})
|
|
})
|
|
}
|
|
func (web *Frame) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
m := web.m
|
|
|
|
index := r.Header.Get("index.module") == ""
|
|
if index {
|
|
// 解析地址
|
|
if ip := r.Header.Get("X-Forwarded-For"); ip != "" {
|
|
r.Header.Set(ice.MSG_USERIP, ip)
|
|
} else if ip := r.Header.Get("X-Real-Ip"); ip != "" {
|
|
r.Header.Set(ice.MSG_USERIP, ip)
|
|
} else if strings.HasPrefix(r.RemoteAddr, "[") {
|
|
r.Header.Set(ice.MSG_USERIP, strings.Split(r.RemoteAddr, "]")[0][1:])
|
|
} else {
|
|
r.Header.Set(ice.MSG_USERIP, strings.Split(r.RemoteAddr, ":")[0])
|
|
}
|
|
m.Info("").Info("%s %s %s", r.Header.Get(ice.MSG_USERIP), r.Method, r.URL)
|
|
|
|
// 解析地址
|
|
r.Header.Set("index.module", "some")
|
|
r.Header.Set("index.path", r.URL.Path)
|
|
r.Header.Set("index.url", r.URL.String())
|
|
}
|
|
|
|
if index && kit.Right(m.Conf(ice.WEB_SERVE, "meta.logheaders")) {
|
|
for k, v := range r.Header {
|
|
m.Info("%s: %v", k, kit.Format(v))
|
|
}
|
|
m.Info(" ")
|
|
}
|
|
|
|
web.ServeMux.ServeHTTP(w, r)
|
|
|
|
if index && kit.Right(m.Conf(ice.WEB_SERVE, "meta.logheaders")) {
|
|
for k, v := range w.Header() {
|
|
m.Info("%s: %v", k, kit.Format(v))
|
|
}
|
|
m.Info(" ")
|
|
}
|
|
}
|
|
|
|
func (web *Frame) Spawn(m *ice.Message, c *ice.Context, arg ...string) ice.Server {
|
|
return &Frame{}
|
|
}
|
|
func (web *Frame) Begin(m *ice.Message, arg ...string) ice.Server {
|
|
web.send = map[string]*ice.Message{}
|
|
return web
|
|
}
|
|
func (web *Frame) Start(m *ice.Message, arg ...string) bool {
|
|
m.Travel(func(p *ice.Context, s *ice.Context) {
|
|
if w, ok := s.Server().(*Frame); ok {
|
|
if w.ServeMux != nil {
|
|
return
|
|
}
|
|
w.ServeMux = http.NewServeMux()
|
|
|
|
// 静态路由
|
|
msg := m.Spawns(s)
|
|
m.Confm(ice.WEB_SERVE, ice.Meta("static"), func(key string, value string) {
|
|
m.Log("route", "%s <- %s <- %s", s.Name, key, value)
|
|
w.Handle(key, http.StripPrefix(key, http.FileServer(http.Dir(value))))
|
|
})
|
|
|
|
// 级联路由
|
|
route := "/" + s.Name + "/"
|
|
if n, ok := p.Server().(*Frame); ok && n.ServeMux != nil {
|
|
msg.Log("route", "%s <= %s", p.Name, route)
|
|
n.Handle(route, http.StripPrefix(path.Dir(route), w))
|
|
}
|
|
|
|
// 命令路由
|
|
m.Travel(func(p *ice.Context, sub *ice.Context, k string, x *ice.Command) {
|
|
if s == sub && k[0] == '/' {
|
|
msg.Log("route", "%s <- %s", s.Name, k)
|
|
w.HandleCmd(msg, k, x)
|
|
}
|
|
})
|
|
}
|
|
})
|
|
|
|
// TODO simple
|
|
m.Richs(ice.WEB_SPIDE, nil, arg[0], func(key string, value map[string]interface{}) {
|
|
client := value["client"].(map[string]interface{})
|
|
|
|
// 服务地址
|
|
port := m.Cap(ice.CTX_STREAM, client["hostname"])
|
|
m.Log("serve", "listen %s %s %v", arg[0], port, m.Conf(ice.CLI_RUNTIME, "node"))
|
|
|
|
// 启动服务
|
|
web.m, web.Server = m, &http.Server{Addr: port, Handler: web}
|
|
m.Event(ice.SERVE_START, arg[0])
|
|
m.Log("serve", "listen %s", web.Server.ListenAndServe())
|
|
m.Event(ice.SERVE_CLOSE, arg[0])
|
|
})
|
|
return true
|
|
}
|
|
func (web *Frame) Close(m *ice.Message, arg ...string) bool {
|
|
return true
|
|
}
|
|
|
|
var Index = &ice.Context{Name: "web", Help: "网络模块",
|
|
Caches: map[string]*ice.Cache{},
|
|
Configs: map[string]*ice.Config{
|
|
ice.WEB_SPIDE: {Name: "spide", Help: "蜘蛛侠", Value: kit.Data(kit.MDB_SHORT, "client.name")},
|
|
ice.WEB_SERVE: {Name: "serve", Help: "服务器", Value: kit.Data(
|
|
"static", map[string]interface{}{"/": "usr/volcanos/",
|
|
"/static/volcanos/": "usr/volcanos/",
|
|
"/publish/": "usr/publish/",
|
|
},
|
|
"volcanos", kit.Dict("path", "usr/volcanos",
|
|
"repos", "https://github.com/shylinux/volcanos",
|
|
"branch", "master"),
|
|
"template", map[string]interface{}{"path": "usr/template", "list": []interface{}{
|
|
`{{define "raw"}}{{.Result}}{{end}}`,
|
|
}},
|
|
"logheaders", "false",
|
|
)},
|
|
ice.WEB_SPACE: {Name: "space", Help: "空间站", Value: kit.Data(kit.MDB_SHORT, "name",
|
|
"redial.a", 3000, "redial.b", 1000, "redial.c", 1000,
|
|
"buffer.r", 4096, "buffer.w", 4096,
|
|
"timeout.c", "30s",
|
|
)},
|
|
ice.WEB_DREAM: {Name: "dream", Help: "梦想家", Value: kit.Data("path", "usr/local/work",
|
|
"cmd", []interface{}{ice.CLI_SYSTEM, "ice.sh", "start", ice.WEB_SPACE, "connect"},
|
|
)},
|
|
ice.WEB_FAVOR: {Name: "favor", Help: "收藏夹", Value: kit.Data(kit.MDB_SHORT, kit.MDB_NAME)},
|
|
ice.WEB_CACHE: {Name: "cache", Help: "缓存池", Value: kit.Data(
|
|
kit.MDB_SHORT, "text", "path", "var/file", "store", "var/data", "limit", "30", "least", "10", "fsize", "100000",
|
|
)},
|
|
ice.WEB_STORY: {Name: "story", Help: "故事会", Value: kit.Dict(
|
|
kit.MDB_META, kit.Dict(kit.MDB_SHORT, "data"),
|
|
"head", kit.Data(kit.MDB_SHORT, "story"),
|
|
"mime", kit.Dict("md", "txt"),
|
|
)},
|
|
ice.WEB_SHARE: {Name: "share", Help: "共享链", Value: kit.Data("template", share_template)},
|
|
ice.WEB_ROUTE: {Name: "route", Help: "路由", Value: kit.Data()},
|
|
ice.WEB_PROXY: {Name: "proxy", Help: "代理", Value: kit.Data()},
|
|
ice.WEB_GROUP: {Name: "group", Help: "分组", Value: kit.Data()},
|
|
ice.WEB_LABEL: {Name: "label", Help: "标签", Value: kit.Data()},
|
|
},
|
|
Commands: map[string]*ice.Command{
|
|
ice.ICE_INIT: {Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
m.Cmd(ice.CTX_CONFIG, "load", "web.json")
|
|
|
|
if m.Richs(ice.WEB_SPIDE, nil, "self", nil) == nil {
|
|
m.Cmd(ice.WEB_SPIDE, "add", "self", kit.Select("http://:9020", m.Conf(ice.CLI_RUNTIME, "conf.ctx_self")))
|
|
}
|
|
if m.Richs(ice.WEB_SPIDE, nil, "dev", nil) == nil {
|
|
m.Cmd(ice.WEB_SPIDE, "add", "dev", kit.Select("http://mac.local:9020", m.Conf(ice.CLI_RUNTIME, "conf.ctx_dev")))
|
|
}
|
|
if m.Richs(ice.WEB_SPIDE, nil, "shy", nil) == nil {
|
|
m.Cmd(ice.WEB_SPIDE, "add", "shy", kit.Select("https://shylinux.com:443", m.Conf(ice.CLI_RUNTIME, "conf.ctx_shy")))
|
|
}
|
|
}},
|
|
ice.ICE_EXIT: {Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
p := m.Conf(ice.WEB_CACHE, "meta.store")
|
|
m.Richs(ice.WEB_CACHE, nil, "*", func(key string, value map[string]interface{}) {
|
|
if f, _, e := kit.Create(path.Join(p, key[:2], key)); e == nil {
|
|
defer f.Close()
|
|
f.WriteString(kit.Formats(value))
|
|
}
|
|
})
|
|
// m.Conf(ice.WEB_CACHE, "hash", kit.Dict())
|
|
m.Cmd(ice.CTX_CONFIG, "save", kit.Keys(m.Cap(ice.CTX_FOLLOW), "json"),
|
|
ice.WEB_SPIDE, ice.WEB_FAVOR, ice.WEB_CACHE, ice.WEB_STORY, ice.WEB_SHARE)
|
|
|
|
m.Done()
|
|
m.Richs(ice.WEB_SPACE, nil, "*", func(key string, value map[string]interface{}) {
|
|
if kit.Format(value["type"]) == "master" {
|
|
m.Done()
|
|
}
|
|
})
|
|
}},
|
|
|
|
ice.WEB_SPIDE: {Name: "spide", Help: "蜘蛛侠", List: kit.List(
|
|
kit.MDB_INPUT, "text", "name", "name",
|
|
kit.MDB_INPUT, "button", "value", "查看", "action", "auto",
|
|
kit.MDB_INPUT, "button", "value", "返回", "cb", "Last",
|
|
), Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) == 0 {
|
|
// 爬虫列表
|
|
m.Richs(ice.WEB_SPIDE, nil, "*", func(key string, value map[string]interface{}) {
|
|
m.Push(key, value["client"], []string{"name", "method", "url"})
|
|
})
|
|
m.Sort("name")
|
|
return
|
|
}
|
|
if len(arg) == 1 {
|
|
// 爬虫详情
|
|
m.Richs(ice.WEB_SPIDE, nil, arg[0], func(key string, value map[string]interface{}) {
|
|
m.Push("detail", value)
|
|
})
|
|
return
|
|
}
|
|
|
|
switch arg[0] {
|
|
case "add":
|
|
// 添加爬虫
|
|
if uri, e := url.Parse(arg[2]); e == nil && arg[2] != "" {
|
|
dir, file := path.Split(uri.EscapedPath())
|
|
m.Rich(ice.WEB_SPIDE, nil, kit.Dict(
|
|
"cookie", kit.Dict(),
|
|
"header", kit.Dict(),
|
|
"client", kit.Dict(
|
|
"name", arg[1],
|
|
"logheaders", false,
|
|
"timeout", "100s",
|
|
"method", "POST",
|
|
"protocol", uri.Scheme,
|
|
"hostname", uri.Host,
|
|
"path", dir,
|
|
"file", file,
|
|
"query", uri.RawQuery,
|
|
"url", arg[2],
|
|
),
|
|
))
|
|
m.Log(ice.LOG_CREATE, "%s: %v", arg[1], arg[2:])
|
|
}
|
|
default:
|
|
// spide shy [cache] [POST|GET] uri file|data|form|part|json arg...
|
|
m.Richs(ice.WEB_SPIDE, nil, arg[0], func(key string, value map[string]interface{}) {
|
|
client := value["client"].(map[string]interface{})
|
|
// 缓存数据
|
|
cache := ""
|
|
switch arg[1] {
|
|
case "raw":
|
|
cache, arg = arg[1], arg[1:]
|
|
case "msg":
|
|
cache, arg = arg[1], arg[1:]
|
|
case "cache":
|
|
cache, arg = arg[1], arg[1:]
|
|
}
|
|
|
|
// 请求方法
|
|
method := kit.Select("POST", client["method"])
|
|
switch arg = arg[1:]; arg[0] {
|
|
case "POST":
|
|
method, arg = "POST", arg[1:]
|
|
case "GET":
|
|
method, arg = "GET", arg[1:]
|
|
}
|
|
|
|
// 请求地址
|
|
uri, arg := arg[0], arg[1:]
|
|
|
|
// 请求数据
|
|
head := map[string]string{}
|
|
body, ok := m.Optionv("body").(io.Reader)
|
|
if !ok && len(arg) > 0 && method != "GET" {
|
|
switch arg[0] {
|
|
case "file":
|
|
if f, e := os.Open(arg[1]); m.Warn(e != nil, "%s", e) {
|
|
return
|
|
} else {
|
|
defer f.Close()
|
|
body, arg = f, arg[2:]
|
|
}
|
|
case "data":
|
|
body, arg = bytes.NewBufferString(arg[1]), arg[2:]
|
|
case "form":
|
|
data := []string{}
|
|
for i := 1; i < len(arg)-1; i += 2 {
|
|
data = append(data, url.QueryEscape(arg[i])+"="+url.QueryEscape(arg[i+1]))
|
|
}
|
|
body = bytes.NewBufferString(strings.Join(data, "&"))
|
|
head["Content-Type"] = "application/x-www-form-urlencoded"
|
|
case "part":
|
|
buf := &bytes.Buffer{}
|
|
mp := multipart.NewWriter(buf)
|
|
for i := 1; i < len(arg)-1; i += 2 {
|
|
if strings.HasPrefix(arg[i+1], "@") {
|
|
if f, e := os.Open(arg[i+1][1:]); m.Assert(e) {
|
|
defer f.Close()
|
|
if p, e := mp.CreateFormFile(arg[i], path.Base(arg[i+1][1:])); m.Assert(e) {
|
|
io.Copy(p, f)
|
|
}
|
|
}
|
|
} else {
|
|
mp.WriteField(arg[i], arg[i+1])
|
|
}
|
|
}
|
|
mp.Close()
|
|
head["Content-Type"] = mp.FormDataContentType()
|
|
body = buf
|
|
case "json":
|
|
arg = arg[1:]
|
|
fallthrough
|
|
default:
|
|
data := map[string]interface{}{}
|
|
for i := 0; i < len(arg)-1; i += 2 {
|
|
kit.Value(data, arg[i], arg[i+1])
|
|
}
|
|
if b, e := json.Marshal(data); m.Assert(e) {
|
|
head["Content-Type"] = "application/json"
|
|
body = bytes.NewBuffer(b)
|
|
}
|
|
m.Log(ice.LOG_EXPORT, "json: %s", kit.Format(data))
|
|
}
|
|
arg = arg[:0]
|
|
} else {
|
|
body = bytes.NewBuffer([]byte{})
|
|
}
|
|
|
|
// 构造请求
|
|
uri = kit.MergeURL2(kit.Format(client["url"]), uri, arg)
|
|
req, e := http.NewRequest(method, uri, body)
|
|
m.Info("%s %s", req.Method, req.URL)
|
|
m.Assert(e)
|
|
|
|
// 请求参数
|
|
for k, v := range head {
|
|
req.Header.Set(k, v)
|
|
}
|
|
if list, ok := m.Optionv("header").([]string); ok {
|
|
for i := 0; i < len(list)-1; i += 2 {
|
|
req.Header.Set(list[i], list[i+1])
|
|
}
|
|
}
|
|
kit.Fetch(client["header"], func(key string, value string) {
|
|
req.Header.Set(key, value)
|
|
})
|
|
kit.Fetch(client["cookie"], func(key string, value string) {
|
|
req.AddCookie(&http.Cookie{Name: key, Value: value})
|
|
m.Info("%s: %s", key, value)
|
|
})
|
|
if method == "POST" {
|
|
m.Info("%s: %s", req.Header.Get("Content-Length"), req.Header.Get("Content-Type"))
|
|
}
|
|
|
|
// 请求代理
|
|
web := m.Target().Server().(*Frame)
|
|
if web.Client == nil {
|
|
web.Client = &http.Client{Timeout: kit.Duration(kit.Format(client["timeout"]))}
|
|
}
|
|
|
|
// 发送请求
|
|
res, e := web.Client.Do(req)
|
|
if m.Warn(e != nil, "%s", e) {
|
|
m.Set("result")
|
|
return
|
|
}
|
|
|
|
// 验证结果
|
|
if m.Cost("%s %s: %s", res.Status, res.Header.Get("Content-Length"), res.Header.Get("Content-Type")); m.Warn(res.StatusCode != http.StatusOK, "%s", res.Status) {
|
|
if cache != "" {
|
|
m.Set("result")
|
|
}
|
|
return
|
|
}
|
|
|
|
// 缓存参数
|
|
for _, v := range res.Cookies() {
|
|
kit.Value(client, "cookie."+v.Name, v.Value)
|
|
m.Info("%s: %s", v.Name, v.Value)
|
|
}
|
|
|
|
switch cache {
|
|
case "msg":
|
|
var data map[string][]string
|
|
m.Assert(json.NewDecoder(res.Body).Decode(&data))
|
|
m.Info("res: %s", kit.Formats(data))
|
|
if len(data["append"]) > 0 {
|
|
for i := range data[data["append"][0]] {
|
|
for _, k := range data["append"] {
|
|
m.Push(k, data[k][i])
|
|
}
|
|
}
|
|
}
|
|
|
|
case "raw":
|
|
if b, e := ioutil.ReadAll(res.Body); m.Assert(e) {
|
|
m.Echo(string(b))
|
|
}
|
|
case "cache":
|
|
// 缓存结果
|
|
m.Optionv("response", res)
|
|
m.Echo(m.Cmd(ice.WEB_CACHE, "download", res.Header.Get("Content-Type"), uri).Append("data"))
|
|
default:
|
|
if strings.HasPrefix(res.Header.Get("Content-Type"), "text/html") {
|
|
b, _ := ioutil.ReadAll(res.Body)
|
|
m.Echo(string(b))
|
|
break
|
|
}
|
|
|
|
// 解析结果
|
|
var data interface{}
|
|
m.Assert(json.NewDecoder(res.Body).Decode(&data))
|
|
data = kit.KeyValue(map[string]interface{}{}, "", data)
|
|
m.Info("res: %s", kit.Formats(data))
|
|
kit.Fetch(data, func(key string, value interface{}) {
|
|
switch value := value.(type) {
|
|
case []interface{}:
|
|
m.Push(key, value)
|
|
default:
|
|
m.Push(key, kit.Format(value))
|
|
}
|
|
})
|
|
}
|
|
})
|
|
}
|
|
}},
|
|
ice.WEB_SERVE: {Name: "serve [shy|dev|self]", Help: "服务器", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
// 节点信息
|
|
m.Conf(ice.CLI_RUNTIME, "node.name", m.Conf(ice.CLI_RUNTIME, "boot.hostname"))
|
|
m.Conf(ice.CLI_RUNTIME, "node.type", ice.WEB_SERVER)
|
|
|
|
switch kit.Select("def", arg, 0) {
|
|
case "shy":
|
|
// 连接根服务
|
|
m.Richs(ice.WEB_SPIDE, nil, "shy", func(key string, value map[string]interface{}) {
|
|
m.Cmd(ice.WEB_SPACE, "connect", "shy")
|
|
})
|
|
fallthrough
|
|
case "dev":
|
|
// 系统初始化
|
|
m.Event(ice.SYSTEM_INIT)
|
|
fallthrough
|
|
case "self":
|
|
// 启动服务
|
|
m.Target().Start(m, "self")
|
|
fallthrough
|
|
default:
|
|
// 连接上游服务
|
|
m.Richs(ice.WEB_SPIDE, nil, "dev", func(key string, value map[string]interface{}) {
|
|
m.Cmd(ice.WEB_SPACE, "connect", "dev")
|
|
})
|
|
}
|
|
}},
|
|
ice.WEB_SPACE: {Name: "space", Help: "空间站", Meta: kit.Dict("exports", []string{"pod", "name"}), List: kit.List(
|
|
kit.MDB_INPUT, "text", "name", "name",
|
|
kit.MDB_INPUT, "button", "value", "查看", "action", "auto",
|
|
kit.MDB_INPUT, "button", "value", "返回", "cb", "Last",
|
|
), Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) == 0 {
|
|
// 节点列表
|
|
m.Richs(ice.WEB_SPACE, nil, "*", func(key string, value map[string]interface{}) {
|
|
m.Push(key, value, []string{"time", "type", "name", "user"})
|
|
})
|
|
m.Sort("name")
|
|
return
|
|
}
|
|
|
|
web := m.Target().Server().(*Frame)
|
|
switch arg[0] {
|
|
case "auth":
|
|
m.Richs(ice.WEB_SPACE, nil, arg[1], func(key string, value map[string]interface{}) {
|
|
sessid := kit.Format(kit.Value(value, "sessid"))
|
|
if value["user"] = arg[2]; sessid == "" || m.Cmdx(ice.AAA_SESS, "check", sessid) != arg[1] {
|
|
sessid = m.Cmdx(ice.AAA_SESS, "create", arg[2:])
|
|
value["sessid"] = sessid
|
|
}
|
|
m.Cmd(ice.WEB_SPACE, arg[1], "sessid", sessid)
|
|
})
|
|
|
|
case "share":
|
|
switch arg[1] {
|
|
case "add":
|
|
m.Cmdy(ice.WEB_SPIDE, "self", path.Join("/space/share/add", path.Join(arg[2:]...)))
|
|
default:
|
|
m.Richs(ice.WEB_SPIDE, nil, m.Option("_dev"), func(key string, value map[string]interface{}) {
|
|
m.Log(ice.LOG_CREATE, "dev: %s share: %s", m.Option("_dev"), arg[1])
|
|
value["share"] = arg[1]
|
|
})
|
|
}
|
|
|
|
case "connect":
|
|
// 基本信息
|
|
dev := kit.Select("dev", arg, 1)
|
|
node := m.Conf(ice.CLI_RUNTIME, "node.type")
|
|
name := m.Conf(ice.CLI_RUNTIME, "node.name")
|
|
user := m.Conf(ice.CLI_RUNTIME, "boot.username")
|
|
|
|
m.Hold(1).Gos(m, func(m *ice.Message) {
|
|
m.Richs(ice.WEB_SPIDE, nil, dev, func(key string, value map[string]interface{}) {
|
|
proto := kit.Select("ws", "wss", kit.Format(kit.Value(value, "client.protocol")) == "https")
|
|
host := kit.Format(kit.Value(value, "client.hostname"))
|
|
|
|
for i := 0; i < m.Confi(ice.WEB_SPACE, "meta.redial.c"); i++ {
|
|
if u, e := url.Parse(kit.MergeURL(proto+"://"+host+"/space/", "node", node, "name", name, "user", user, "share", value["share"])); m.Assert(e) {
|
|
if s, e := net.Dial("tcp", host); !m.Warn(e != nil, "%s", e) {
|
|
if s, _, e := websocket.NewClient(s, u, nil, m.Confi(ice.WEB_SPACE, "meta.buffer.r"), m.Confi(ice.WEB_SPACE, "meta.buffer.w")); !m.Warn(e != nil, "%s", e) {
|
|
|
|
// 连接成功
|
|
m.Rich(ice.WEB_SPACE, nil, kit.Dict(kit.MDB_TYPE, ice.WEB_MASTER, kit.MDB_NAME, dev, "user", kit.Value(value, "client.hostname")))
|
|
m.Log(ice.LOG_CMDS, "%d conn %s success %s", i, dev, u)
|
|
if i = 0; web.HandleWSS(m, true, s, dev) {
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
// 断线重连
|
|
sleep := time.Duration(rand.Intn(m.Confi(ice.WEB_SPACE, "meta.redial.a"))*i+i*m.Confi(ice.WEB_SPACE, "meta.redial.b")) * time.Millisecond
|
|
m.Info("%d sleep: %s reconnect: %s", i, sleep, u)
|
|
time.Sleep(sleep)
|
|
}
|
|
}
|
|
})
|
|
m.Done()
|
|
})
|
|
|
|
default:
|
|
if len(arg) == 1 {
|
|
// 节点详情
|
|
m.Richs(ice.WEB_SPACE, nil, arg[0], func(key string, value map[string]interface{}) {
|
|
m.Push("detail", value)
|
|
})
|
|
break
|
|
}
|
|
|
|
if arg[0] == "" {
|
|
// 本地命令
|
|
m.Cmdy(arg[1:])
|
|
break
|
|
}
|
|
|
|
target := strings.Split(arg[0], ".")
|
|
m.Warn(m.Richs(ice.WEB_SPACE, nil, target[0], func(key string, value map[string]interface{}) {
|
|
if socket, ok := value["socket"].(*websocket.Conn); ok {
|
|
// 构造路由
|
|
id := kit.Format(c.ID())
|
|
m.Optionv(ice.MSG_SOURCE, []string{id})
|
|
m.Optionv(ice.MSG_TARGET, target[1:])
|
|
for _, k := range []string{"top", "hot"} {
|
|
if m.Options(k) {
|
|
m.Option(k, m.Option(k))
|
|
}
|
|
}
|
|
m.Set(ice.MSG_DETAIL, arg[1:]...)
|
|
m.Info("send %s %s", id, m.Format("meta"))
|
|
|
|
// 下发命令
|
|
m.Target().Server().(*Frame).send[id] = m
|
|
socket.WriteMessage(MSG_MAPS, []byte(m.Format("meta")))
|
|
t := time.AfterFunc(kit.Duration(m.Conf(ice.WEB_SPACE, "meta.timeout.c")), func() {
|
|
m.Log(ice.LOG_WARN, "timeout")
|
|
m.Back(nil)
|
|
})
|
|
m.Call(true, func(msg *ice.Message) *ice.Message {
|
|
// 返回结果
|
|
m.Copy(msg).Log("cost", "%s: %s %v", m.Format("cost"), arg[0], arg[1:])
|
|
t.Stop()
|
|
return nil
|
|
})
|
|
}
|
|
}) == nil, "not found %s", arg[0])
|
|
}
|
|
}},
|
|
ice.WEB_DREAM: {Name: "dream", Help: "梦想家", Meta: kit.Dict("exports", []string{"you", "name"},
|
|
"detail", []interface{}{"启动", "停止"},
|
|
), List: kit.List(
|
|
kit.MDB_INPUT, "text", "value", "", "name", "name",
|
|
kit.MDB_INPUT, "button", "value", "创建", "action", "auto",
|
|
), Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) > 1 {
|
|
if !m.Right(cmd, arg[1]) {
|
|
return
|
|
}
|
|
switch arg[1] {
|
|
case "启动":
|
|
arg = arg[:1]
|
|
case "停止", "stop":
|
|
m.Cmd(ice.WEB_SPACE, arg[0], "exit", "1")
|
|
time.Sleep(time.Second * 3)
|
|
m.Event(ice.DREAM_CLOSE, arg[0])
|
|
arg = arg[:0]
|
|
}
|
|
}
|
|
|
|
if len(arg) > 0 {
|
|
// 规范命名
|
|
if !strings.Contains(arg[0], "-") || !strings.HasPrefix(arg[0], "20") {
|
|
arg[0] = m.Time("20060102-") + arg[0]
|
|
}
|
|
|
|
// 创建目录
|
|
p := path.Join(m.Conf(ice.WEB_DREAM, "meta.path"), arg[0])
|
|
if _, e := os.Stat(p); e != nil {
|
|
os.MkdirAll(p, 0777)
|
|
}
|
|
|
|
if m.Richs(ice.WEB_SPACE, nil, arg[0], nil) == nil {
|
|
// 启动任务
|
|
m.Option("cmd_dir", p)
|
|
m.Option("cmd_type", "daemon")
|
|
m.Option("cmd_env", "ctx_log", "boot.log")
|
|
m.Cmd(m.Confv(ice.WEB_DREAM, "meta.cmd"), "self", arg[0])
|
|
time.Sleep(time.Second * 3)
|
|
m.Event(ice.DREAM_START, arg...)
|
|
}
|
|
}
|
|
|
|
// 任务列表
|
|
m.Cmdy("nfs.dir", m.Conf(ice.WEB_DREAM, "meta.path"), "time name")
|
|
m.Table(func(index int, value map[string]string, head []string) {
|
|
if m.Richs(ice.WEB_SPACE, nil, value["name"], func(key string, value map[string]interface{}) {
|
|
m.Push("type", value["type"])
|
|
m.Push("status", "start")
|
|
}) == nil {
|
|
m.Push("type", "none")
|
|
m.Push("status", "stop")
|
|
}
|
|
})
|
|
m.Sort("name")
|
|
m.Sort("status")
|
|
}},
|
|
|
|
ice.WEB_FAVOR: {Name: "favor", Help: "收藏夹", Meta: kit.Dict("remote", "you", "exports", []string{"hot", "favor"},
|
|
"detail", []string{"执行", "编辑", "收录", "导出", "下载"}), List: kit.List(
|
|
kit.MDB_INPUT, "text", "name", "hot", "action", "auto",
|
|
kit.MDB_INPUT, "text", "name", "id", "action", "auto",
|
|
kit.MDB_INPUT, "button", "value", "查看", "action", "auto",
|
|
kit.MDB_INPUT, "button", "value", "返回", "cb", "Last",
|
|
), Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) > 3 {
|
|
id := kit.Select(arg[0], kit.Select(m.Option("id"), arg[3], arg[2] == "id"))
|
|
switch arg[1] {
|
|
case "modify":
|
|
// 编辑收藏
|
|
m.Richs(ice.WEB_FAVOR, nil, m.Option("hot"), func(key string, value map[string]interface{}) {
|
|
m.Grows(ice.WEB_FAVOR, kit.Keys(kit.MDB_HASH, key), "id", id, func(index int, value map[string]interface{}) {
|
|
m.Info("modify favor: %s index: %d value: %v->%v", key, index, value[arg[2]], arg[3])
|
|
kit.Value(value, arg[2], arg[3])
|
|
})
|
|
})
|
|
arg = []string{m.Option("hot")}
|
|
|
|
case "收录":
|
|
m.Richs(ice.WEB_FAVOR, nil, m.Option("hot"), func(key string, value map[string]interface{}) {
|
|
m.Grows(ice.WEB_FAVOR, kit.Keys(kit.MDB_HASH, key), "id", id, func(index int, value map[string]interface{}) {
|
|
m.Cmd(ice.WEB_STORY, "add", value["type"], m.Option("top"), value["text"])
|
|
})
|
|
})
|
|
arg = []string{m.Option("hot")}
|
|
|
|
case "执行":
|
|
m.Event(ice.FAVOR_START, m.Option("you"), kit.Select(m.Option("hot"), arg[3], arg[2] == "favor"))
|
|
arg = arg[:0]
|
|
case "导出":
|
|
arg = []string{"export", m.Option("hot")}
|
|
}
|
|
}
|
|
|
|
if len(arg) > 0 {
|
|
switch arg[0] {
|
|
case "import":
|
|
if len(arg) == 2 {
|
|
m.Cmdy(ice.MDB_IMPORT, ice.WEB_FAVOR, kit.MDB_HASH, kit.MDB_HASH, arg[1])
|
|
} else {
|
|
m.Richs(ice.WEB_FAVOR, nil, arg[2], func(key string, value map[string]interface{}) {
|
|
m.Cmdy(ice.MDB_IMPORT, ice.WEB_FAVOR, kit.Keys(kit.MDB_HASH, key), kit.MDB_LIST, arg[1])
|
|
})
|
|
}
|
|
return
|
|
case "export":
|
|
if len(arg) == 1 {
|
|
m.Cmdy(ice.MDB_EXPORT, ice.WEB_FAVOR, kit.MDB_HASH, kit.MDB_HASH, "favor.json")
|
|
} else {
|
|
m.Option("cache.limit", "1000")
|
|
m.Richs(ice.WEB_FAVOR, nil, arg[1], func(key string, value map[string]interface{}) {
|
|
m.Cmdy(ice.MDB_EXPORT, ice.WEB_FAVOR, kit.Keys(kit.MDB_HASH, key), kit.MDB_LIST, arg[1]+".csv")
|
|
})
|
|
}
|
|
return
|
|
}
|
|
}
|
|
|
|
if len(arg) == 0 {
|
|
// 收藏门类
|
|
m.Richs(ice.WEB_FAVOR, nil, "*", func(key string, value map[string]interface{}) {
|
|
m.Push("time", kit.Value(value, "meta.time"))
|
|
m.Push("favor", kit.Value(value, "meta.name"))
|
|
m.Push("count", kit.Value(value, "meta.count"))
|
|
})
|
|
m.Sort("favor")
|
|
return
|
|
}
|
|
|
|
// 创建收藏
|
|
favor := ""
|
|
if m.Richs(ice.WEB_FAVOR, nil, arg[0], func(key string, value map[string]interface{}) {
|
|
favor = key
|
|
}) == nil && len(arg) > 1 {
|
|
favor = m.Rich(ice.WEB_FAVOR, nil, kit.Data(kit.MDB_NAME, arg[0]))
|
|
m.Log(ice.LOG_CREATE, "favor: %s name: %s", favor, arg[0])
|
|
}
|
|
|
|
if len(arg) == 1 {
|
|
m.Option("cache.limit", 30)
|
|
// 收藏列表
|
|
m.Grows(ice.WEB_FAVOR, kit.Keys(kit.MDB_HASH, favor), "", "", func(index int, value map[string]interface{}) {
|
|
m.Push(kit.Format(index), value, []string{kit.MDB_TIME, kit.MDB_ID, kit.MDB_TYPE, kit.MDB_NAME, kit.MDB_TEXT})
|
|
})
|
|
return
|
|
}
|
|
|
|
if len(arg) == 2 {
|
|
// 收藏详情
|
|
m.Grows(ice.WEB_FAVOR, kit.Keys(kit.MDB_HASH, favor), "id", arg[1], func(index int, value map[string]interface{}) {
|
|
m.Push("detail", value)
|
|
})
|
|
return
|
|
}
|
|
if arg[1] == "file" {
|
|
arg[1] = kit.MIME_FILE
|
|
}
|
|
|
|
// 添加收藏
|
|
extra := kit.Dict()
|
|
for i := 4; i < len(arg)-1; i += 2 {
|
|
kit.Value(extra, arg[i], arg[i+1])
|
|
}
|
|
index := m.Grow(ice.WEB_FAVOR, kit.Keys(kit.MDB_HASH, favor), kit.Dict(
|
|
kit.MDB_TYPE, arg[1], kit.MDB_NAME, arg[2], kit.MDB_TEXT, kit.Select("", arg, 3),
|
|
"extra", extra,
|
|
))
|
|
m.Log(ice.LOG_INSERT, "favor: %s index: %d name: %s text: %s", favor, index, arg[2], kit.Select("", arg, 3))
|
|
m.Echo("%d", index)
|
|
}},
|
|
ice.WEB_CACHE: {Name: "cache", Help: "缓存池", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) == 0 {
|
|
// 记录列表
|
|
m.Grows(ice.WEB_CACHE, nil, "", "", func(index int, value map[string]interface{}) {
|
|
m.Push(kit.Format(index), value, []string{kit.MDB_TIME, kit.MDB_SIZE, kit.MDB_TYPE, kit.MDB_NAME, kit.MDB_TEXT})
|
|
m.Push(kit.MDB_LINK, kit.Format(m.Conf(ice.WEB_SHARE, "meta.template.download"), value["data"], kit.Short(value["data"])))
|
|
|
|
})
|
|
return
|
|
}
|
|
|
|
switch arg[0] {
|
|
case "catch":
|
|
if f, e := os.Open(arg[2]); m.Assert(e) {
|
|
defer f.Close()
|
|
|
|
h := kit.Hashs(f)
|
|
if o, p, e := kit.Create(path.Join(m.Conf(ice.WEB_CACHE, ice.Meta("path")), h[:2], h)); m.Assert(e) {
|
|
defer o.Close()
|
|
f.Seek(0, os.SEEK_SET)
|
|
|
|
if n, e := io.Copy(o, f); m.Assert(e) {
|
|
m.Log(ice.LOG_IMPORT, "%s: %s", kit.FmtSize(n), p)
|
|
arg = kit.Simple(arg[0], arg[1], path.Base(arg[2]), p, p, n)
|
|
}
|
|
}
|
|
}
|
|
fallthrough
|
|
case "upload", "download":
|
|
// 打开文件
|
|
if m.R != nil {
|
|
if f, h, e := m.R.FormFile(kit.Select("upload", arg, 1)); e == nil {
|
|
defer f.Close()
|
|
|
|
// 创建文件
|
|
file := kit.Hashs(f)
|
|
if o, p, e := kit.Create(path.Join(m.Conf(ice.WEB_CACHE, ice.Meta("path")), file[:2], file)); m.Assert(e) {
|
|
defer o.Close()
|
|
|
|
// 保存文件
|
|
f.Seek(0, os.SEEK_SET)
|
|
if n, e := io.Copy(o, f); m.Assert(e) {
|
|
m.Info("upload: %s file: %s", kit.FmtSize(n), p)
|
|
arg = kit.Simple(arg[0], h.Header.Get("Content-Type"), h.Filename, p, p, n)
|
|
}
|
|
}
|
|
}
|
|
} else if r, ok := m.Optionv("response").(*http.Response); ok {
|
|
if buf, e := ioutil.ReadAll(r.Body); m.Assert(e) {
|
|
file := kit.Hashs(string(buf))
|
|
if o, p, e := kit.Create(path.Join(m.Conf(ice.WEB_CACHE, ice.Meta("path")), file[:2], file)); m.Assert(e) {
|
|
defer o.Close()
|
|
if n, e := o.Write(buf); m.Assert(e) {
|
|
m.Info("download: %s file: %s", kit.FmtSize(int64(n)), p)
|
|
arg = kit.Simple(arg[0], arg[1], arg[2], p, p, n)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
fallthrough
|
|
case "add":
|
|
// 添加数据
|
|
size := kit.Select(kit.Format(len(arg[3])), arg, 5)
|
|
data := kit.Dict(
|
|
kit.MDB_TYPE, arg[1], kit.MDB_NAME, arg[2], kit.MDB_TEXT, arg[3],
|
|
kit.MDB_FILE, kit.Select("", arg, 4), kit.MDB_SIZE, size,
|
|
)
|
|
h := m.Rich(ice.WEB_CACHE, nil, data)
|
|
m.Log(ice.LOG_CREATE, "cache: %s %s: %s", h, arg[1], arg[2])
|
|
|
|
// 保存数据
|
|
if arg[0] == "add" && len(arg) == 4 {
|
|
p := path.Join(m.Conf(ice.WEB_CACHE, ice.Meta("path")), h[:2], h)
|
|
if m.Cmd("nfs.save", p, arg[3]); kit.Int(size) > 512 {
|
|
data["text"], data["file"], arg[3] = p, p, p
|
|
}
|
|
}
|
|
|
|
// 添加记录
|
|
m.Grow(ice.WEB_CACHE, nil, kit.Dict(
|
|
kit.MDB_TYPE, arg[1], kit.MDB_NAME, arg[2], kit.MDB_TEXT, arg[3],
|
|
kit.MDB_SIZE, size, "data", h,
|
|
))
|
|
|
|
// 返回结果
|
|
m.Push("time", m.Time())
|
|
m.Push("type", arg[1])
|
|
m.Push("name", arg[2])
|
|
m.Push("text", arg[3])
|
|
m.Push("size", size)
|
|
m.Push("data", h)
|
|
}
|
|
}},
|
|
ice.WEB_STORY: {Name: "story", Help: "故事会", Meta: kit.Dict("remote", "you", "exports", []string{"top", "story"},
|
|
"detail", []string{"归档", "共享", "导出", "下载"}), List: kit.List(
|
|
kit.MDB_INPUT, "text", "name", "top", "action", "auto",
|
|
kit.MDB_INPUT, "text", "name", "list", "action", "auto",
|
|
kit.MDB_INPUT, "button", "value", "查看", "action", "auto",
|
|
kit.MDB_INPUT, "button", "value", "返回", "cb", "Last",
|
|
), Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) > 1 {
|
|
switch arg[1] {
|
|
case "共享":
|
|
switch arg[2] {
|
|
case "story", "list":
|
|
pod := ""
|
|
if list := kit.Simple(m.Optionv("_source")); len(list) > 0 {
|
|
pod = strings.Join(list[1:], ".")
|
|
}
|
|
msg := m.Cmd(ice.WEB_STORY, "index", arg[3])
|
|
m.Cmdy(ice.WEB_SPACE, "share", "add", ice.TYPE_STORY,
|
|
msg.Append("story"), arg[3], "pod", pod, "data", arg[3])
|
|
return
|
|
}
|
|
case "导出":
|
|
switch arg[2] {
|
|
case "story", "list":
|
|
arg = []string{ice.STORY_WATCH, arg[3], m.Option("story")}
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(arg) == 0 {
|
|
// 故事列表
|
|
m.Richs(ice.WEB_STORY, "head", "*", func(key string, value map[string]interface{}) {
|
|
m.Push(key, value, []string{"time", "story", "count"})
|
|
})
|
|
m.Sort("time", "time_r")
|
|
return
|
|
}
|
|
|
|
// head list data time text file
|
|
switch arg[0] {
|
|
case ice.STORY_PULL:
|
|
// 起止节点
|
|
prev, begin, end := "", arg[2], ""
|
|
m.Richs(ice.WEB_STORY, "head", arg[2], func(key string, val map[string]interface{}) {
|
|
prev, end = kit.Format(val["list"]), kit.Format(kit.Value(val, kit.Keys("remote", arg[1], "pull", "list")))
|
|
})
|
|
|
|
pull := end
|
|
var first map[string]interface{}
|
|
for begin != end {
|
|
if m.Cmd(ice.WEB_SPIDE, arg[1], "msg", "/story/pull", "begin", begin, "end", end).Table(func(index int, value map[string]string, head []string) {
|
|
if m.Richs(ice.WEB_CACHE, nil, value["data"], nil) == nil {
|
|
// 导入缓存
|
|
m.Log(ice.LOG_IMPORT, "%v: %v", value["data"], value["save"])
|
|
if node := kit.UnMarshal(value["save"]); kit.Format(kit.Value(node, "file")) != "" {
|
|
m.Cmd(ice.WEB_SPIDE, arg[1], "cache", "GET", "/story/download/"+value["data"])
|
|
} else {
|
|
m.Conf(ice.WEB_CACHE, kit.Keys("hash", value["data"]), node)
|
|
}
|
|
}
|
|
|
|
node := kit.UnMarshal(value["node"]).(map[string]interface{})
|
|
if m.Richs(ice.WEB_STORY, nil, value["list"], nil) == nil {
|
|
// 导入节点
|
|
m.Log(ice.LOG_IMPORT, "%v: %v", value["list"], value["node"])
|
|
m.Conf(ice.WEB_STORY, kit.Keys("hash", value["list"]), node)
|
|
}
|
|
|
|
if first == nil {
|
|
if m.Richs(ice.WEB_STORY, "head", node["story"], nil) == nil {
|
|
// 自动创建
|
|
h := m.Rich(ice.WEB_STORY, "head", kit.Dict(
|
|
"scene", node["scene"], "story", node["story"],
|
|
"count", node["count"], "list", value["list"],
|
|
))
|
|
m.Log(ice.LOG_CREATE, "%v: %v", h, node["story"])
|
|
}
|
|
|
|
pull, first = kit.Format(value["list"]), node
|
|
m.Richs(ice.WEB_STORY, "head", node["story"], func(key string, val map[string]interface{}) {
|
|
prev = kit.Format(val["list"])
|
|
if kit.Int(node["count"]) > kit.Int(kit.Value(val, kit.Keys("remote", arg[1], "pull", "count"))) {
|
|
// 更新分支
|
|
m.Log(ice.LOG_IMPORT, "%v: %v", pull, arg[1])
|
|
kit.Value(val, kit.Keys("remote", arg[1], "pull"), kit.Dict(
|
|
"time", node["time"], "list", pull, "count", node["count"],
|
|
))
|
|
}
|
|
})
|
|
}
|
|
|
|
if prev == kit.Format(node["prev"]) || prev == kit.Format(node["pull"]) {
|
|
// 快速合并
|
|
m.Log(ice.LOG_IMPORT, "%v: %v", pull, arg[2])
|
|
m.Richs(ice.WEB_STORY, "head", node["story"], func(key string, val map[string]interface{}) {
|
|
val["count"] = first["count"]
|
|
val["time"] = first["time"]
|
|
val["list"] = pull
|
|
})
|
|
prev = pull
|
|
}
|
|
|
|
begin = kit.Format(node["prev"])
|
|
}).Append("list") == "" {
|
|
break
|
|
}
|
|
}
|
|
if !m.Warn(prev != pull, "unmerge: %s", pull) {
|
|
m.Echo(pull)
|
|
}
|
|
|
|
case ice.STORY_PUSH:
|
|
// 更新分支
|
|
m.Cmdx(ice.WEB_STORY, "pull", arg[1:])
|
|
|
|
// 查询索引
|
|
push, list := "", m.Cmd(ice.WEB_STORY, "index", arg[2]).Append("list")
|
|
m.Richs(ice.WEB_STORY, "head", arg[2], func(key string, val map[string]interface{}) {
|
|
push = kit.Format(kit.Value(val, kit.Keys("remote", arg[1], "push", "list")))
|
|
})
|
|
|
|
// 查询节点
|
|
nodes := []string{}
|
|
for list != "" && list != push {
|
|
m.Richs(ice.WEB_STORY, nil, list, func(key string, value map[string]interface{}) {
|
|
nodes, list = append(nodes, list), kit.Format(value["prev"])
|
|
})
|
|
}
|
|
|
|
for _, v := range kit.Revert(nodes) {
|
|
m.Richs(ice.WEB_STORY, nil, v, func(list string, node map[string]interface{}) {
|
|
m.Richs(ice.WEB_CACHE, nil, node["data"], func(data string, save map[string]interface{}) {
|
|
// 推送节点
|
|
m.Log(ice.LOG_EXPORT, "%s: %s", v, kit.Format(node))
|
|
m.Cmd(ice.WEB_SPIDE, arg[1], "/story/push",
|
|
"list", v, "node", kit.Format(node),
|
|
"data", node["data"], "save", kit.Format(save),
|
|
)
|
|
|
|
if kit.Format(save["file"]) != "" {
|
|
// 推送缓存
|
|
m.Cmd(ice.WEB_SPIDE, arg[1], "/story/upload",
|
|
"part", "upload", "@"+kit.Format(save["file"]),
|
|
)
|
|
}
|
|
})
|
|
})
|
|
}
|
|
|
|
// 更新分支
|
|
m.Cmd(ice.WEB_STORY, "pull", arg[1:])
|
|
|
|
case ice.STORY_WATCH:
|
|
msg := m.Cmd(ice.WEB_STORY, "index", arg[1])
|
|
name := kit.Select(arg[1], arg, 2)
|
|
os.Remove(kit.Keys(name, "bak"))
|
|
os.Rename(name, kit.Keys(name, "bak"))
|
|
if msg.Append("file") != "" {
|
|
p := path.Dir(name)
|
|
e := os.MkdirAll(p, 0777)
|
|
m.Log("what", "%v", e)
|
|
|
|
e = os.Link(msg.Append("file"), name)
|
|
m.Log("what", "%v", e)
|
|
m.Log(ice.LOG_EXPORT, "%s: %s", msg.Append("file"), name)
|
|
} else {
|
|
if f, p, e := kit.Create(name); m.Assert(e) {
|
|
defer f.Close()
|
|
f.WriteString(msg.Append("text"))
|
|
m.Log(ice.LOG_EXPORT, "%s: %s", msg.Append("text"), p)
|
|
}
|
|
}
|
|
|
|
case ice.STORY_CATCH:
|
|
if last := m.Richs(ice.WEB_STORY, "head", arg[2], nil); last != nil {
|
|
if t, e := time.ParseInLocation(ice.ICE_TIME, kit.Format(last["time"]), time.Local); e == nil {
|
|
if s, e := os.Stat(arg[2]); e == nil && s.ModTime().Before(t) {
|
|
m.Info("%s last: %s", arg[2], kit.Format(t))
|
|
m.Echo("%s", last["list"])
|
|
break
|
|
}
|
|
}
|
|
}
|
|
fallthrough
|
|
case "add", ice.STORY_UPLOAD:
|
|
pull := ""
|
|
if arg[0] == ice.STORY_CATCH {
|
|
pull, arg = kit.Select("", arg, 3), arg[:3]
|
|
m.Richs(ice.WEB_STORY, "head", pull, func(key string, value map[string]interface{}) {
|
|
// 合并分支
|
|
pull = kit.Format(kit.Value(value, kit.Keys("remote", pull, "pull", "list")))
|
|
})
|
|
}
|
|
|
|
// 保存数据
|
|
if m.Richs(ice.WEB_CACHE, nil, kit.Select("", arg, 3), func(key string, value map[string]interface{}) {
|
|
if len(arg) > 3 {
|
|
arg[3] = key
|
|
}
|
|
}) == nil {
|
|
m.Cmdy(ice.WEB_CACHE, arg)
|
|
arg = []string{arg[0], m.Append("type"), m.Append("name"), m.Append("data")}
|
|
}
|
|
|
|
// 查询索引
|
|
head, prev, value, count := "", "", map[string]interface{}{}, 0
|
|
m.Richs(ice.WEB_STORY, "head", arg[2], func(key string, val map[string]interface{}) {
|
|
head, prev, value, count = key, kit.Format(val["list"]), val, kit.Int(val["count"])
|
|
m.Log("info", "head: %v prev: %v count: %v", head, prev, count)
|
|
})
|
|
|
|
if last := m.Richs(ice.WEB_STORY, nil, prev, nil); prev != "" && last != nil && last["data"] == arg[3] {
|
|
// 重复提交
|
|
break
|
|
}
|
|
|
|
// 添加节点
|
|
list := m.Rich(ice.WEB_STORY, nil, kit.Dict(
|
|
"scene", arg[1], "story", arg[2], "count", count+1, "data", arg[3], "prev", prev, "pull", pull,
|
|
))
|
|
m.Log(ice.LOG_CREATE, "story: %s %s: %s", list, arg[1], arg[2])
|
|
m.Push("list", list)
|
|
m.Push("pull", pull)
|
|
|
|
if head == "" {
|
|
// 添加索引
|
|
m.Rich(ice.WEB_STORY, "head", kit.Dict("scene", arg[1], "story", arg[2], "count", count+1, "list", list))
|
|
} else {
|
|
// 更新索引
|
|
value["count"] = count + 1
|
|
value["time"] = m.Time()
|
|
value["list"] = list
|
|
}
|
|
m.Echo(list)
|
|
|
|
case ice.STORY_DOWNLOAD:
|
|
// 下载文件
|
|
m.Cmdy(ice.WEB_STORY, "index", arg[1])
|
|
m.Push("_output", kit.Select("file", "result", m.Append("file") == ""))
|
|
|
|
case "commit":
|
|
// 查询索引
|
|
head := kit.Hashs(arg[1])
|
|
prev := m.Conf("story", ice.Meta("head", head, "list"))
|
|
m.Log("info", "head: %v prev: %v", head, prev)
|
|
|
|
// 查询节点
|
|
menu := map[string]string{}
|
|
for i := 2; i < len(arg); i++ {
|
|
if i < len(arg)-1 && m.Confs("story", kit.Keys("hash", arg[i+1])) {
|
|
menu[arg[i]] = arg[i+1]
|
|
i++
|
|
} else if head := kit.Hashs(arg[i]); m.Confs("story", kit.Keys("meta", "head", head)) {
|
|
menu[arg[i]] = m.Conf(ice.WEB_STORY, ice.Meta("head", head, "list"))
|
|
} else {
|
|
m.Error(true, "not found %v", arg[i])
|
|
return
|
|
}
|
|
}
|
|
|
|
// 添加节点
|
|
meta := map[string]interface{}{
|
|
"time": m.Time(),
|
|
"scene": "commit",
|
|
"story": arg[1],
|
|
"list": menu,
|
|
"prev": prev,
|
|
}
|
|
list := m.Rich("story", nil, meta)
|
|
m.Log("info", "list: %v meta: %v", list, kit.Format(meta))
|
|
|
|
// 添加索引
|
|
m.Conf("story", ice.Meta("head", head), map[string]interface{}{
|
|
"time": m.Time(), "scene": "commit", "story": arg[1], "list": list,
|
|
})
|
|
m.Echo(list)
|
|
|
|
case "history":
|
|
// 历史记录
|
|
list := m.Cmd(ice.WEB_STORY, "index", arg[1]).Append("list")
|
|
for i := 0; i < 10 && list != ""; i++ {
|
|
m.Confm(ice.WEB_STORY, kit.Keys("hash", list), func(value map[string]interface{}) {
|
|
// 直连节点
|
|
val := m.Confm(ice.WEB_CACHE, kit.Keys("hash", value["data"]))
|
|
if val == nil {
|
|
data := kit.Format(value["data"])
|
|
if f, e := os.Open(path.Join(m.Conf(ice.WEB_CACHE, "meta.store"), data[:2], data)); e != nil ||
|
|
json.NewDecoder(f).Decode(&val) != nil {
|
|
return
|
|
}
|
|
}
|
|
m.Push(list, value, []string{"key", "time", "count", "scene", "story"})
|
|
|
|
m.Push("drama", val["text"])
|
|
m.Push("data", value["data"])
|
|
|
|
// 复合节点
|
|
kit.Fetch(value["list"], func(key string, val string) {
|
|
m.Push(list, value, []string{"key", "time", "count"})
|
|
|
|
node := m.Confm(ice.WEB_STORY, kit.Keys("hash", val))
|
|
m.Push("scene", node["scene"])
|
|
m.Push("story", kit.Keys(kit.Format(value["story"]), key))
|
|
|
|
m.Push("drama", m.Conf(ice.WEB_CACHE, kit.Keys("hash", node["data"], "text")))
|
|
m.Push("data", node["data"])
|
|
})
|
|
|
|
list = kit.Format(value["prev"])
|
|
})
|
|
}
|
|
|
|
case "index":
|
|
// 查询索引
|
|
if m.Richs(ice.WEB_STORY, "head", arg[1], func(key string, value map[string]interface{}) {
|
|
arg[1] = kit.Format(value["list"])
|
|
}) == nil {
|
|
arg[1] = kit.Select(arg[1], m.Conf(ice.WEB_STORY, kit.Keys("head.hash", arg[1], "list")))
|
|
}
|
|
// 查询节点
|
|
if node := m.Confm(ice.WEB_STORY, kit.Keys("hash", arg[1])); node != nil {
|
|
m.Push("list", arg[1])
|
|
m.Push(arg[1], node, []string{"scene", "story"})
|
|
arg[1] = kit.Format(node["data"])
|
|
}
|
|
|
|
// 查询数据
|
|
node := m.Confm(ice.WEB_CACHE, kit.Keys("hash", arg[1]))
|
|
if node == nil {
|
|
if f, e := os.Open(path.Join(m.Conf(ice.WEB_CACHE, "meta.store"), arg[1][:2], arg[1])); e != nil ||
|
|
json.NewDecoder(f).Decode(&node) != nil {
|
|
return
|
|
}
|
|
}
|
|
m.Push("data", arg[1])
|
|
m.Push(arg[1], node, []string{"text", "time", "size", "type", "name", "file"})
|
|
m.Echo("%s", node["text"])
|
|
default:
|
|
if len(arg) == 1 {
|
|
if _, e := os.Stat(arg[0]); e == nil {
|
|
if scene := m.Conf(ice.WEB_STORY, kit.Keys("mime", strings.TrimPrefix(path.Ext(arg[0]), "."))); scene != "" {
|
|
m.Cmd(ice.WEB_STORY, ice.STORY_CATCH, scene, arg[0])
|
|
}
|
|
}
|
|
|
|
m.Cmd(ice.WEB_STORY, "history", arg).Table(func(index int, value map[string]string, head []string) {
|
|
m.Push("time", value["time"])
|
|
m.Push("list", value["key"])
|
|
m.Push("scene", value["scene"])
|
|
m.Push("story", value["story"])
|
|
m.Push("drama", value["drama"])
|
|
m.Push("link", kit.Format(m.Conf(ice.WEB_SHARE, "meta.template.download"),
|
|
kit.Format(value["data"])+"&pod="+m.Conf(ice.CLI_RUNTIME, "node.name"), kit.Short(value["data"])))
|
|
})
|
|
break
|
|
}
|
|
m.Richs(ice.WEB_STORY, nil, arg[1], func(key string, value map[string]interface{}) {
|
|
m.Push("detail", value)
|
|
})
|
|
}
|
|
}},
|
|
ice.WEB_SHARE: {Name: "share", Help: "共享链", List: kit.List(
|
|
kit.MDB_INPUT, "text", "name", "share", "action", "auto",
|
|
kit.MDB_INPUT, "button", "value", "查看", "action", "auto",
|
|
kit.MDB_INPUT, "button", "value", "返回", "cb", "Last",
|
|
), Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) == 0 {
|
|
// 共享列表
|
|
m.Grows(ice.WEB_SHARE, nil, "", "", func(key int, value map[string]interface{}) {
|
|
m.Push(kit.Format(key), value, []string{kit.MDB_TIME, "share", kit.MDB_TYPE, kit.MDB_NAME, kit.MDB_TEXT})
|
|
m.Push("link", fmt.Sprintf(m.Conf(ice.WEB_SHARE, "meta.template.share"), value["share"], value["share"]))
|
|
})
|
|
return
|
|
}
|
|
if len(arg) == 1 {
|
|
m.Richs(ice.WEB_SHARE, nil, arg[0], func(key string, value map[string]interface{}) {
|
|
m.Push("detail", value)
|
|
})
|
|
return
|
|
}
|
|
|
|
switch arg[0] {
|
|
case "add":
|
|
arg = arg[1:]
|
|
fallthrough
|
|
default:
|
|
// 创建共享
|
|
extra := kit.Dict()
|
|
for i := 3; i < len(arg)-1; i += 2 {
|
|
kit.Value(extra, arg[i], arg[i+1])
|
|
}
|
|
|
|
h := m.Rich(ice.WEB_SHARE, nil, kit.Dict(
|
|
kit.MDB_TYPE, arg[0], kit.MDB_NAME, arg[1], kit.MDB_TEXT, kit.Select("", arg, 2),
|
|
"extra", extra,
|
|
))
|
|
m.Grow(ice.WEB_SHARE, nil, kit.Dict(
|
|
kit.MDB_TYPE, arg[0], kit.MDB_NAME, arg[1], kit.MDB_TEXT, kit.Select("", arg, 2),
|
|
"share", h,
|
|
))
|
|
m.Log(ice.LOG_CREATE, "share: %s extra: %s", h, kit.Format(extra))
|
|
m.Echo(h)
|
|
}
|
|
}},
|
|
|
|
ice.WEB_ROUTE: {Name: "route", Help: "路由", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
}},
|
|
ice.WEB_PROXY: {Name: "proxy", Help: "代理", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
}},
|
|
ice.WEB_GROUP: {Name: "group", Help: "分组", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
}},
|
|
ice.WEB_LABEL: {Name: "label", Help: "标签", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
}},
|
|
|
|
"/share/": {Name: "/share/", Help: "共享链", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
switch arg[0] {
|
|
case "local":
|
|
http.ServeFile(m.W, m.R, m.Cmdx(arg[1], path.Join(arg[2:]...)))
|
|
|
|
case "login":
|
|
Cookie(m, m.Cmdx(ice.AAA_USER, "login", m.Option("username"), m.Option("password")))
|
|
|
|
default:
|
|
m.Richs(ice.WEB_SHARE, nil, arg[0], func(key string, value map[string]interface{}) {
|
|
m.Info("share %s %v", arg, kit.Format(value))
|
|
switch value["type"] {
|
|
case ice.TYPE_STORY:
|
|
if m.Cmdy(ice.WEB_STORY, "index", kit.Value(value, "text")).Append("text") == "" {
|
|
m.Cmdy(ice.WEB_SPACE, kit.Value(value, "extra.pod"), ice.WEB_STORY, "index", kit.Value(value, "text"))
|
|
}
|
|
|
|
p := path.Join("tmp/file", m.Append("data"))
|
|
if _, e := os.Stat(p); e == nil {
|
|
m.Append("_output", "file")
|
|
m.Append("file", p)
|
|
break
|
|
}
|
|
|
|
m.Set("result")
|
|
m.Render(m.Conf(ice.WEB_SHARE, "meta.template.story.prefix"))
|
|
m.Cmdy("web.wiki._text", m.Append("file"))
|
|
m.Render(m.Conf(ice.WEB_SHARE, "meta.template.story.suffix"))
|
|
m.Push("_output", "result")
|
|
|
|
if f, p, e := kit.Create(p); e == nil {
|
|
defer f.Close()
|
|
if n, e := f.WriteString(m.Result()); e == nil {
|
|
m.Log(ice.LOG_EXPORT, "%d: %s", n, p)
|
|
}
|
|
}
|
|
|
|
case "active":
|
|
m.Push("_output", "qrcode")
|
|
m.Echo(kit.Format(value))
|
|
|
|
case "qrcode":
|
|
m.Push("_output", "qrcode")
|
|
m.Echo("%s", value["text"])
|
|
|
|
default:
|
|
if m.Cmdy(ice.WEB_STORY, "index", value["data"]); m.Append("file") != "" {
|
|
m.Push("_output", "file")
|
|
} else {
|
|
m.Push("_output", "result")
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
}},
|
|
"/story/": {Name: "/story/", Help: "故事会", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
switch arg[0] {
|
|
case ice.STORY_PULL:
|
|
// 下载节点
|
|
list := m.Cmd(ice.WEB_STORY, "index", m.Option("begin")).Append("list")
|
|
for i := 0; i < 10 && list != "" && list != m.Option("end"); i++ {
|
|
if m.Richs(ice.WEB_STORY, nil, list, func(key string, value map[string]interface{}) {
|
|
// 节点信息
|
|
m.Push("list", key)
|
|
m.Push("node", kit.Format(value))
|
|
m.Push("data", value["data"])
|
|
m.Push("save", kit.Format(m.Richs(ice.WEB_CACHE, nil, value["data"], nil)))
|
|
list = kit.Format(value["prev"])
|
|
}) == nil {
|
|
break
|
|
}
|
|
}
|
|
|
|
case ice.STORY_PUSH:
|
|
// 上传节点
|
|
if m.Richs(ice.WEB_CACHE, nil, m.Option("data"), nil) == nil {
|
|
// 导入缓存
|
|
m.Log(ice.LOG_IMPORT, "%v: %v", m.Option("data"), m.Option("save"))
|
|
node := kit.UnMarshal(m.Option("save"))
|
|
m.Conf(ice.WEB_CACHE, kit.Keys("hash", m.Option("data")), node)
|
|
}
|
|
|
|
node := kit.UnMarshal(m.Option("node")).(map[string]interface{})
|
|
if m.Richs(ice.WEB_STORY, nil, m.Option("list"), nil) == nil {
|
|
// 导入节点
|
|
m.Log(ice.LOG_IMPORT, "%v: %v", m.Option("list"), m.Option("node"))
|
|
m.Conf(ice.WEB_STORY, kit.Keys("hash", m.Option("list")), node)
|
|
}
|
|
|
|
if head := m.Richs(ice.WEB_STORY, "head", node["story"], nil); head == nil {
|
|
// 自动创建
|
|
h := m.Rich(ice.WEB_STORY, "head", kit.Dict(
|
|
"scene", node["scene"], "story", node["story"],
|
|
"count", node["count"], "list", m.Option("list"),
|
|
))
|
|
m.Log(ice.LOG_CREATE, "%v: %v", h, node["story"])
|
|
} else if head["list"] == kit.Format(node["prev"]) || head["list"] == kit.Format(node["pull"]) {
|
|
// 快速合并
|
|
head["list"] = m.Option("list")
|
|
head["count"] = node["count"]
|
|
head["time"] = node["time"]
|
|
} else {
|
|
// 推送失败
|
|
}
|
|
|
|
case ice.STORY_UPLOAD:
|
|
// 上传数据
|
|
m.Cmdy(ice.WEB_CACHE, "upload")
|
|
|
|
case ice.STORY_DOWNLOAD:
|
|
// 下载数据
|
|
m.Cmdy(ice.WEB_STORY, "index", arg[1])
|
|
m.Push("_output", kit.Select("file", "result", m.Append("file") == ""))
|
|
}
|
|
}},
|
|
"/space/": {Name: "/space/", Help: "空间站", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
list := strings.Split(cmd, "/")
|
|
|
|
switch list[2] {
|
|
case "login":
|
|
m.Option(ice.MSG_SESSID, Cookie(m, m.Cmdx(ice.AAA_USER, "login", m.Option("username"), m.Option("password"))))
|
|
return
|
|
case "share":
|
|
m.Cmdy(ice.WEB_SHARE, list[3:])
|
|
return
|
|
}
|
|
|
|
if s, e := websocket.Upgrade(m.W, m.R, nil, m.Confi(ice.WEB_SPACE, "meta.buffer.r"), m.Confi(ice.WEB_SPACE, "meta.buffer.w")); m.Assert(e) {
|
|
m.Option("name", strings.Replace(m.Option("name"), ".", "_", -1))
|
|
if !m.Options("name") {
|
|
m.Option("name", kit.Hashs("uniq"))
|
|
}
|
|
|
|
// 共享空间
|
|
share := m.Option("share")
|
|
if m.Richs(ice.WEB_SHARE, nil, share, nil) == nil {
|
|
share = m.Cmdx(ice.WEB_SHARE, "add", m.Option("node"), m.Option("name"), m.Option("user"))
|
|
}
|
|
|
|
// 添加节点
|
|
h := m.Rich(ice.WEB_SPACE, nil, kit.Dict(
|
|
kit.MDB_TYPE, m.Option("node"),
|
|
kit.MDB_NAME, m.Option("name"),
|
|
kit.MDB_USER, m.Option("user"),
|
|
"sessid", m.Option("sessid"),
|
|
"share", share, "socket", s,
|
|
))
|
|
m.Log(ice.LOG_CREATE, "space: %s share: %s", m.Option(kit.MDB_NAME), share)
|
|
|
|
m.Gos(m, func(m *ice.Message) {
|
|
// 监听消息
|
|
m.Event(ice.SPACE_START, m.Option("node"), m.Option("name"))
|
|
m.Target().Server().(*Frame).HandleWSS(m, false, s, m.Option("name"))
|
|
m.Log(ice.LOG_CLOSE, "%s: %s", m.Option(kit.MDB_NAME), kit.Format(m.Confv(ice.WEB_SPACE, kit.Keys(kit.MDB_HASH, h))))
|
|
m.Event(ice.SPACE_CLOSE, m.Option("node"), m.Option("name"))
|
|
m.Confv(ice.WEB_SPACE, kit.Keys(kit.MDB_HASH, h), "")
|
|
})
|
|
|
|
// 共享空间
|
|
if share != m.Option("share") {
|
|
m.Cmd(ice.WEB_SPACE, m.Option("name"), ice.WEB_SPACE, "share", share)
|
|
}
|
|
m.Echo(share)
|
|
}
|
|
}},
|
|
"/static/volcanos/plugin/github.com/": {Name: "/space/", Help: "空间站", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
file := strings.TrimPrefix(cmd, "/static/volcanos/")
|
|
if _, e := os.Stat(path.Join("usr/volcanos", file)); e != nil {
|
|
m.Cmd("cli.system", "git", "clone", "https://"+strings.Join(strings.Split(cmd, "/")[4:7], "/"),
|
|
path.Join("usr/volcanos", strings.Join(strings.Split(cmd, "/")[3:7], "/")))
|
|
}
|
|
|
|
m.Push("_output", "void")
|
|
http.ServeFile(m.W, m.R, path.Join("usr/volcanos", file))
|
|
}},
|
|
"/plugin/github.com/": {Name: "/space/", Help: "空间站", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if _, e := os.Stat(path.Join("usr/volcanos", cmd)); e != nil {
|
|
m.Cmd("cli.system", "git", "clone", "https://"+strings.Join(strings.Split(cmd, "/")[2:5], "/"),
|
|
path.Join("usr/volcanos", strings.Join(strings.Split(cmd, "/")[1:5], "/")))
|
|
}
|
|
|
|
m.Push("_output", "void")
|
|
http.ServeFile(m.W, m.R, path.Join("usr/volcanos", cmd))
|
|
}},
|
|
},
|
|
}
|
|
|
|
func init() { ice.Index.Register(Index, &Frame{}) }
|