package templates

import (
	"strconv"
	"track-gopher/models"
	"track-gopher/db"
	"track-gopher/derby"
)

templ Heats(groups []models.Group, racers []models.Racer, selectedGroupID int64, db *db.DB) {
	@Layout("Race Heats") {
		<div class="container py-4">
			<h1 class="mb-4">Race Heats Generator</h1>
			
			<div class="mb-4">
				<label for="group-select" class="form-label">Select Group</label>
				<select 
					id="group-select" 
					class="form-select"
					hx-get="/heats"
					hx-target="#heats-container"
					hx-trigger="change"
					hx-include="this"
					name="group_id"
				>
					<option value="">Select a group</option>
					for _, group := range groups {
						<option value={ strconv.FormatInt(group.ID, 10) } selected?={ group.ID == selectedGroupID }>
							{ group.Name }
						</option>
					}
				</select>
			</div>
			
			<div id="heats-container">
				if selectedGroupID > 0 {
					@HeatsContent(selectedGroupID, groups, db)
				} else {
					<div class="alert alert-info">
						<p class="mb-0">Please select a group to generate heats.</p>
					</div>
				}
			</div>
		</div>
	}
}

templ HeatsContent(groupID int64, groups []models.Group, db *db.DB) {
	// Filter racers by group
	{{ groupRacers, _ := db.GetRacersByGroup(groupID) }}
	
	<div class="card">
		<div class="card-body">
			<div class="d-flex justify-content-between align-items-center mb-4">
				<h2 class="card-title">
					Heats for { getGroupName(groups, groupID) } ({ strconv.Itoa(len(groupRacers)) } racers)
				</h2>
				<div>
					<button 
						class="btn btn-primary me-2"
						hx-post={ "/api/heats/generate?group_id=" + strconv.FormatInt(groupID, 10) }
						hx-target="#heats-list"
						hx-swap="innerHTML"
					>
						Regenerate Heats
					</button>
					<button 
						class="btn btn-success"
						hx-post={ "/api/heats/save?group_id=" + strconv.FormatInt(groupID, 10) }
						hx-target="#save-status"
					>
						Save Heats
					</button>
				</div>
			</div>
			
			<div id="save-status" class="mb-4"></div>
			
			if len(groupRacers) == 0 {
				<div class="alert alert-warning mb-4">
					<p class="mb-0">No racers in this group. Add racers to generate heats.</p>
				</div>
			} else {
				<div id="heats-list">
					@GeneratedHeats(groupRacers)
				</div>
			}
		</div>
	</div>
}

templ GeneratedHeats(racers []models.Racer) {
	// Convert db.Racer to derby.Racer for heat generation
	{{
        var derbyRacers []derby.Racer
	    for _, racer := range racers {
		    derbyRacers = append(derbyRacers, derby.CreateRacer(racer.ID, racer.FirstName, racer.LastName, racer.CarNumber))
	    }
	
	    var heats = derby.GenerateHeats(derbyRacers)
    }}
	if len(heats) == 0 {
		<div class="alert alert-warning">
			<p class="mb-0">No heats could be generated. Please add more racers.</p>
		</div>
	} else {
		<div class="row row-cols-1 row-cols-md-2 row-cols-lg-3 g-4">
			for _, heat := range heats {
				<div class="col">
					<div class="card h-100">
						<div class="card-body">
							<h3 class="card-title mb-3">Heat { strconv.Itoa(heat.HeatNum) }</h3>
							<table class="table table-sm">
								<thead>
									<tr>
										<th>Lane</th>
										<th>Racer</th>
										<th>Car #</th>
									</tr>
								</thead>
								<tbody>
									<tr>
										<td>1</td>
										<td>{ getRacerName(racers, heat.Lane1ID) }</td>
										<td>{ getRacerCarNumber(racers, heat.Lane1ID) }</td>
									</tr>
									<tr>
										<td>2</td>
										<td>{ getRacerName(racers, heat.Lane2ID) }</td>
										<td>{ getRacerCarNumber(racers, heat.Lane2ID) }</td>
									</tr>
									<tr>
										<td>3</td>
										<td>{ getRacerName(racers, heat.Lane3ID) }</td>
										<td>{ getRacerCarNumber(racers, heat.Lane3ID) }</td>
									</tr>
									<tr>
										<td>4</td>
										<td>{ getRacerName(racers, heat.Lane4ID) }</td>
										<td>{ getRacerCarNumber(racers, heat.Lane4ID) }</td>
									</tr>
								</tbody>
							</table>
						</div>
					</div>
				</div>
			}
		</div>
	}
}

func getGroupName(groups []models.Group, groupID int64) string {
	for _, group := range groups {
		if group.ID == groupID {
			return group.Name
		}
	}
	return "Unknown Group"
}

func getRacerName(racers []models.Racer, racerID *int64) string {
	if racerID == nil {
		return "Empty"
	}
	
	for _, racer := range racers {
		if racer.ID == *racerID {
			return racer.FirstName + " " + racer.LastName
		}
	}
	return "Unknown Racer"
}

func getRacerCarNumber(racers []models.Racer, racerID *int64) string {
	if racerID == nil {
		return "-"
	}
	
	for _, racer := range racers {
		if racer.ID == *racerID {
			return racer.CarNumber
		}
	}
	return "-"
}