sync.ps1

param($profileName) 

# If /? is specified then print help
if ($args -contains "/?") {
    Write-Output "Usage: sync.ps1 [profileName] [-wipe] [-prod] [-force] [-nopref] [-precompile]"
    Write-Output "profileName: The name of the profile to sync"
    Write-Output "-wipe: Wipe all files on the board before syncing"
    Write-Output "-prod: Sync only the minimum files for flashing, with version 1.0.0"
    Write-Output "-force: Sync all files, regardless of last edit time"
    Write-Output "-nopref: Skip sending prefs.json"
    Write-Output "-precompile: Precompile .py files to .mpy files"
    exit 0
}

# Globals
Remove-Module -Name serial-toys 2>$null
Import-Module .\serial-toys.psm1


# Set env var for baud
$env:AMPY_BAUD = 115200

# Load the profile
try {

    if ($null -eq $profileName) {
        throw "Profile Name Required!"
    }

    if (!(Test-Path -Path ".\profiles\$profileName.json")) {
        # list all available profiles
        $profiles = Get-ChildItem -Path ".\profiles" -Filter "*.json" | Select-Object -ExpandProperty BaseName
        Write-Output "Available profiles: $profiles"

        throw "Profile file .\profiles\$profileName.json does not exist!"
    }
    
    $activeProfile = Get-Content -Raw ".\profiles\$profileName.json" | ConvertFrom-Json 
    # load active modules
    $activeModules = $activeProfile.activeModules
    Write-Host "Flashing Profile: $profileName"
    Write-Host "Active Modules: $activeModules"

    $dest = "$PSScriptRoot\profile.json"
    Write-Output "Placeholder" > $dest
    Invoke-Expression "xcopy .\profiles\$profileName.json $dest /Y"
}
catch {
    Write-Error "Could not load profile from file .\profiles\$profileName.json"
    Write-Error "Please specify the profile name as a parameter to this script" -ErrorAction Stop
}

# Connect to the board
$port = Find-MicrocontrollerPort
if ($port -eq "COM1")
{
    Write-Error "Board not detected. Aborting." -ErrorAction Stop
}

$MAX = 0
$MAXEDITTIME = 0

# check basic connectivity to the board
Write-Host "Checking basic connectivity on $port to the board.."
$job = Start-Job -ScriptBlock { ampy --port $using:port ls } 
Wait-Job -Job $job -Timeout 10 | Out-Null
if ($job.State -eq "Running") {
    Stop-Job -Job $job
    Write-Error "Could not connect - timed out" -ErrorAction Stop
} else {
    $output = Receive-Job -Job $job

    if (-not $output -or $output -match "Error" -or $output -match "Traceback") {
        Write-Output "Error: $LASTEXITCODE $output"
        Write-Error "Could not connect to the board. Please check the connection." -ErrorAction Stop
    } 
}

# if the user specified -wipe or -prod, delete all files on the board
if ($args -contains "-wipe" -or $args -contains "-prod") {
    Write-Host "Wipe option specified. Deleting all files on the board.."

    # wipe all files on the board, recursively
    $files = ampy --port $port ls -r
    $i = 0
    foreach ($f in $files) {
        $fn = $f -replace "\s", ""
        $i++
        # Use Write Progress to show progress
        $percentComplete = [math]::Round(($i / $files.Count) * 100)
        Write-Progress "Deleting:" -Status "($percentComplete%) Deleting file $fn.." -PercentComplete (($i / $files.Count) * 100)  -Id 1

        # skip prefs.json if -nopref is specified
        if ($args -contains "-nopref" -and $fn -eq "prefs.json") {
            Write-Output "Skipping deleting prefs.json (-nopref specified)"
            continue;
        }
        ampy --port $port rm $fn > $null
    }
    Write-Output "All files deleted."
}

# if the user added -prod to the command line, send a new file to the board containing 1.0.0 with the file name version
if ($args -contains "-prod") {
    Write-Host "Prod option specified. Will upload minimum files for flashing, with version 1.0.0."
    Write-Output 0 | Out-File -Encoding ascii .\lastedit.dat
    $activeModules = @("basic", "ota", "wifi", "web")
} elseif ($args -contains "-force") {
    Write-Output 0 | Out-File -Encoding ascii .\lastedit.dat
    Write-Output "Force option specified. All files will be copied!"
} elseif ($args -contains "-wipe") {
    Write-Output 0 | Out-File -Encoding ascii .\lastedit.dat
    Write-Output "Wipe option specified. All files will be copied!"
} else {
    Write-Host "Checking when board was last updated.."
    Remove-Item ./lastedit.dat
    ampy --port $port get lastedit.dat > lastedit.dat 2> $null

    $MAX = Get-Content -Path .\lastedit.dat
    $MAXEDITTIME = $MAX

    if ((Get-Item "lastedit.dat").length -eq 0) {
        Write-Output "The board does not have a lastedit.dat file, so all files will be copied."
        Write-Host "Press any key to continue..."
        $Host.UI.RawUI.ReadKey("NoEcho,IncludeKeyDown") | Out-Null

        Write-Output 0 | Out-File -Encoding ascii .\lastedit.dat
    }
    else {
        Write-Output "Last edit time on board: $MAX"
    }
}

# send all files to the device
$files = Get-ChildItem . -name -recurse -include *.py, *.html, *.sh, *.js, *.cfg, *.crt, *.key, *.c, *.raw, profile.json, *.json
$sent = 0

# Remove all .venv files from $files
$files = $files | Where-Object {$_ -notlike ".venv*"}
$files = $files | Where-Object {$_ -notlike ".vscode*"}

# Create an array to store the files to send
$filesToSend = @()

for ($i = 0; $i -lt $files.Count; $i++) {
    $f = $files[$i]

    $percentComplete = [math]::Round(($i / $files.Count) * 100)
    Write-Progress "Processing files:" -Status "($percentComplete%) Checking $f..." -PercentComplete $percentComplete -Id 1

    $LE = (Get-ChildItem $f).LastWriteTimeUtc | Get-Date -UFormat %s

    if ($LE -gt $MAX) {

        # Skip unchanged files
        if ($MAXEDITTIME -lt $LE)
        {
            $MAXEDITTIME = $LE
        }

        # Skip files from inactive modules
        $activeModule = $False
        $rootFile = $False
        if ($activeModules.Contains("$((Get-Item $f).Directory.Name)")) {
            $activeModule = $True
        }
        if (!$f.Contains("\")) {
            $rootFile = $True
        }
        if ($f.Contains("board_system")) {
            $rootFile = $True
        }
        if ($f.Contains("ulib")) {
            $rootFile = $True
        }        

        # Skip .package
        if ($f.Contains(".package")) {
            continue;
        }

        if (!$rootFile -and !$activeModule)
        {
            continue;
        }

        # Skip prefs.json if "-no-prefs" is specified
        if ($args -contains "-nopref" -and $f -eq "prefs.json") {
            Write-Output "Skipping sending prefs.json (-nopref specified)"
            continue;
        }

        # Add the file to the array to send
        $filesToSend += $f
    }
}

# Let user know how many files will be transferred
Write-Output "Files to send: $($filesToSend.Count)"

# Send all the identified files
for ($i = 0; $i -lt $filesToSend.Count; $i++) {
    $f = $filesToSend[$i]

    $percentComplete = [math]::Round(($i / $filesToSend.Count) * 100)

    # Ok send the file, all conditions satisfied
    Write-Progress "Uploading files:" -Status "($percentComplete%) Processing file $f..." -PercentComplete $percentComplete -Id 1

    # MAKE SURE PATH EXISTS ON DEVICE
    $bits = $f.ToString() -split '\\'
    $dir = ""
    for ($j = 0; $j -lt $bits.Count - 1; $j++) {
        if ($j -gt 0) {
            $dir = $dir + "/" + $bits[$j]
        }
        else {
            $dir = $bits[$j]
        }

        ampy --port $port mkdir $dir > $null  2>&1
    }

    # SEND THE FILE
    $fn = "$($f)"
    $fnn = $fn -replace "\\", "/"

    # Only precompile if -precompile is specified
    if ($args -contains "-precompile") {
        # if the file is a .py file cross compile it, skip main.py, boot.py
        if ($fn -like "*.py" -and $fn -ne "main.py" -and $fn -ne "boot.py") {
            Write-Progress "Uploading files:" -Status "($percentComplete%) Cross Compiling file $fn..." -PercentComplete $percentComplete -Id 1
            python -m mpy_cross -march=xtensawin $fn
            $fnn = $fnn -replace ".py", ".mpy"
        }
    }

    # send the file using ampy
    Write-Progress "Uploading files:" -Status "($percentComplete%) Sending file $fnn..." -PercentComplete $percentComplete -Id 1
    ampy --port $port put $fnn $fnn

    if (!($?)) {
        Write-Output "Failed to send file to the board, attempting to delete and send again.."
        $boardFile = "$(ampy --port $port get $fnn)" 
        Write-Output "File on microcontoller is $($boardFile.length) bytes"
        Write-Output "File on disk is $((Get-Item $fnn).length) bytes"

        Write-Output "Deleting file on microcontroller:"
        ampy --port $port rm $fnn
        Write-Output "Trying another copy:"
        ampy --port $port put $fnn $fnn
        if (!($?)) {
            Write-Output "Failed again. Giving up."
            exit 3
        }
        Write-Output "Success, moving to next file"
    }
    else {
        $percentComplete = [math]::Round((($i+1) / $filesToSend.Count) * 100)
        Write-Progress "Uploading files:" -Status "($percentComplete%) Sent $fnn!" -PercentComplete $percentComplete -Id 1
    }
    $sent++
}

if ($sent -gt 0) {
    # Write "8.1.0" to the version file
    Write-Output "8.1.0" | Out-File -Encoding ascii .\version

    Write-Host "Uploading new version file to board.."
    ampy --port $port put version

    # record the last time a file was edited
    $MAXEDITTIME = [math]::Round($MAXEDITTIME)
    Write-Output $MAXEDITTIME | Out-File -Encoding ascii .\lastedit.dat
    Write-Host "Uploading lastedit.dat file to board.."
    ampy --port $port put lastedit.dat
} else {
    Write-Output "No changes since last sync."
}

# if the user added -prod to the command line, send a new file to the board containing 1.0.0 with the file name version
if ($args -contains "-prod") {
    Write-Output "1.0.0" | Out-File -Encoding ascii .\version
    Write-Host "Uploading 1.0.0 version file to board, so it will auto update.."
    ampy --port $port put version
    Write-Output "8.1.0" | Out-File -Encoding ascii .\version
}

# Clean up
Remove-Item profile.json

Write-Output "Rebooting..."
Restart-Microcontroller $port

# Remove the progress bar
Write-Progress -Id 1 -Completed "Clearing Progress Bar"

Show-SerialLog $port