Drivers and Uefi download tasks

Author: SuperJMN

Source/Deployer.Raspberry/Scripts/deployment.txt

@@ -1,3 +1,5 @@
 DisplayMarkdown "Core\license.md"
-DeployWindows2
-InjectDrivers "Bla"
+UefiDownload
+DriversDownload
+DeployWindows
+InjectDrivers "Downloaded\Drivers"

Source/Deployer.Raspberry/Tasks/DriversDownload.cs

@@ -0,0 +1,99 @@
+using System;
+using System.Collections.Generic;
+using System.Globalization;
+using System.IO;
+using System.IO.Compression;
+using System.Linq;
+using System.Threading.Tasks;
+using Deployer.Execution;
+
+namespace Deployer.Raspberry.Tasks
+{
+    public class DriversDownload : IDeploymentTask
+    {
+        private readonly IGitHubDownloader downloader;
+        private readonly IFileSystemOperations operations;
+
+        public DriversDownload(IGitHubDownloader downloader, IFileSystemOperations operations)
+        {
+            this.downloader = downloader;
+            this.operations = operations;
+        }
+
+        public async Task Execute()
+        {
+            using (var stream = await downloader.OpenZipStream("https://github.com/andreiw/RaspberryPiPkg"))
+            {
+                var zipArchive = new ZipArchive(stream, ZipArchiveMode.Read);
+
+                var root = zipArchive.Entries.First(x => x.FullName.EndsWith("Drivers/"));
+
+                var contents = zipArchive.Entries.Where(x => x.FullName.StartsWith(root.FullName) && !x.FullName.EndsWith("/"));
+                await ExtractContents(@"Downloaded\Drivers", root, contents);
+            }
+        }
+
+        private async Task ExtractContents(string destination, ZipArchiveEntry baseEntry,
+            IEnumerable<ZipArchiveEntry> entries)
+        {
+            foreach (var entry in entries)
+            {
+                var filePath = entry.FullName.Substring(baseEntry.FullName.Length);
+
+                var destFile = Path.Combine(destination, filePath.Replace("/", "\\"));
+                var dir = Path.GetDirectoryName(destFile);
+                if (!operations.DirectoryExists(dir))
+                {
+                    operations.CreateDirectory(dir);
+                }
+
+                using (var destStream = File.Open(destFile, FileMode.OpenOrCreate))
+                using (var stream = entry.Open())
+                {
+                    await stream.CopyToAsync(destStream);
+                }
+            }
+        }
+
+        private ZipArchiveEntry GetMostRecentDirEntry(ZipArchive p)
+        {
+            var dirs = from e in p.Entries
+                where e.FullName.EndsWith("/")
+                select e;
+
+            var splitted = from e in dirs
+                select new
+                {
+                    e,
+                    Parts = e.FullName.Split('/'),
+                };
+
+            var parsed = from r in splitted
+                select new
+                {
+                    r.e,
+                    Date = FirstParseableOrNull(r.Parts),
+                };
+
+            return parsed.OrderByDescending(x => x.Date).First().e;
+        }
+
+        private DateTime? FirstParseableOrNull(string[] parts)
+        {
+            foreach (var part in parts)
+            {
+                var candidate = part.Split('-');
+                if (candidate.Length > 1)
+                {
+                    var datePart = candidate[0];
+                    if (DateTime.TryParseExact(datePart, "yyyyMMMdd", CultureInfo.InvariantCulture, DateTimeStyles.None, out var date))
+                    {
+                        return date;
+                    }
+                }
+            }
+
+            return null;
+        }
+    }
+}

Source/Deployer.Raspberry/Tasks/UefiDownload.cs

@@ -0,0 +1,99 @@
+using System;
+using System.Collections.Generic;
+using System.Globalization;
+using System.IO;
+using System.IO.Compression;
+using System.Linq;
+using System.Threading.Tasks;
+using Deployer.Execution;
+
+namespace Deployer.Raspberry.Tasks
+{
+    public class UefiDownload : IDeploymentTask
+    {
+        private readonly IGitHubDownloader downloader;
+        private readonly IFileSystemOperations operations;
+
+        public UefiDownload(IGitHubDownloader downloader, IFileSystemOperations operations)
+        {
+            this.downloader = downloader;
+            this.operations = operations;
+        }
+
+        public async Task Execute()
+        {
+            using (var stream = await downloader.OpenZipStream("https://github.com/andreiw/RaspberryPiPkg"))
+            {
+                var zipArchive = new ZipArchive(stream, ZipArchiveMode.Read);
+
+                var mostRecentFolderEntry = GetMostRecentDirEntry(zipArchive);
+
+                var contents = zipArchive.Entries.Where(x => x.FullName.StartsWith(mostRecentFolderEntry.FullName) && !x.FullName.EndsWith("/"));
+                await ExtractContents(@"Downloaded\UEFI", mostRecentFolderEntry, contents);
+            }
+        }
+
+        private async Task ExtractContents(string destination, ZipArchiveEntry baseEntry,
+            IEnumerable<ZipArchiveEntry> entries)
+        {
+            foreach (var entry in entries)
+            {
+                var filePath = entry.FullName.Substring(baseEntry.FullName.Length);
+
+                var destFile = Path.Combine(destination, filePath.Replace("/", "\\"));
+                var dir = Path.GetDirectoryName(destFile);
+                if (!operations.DirectoryExists(dir))
+                {
+                    operations.CreateDirectory(dir);
+                }
+
+                using (var destStream = File.Open(destFile, FileMode.OpenOrCreate))
+                using (var stream = entry.Open())
+                {
+                    await stream.CopyToAsync(destStream);
+                }
+            }
+        }
+
+        private ZipArchiveEntry GetMostRecentDirEntry(ZipArchive p)
+        {
+            var dirs = from e in p.Entries
+                where e.FullName.EndsWith("/")
+                select e;
+
+            var splitted = from e in dirs
+                select new
+                {
+                    e,
+                    Parts = e.FullName.Split('/'),
+                };
+
+            var parsed = from r in splitted
+                select new
+                {
+                    r.e,
+                    Date = FirstParseableOrNull(r.Parts),
+                };
+
+            return parsed.OrderByDescending(x => x.Date).First().e;
+        }
+
+        private DateTime? FirstParseableOrNull(string[] parts)
+        {
+            foreach (var part in parts)
+            {
+                var candidate = part.Split('-');
+                if (candidate.Length > 1)
+                {
+                    var datePart = candidate[0];
+                    if (DateTime.TryParseExact(datePart, "yyyyMMMdd", CultureInfo.InvariantCulture, DateTimeStyles.None, out var date))
+                    {
+                        return date;
+                    }
+                }
+            }
+
+            return null;
+        }
+    }
+}

Source/Deployer.Tests/Deployer.Tests.csproj

@@ -17,6 +17,7 @@
   </ItemGroup>
 
   <ItemGroup>
+    <ProjectReference Include="..\Deployer.Raspberry\Deployer.Raspberry.csproj" />
     <ProjectReference Include="..\Deployer\Deployer.csproj" />
   </ItemGroup>
 

Source/Deployer.Tests/Tasks/DriversDownloadTests.cs

@@ -0,0 +1,16 @@
+using System.Threading.Tasks;
+using Deployer.Raspberry.Tasks;
+using Xunit;
+
+namespace Deployer.Tests.Tasks
+{
+    public class DriversDownloadTests
+    {
+        [Fact]
+        public async Task Test()
+        {
+            var task = new DriversDownload(new GitHubDownloader(), new FileSystemOperations());
+            await task.Execute();
+        }
+    }
+}

Source/Deployer.Tests/Tasks/UefiDownloadTests.cs

@@ -0,0 +1,16 @@
+using System.Threading.Tasks;
+using Deployer.Raspberry.Tasks;
+using Xunit;
+
+namespace Deployer.Tests.Tasks
+{
+    public class UefiDownloadTests
+    {
+        [Fact]
+        public async Task Test()
+        {
+            var task = new UefiDownload(new GitHubDownloader(), new FileSystemOperations());
+            await task.Execute();
+        }
+    }
+}

Source/Deployer/Execution/Testing/TestFileSystemOperations.cs

@@ -28,5 +28,9 @@ public bool DirectoryExists(string path)
         {
             return Directory.Exists(path);
         }
+
+        public void CreateDirectory(string path)
+        {
+        }
     }
 }

Source/Deployer/FileSystemOperations.cs

@@ -30,5 +30,10 @@ public bool DirectoryExists(string path)
         {
             return Directory.Exists(path);
         }
+
+        public void CreateDirectory(string path)
+        {
+            Directory.CreateDirectory(path);
+        }
     }
 }

Source/Deployer/IFileSystemOperations.cs

@@ -8,5 +8,6 @@ public interface IFileSystemOperations
         Task CopyDirectory(string source, string destination);
         Task DeleteDirectory(string path);
         bool DirectoryExists(string path);
+        void CreateDirectory(string path);
     }
 }