Updated flasher script to work with submodules for libraries

flash.ps1

@@ -5,6 +5,15 @@ param(
   [String[]]$Libs
 )
 
+Function Find-LibraryFile {
+  [cmdletbinding()]
+  param(
+    [String]$Library,
+    [String]$SearchPath = $(Get-Location)
+  )
+  return (Get-ChildItem -Path $SearchPath -Recurse -Name "$Library.py").FullName
+}
+
 Function Write-AmpyFile {
   [cmdletbinding()]
   param(
@@ -28,31 +37,38 @@ Function Reset-AmpyDevice {
   ampy -p $COMPort reset
 }
 
-$BaseFiles=(
+$BaseFiles = (
   'boot',
   'main',
   'uPyConfig'
 )
-$ExtraFiles=(
+$ExtraFiles = (
   'init_sample'
 )
-$CfgFiles=(
+$CfgFiles = (
   'wifi_cfg',
   'webrepl_cfg'
 )
 
 $BaseFiles | % {
-    Write-AmpyFile -Filename "$_.py" -COMPort $COMPort
+  $Lib = Find-LibraryFile -Library $_
+  Write-AmpyFile -Filename "$Lib" -COMPort $COMPort
 }
 $CfgFiles | % {
   If (Get-ChildItem "$_.py" -ErrorAction SilentlyContinue) {
     Write-AmpyFile -Filename "$_.py" -COMPort $COMPort
   }
 }
-If ($Extras) { $ExtraFiles | % {
-    Write-AmpyFile -Filename "$_.py" -COMPort $COMPort
-}}
-If ($Libs.Count -gt 0) { $Libs | %{
-    Write-AmpyFile -Filename "$_.py" -COMPort $COMPort
-}}
+If ($Extras) {
+  $ExtraFiles | % {
+    $Lib = Find-LibraryFile -Library $_
+    Write-AmpyFile -Filename "$Lib" -COMPort $COMPort
+  }
+}
+If ($Libs.Count -gt 0) {
+  $Libs | % {
+    $Lib = Find-LibraryFile -Library $_
+    Write-AmpyFile -Filename "$Lib" -COMPort $COMPort
+  }
+}
 If ($ResetAfterFlash) { Reset-AmpyDevice -COMPort $COMPort }