chirpnest/sql.go

package main

import (
	"log"
	"strconv"
)

func initDB() {
	log.Println("Setting up database...")

	tx, err := db.Begin()
	if err != nil {
		log.Fatal(err)
	}
	defer tx.Rollback()

	_, err = tx.Exec("set foreign_key_checks = 0")
	if err != nil {
		log.Fatal(err)
	}

	var name string
	rows, err := db.Query("show tables")
	if err != nil {
		log.Fatal(err)
	}
	for rows.Next() {
		err = rows.Scan(&name)
		if err != nil {
			log.Fatal(err)
		}
		_, err = tx.Exec("drop table " + name)
		if err != nil {
			log.Fatal(err)
		}
	}
	err = rows.Err()
	if err != nil {
		log.Fatal(err)
	}
	rows.Close()

	_, err = tx.Exec("set foreign_key_checks = 1")
	if err != nil {
		log.Fatal(err)
	}

	err = tx.Commit()
	if err != nil {
		log.Fatal(err)
	}

	log.Println("Database cleaned up")

	_, err = db.Exec(`CREATE TABLE code_obj_type (
							 id			SMALLINT(5) UNSIGNED NOT NULL
							,intl_id	VARCHAR(32) NOT NULL
							,name		VARCHAR(80) NOT NULL
							,PRIMARY KEY (id)
						  ) ENGINE = InnoDB CHARSET=utf8 COLLATE utf8_unicode_ci;`)
	if err != nil {
		log.Fatal(err)
	}

	_, err = db.Exec(`CREATE TABLE code_obj_sub_type (
							 id 			SMALLINT(5) UNSIGNED NOT NULL
							,intl_id 	VARCHAR(32) NOT NULL
							,name 		VARCHAR(80) NOT NULL
							,obj_type_id SMALLINT UNSIGNED NOT NULL
							,PRIMARY KEY (id)
							,FOREIGN KEY (obj_type_id) REFERENCES code_obj_type(id) ON DELETE CASCADE
						  ) ENGINE = InnoDB CHARSET=utf8 COLLATE utf8_unicode_ci;`)
	if err != nil {
		log.Fatal(err)
	}

	_, err = db.Exec(`CREATE TABLE obj (
							 id				BIGINT UNSIGNED NOT NULL AUTO_INCREMENT
							,obj_type_id	SMALLINT UNSIGNED NOT NULL
							,PRIMARY KEY (id)
						  ) ENGINE = InnoDB AUTO_INCREMENT = 0 CHARSET=utf8 COLLATE utf8_unicode_ci;`)
	if err != nil {
		log.Fatal(err)
	}

	_, err = db.Exec(`CREATE TABLE obj_user (
							 obj_id			BIGINT UNSIGNED NOT NULL
							,tg_id          BIGINT UNSIGNED NOT NULL
							,user_id		VARCHAR(32) NOT NULL
							,name			VARCHAR(80) NOT NULL
							,guild_id		BIGINT UNSIGNED
							,last_msg		TIMESTAMP
							,busy_until		TIMESTAMP
							,role			ENUM('commander', 'bartender', 'squire', 'none')
							,FOREIGN KEY (obj_id) REFERENCES obj(id) ON DELETE CASCADE
						  ) ENGINE = InnoDB CHARSET=utf8 COLLATE utf8_unicode_ci;`)
	if err != nil {
		log.Fatal(err)
	}

	_, err = db.Exec(`CREATE TABLE obj_guild (
							 obj_id				BIGINT UNSIGNED NOT NULL AUTO_INCREMENT
							,tag				VARCHAR(32) NOT NULL
							,name				VARCHAR(80) NOT NULL
							,chat_id			BIGINT NOT NULL
							,deposit_chat_id	BIGINT NOT NULL
							,FOREIGN KEY (obj_id) REFERENCES obj(id) ON DELETE CASCADE
						  ) ENGINE = InnoDB CHARSET=utf8 COLLATE utf8_unicode_ci;`)
	if err != nil {
		log.Fatal(err)
	}

	_, err = db.Exec(`CREATE TABLE obj_msg (
							 obj_id         BIGINT UNSIGNED NOT NULL
							,msg_id			BIGINT NOT NULL
							,chat_id        BIGINT NOT NULL
							,sender_user_id BIGINT NOT NULL
							,date           TIMESTAMP NOT NULL
							,text           VARCHAR(4096) NOT NULL
							,FOREIGN KEY (obj_id) REFERENCES obj(id) ON DELETE CASCADE
							,UNIQUE KEY (msg_id, chat_id, sender_user_id)
						  ) ENGINE = InnoDB CHARSET=utf8 COLLATE utf8_unicode_ci;`)
	if err != nil {
		log.Fatal(err)
	}

	_, err = db.Exec(`CREATE TABLE obj_msg_report (
							 obj_id         BIGINT UNSIGNED NOT NULL
							,war_date       TIMESTAMP NOT NULL
							,attack         SMALLINT UNSIGNED NOT NULL
							,defense        SMALLINT UNSIGNED NOT NULL
							,gold           SMALLINT UNSIGNED NOT NULL
							,stock			SMALLINT UNSIGNED NOT NULL
							,exp			SMALLINT UNSIGNED NOT NULL
							,stamina		BOOLEAN NOT NULL
							,crit			BOOLEAN NOT NULL
							,FOREIGN KEY (obj_id) REFERENCES obj(id) ON DELETE CASCADE
							,UNIQUE KEY (msg_id, chat_id, sender_user_id)
						  ) ENGINE = InnoDB CHARSET=utf8 COLLATE utf8_unicode_ci;`)
	if err != nil {
		log.Fatal(err)
	}

	_, err = db.Exec(`INSERT INTO code_obj_type (id, intl_id, name)
						  VALUES (1, "user", "User")
						        ,(2, "guild", "Guild")
						        ,(3, "msg", "Message");`)
	if err != nil {
		log.Fatal(err)
	}

	_, err = db.Exec(`INSERT INTO code_obj_sub_type (id, intl_id, name, obj_type_id)
						  VALUES (1, "unprocessed", "Unprocessed", 3)
						        ,(2, "unknown", "Unknown", 3)
						        ,(3, "report", "War report", 3)
						        ,(4, "g_report", "Guild War report", 3)
						        ,(5, "quest", "Quest result", 3)
						        ,(6, "fight", "Fight result", 3)
						        ,(7, "hero", "Hero summary", 3)
						        ,(8, "me", "Hero short summary", 3)
						        ,(9, "inv", "Inventory", 3);`)
	if err != nil {
		log.Fatal(err)
	}

	log.Println("Database set up")
}

func putUnprocessedMsg(m ChatWarsMessage) (int64, error) {
	res, err := db.Exec(`INSERT INTO obj (obj_type_id, obj_sub_type_id)
						          VALUES (3          , 1);`)
	if err != nil {
		return nil, err
	}

	objId, err := res.LastInsertId()
	if err != nil {
		return nil, err
	}

	stmt, err := db.Prepare(`INSERT INTO obj_msg	(obj_id, msg_id, chat_id, sender_user_id, date            , text)
							 VALUES 				(?,      ?,      ?,       ?,              FROM_UNIXTIME(?), ?);`)
	if err != nil {
		return nil, err
	}
	defer stmt.Close()

	_, err = stmt.Exec(objId, m.ID64, m.ChatID64, m.SenderUserID64, m.Date, m.Text)
	if err != nil {
		return nil, err
	}

	return objId, nil
}

func getMsg(objId int64) (ChatWarsMessage, error) {
	var m ChatWarsMessage

	stmt, err := db.Prepare(`SELECT om.msg_id, om.chat_id, om.sender_user_id, UNIX_TIMESTAMP(om.date), om.text FROM obj_msg om WHERE om.id = ?`)
	if err != nil {
		return nil, err
	}
	defer stmt.Close()

	err = stmtOut.QueryRow(objId).Scan(&m.ID64, &m.ChatID64, &m.SenderUserID64, &m.Date, &m.Text)
	if err != nil {
		return nil, err
	}

	return m, nil
}