bamort/backend/gsmaster/misclookup.go

package gsmaster

import (
	"bamort/database"
	"bamort/models"
	"fmt"
	"sort"
	"strings"
)

// GetMiscLookupByKey retrieves values for a key using the default game system id.
// Optional order parameter can be: "id", "value", "source", "source_value" (defaults to "value").
func GetMiscLookupByKey(key string, order ...string) ([]models.MiscLookup, error) {
	gs := models.GetGameSystem(0, "")
	return GetMiscLookupByKeyForSystem(key, gs.ID, order...)
}

// GetMiscLookupByKeyForSystem retrieves values for a key filtered by the provided game system id.
// Optional order parameter can be: "id", "value", "source", "source_value" (defaults to "value").
func GetMiscLookupByKeyForSystem(key string, gameSystemId uint, order ...string) ([]models.MiscLookup, error) {
	var items []models.MiscLookup

	orderKey := "value"
	orderBy := "value ASC"
	if len(order) > 0 && order[0] != "" {
		switch order[0] {
		case "id":
			orderKey = "id"
			orderBy = "id ASC"
		case "value":
			orderKey = "value"
			orderBy = "value ASC"
		case "source":
			orderKey = "source"
			orderBy = "source_id ASC, value ASC"
		case "source_value":
			orderKey = "source"
			orderBy = "source_id ASC, value ASC"
		default:
			orderKey = "value"
			orderBy = "value ASC"
		}
	}

	gs := models.GetGameSystem(gameSystemId, "")
	if gs == nil {
		gs = models.GetGameSystem(0, "")
	}

	query := database.DB.Where("`key` = ?", key)
	if gs.ID != 0 {
		query = query.Where("game_system_id = ? OR game_system_id = 0 OR game_system_id IS NULL", gs.ID)
	} else {
		query = query.Where("game_system_id = 0 OR game_system_id IS NULL")
	}

	if err := query.Order(orderBy).Find(&items).Error; err != nil {
		return items, err
	}

	if gs.ID != 0 {
		for i := range items {
			if items[i].GameSystemId == 0 {
				items[i].GameSystemId = gs.ID
				database.DB.Model(&items[i]).Update("game_system_id", gs.ID)
			}
		}
	}

	if key == "faiths" {
		existingValues := make(map[string]struct{}, len(items))
		for i := range items {
			trimmed := strings.TrimSpace(items[i].Value)
			if trimmed == "" {
				continue
			}
			existingValues[trimmed] = struct{}{}
		}

		believeItems, err := collectBeliefsForSystem(gs)
		if err != nil {
			return items, err
		}

		additions := make([]models.MiscLookup, 0, len(believeItems))
		for _, believe := range believeItems {
			value := strings.TrimSpace(believe.Name)
			if value == "" {
				continue
			}

			if _, exists := existingValues[value]; exists {
				continue
			}
			existingValues[value] = struct{}{}

			addition := models.MiscLookup{
				ID:           believe.ID,
				Key:          key,
				Value:        value,
				SourceID:     believe.SourceID,
				PageNumber:   believe.PageNumber,
				GameSystem:   believe.GameSystem,
				GameSystemId: believe.GameSystemId,
			}

			if addition.GameSystemId == 0 && gs.ID != 0 {
				addition.GameSystemId = gs.ID
			}

			additions = append(additions, addition)
		}

		if len(additions) > 0 {
			items = append(items, additions...)
			sortMiscLookup(items, orderKey)
		}
	}

	return items, nil
}

func collectBeliefsForSystem(gs *models.GameSystem) ([]models.Believe, error) {
	var believes []models.Believe
	if gs == nil {
		return believes, nil
	}

	query := database.DB.Model(&models.Believe{})
	if gs.ID != 0 {
		query = query.Where("game_system_id = ?", gs.ID)
	}
	if gs.Name != "" {
		query = query.Or("game_system = ?", gs.Name)
	}

	if err := query.Find(&believes).Error; err != nil {
		return believes, err
	}

	return believes, nil
}

func sortMiscLookup(items []models.MiscLookup, orderKey string) {
	sort.SliceStable(items, func(i, j int) bool {
		switch orderKey {
		case "id":
			return items[i].ID < items[j].ID
		case "source":
			if items[i].SourceID == items[j].SourceID {
				return items[i].Value < items[j].Value
			}
			return items[i].SourceID < items[j].SourceID
		default:
			return items[i].Value < items[j].Value
		}
	})
}

/*
// PopulateMiscLookupData populates initial misc lookup data if table is empty
func PopulateMiscLookupData() error {
	// Check if data already exists
	var count int64
	if err := database.DB.Model(&models.MiscLookup{}).Count(&count).Error; err != nil {
		return err
	}
	if count > 0 {
		return nil // Data already exists
	}

	// Define initial data
	initialData := []struct {
		key    string
		values []string
	}{
		{
			key:    "gender",
			values: []string{"männlich", "weiblich", "divers"},
		},
		{
			key:    "races",
			values: []string{"Mensch", "Elf", "Zwerg", "Gnom", "Halbling"},
		},
		{
			key: "origins",
			values: []string{
				"Albai", "Aran", "Chryseia", "Clanngadarn", "Erainn",
				"Eschar", "Fuardain", "Ikengabecken", "KanThaiPan", "Küstenstaaten",
				"Moravod", "Nahuatlan", "Rawindra", "Twyneddin", "Valian",
			},
		},
		{
			key:    "social_classes",
			values: []string{"Unfrei", "Volk", "Mittelschicht", "Adel"},
		},
		{
			key:    "faiths",
			values: []string{"Keine", "Nathir", "Deis Albai", "Mahal", "Druide"},
		},
		{
			key:    "handedness",
			values: []string{"rechts", "links", "beidhändig"},
		},
		{
			key:    "social_class_bonus",
			values: []string{"Unfrei|Halbwelt|2", "Volk|Alltag|2", "Mittelschicht|Wissen|2", "Adel|Sozial|2"},
		},
	}

	// Insert data
	for _, item := range initialData {
		for _, value := range item.values {
			misc := models.MiscLookup{
				Key:      item.key,
				Value:    value,
				SourceID: 1,
			}
			if err := database.DB.Create(&misc).Error; err != nil {
				return err
			}
		}
	}

	return nil
}
*/

// GetSocialClassBonusPoints retrieves bonus learning points for a social class from database
func GetSocialClassBonusPoints(socialClass string) (map[string]int, error) {
	bonuses, err := GetMiscLookupByKey("social_class_bonus")
	if err != nil {
		return nil, err
	}

	result := make(map[string]int)
	for _, bonus := range bonuses {
		// Parse format: "SocialClass|Category|Points"
		parts := strings.Split(bonus.Value, "|")
		if len(parts) == 3 && parts[0] == socialClass {
			category := parts[1]
			points := 0
			fmt.Sscanf(parts[2], "%d", &points)
			result[category] = points
		}
	}

	return result, nil
}