param (
    [switch]$doNotBuildBaseFap = $false,
    [switch]$buildCustomFontFap = $false
)

function Get-LatestDirectory {
    param (
        $Path
    )

    Get-ChildItem -Path $Path | Where-Object {$_.PSIsContainer} | Sort-Object LastWriteTime -Descending | Select-Object -First 1
}

$build_commands = @(
    [PSCustomObject]@{
        Name = "Official Dev";
        FbtSwitch = "od";
        FirmwarePath = "flipperzero-firmware_official_dev";
        ArtifactName = "totp_official-dev_fw{FEATURES_SUFFIX}.fap";
    }
    [PSCustomObject]@{
        Name = "Official Stable";
        FbtSwitch = "os";
        FirmwarePath = "flipperzero-firmware_official_stable";
        ArtifactName = "totp_official-stable_fw{FEATURES_SUFFIX}.fap";
    }
    [PSCustomObject]@{
        Name = "Xtreme \ Unleashed";
        FbtSwitch = "x";
        FirmwarePath = "flipperzero-firmware_xtreme";
        ArtifactName = "totp_xtreme_unleashed_fw{FEATURES_SUFFIX}.fap";
    }
)

Push-Location $PSScriptRoot

if (!(Test-Path -PathType Container "build")) {
    New-Item -ItemType Directory -Path "build"
} else {
    Remove-Item "build\*" -Recurse -Force
}

function Build-Run {
    param (
        [string]$FeaturesSuffix,
        [string[]]$CppDefine,
        [string]$Subfolder
    )

    foreach ($build_command in $build_commands) {
        Write-Host "Building $($build_command.Name)"
        $build_path = Join-Path -Path $build_command.FirmwarePath -ChildPath "build"

        $fbt_args = @($build_command.FbtSwitch, "COMPACT=1", "DEBUG=0", "VERBOSE=0", "fap_totp", "--clean")
        if ($CppDefine.Length -gt 0) {
            $CppDefine | ForEach-Object {
                $fbt_args += '-D'
                $fbt_args += $_
            }
        }

        Invoke-Expression -Command "./fbt.ps1 $fbt_args"

        $latest_dir = Get-LatestDirectory -Path $build_path
        $build_output_folder = "build"
        if ($Subfolder -ne $null -and $Subfolder -ne '') {
            $build_output_folder = Join-Path $build_output_folder $Subfolder
            if (!(Test-Path -PathType Container $build_output_folder)) {
                New-Item -ItemType Directory -Path $build_output_folder
            }
        }

        $build_output_artifact = Join-Path $build_output_folder "$($build_command.ArtifactName -replace '{FEATURES_SUFFIX}',$FeaturesSuffix)"

        Copy-Item "$build_path\$latest_dir\.extapps\totp.fap" -Destination $build_output_artifact

        Write-Host "Artifacts for $($build_command.Name) stored at $build_output_artifact"
    }
}

if ($doNotBuildBaseFap -eq $false) {
    Write-Information 'Building with all the features enables'
    Build-Run -FeaturesSuffix ''

    Write-Information 'Building with BadBT but without BadBT icon'
    Build-Run -FeaturesSuffix '_badbt-wo-icon' -CppDefine TOTP_NO_AUTOMATION_ICONS

    Write-Information 'Building without BadBT'
    Build-Run -FeaturesSuffix '_no-badbt' -CppDefine TOTP_NO_BADBT_TYPE,TOTP_NO_AUTOMATION_ICONS
}

if ($buildCustomFontFap -eq $true) {
    $custom_fonts = @(
        [PSCustomObject]@{
            CDEF = "TOTP_FONT_REDHATMONO";
            Subfolder = 'Redhat_Mono'
        },
        [PSCustomObject]@{
            CDEF = "TOTP_FONT_BEDSTEAD";
            Subfolder = 'Bedstead'
        },
        [PSCustomObject]@{
            CDEF = "TOTP_FONT_ZECTOR";
            Subfolder = 'Zector'
        },
        [PSCustomObject]@{
            CDEF = "TOTP_FONT_712SERIF";
            Subfolder = '7-12_Serif'
        },
        [PSCustomObject]@{
            CDEF = "TOTP_FONT_GRAPH35PIX";
            Subfolder = 'Graph35_pix'
        },
        [PSCustomObject]@{
            CDEF = "TOTP_FONT_KARMAFUTURE";
            Subfolder = 'Karma_future'
        },
        [PSCustomObject]@{
            CDEF = "TOTP_FONT_FUNCLIMBING";
            Subfolder = 'Funclimbing'
        },
        [PSCustomObject]@{
            CDEF = "TOTP_FONT_DPCOMIC";
            Subfolder = 'DPComic'
        },
        [PSCustomObject]@{
            CDEF = "TOTP_FONT_PIXELFLAG";
            Subfolder = 'Pixel_flag'
        }
    )

    foreach ($custom_font in $custom_fonts) {
        Write-Information "Custom font ($($custom_font.Subfolder)): Building with all the features enables"
        Build-Run -FeaturesSuffix '' -CppDefine TOTP_FONT=$($custom_font.CDEF) -Subfolder $($custom_font.Subfolder)

        Write-Information "Custom font ($($custom_font.Subfolder)): Building with BadBT but without BadBT icon"
        Build-Run -FeaturesSuffix '_badbt-wo-icon' -CppDefine TOTP_NO_AUTOMATION_ICONS,TOTP_FONT=$($custom_font.CDEF) -Subfolder $($custom_font.Subfolder)

        Write-Information "Custom font ($($custom_font.Subfolder)): Building without BadBT"
        Build-Run -FeaturesSuffix '_no-badbt' -CppDefine TOTP_NO_BADBT_TYPE,TOTP_NO_AUTOMATION_ICONS,TOTP_FONT=$($custom_font.CDEF) -Subfolder $($custom_font.Subfolder)
    }
}

Pop-Location