# Function to ensure log directory exists
function Ensure-LogDirectory {
    $logDir = "C:\swh-resources\logs"
    if (-not (Test-Path -Path $logDir)) {
        New-Item -ItemType Directory -Force -Path $logDir | Out-Null
        Write-Host "Created log directory: $logDir"
    }
}

# Function to log messages
function Write-Log {
    param (
        [string]$Message,
        [switch]$Warning,
        [switch]$Error
    )

    Ensure-LogDirectory
    $logFile = "C:\swh-resources\logs\timelog.txt"
    $timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
    "$timestamp - $Message" | Out-File -FilePath $logFile -Append

    # Also output to console with appropriate styling
    if ($Error) {
        Write-Error $Message
    }
    elseif ($Warning) {
        Write-Warning $Message
    }
    else {
        Write-Host $Message
    }
}

# Function to get timezone and time data
function Get-TimeZoneData {
    $apiUrl = "http://ip-api.com/json/?fields=timezone,status,message"
    Write-Log "Attempting to fetch timezone data from IP-API..."

    try {
        $response = Invoke-RestMethod -Uri $apiUrl -Method Get

        if ($response.status -eq "success") {
            # Get current UTC time
            $utcNow = [DateTime]::UtcNow
            Write-Log "Current UTC time: $utcNow"

            # Create response object
            $timeData = @{
                timezone = $response.timezone
                datetime = $utcNow.ToString("o") # ISO 8601 format
            }

            Write-Log "Successfully fetched time data from IP-API"
            Write-Log "Detected timezone: $($timeData.timezone)"
            return $timeData
        } else {
            throw "API returned error status: $($response.message)"
        }
    }
    catch {
        Write-Log "Failed to fetch data from IP-API: $_" -Error
        return $null
    }
}

# Function to set system time
function Set-SystemTime {
    param (
        [DateTime]$NewTime
    )

    try {
        Set-Date -Date $NewTime
        Write-Log "System time updated successfully to $NewTime"
    }
    catch {
        Write-Log "Failed to set system time: $_" -Error
    }
}

# Function to set timezone
function Set-SystemTimeZone {
    param (
        [string]$TimeZoneId
    )

    try {
        Set-TimeZone -Id $TimeZoneId
        Write-Log "System timezone updated successfully to $TimeZoneId"
    }
    catch {
        Write-Log "Failed to set system timezone: $_" -Error
    }
}

# Enhanced timezone mapping function
function Get-WindowsTimeZone {
    param (
        [string]$ApiTimeZone
    )

    Write-Log "Attempting to map timezone: $ApiTimeZone"

    # Comprehensive mapping of IANA to Windows timezone IDs
    $timeZoneMap = @{
        # North America
        "America/Chicago" = "Central Standard Time"
        "America/New_York" = "Eastern Standard Time"
        "America/Los_Angeles" = "Pacific Standard Time"
        "America/Denver" = "Mountain Standard Time"
        "America/Phoenix" = "US Mountain Standard Time"
        "America/Anchorage" = "Alaskan Standard Time"
        "America/Toronto" = "Eastern Standard Time"
        "America/Vancouver" = "Pacific Standard Time"
        "America/Mexico_City" = "Central Standard Time (Mexico)"

        # Europe
        "Europe/London" = "GMT Standard Time"
        "Europe/Paris" = "Romance Standard Time"
        "Europe/Berlin" = "W. Europe Standard Time"
        "Europe/Moscow" = "Russian Standard Time"
        "Europe/Amsterdam" = "W. Europe Standard Time"
        "Europe/Rome" = "W. Europe Standard Time"
        "Europe/Madrid" = "Romance Standard Time"

        # Asia
        "Asia/Tokyo" = "Tokyo Standard Time"
        "Asia/Shanghai" = "China Standard Time"
        "Asia/Hong_Kong" = "China Standard Time"
        "Asia/Singapore" = "Singapore Standard Time"
        "Asia/Dubai" = "Arabian Standard Time"
        "Asia/Kolkata" = "India Standard Time"
        "Asia/Bangkok" = "SE Asia Standard Time"
        "Asia/Seoul" = "Korea Standard Time"

        # Australia and Pacific
        "Australia/Sydney" = "AUS Eastern Standard Time"
        "Australia/Melbourne" = "AUS Eastern Standard Time"
        "Australia/Perth" = "W. Australia Standard Time"
        "Pacific/Auckland" = "New Zealand Standard Time"
        "Pacific/Honolulu" = "Hawaiian Standard Time"

        # Africa and Middle East
        "Africa/Cairo" = "Egypt Standard Time"
        "Africa/Johannesburg" = "South Africa Standard Time"
        "Africa/Lagos" = "W. Central Africa Standard Time"
        "Africa/Nairobi" = "E. Africa Standard Time"
    }

    if ($timeZoneMap.ContainsKey($ApiTimeZone)) {
        $windowsTimeZone = $timeZoneMap[$ApiTimeZone]
        Write-Log "Found Windows timezone mapping: $windowsTimeZone"
        return $windowsTimeZone
    }

    # If no direct mapping found, try to find a similar timezone
    Write-Log "No direct mapping found, attempting to find similar timezone..." -Warning
    $timeZoneName = $ApiTimeZone.Split('/')[-1].Replace('_', ' ')
    $allTimeZones = [System.TimeZoneInfo]::GetSystemTimeZones()

    $similarTimeZone = $allTimeZones | Where-Object {
        $_.DisplayName -like "*$timeZoneName*" -or
        $_.StandardName -like "*$timeZoneName*" -or
        $_.Id -like "*$timeZoneName*"
    } | Select-Object -First 1

    if ($similarTimeZone) {
        Write-Log "Found similar Windows timezone: $($similarTimeZone.Id)"
        return $similarTimeZone.Id
    }

    Write-Log "Could not find a matching Windows timezone for ${ApiTimeZone}" -Error
    return $null
}

# Main script
Write-Log "=== Time Synchronization Script Started ==="

$timeData = Get-TimeZoneData

if ($timeData) {
    # Set system time
    $newDateTime = [DateTime]::Parse($timeData.datetime)
    Set-SystemTime -NewTime $newDateTime

    # Get and set Windows timezone
    $windowsTimeZone = Get-WindowsTimeZone -ApiTimeZone $timeData.timezone

    if ($windowsTimeZone) {
        Set-SystemTimeZone -TimeZoneId $windowsTimeZone
    }
    else {
        Write-Log "Couldn't find a matching Windows timezone. Please set manually." -Warning
    }
}

# Enable Windows Time service
Write-Log "Configuring Windows Time service..."
Set-Service -Name W32Time -StartupType Automatic
Start-Service W32Time
Write-Log "Windows Time service enabled and started"

# Set time synchronization to use Windows Time service
Write-Log "Configuring time synchronization..."
$configResult = w32tm /config /manualpeerlist:time.windows.com /syncfromflags:manual /reliable:yes /update
Write-Log "Time synchronization configured: $configResult"

# Force time sync
Write-Log "Forcing time synchronization..."
$syncResult = w32tm /resync /force
Write-Log "Time sync result: $syncResult"

# Set time zone to auto-detect
Set-ItemProperty -Path "HKLM:\SYSTEM\CurrentControlSet\Control\TimeZoneInformation" -Name "DynamicDaylightTimeDisabled" -Value 0
Write-Log "Time zone auto-detect enabled"

Write-Log "Time and time zone have been synchronized."
Write-Log "=== Time Synchronization Script Completed ==="