forked from x/icebergs
133 lines
3.7 KiB
Go
133 lines
3.7 KiB
Go
package input
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"encoding/csv"
|
|
"github.com/shylinux/icebergs"
|
|
"github.com/shylinux/icebergs/base/cli"
|
|
"github.com/shylinux/toolkits"
|
|
"math/rand"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
)
|
|
|
|
var Index = &ice.Context{Name: "input", Help: "输入法",
|
|
Caches: map[string]*ice.Cache{},
|
|
Configs: map[string]*ice.Config{
|
|
"input": {Name: "input", Help: "输入法", Value: kit.Data(
|
|
"store", "var/input/", "limit", "2000", "least", "1000",
|
|
kit.MDB_SHORT, "code",
|
|
)},
|
|
},
|
|
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", "input.json")
|
|
}},
|
|
|
|
ice.ICE_EXIT: {Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
m.Cmd(ice.CTX_CONFIG, "save", "input.json", "cli.input.input")
|
|
}},
|
|
|
|
"input": {Name: "input load|list", Help: "输入法", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
count := kit.Int(m.Conf("input", "meta.count"))
|
|
if len(arg) == 0 {
|
|
arg = append(arg, "list", kit.Format(count-rand.Intn(count)))
|
|
}
|
|
|
|
switch arg[0] {
|
|
case "load":
|
|
// 加载词库
|
|
if f, e := os.Open(arg[1]); m.Assert(e) {
|
|
bio := bufio.NewScanner(f)
|
|
for bio.Scan() {
|
|
if strings.HasPrefix(bio.Text(), "#") {
|
|
continue
|
|
}
|
|
line := kit.Split(bio.Text(), " \t")
|
|
m.Grow("input", nil, kit.Dict(
|
|
"text", line[0], "code", line[1], "weight", line[2],
|
|
))
|
|
}
|
|
}
|
|
case "push":
|
|
m.Rich("input", nil, kit.Dict(
|
|
"id", "0", "text", arg[1], "code", arg[2], "weight", kit.Select("99990000", arg, 3),
|
|
))
|
|
case "list":
|
|
// 词汇列表
|
|
m.Option("cache.offend", kit.Select("0", arg, 1))
|
|
m.Option("cache.limit", kit.Select("10", arg, 2))
|
|
m.Grows("input", nil, "", "", func(index int, value map[string]interface{}) {
|
|
m.Push("", value, []string{"id", "code", "text", "weight"})
|
|
})
|
|
}
|
|
}},
|
|
"match": {Name: "match [word [method]]", Help: "五笔字码", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
|
|
if len(arg) == 0 {
|
|
// 收藏列表
|
|
m.Cmdy(ice.WEB_FAVOR, "input.word")
|
|
return
|
|
}
|
|
|
|
m.Info("add %s: %s", arg[0], kit.Hashs(arg[0]))
|
|
m.Richs("input", nil, arg[0], func(key string, value map[string]interface{}) {
|
|
m.Push(key, value, []string{"id", "code", "text", "weight"})
|
|
})
|
|
|
|
// 搜索方法
|
|
method := "word"
|
|
if len(arg) > 1 {
|
|
method = arg[1]
|
|
}
|
|
switch method {
|
|
case "line":
|
|
case "word":
|
|
arg[0] = "^" + arg[0] + ","
|
|
}
|
|
|
|
// 字段列表
|
|
field := m.Confm("input", "meta.field")
|
|
if field == nil {
|
|
field = map[string]interface{}{}
|
|
head := []string{}
|
|
if f, e := os.Open(path.Join(m.Conf("input", "meta.store"), "input.csv")); m.Assert(e) {
|
|
defer f.Close()
|
|
|
|
bio := csv.NewReader(f)
|
|
head, e = bio.Read()
|
|
}
|
|
for i, k := range head {
|
|
field[k] = i
|
|
}
|
|
m.Conf("input", "meta.field", field)
|
|
}
|
|
|
|
// 搜索词汇
|
|
bio := csv.NewReader(bytes.NewBufferString(m.Cmdx(ice.CLI_SYSTEM, "grep", "-rh", arg[0], m.Conf("input", "meta.store"))))
|
|
for i := 0; i < kit.Int(kit.Select("100", arg, 2)); i++ {
|
|
if line, e := bio.Read(); e != nil {
|
|
break
|
|
} else if len(line) < 3 {
|
|
} else {
|
|
if method == "word" && i == 0 {
|
|
// 添加收藏
|
|
m.Cmd(ice.WEB_FAVOR, "input.word", "input",
|
|
line[kit.Int(field["code"])], line[kit.Int(field["text"])],
|
|
"id", line[kit.Int(field["id"])], "weight", line[kit.Int(field["weight"])],
|
|
)
|
|
}
|
|
for _, k := range []string{"id", "code", "text", "weight"} {
|
|
// 输出词汇
|
|
m.Push(k, line[kit.Int(field[k])])
|
|
}
|
|
}
|
|
}
|
|
m.Sort("weight", "int_r")
|
|
}},
|
|
},
|
|
}
|
|
|
|
func init() { cli.Index.Register(Index, nil) }
|