chirpnest/obj.go

625 lines
15 KiB
Go
Raw Normal View History

2019-05-25 09:25:11 +02:00
package main
2019-05-25 09:37:45 +02:00
import (
2019-06-02 09:00:16 +02:00
"fmt"
2019-06-03 10:31:54 +02:00
"regexp"
2019-05-25 09:37:45 +02:00
"strconv"
2019-06-06 09:54:29 +02:00
"strings"
2019-05-31 05:15:19 +02:00
"sync"
2019-05-25 09:37:45 +02:00
"time"
)
2019-05-25 09:25:11 +02:00
var (
2019-06-09 14:17:05 +02:00
cacheObjCastle *sync.Map
cacheObjGuild *sync.Map
cacheObjUser *sync.Map
cacheObjItem *sync.Map
2019-06-09 14:17:59 +02:00
cacheObjMsg *sync.Map
2019-05-25 09:25:11 +02:00
)
func getObjTypeId(objId int64) (int64, error) {
var objTypeId int64
stmt, err := db.Prepare(`SELECT o.obj_type_id FROM obj o WHERE o.id = ?`)
if err != nil {
return 0, err
}
defer stmt.Close()
err = stmt.QueryRow(objId).Scan(&objTypeId)
if err != nil {
return 0, err
}
return objTypeId, nil
}
func getObjSubTypeId(objId int64) (int64, error) {
var objSubTypeId int64
stmt, err := db.Prepare(`SELECT o.obj_sub_type_id FROM obj o WHERE o.id = ?`)
if err != nil {
return 0, err
}
defer stmt.Close()
err = stmt.QueryRow(objId).Scan(&objSubTypeId)
if err != nil {
return 0, err
}
return objSubTypeId, nil
}
func setObjSubTypeId(objId int64, objSubTypeID64 int64) error {
stmt, err := db.Prepare(`UPDATE obj o SET o.obj_sub_type_id = ? WHERE o.id = ?;`)
logOnError(err, "setObjSubTypeId : prepare update")
if err != nil {
return err
}
defer stmt.Close()
_, err = stmt.Exec(objSubTypeID64, objId)
logOnError(err, "setObjSubTypeId : exec update")
return err
}
func addObjMsg(msgID64 int64, msgChatID64 int64, msgUserID64 int64, msgSenderUserID64 int64, msgDate time.Time, msgText string) (int64, error) {
tx, err := db.Begin()
logOnError(err, "addObjMsg : start transaction")
if err != nil {
return 0, err
}
res, err := tx.Exec(`INSERT INTO obj (obj_type_id, obj_sub_type_id)
VALUES (` + strconv.Itoa(objTypeMessage) + `,` + strconv.Itoa(objSubTypeMessageUnknown) + `);`)
logOnError(err, "addObjMsg : exec insert obj")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjMsg : rollback insert obj")
return 0, err
}
objId, err := res.LastInsertId()
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjMsg : rollback get lastInsertId")
return 0, err
}
stmt, err := tx.Prepare(`INSERT INTO obj_msg (obj_id, msg_id, chat_id, user_id, sender_user_id, date, text)
VALUES (?, ?, ?, ?, ?, ?, ?);`)
logOnError(err, "addObjMsg : prepare insert obj_msg")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjMsg : rollback prepare insert obj_msg")
return 0, err
}
defer stmt.Close()
_, err = stmt.Exec(objId, msgID64, msgChatID64, msgUserID64, msgSenderUserID64, msgDate, msgText)
logOnError(err, "addObjMsg : exec insert obj_msg")
if err != nil {
err2 := tx.Rollback()
2019-05-31 04:02:21 +02:00
logOnError(err2, "addObjMsg : rollback exec insert obj_msg")
2019-05-25 09:25:11 +02:00
return 0, err
}
err = tx.Commit()
logOnError(err, "addObjMsg : commit")
if err != nil {
return 0, err
}
return objId, nil
}
2019-06-09 14:17:05 +02:00
func getObjMsg(objId int64) (*ChatWarsMessage, error) {
var m *ChatWarsMessage
if v, ok := cacheObjMsg.Load(objId); ok {
m := v.(ChatWarsMessage)
return m, nil
}
stmt, err := db.Prepare(`SELECT om.msg_id, om.chat_id, om.user_id, om.sender_user_id, om.date, om.text FROM obj_msg om WHERE om.obj_id = ?`)
if err != nil {
return m, err
}
defer stmt.Close()
m = new(ChatWarsMessage)
err = stmt.QueryRow(objId).Scan(&m.ID64, &m.ChatID64, &m.UserID64, &m.SenderUserID64, &m.Date, &m.Text)
if err != nil {
return m, err
}
cacheObjMsg.Store(objId, *m)
return m, nil
}
func loadObjMsg() error {
cacheObjMsg = new(sync.Map)
return nil
}
2019-05-25 09:29:24 +02:00
func addObjCastle(logo string, name string) (int64, error) {
2019-05-25 09:25:11 +02:00
tx, err := db.Begin()
logOnError(err, "addObjCastle : start transaction")
if err != nil {
return 0, err
}
res, err := tx.Exec(`INSERT INTO obj (obj_type_id, obj_sub_type_id)
VALUES (` + strconv.Itoa(objTypeCastle) + `,` + strconv.Itoa(objSubTypeCastle) + `);`)
logOnError(err, "addObjCastle : exec insert obj")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjCastle : rollback insert obj")
return 0, err
}
objId, err := res.LastInsertId()
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjCastle : rollback get lastInsertId")
return 0, err
}
stmt, err := tx.Prepare(`INSERT INTO obj_castle (obj_id, logo, name)
VALUES (?, ?, ?);`)
logOnError(err, "addObjCastle : prepare insert obj_castle")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjCastle : rollback prepare insert obj_castle")
return 0, err
}
defer stmt.Close()
_, err = stmt.Exec(objId, logo, name)
logOnError(err, "addObjCastle : exec insert obj_castle")
if err != nil {
err2 := tx.Rollback()
2019-05-31 04:02:21 +02:00
logOnError(err2, "addObjCastle : rollback exec insert obj_castle")
2019-05-25 09:25:11 +02:00
return 0, err
}
err = tx.Commit()
logOnError(err, "addObjCastle : commit")
if err != nil {
return 0, err
}
return objId, nil
}
2019-05-25 09:48:34 +02:00
func getObjCastleID(s string) int64 {
2019-05-31 05:27:44 +02:00
v, _ := cacheObjCastle.Load(s)
2019-05-31 05:26:47 +02:00
c := v.(ChatWarsCastle)
2019-05-31 05:15:19 +02:00
return c.ObjID64
2019-05-25 09:25:11 +02:00
}
func loadObjCastle() error {
var (
id int64
logo string
name string
)
2019-05-31 05:27:44 +02:00
cacheObjCastle = new(sync.Map)
2019-05-25 09:25:11 +02:00
2019-05-25 10:06:33 +02:00
castles, err := db.Query(`SELECT oc.obj_id, oc.logo, oc.name FROM obj_castle oc;`)
2019-05-25 09:25:11 +02:00
if err != nil {
return err
}
defer castles.Close()
for castles.Next() {
2019-05-25 09:37:45 +02:00
err = castles.Scan(&id, &logo, &name)
2019-05-25 09:25:11 +02:00
if err != nil {
return err
}
c := new(ChatWarsCastle)
c.ObjID64 = id
c.Logo = logo
c.Name = name
2019-05-31 05:15:19 +02:00
cacheObjCastle.Store(logo, *c)
cacheObjCastle.Store(name, *c)
2019-05-25 09:25:11 +02:00
}
return nil
}
2019-05-25 10:56:13 +02:00
func addObjGuild(tag string, name string) (int64, error) {
tx, err := db.Begin()
logOnError(err, "addObjGuild : start transaction")
if err != nil {
return 0, err
}
res, err := tx.Exec(`INSERT INTO obj (obj_type_id, obj_sub_type_id)
VALUES (` + strconv.Itoa(objTypeGuild) + `,` + strconv.Itoa(objSubTypeGuild) + `);`)
logOnError(err, "addObjGuild : exec insert obj")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjGuild : rollback insert obj")
return 0, err
}
objId, err := res.LastInsertId()
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjGuild : rollback get lastInsertId")
return 0, err
}
stmt, err := tx.Prepare(`INSERT INTO obj_guild (obj_id, tag, name, chat_id, deposit_chat_id)
VALUES (?, ?, ?, NULL, NULL);`)
logOnError(err, "addObjGuild : prepare insert obj_guild")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjGuild : rollback prepare insert obj_guild")
return 0, err
}
defer stmt.Close()
_, err = stmt.Exec(objId, tag, name)
logOnError(err, "addObjGuild : exec insert obj_guild")
if err != nil {
err2 := tx.Rollback()
2019-05-31 04:02:21 +02:00
logOnError(err2, "addObjGuild : rollback exec insert obj_guild")
2019-05-25 10:56:13 +02:00
return 0, err
}
err = tx.Commit()
logOnError(err, "addObjGuild : commit")
if err != nil {
return 0, err
}
return objId, nil
}
func getObjGuildID(s string) int64 {
2019-05-31 05:26:47 +02:00
if v, ok := cacheObjGuild.Load(s); ok {
g := v.(ChatWarsGuild)
2019-05-25 10:56:13 +02:00
return g.ObjID64
} else {
objID64, err := addObjGuild(s, ``)
logOnError(err, "getObjGuildID")
2019-05-31 05:26:47 +02:00
g := new(ChatWarsGuild)
g.ObjID64 = objID64
g.Tag = s
g.Name = ``
cacheObjGuild.Store(s, *g)
2019-05-31 05:15:19 +02:00
return objID64
2019-05-25 10:56:13 +02:00
}
}
func loadObjGuild() error {
var (
id int64
tag string
name string
)
2019-05-31 05:27:44 +02:00
cacheObjGuild = new(sync.Map)
2019-05-25 10:56:13 +02:00
guilds, err := db.Query(`SELECT og.obj_id, og.tag, og.name FROM obj_guild og;`)
if err != nil {
return err
}
defer guilds.Close()
for guilds.Next() {
err = guilds.Scan(&id, &tag, &name)
if err != nil {
return err
}
g := new(ChatWarsGuild)
g.ObjID64 = id
g.Tag = tag
g.Name = name
2019-05-31 05:15:19 +02:00
cacheObjGuild.Store(tag, *g)
2019-05-25 10:56:13 +02:00
}
return nil
}
func addObjUser(name string) (int64, error) {
tx, err := db.Begin()
logOnError(err, "addObjUser : start transaction")
if err != nil {
return 0, err
}
res, err := tx.Exec(`INSERT INTO obj (obj_type_id, obj_sub_type_id)
VALUES (` + strconv.Itoa(objTypeUser) + `,` + strconv.Itoa(objSubTypeUser) + `);`)
logOnError(err, "addObjUser : exec insert obj")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjUser : rollback insert obj")
return 0, err
}
objId, err := res.LastInsertId()
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjUser : rollback get lastInsertId")
return 0, err
}
stmt, err := tx.Prepare(`INSERT INTO obj_user (obj_id, name)
VALUES (?, ?);`)
logOnError(err, "addObjUser : prepare insert obj_user")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjUser : rollback prepare insert obj_user")
return 0, err
}
defer stmt.Close()
_, err = stmt.Exec(objId, name)
logOnError(err, "addObjUser : exec insert obj_user")
if err != nil {
err2 := tx.Rollback()
2019-05-31 04:02:21 +02:00
logOnError(err2, "addObjUser : rollback exec insert obj_user")
2019-05-25 10:56:13 +02:00
return 0, err
}
err = tx.Commit()
logOnError(err, "addObjUser : commit")
if err != nil {
return 0, err
}
return objId, nil
}
func getObjUserID(s string) int64 {
2019-05-31 05:26:47 +02:00
if v, ok := cacheObjUser.Load(s); ok {
u := v.(ChatWarsUser)
2019-05-31 05:28:16 +02:00
return u.ObjID64
2019-05-25 10:56:13 +02:00
} else {
2019-05-25 11:02:24 +02:00
objID64, err := addObjUser(s)
2019-05-25 10:56:13 +02:00
logOnError(err, "getObjUserID")
2019-05-31 05:26:47 +02:00
u := new(ChatWarsUser)
u.ObjID64 = objID64
u.Name = s
cacheObjUser.Store(s, *u)
2019-05-31 05:15:19 +02:00
return objID64
2019-05-25 10:56:13 +02:00
}
}
func loadObjUser() error {
var (
id int64
name string
)
2019-05-31 05:27:44 +02:00
cacheObjUser = new(sync.Map)
2019-05-25 10:56:13 +02:00
2019-05-25 10:59:14 +02:00
users, err := db.Query(`SELECT ou.obj_id, ou.name FROM obj_user ou;`)
2019-05-25 10:56:13 +02:00
if err != nil {
return err
}
defer users.Close()
for users.Next() {
2019-05-25 10:59:14 +02:00
err = users.Scan(&id, &name)
2019-05-25 10:56:13 +02:00
if err != nil {
return err
}
2019-05-25 10:59:14 +02:00
u := new(ChatWarsUser)
u.ObjID64 = id
u.Name = name
2019-05-31 05:15:19 +02:00
cacheObjUser.Store(name, *u)
2019-05-25 10:56:13 +02:00
}
return nil
}
2019-05-30 07:49:11 +02:00
func getObjMsgDate(objID64 int64) (time.Time, error) {
2019-06-09 14:17:59 +02:00
m, err := getObjMsg(objID64)
2019-05-30 07:49:11 +02:00
if err != nil {
2019-05-30 08:08:39 +02:00
return time.Now(), err
} else {
return m.Date, nil
2019-05-30 07:49:11 +02:00
}
}
2019-05-31 04:02:21 +02:00
2019-05-31 04:05:58 +02:00
func addObjXP(userID64 int64, expNow int64, expLvl int64, level int64, date time.Time) (int64, error) {
2019-05-31 04:02:21 +02:00
tx, err := db.Begin()
logOnError(err, "addObjXP : start transaction")
if err != nil {
return 0, err
}
res, err := tx.Exec(`INSERT INTO obj (obj_type_id, obj_sub_type_id)
VALUES (` + strconv.Itoa(objTypeExperience) + `,` + strconv.Itoa(objSubTypeExperience) + `);`)
logOnError(err, "addObjXP : exec insert obj")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjXP : rollback insert obj")
return 0, err
}
objId, err := res.LastInsertId()
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjXP : rollback get lastInsertId")
return 0, err
}
stmt, err := tx.Prepare(`INSERT INTO obj_xp (obj_id, user_id, val, target, level, date)
VALUES (?, ?, ?, ?, ?, ?);`)
logOnError(err, "addObjXP : prepare insert obj_xp")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjXP : rollback prepare insert obj_xp")
return 0, err
}
defer stmt.Close()
2019-05-31 04:05:58 +02:00
_, err = stmt.Exec(objId, userID64, expNow, expLvl, level, date)
2019-05-31 04:02:21 +02:00
logOnError(err, "addObjXP : exec insert obj_xp")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjXP : rollback exec insert obj_xp")
return 0, err
}
err = tx.Commit()
logOnError(err, "addObjXP : commit")
if err != nil {
return 0, err
}
return objId, nil
}
2019-06-02 08:48:16 +02:00
2019-06-02 15:23:10 +02:00
func addObjItem(code string, name string, itemTypeID64 int64, weight int) (int64, error) {
2019-06-02 08:48:16 +02:00
tx, err := db.Begin()
logOnError(err, "addObjItem : start transaction")
if err != nil {
return 0, err
}
res, err := tx.Exec(`INSERT INTO obj (obj_type_id, obj_sub_type_id)
2019-06-02 09:01:04 +02:00
VALUES (` + strconv.Itoa(objTypeItem) + `,` + fmt.Sprintf("%d", itemTypeID64) + `);`)
2019-06-02 08:48:16 +02:00
logOnError(err, "addObjItem : exec insert obj")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjItem : rollback insert obj")
return 0, err
}
objId, err := res.LastInsertId()
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjItem : rollback get lastInsertId")
return 0, err
}
stmt, err := tx.Prepare(`INSERT INTO obj_item (obj_id, intl_id, name, weight)
VALUES (?, ?, ?, ?);`)
logOnError(err, "addObjItem : prepare insert obj_item")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjItem : rollback prepare insert obj_item")
return 0, err
}
defer stmt.Close()
_, err = stmt.Exec(objId, code, name, weight)
logOnError(err, "addObjItem : exec insert obj_item")
if err != nil {
err2 := tx.Rollback()
logOnError(err2, "addObjItem : rollback exec insert obj_item")
return 0, err
}
err = tx.Commit()
logOnError(err, "addObjItem : commit")
if err != nil {
return 0, err
}
2019-06-02 15:30:18 +02:00
c := new(ChatWarsItem)
c.ObjID64 = objId
c.Code = code
2019-06-06 10:07:38 +02:00
c.Name = name
2019-06-02 15:30:18 +02:00
c.Weight = weight
2019-06-02 15:31:13 +02:00
cacheObjItem.Store(code, *c)
2019-06-06 10:07:38 +02:00
cacheObjItem.Store(strings.ToUpper(name), *c)
2019-06-02 15:30:18 +02:00
2019-06-02 08:48:16 +02:00
return objId, nil
}
2019-06-02 15:23:10 +02:00
func getObjItemID(c string, n string) int64 {
2019-06-03 03:40:37 +02:00
i := silentGetObjItemID(c, n)
if i == 0 {
w := TGCommand{
Type: commandSendMsg,
Text: fmt.Sprintf("Object unknown : %s - %s\n", c, n),
ToUserID64: cfg.Bot.Admin,
}
TGCmdQueue <- w
}
return i
}
2019-06-06 09:54:29 +02:00
func silentGetObjItemID(code string, name string) int64 {
2019-06-06 09:55:00 +02:00
c := strings.ToLower(code)
n := strings.ToUpper(name)
2019-06-02 15:28:28 +02:00
if v, ok := cacheObjItem.Load(c); ok {
2019-06-02 12:48:20 +02:00
i := v.(ChatWarsItem)
2019-06-02 08:48:16 +02:00
return i.ObjID64
}
2019-06-03 10:31:28 +02:00
if v, ok := cacheObjItem.Load(n); ok {
i := v.(ChatWarsItem)
return i.ObjID64
}
2019-06-07 12:54:26 +02:00
if ok, _ := regexp.MatchString(`(u|a|e)[0-9]+[a-z]{0,1}`, c); ok {
2019-06-07 10:00:20 +02:00
r := regexp.MustCompile(`^((?P<Modifier>⚡\+[0-9]+) ){0,1}(?P<BaseName>.+?)( \+(?P<Atk>[0-9]+)⚔){0,1}( \+(?P<Def>[0-9]+)🛡){0,1}( \+(?P<Mana>[0-9]+)💧){0,1}$`)
2019-06-03 10:31:28 +02:00
n2 := r.ReplaceAllString(n, "${BaseName}")
if v, ok := cacheObjItem.Load(n2); ok {
i := v.(ChatWarsItem)
return i.ObjID64
}
2019-06-06 13:38:00 +02:00
fmt.Printf("silentGetObjItemID(unique) : Modifier : `%s`\n", r.ReplaceAllString(n, "${Modifier}"))
fmt.Printf("silentGetObjItemID(unique) : BaseName : `%s`\n", r.ReplaceAllString(n, "${BaseName}"))
fmt.Printf("silentGetObjItemID(unique) : Atk : `%s`\n", r.ReplaceAllString(n, "${Atk}"))
fmt.Printf("silentGetObjItemID(unique) : Def : `%s`\n", r.ReplaceAllString(n, "${Def}"))
fmt.Printf("silentGetObjItemID(unique) : Mana : `%s`\n", r.ReplaceAllString(n, "${Mana}"))
2019-06-03 10:31:28 +02:00
}
if len(c) == 0 {
2019-06-07 10:00:20 +02:00
r := regexp.MustCompile(`^((?P<Modifier>⚡\+[0-9]+) ){0,1}(?P<BaseName>.+?)( \+(?P<Atk>[0-9]+)⚔){0,1}( \+(?P<Def>[0-9]+)🛡){0,1}( \+(?P<Mana>[0-9]+)💧){0,1}$`)
2019-06-03 10:31:28 +02:00
n2 := r.ReplaceAllString(n, "${BaseName}")
if v, ok := cacheObjItem.Load(n2); ok {
i := v.(ChatWarsItem)
return i.ObjID64
}
2019-06-06 13:38:00 +02:00
fmt.Printf("silentGetObjItemID(null) : Modifier : `%s`\n", r.ReplaceAllString(n, "${Modifier}"))
fmt.Printf("silentGetObjItemID(null) : BaseName : `%s`\n", r.ReplaceAllString(n, "${BaseName}"))
fmt.Printf("silentGetObjItemID(null) : Atk : `%s`\n", r.ReplaceAllString(n, "${Atk}"))
fmt.Printf("silentGetObjItemID(null) : Def : `%s`\n", r.ReplaceAllString(n, "${Def}"))
fmt.Printf("silentGetObjItemID(null) : Mana : `%s`\n", r.ReplaceAllString(n, "${Mana}"))
2019-06-03 10:31:28 +02:00
}
return 0
2019-06-02 08:48:16 +02:00
}
func loadObjItem() error {
var (
id int64
intl_id string
name string
weight int
)
cacheObjItem = new(sync.Map)
items, err := db.Query(`SELECT oi.obj_id, oi.intl_id, oi.name, oi.weight FROM obj_item oi;`)
if err != nil {
return err
}
defer items.Close()
for items.Next() {
err = items.Scan(&id, &intl_id, &name, &weight)
if err != nil {
return err
}
c := new(ChatWarsItem)
c.ObjID64 = id
c.Code = intl_id
2019-06-06 10:02:41 +02:00
c.Name = name
2019-06-02 08:48:16 +02:00
c.Weight = weight
cacheObjItem.Store(intl_id, *c)
2019-06-06 10:02:41 +02:00
cacheObjItem.Store(strings.ToUpper(name), *c)
2019-06-02 08:48:16 +02:00
}
return nil
}