Squads and chats in DataCache, squads rework

Work in progress, bugs may vary

lib/migrations/18_add_pokememes_wealth.go

@@ -8,7 +8,7 @@ import (
 	"database/sql"
 )
 
-// AddPokememesWealth prepares database for latest game update in mid-October
+// AddPokememesWealthUp prepares database for latest game update in mid-October
 func AddPokememesWealthUp(tx *sql.Tx) error {
 	_, err := tx.Exec("ALTER TABLE `profiles` ADD COLUMN `pokememes_wealth` INT(11) NOT NULL DEFAULT 0 COMMENT 'Стоимость покемонов на руках' AFTER `wealth`;")
 	if err != nil {

lib/migrations/31_change_squads_table.go

@@ -0,0 +1,78 @@
+// i2_bot – Instinct PokememBro Bot
+// Copyright (c) 2017 Vladimir "fat0troll" Hodakov
+
+package migrations
+
+import (
+	"database/sql"
+)
+
+// ChangeSquadsTableUp changes `sqauds` to new format
+func ChangeSquadsTableUp(tx *sql.Tx) error {
+	request := "ALTER TABLE `squads` DROP COLUMN `flood_chat_id`"
+	_, err := tx.Exec(request)
+	if err != nil {
+		return err
+	}
+
+	request = "ALTER TABLE `squads` DROP COLUMN `author_id`"
+	_, err = tx.Exec(request)
+	if err != nil {
+		return err
+	}
+
+	request = "ALTER TABLE `squads` ADD COLUMN `min_level` int(11) NOT NULL DEFAULT 0 AFTER `chat_id`"
+	_, err = tx.Exec(request)
+	if err != nil {
+		return err
+	}
+
+	request = "ALTER TABLE `squads` ADD COLUMN `max_level` int(11) NOT NULL DEFAULT 0 AFTER `min_level`"
+	_, err = tx.Exec(request)
+	if err != nil {
+		return err
+	}
+
+	request = "ALTER TABLE `squads` ADD COLUMN `invite_link` varchar(191) NOT NULL DEFAULT 'https://example.com' AFTER `max_level`"
+	_, err = tx.Exec(request)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// ChangeSquadsTableDown reverts `squads` to old format
+func ChangeSquadsTableDown(tx *sql.Tx) error {
+	request := "ALTER TABLE `squads` ADD COLUMN `flood_chat_id` int(11) NOT NULL AFTER `chat_id`"
+	_, err := tx.Exec(request)
+	if err != nil {
+		return err
+	}
+
+	request = "ALTER TABLE `squads` ADD COLUMN `author_id` int(11) NOT NULL AFTER `flood_chat_id"
+	_, err = tx.Exec(request)
+	if err != nil {
+		return err
+	}
+
+	request = "ALTER TABLE `squads` DROP COLUMN `min_level`"
+	_, err = tx.Exec(request)
+	if err != nil {
+		return err
+	}
+
+	request = "ALTER TABLE `squads` DROP COLUMN `max_level`"
+	_, err = tx.Exec(request)
+	if err != nil {
+		return err
+	}
+
+	request = "ALTER TABLE `squads` DROP COLUMN `invite_link`"
+	_, err = tx.Exec(request)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

lib/migrations/migrations.go

@@ -41,6 +41,7 @@ func (m *Migrations) Init() {
 	goose.AddNamedMigration("28_fix_locations.go", FixLocationsUp, FixLocationsDown)
 	goose.AddNamedMigration("29_fix_leagues_names.go", FixLeaguesNamesUp, FixLeaguesNamesDown)
 	goose.AddNamedMigration("30_create_alarms.go", CreateAlarmsUp, CreateAlarmsUp)
+	goose.AddNamedMigration("31_change_squads_table.go", ChangeSquadsTableUp, ChangeSquadsTableDown)
 }
 
 // Migrate migrates database to current version