package migration

import (
	"database/sql"

	"github.com/pressly/goose"
)

func init() {
	goose.AddMigration(upAddMoreTags, downAddMoreTags)
}

func upAddMoreTags(tx *sql.Tx) error {
	_, err := tx.Exec(`
alter table media_file
	add mbz_track_id varchar(255);
alter table media_file
	add mbz_album_id varchar(255);
alter table media_file
	add mbz_artist_id varchar(255);
alter table media_file
	add mbz_album_artist_id varchar(255);
alter table media_file
	add mbz_album_type varchar(255);
alter table media_file
	add mbz_album_comment varchar(255);
alter table media_file
	add catalog_num varchar(255);

alter table album
	add mbz_album_id varchar(255);
alter table album
	add mbz_album_artist_id varchar(255);
alter table album
	add mbz_album_type varchar(255);
alter table album
	add mbz_album_comment varchar(255);
alter table album
	add catalog_num varchar(255);

create index if not exists album_mbz_album_type
	on album (mbz_album_type);

alter table artist
	add mbz_artist_id varchar(255);

`)
	if err != nil {
		return err
	}
	notice(tx, "A full rescan needs to be performed to import more tags")
	return forceFullRescan(tx)
}

func downAddMoreTags(tx *sql.Tx) error {
	return nil
}