package customization

import (
	"log"

	function "example.com/database/function"
	group "example.com/database/group"
	itemtag "example.com/database/itemtag"

	"gorm.io/gorm"
)

type Customization struct {
	gorm.Model
	Name                string              `gorm:"primaryKey uniqueIndex" json:"name"`
	Functions           []function.Function `gorm:"many2many:customization_function_associations" json:"functions"`
	FlavorText          string              `json:"flavor_text"`
	RulesDescription    string              `json:"rules_description"`
	PhysrepRequirements string              `json:"physrep_requirements"`
	Tags                []itemtag.ItemTag   `gorm:"many2many:customization_tag_associations" json:"tags"`              // Unique
	Visibility          []group.Group       `gorm:"many2many:customization_visibility_associations" json:"visibility"` // Unique
}

func CreateDatabaseCustomization(db *gorm.DB, customization Customization) error {
	result := db.Create(&customization)
	if result.Error != nil {
		return result.Error
	}
	return nil
}

func GetDatabaseCustomization(db *gorm.DB, inputCustomization string) Customization {
	var outputCustomization Customization
	result := db.Model(&Customization{}).Where("name = ?", inputCustomization).Take(&outputCustomization)
	if result.Error != nil {
		return Customization{}
	}
	db.Model(&outputCustomization).Association("Functions").Find(&outputCustomization.Functions)
	db.Model(&outputCustomization).Association("Tags").Find(&outputCustomization.Tags)
	db.Model(&outputCustomization).Association("Visibility").Find(&outputCustomization.Visibility)
	return outputCustomization
}

func GetDatabaseCustomizations(db *gorm.DB, inputCustomizations []string) []Customization {
	var outputCustomizations []Customization
	for _, inputCustomization := range inputCustomizations {
		outputCustomizations = append(outputCustomizations, GetDatabaseCustomization(db, inputCustomization))
	}
	return outputCustomizations
}

func GetAllDatabaseCustomizations(db *gorm.DB) []Customization {
	var outputCustomizations []Customization
	result := db.Find(&outputCustomizations)
	if result.Error != nil {
		log.Println(result.Error)
	}
	for index, outputCustomization := range outputCustomizations {
		outputCustomizations[index] = GetDatabaseCustomization(db, outputCustomization.Name)
	}
	return outputCustomizations
}

func UpdateDatabaseCustomization(db *gorm.DB, customization Customization) error {
	result := db.Save(&customization)
	if result.Error != nil {
		return result.Error
	}
	return nil
}