package migrate

import (
	ticketmodel "github.com/tidbcloud/aisupport/services/ticket-manager/models/ticket"

	"gorm.io/gorm"
)

var _ = addRecord("202607101400", removeEngineerExpertiseAndWorkloadIndex)

type engineerWithExpertise struct {
	ExpertiseJSON string `gorm:"column:expertise_json"`
}

func (engineerWithExpertise) TableName() string {
	return "engineer_engineers"
}

func removeEngineerExpertiseAndWorkloadIndex(db *gorm.DB) error {
	if db.Migrator().HasColumn(&engineerWithExpertise{}, "expertise_json") {
		if err := db.Migrator().DropColumn(&engineerWithExpertise{}, "expertise_json"); err != nil {
			return err
		}
	}
	if db.Migrator().HasIndex(&ticketmodel.Ticket{}, "idx_ticket_assignee_resolved_at") {
		return db.Migrator().DropIndex(&ticketmodel.Ticket{}, "idx_ticket_assignee_resolved_at")
	}
	return nil
}