icebergs/core/mall/asset.go

package mall

import (
	ice "github.com/shylinux/icebergs"
	"github.com/shylinux/icebergs/base/ctx"
	"github.com/shylinux/icebergs/base/mdb"
	"github.com/shylinux/icebergs/base/web"
	kit "github.com/shylinux/toolkits"
)

func _sub_key(m *ice.Message, account string) string {
	return kit.Keys(kit.MDB_HASH, kit.Hashs(account))
}
func _asset_list(m *ice.Message, account string, id string) {
	if account == "" {
		m.Option(mdb.FIELDS, "time,account,amount,count")
	} else {
		m.Option(mdb.FIELDS, kit.Select("time,id,type,amount,name,text", mdb.DETAIL, id != ""))
	}
	m.Cmdy(mdb.SELECT, ASSET, "", mdb.ZONE, account, id)
	if id != "" {
		m.PushAction(mdb.PLUGIN)
	}
}

func _asset_create(m *ice.Message, account string) {
	m.Cmdy(mdb.INSERT, ASSET, "", mdb.HASH, ACCOUNT, account)
}
func _asset_insert(m *ice.Message, account string, arg ...string) {
	msg := m.Cmd(mdb.SELECT, ASSET, "", mdb.HASH, ACCOUNT, account)
	amount := kit.Int(msg.Append(AMOUNT))
	for i := 0; i < len(arg)-1; i += 2 {
		if arg[i] == "amount" {
			amount += kit.Int(arg[i+1])
		}
	}
	m.Cmdy(mdb.INSERT, ASSET, _sub_key(m, account), mdb.LIST, arg)
	m.Cmdy(mdb.MODIFY, ASSET, "", mdb.HASH, ACCOUNT, account, AMOUNT, amount)
}
func _asset_modify(m *ice.Message, account, id, field, value string, arg ...string) {
	m.Cmdy(mdb.MODIFY, ASSET, _sub_key(m, account), mdb.LIST, kit.MDB_ID, id, field, value, arg)
}
func _asset_export(m *ice.Message, file string) {
	m.Option(mdb.FIELDS, "account,id,time,type,name,text,amount,extra")
	m.Cmdy(mdb.EXPORT, ASSET, "", mdb.ZONE, file)
}
func _asset_import(m *ice.Message, file string) {
	m.Option(mdb.FIELDS, "account")
	m.Cmdy(mdb.IMPORT, ASSET, "", mdb.ZONE, file)
}
func _asset_inputs(m *ice.Message, field, value string) {
	switch field {
	case "from", "to", ACCOUNT:
		m.Cmdy(mdb.INPUTS, ASSET, "", mdb.HASH, ACCOUNT, value)
	default:
		m.Cmdy(mdb.INPUTS, ASSET, _sub_key(m, m.Option(kit.MDB_ZONE)), mdb.LIST, field, value)
	}
}

const (
	ACCOUNT = "account"
	AMOUNT  = "amount"

	SPEND = "spend"
	TRANC = "tranc"
	BONUS = "bonus"
)
const ASSET = "asset"

func init() {
	Index.Merge(&ice.Context{
		Configs: map[string]*ice.Config{
			ASSET: {Name: ASSET, Help: "资产", Value: kit.Data(kit.MDB_SHORT, ACCOUNT)},
		},
		Commands: map[string]*ice.Command{
			ASSET: {Name: "asset account id auto spend tranc bonus export import", Help: "资产", Action: map[string]*ice.Action{
				SPEND: {Name: "spend account amount time@date name text", Help: "消费", Hand: func(m *ice.Message, arg ...string) {
					_asset_create(m, arg[1])
					if amount := kit.Int(arg[3]); amount > 0 {
						arg[3] = kit.Format(-amount)
					}
					_asset_insert(m, arg[1], kit.Simple(kit.MDB_TYPE, "支出", arg[2:])...)
				}},
				TRANC: {Name: "tranc from to amount time@date text", Help: "转账", Hand: func(m *ice.Message, arg ...string) {
					_asset_create(m, arg[3])
					_asset_insert(m, arg[3], kit.Simple(kit.MDB_TYPE, "转入", kit.MDB_NAME, arg[1], arg[4:])...)

					_asset_create(m, arg[1])
					if amount := kit.Int(arg[5]); amount > 0 {
						arg[5] = kit.Format(-amount)
					}
					_asset_insert(m, arg[1], kit.Simple(kit.MDB_TYPE, "转出", kit.MDB_NAME, arg[3], arg[4:])...)

				}},
				BONUS: {Name: "bonus account amount time@date name text", Help: "收入", Hand: func(m *ice.Message, arg ...string) {
					_asset_create(m, arg[1])
					_asset_insert(m, arg[1], kit.Simple(kit.MDB_TYPE, "收入", arg[2:])...)
				}},

				mdb.MODIFY: {Name: "modify", Help: "编辑", Hand: func(m *ice.Message, arg ...string) {
					_asset_modify(m, m.Option(ACCOUNT), m.Option(kit.MDB_ID), arg[0], arg[1])
				}},
				mdb.EXPORT: {Name: "export file", Help: "导出", Hand: func(m *ice.Message, arg ...string) {
					_asset_export(m, m.Option(kit.MDB_FILE))
				}},
				mdb.IMPORT: {Name: "import file", Help: "导入", Hand: func(m *ice.Message, arg ...string) {
					_asset_import(m, m.Option(kit.MDB_FILE))
				}},
				mdb.INPUTS: {Name: "inputs", Help: "补全", Hand: func(m *ice.Message, arg ...string) {
					switch arg[0] {
					case "pod", "extra.pod":
						m.Cmdy(web.ROUTE)
					case "ctx", "extra.ctx":
						m.Cmdy(ctx.CONTEXT)
					case "cmd", "extra.cmd":
						m.Cmdy(ctx.CONTEXT, kit.Select(m.Option("ctx"), m.Option("extra.ctx")), ctx.COMMAND)
					case "arg":

					default:
						_asset_inputs(m, kit.Select("", arg, 0), kit.Select("", arg, 1))
					}
				}},

				mdb.PLUGIN: {Name: "plugin extra.pod extra.ctx extra.cmd extra.arg", Help: "插件", Hand: func(m *ice.Message, arg ...string) {
					_asset_modify(m, m.Option(ACCOUNT), m.Option(kit.MDB_ID), kit.MDB_TIME, m.Time(), kit.Simple(kit.Dict(arg))...)
				}},
				ctx.COMMAND: {Name: "command", Help: "命令", Hand: func(m *ice.Message, arg ...string) {
					if arg[0] == "run" {
						m.Cmdy(arg[1], arg[2:])
						return
					}
					m.Cmdy(ctx.COMMAND, arg[0])
				}},
			}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
				_asset_list(m, kit.Select("", arg, 0), kit.Select("", arg, 1))
			}},
		},
	}, nil)
}