74 lines
2.3 KiB
Go
74 lines
2.3 KiB
Go
// i2_bot – Instinct PokememBro Bot
|
||
// Copyright (c) 2017 Vladimir "fat0troll" Hodakov
|
||
|
||
package migrations
|
||
|
||
import (
|
||
"database/sql"
|
||
)
|
||
|
||
// CreateWeaponsAndAddWealthUp creates `weapons` table and adds `wealth` column to `profiles`
|
||
func CreateWeaponsAndAddWealthUp(tx *sql.Tx) error {
|
||
request := "CREATE TABLE `weapons` ("
|
||
request += "`id` int(11) NOT NULL AUTO_INCREMENT COMMENT 'ID оружия',"
|
||
request += "`name` varchar(191) NOT NULL COMMENT 'Название оружия',"
|
||
request += "`power` int(11) NOT NULL COMMENT 'Атака оружия',"
|
||
request += "`price` int(11) NOT NULL COMMENT 'Цена в магазине',"
|
||
request += "`created_at` datetime NOT NULL COMMENT 'Добавлено в базу',"
|
||
request += "PRIMARY KEY (`id`),"
|
||
request += "UNIQUE KEY `id` (`id`),"
|
||
request += "KEY `weapons_created_at` (`created_at`)"
|
||
request += ") ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4 COMMENT='Оружие';"
|
||
_, err := tx.Exec(request)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
_, err = tx.Exec("INSERT INTO `weapons` VALUES(NULL, 'Бита', 2, 5, NOW());")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
_, err = tx.Exec("INSERT INTO `weapons` VALUES(NULL, 'Стальная бита', 10, 40, NOW());")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
_, err = tx.Exec("INSERT INTO `weapons` VALUES(NULL, 'Чугунная бита ', 200, 500, NOW());")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
_, err = tx.Exec("INSERT INTO `weapons` VALUES(NULL, 'Титановая бита', 2000, 10000, NOW());")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
_, err = tx.Exec("INSERT INTO `weapons` VALUES(NULL, 'Алмазная бита', 10000, 100000, NOW());")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
_, err = tx.Exec("INSERT INTO `weapons` VALUES(NULL, 'Криптонитовая бита', 100000, 500000, NOW());")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
_, err = tx.Exec("ALTER TABLE `profiles` ADD COLUMN `wealth` INT(11) NOT NULL COMMENT 'Денег на руках' AFTER `pokeballs`;")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// CreateWeaponsAndAddWealthDown drops `weapons` table and `wealth` column of `profiles` table
|
||
func CreateWeaponsAndAddWealthDown(tx *sql.Tx) error {
|
||
_, err := tx.Exec("ALTER TABLE `profiles` DROP COLUMN `wealth`;")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
_, err = tx.Exec("DROP TABLE `weapons`;")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
return nil
|
||
}
|