emozi/cmd/main.go

package main

import (
	"flag"
	"fmt"
	"strconv"
	"time"

	"github.com/fumiama/emozi"
)

func main() {
	dbpath := flag.String("db", emozi.EmoziDatabasePath, "符合规范的查询数据库位置, 不存在则会自动释放到该路径.")
	isencode := flag.String("e", "", "编码汉字序列为颜文字")
	isdecode := flag.String("d", "", "解码颜文字为汉字序列")
	getglobalid := flag.Bool("i", false, "指定汉字-a和带声调的拼音-p以计算其全局唯一ID")
	addoverlay := flag.String("a", "", "添加一个汉字到附加库")
	pinyinfull := flag.String("p", "", "带声调的拼音")
	radical := flag.String("r", "", "指定欲编辑的部首")
	radicalemozi := flag.String("re", "", "指定部首对应的颜文字")
	noRandom := flag.Bool("nr", false, "不随机选取所有读音相近的颜文字")
	showhelp := flag.Bool("h", false, "显示帮助信息")
	forcedecode := flag.Bool("f", false, "强制解码并非由本程序生成的颜文字序列")
	stabilize := flag.Int64("stabilize", 0, "固定附加库中的字到主库")
	deloverlay := flag.Int64("deloverlay", 0, "删除一个附加库中的字")
	delradical := flag.Bool("delradical", false, "删除-r指定的部首的记录")
	flag.Parse()
	defer fmt.Println("程序处理结束")
	if *showhelp {
		fmt.Println("用法: [-h|f|nr] [-db 字.db] [-d 🌹😺‎🐴‫👩] [-e 好] 形声字选择1 形声字选择2 ...") // nolint: go-staticcheck
		flag.PrintDefaults()
		return
	}
	emozi.EmoziDatabasePath = *dbpath
	coder, err := emozi.NewCoder(time.Minute)
	if err != nil {
		fmt.Println("ERROR: emozi.NewCoder:", err)
		return
	}
	defer coder.Close()
	if *isencode != "" {
		rem := flag.Args()
		lst := make([]int, len(rem))
		for i, ns := range rem {
			n, err := strconv.Atoi(ns)
			if err != nil {
				fmt.Println("ERROR: 第" + strconv.Itoa(i+1) + "个形声字参数 '" + ns + "' 非法")
				return
			}
			lst[i] = n
		}
		es, lst, err := coder.Encode(!*noRandom, *isencode, lst...)
		if err != nil {
			fmt.Println("ERROR: coder.Encode:", err)
			return
		}
		fmt.Println("编码结果:", string(es))
		if len(lst) > 0 && len(rem) == 0 {
			fmt.Println("可选形声:", lst)
			fmt.Println("在参数中指定形声字编号(从0开始)以生成不带中括号的编码结果")
		}
	}
	if *isdecode != "" {
		s, err := coder.Decode(emozi.EmoziString(*isdecode), *forcedecode)
		switch {
		case err == emozi.ErrInvalidEmoziString:
			fmt.Println("解码结果: 非本程序直接生成的颜文字序列或序列经过人为修改")
		case err != nil:
			fmt.Println("ERROR: coder.Decode:", err)
			return
		default:
			fmt.Println("解码结果:", s)
		}
	}
	if *addoverlay != "" && *pinyinfull != "" && *radical != "" {
		id, desc, err := coder.AddCharOverlay(*addoverlay, *radical, "", *pinyinfull)
		if err != nil {
			fmt.Println("ERROR: coder.AddCharOverlay:", err)
			return
		}
		fmt.Println("已添加汉字:", *addoverlay, "读音:", desc, "部首:", *radical, "ID:", id)
	}
	if *radical != "" && *radicalemozi != "" {
		rr := []rune(*radical)
		if len(rr) != 1 {
			fmt.Println("ERROR: 非法部首 '" + *radical + "': 长度为" + strconv.Itoa(len(rr)))
			return
		}
		err = coder.AddRadical(rr[0], *radicalemozi)
		if err != nil {
			fmt.Println("ERROR: coder.AddRadical:", err)
			return
		}
		fmt.Println("已添加部首:", *radical, "颜文字:", *radicalemozi)
	}
	if *getglobalid && *addoverlay != "" && *pinyinfull != "" {
		sm, ym, sd, err := emozi.SplitPinyin(*pinyinfull)
		if err != nil {
			fmt.Println("ERROR: coder.SplitPinyin:", err)
			return
		}
		r := []rune(*addoverlay)[0]
		id, _ := emozi.CharGlobalID(r, *pinyinfull)
		fmt.Println("文字:", string(r), "拼音IPA:", sm, ym, sd, "ID:", id)
	}
	if *stabilize != 0 {
		desc, err := coder.StabilizeCharFromOverlay(*stabilize)
		if err != nil {
			fmt.Println("ERROR: coder.StabilizeCharFromOverlay:", err)
			return
		}
		fmt.Println("固定到主库成功:", desc)
	}
	if *delradical && *radical != "" {
		rr := []rune(*radical)
		if len(rr) != 1 {
			fmt.Println("ERROR: 非法部首 '" + *radical + "': 长度为" + strconv.Itoa(len(rr)))
			return
		}
		err = coder.DelRadical(rr[0])
		if err != nil {
			fmt.Println("ERROR: coder.DelRadical:", err)
			return
		}
		fmt.Println("删除部首", string(rr[0]), "成功")
	}
	if *deloverlay != 0 {
		err = coder.DelCharOverlay(*deloverlay)
		if err != nil {
			fmt.Println("ERROR: coder.DelCharOverlay:", err)
			return
		}
		fmt.Println("已删除汉字ID:", *deloverlay)
	}
	if *addoverlay != "" && !*getglobalid {
		r := []rune(*addoverlay)
		if len(r) != 1 {
			fmt.Println("ERROR: 非法汉字参数-a:", *addoverlay)
			return
		}
		lst, err := coder.Lookup(r[0])
		if err != nil {
			fmt.Println("ERROR: coder.Lookup:", err)
			return
		}
		fmt.Println("查询到汉字", *addoverlay, "的记录:")
		for i, x := range lst {
			fmt.Printf("%d)\t%s\n", i, x)
		}
	}
}