azp: fix localcheckoutazure in single file deployment (#533)

Author: ChristopherHX

src/Runner.Server/Controllers/MessageController.cs

@@ -51,6 +51,7 @@
 using Sdk.Actions;
 using Sdk.Pipelines;
 using ExecutionContext = Sdk.Pipelines.ExecutionContext;
+using Microsoft.Extensions.FileProviders;
 
 namespace Runner.Server.Controllers
 {
@@ -2814,7 +2815,13 @@ private HookResponse AzureDevopsMain(string fileRelativePath, string content, st
                 var tasksByNameAndVersion = new Dictionary<string, TaskMetaData>(getOrCreateTaskCache(null), StringComparer.OrdinalIgnoreCase);
                 state.TasksByNameAndVersion = tasksByNameAndVersion;
                 try {
-                    TaskMetaData.Load(tasksByNameAndVersion, Path.Join(Path.GetDirectoryName(Assembly.GetEntryAssembly().Location), "wwwroot", "localcheckoutazure.zip"));
+                    if(!string.IsNullOrWhiteSpace(Assembly.GetEntryAssembly().Location)) {
+                        TaskMetaData.Load(tasksByNameAndVersion, Path.Join(Path.GetDirectoryName(Assembly.GetEntryAssembly().Location), "wwwroot", "localcheckoutazure.zip"));
+                    } else {
+                        var embeddedFileProvider = new ManifestEmbeddedFileProvider(Assembly.GetAssembly(type: typeof(Program))!, "wwwroot");
+                        using var str = embeddedFileProvider.GetFileInfo("localcheckoutazure.zip").CreateReadStream();
+                        TaskMetaData.Load(tasksByNameAndVersion, "embedded://localcheckoutazure.zip", str);
+                    }
                 } catch(Exception ex) {
                     workflowTraceWriter.Error($"Failed to read the localcheckoutazure inbox task: {ex}");
                 }

src/Runner.Server/Controllers/TaskController.cs

@@ -15,6 +15,7 @@
 using Microsoft.AspNetCore.Mvc;
 using Microsoft.Extensions.Caching.Memory;
 using Microsoft.Extensions.Configuration;
+using Microsoft.Extensions.FileProviders;
 using Microsoft.Extensions.Logging;
 using Newtonsoft.Json;
 using Swashbuckle.AspNetCore.Annotations;
@@ -67,6 +68,12 @@ public IActionResult GetTaskArchive(Guid taskid, string version) {
                 handler.GetZip(runid, false, false, archivePath.Substring(prefix.Length), true).GetAwaiter().GetResult();
                 return new EmptyResult();
             }
+            prefix = "embedded://";
+            if(archivePath.StartsWith(prefix, StringComparison.OrdinalIgnoreCase)) {
+                var embeddedFileProvider = new ManifestEmbeddedFileProvider(Assembly.GetAssembly(type: typeof(Program))!, "wwwroot");
+                var str = embeddedFileProvider.GetFileInfo(archivePath.Substring(prefix.Length)).CreateReadStream();
+                return new FileStreamResult(str, "application/zip") { EnableRangeProcessing = true };
+            }
             return new FileStreamResult(System.IO.File.OpenRead(tasksByNameAndVersion[$"{taskid}@{version}"].ArchivePath), "application/zip") { EnableRangeProcessing = true };
         }
     }

src/Sdk/AzurePipelines/TaskMetaData.cs

@@ -45,7 +45,17 @@ public static (List<TaskMetaData>, Dictionary<string, TaskMetaData>) LoadTasks(s
     }
 
     public static TaskMetaData Load(Dictionary<string, TaskMetaData> tasksByNameAndVersion, string filePath) {
-        var task = System.IO.Compression.ZipFile.OpenRead(filePath);
+        using (var task = System.IO.Compression.ZipFile.OpenRead(filePath))
+            return Load(tasksByNameAndVersion, filePath, task);
+    }
+
+    public static TaskMetaData Load(Dictionary<string, TaskMetaData> tasksByNameAndVersion, string filePath, Stream zipStream) {
+        using (var task = new System.IO.Compression.ZipArchive(zipStream))
+            return Load(tasksByNameAndVersion, filePath, task);
+    }
+
+    public static TaskMetaData Load(Dictionary<string, TaskMetaData> tasksByNameAndVersion, string filePath, System.IO.Compression.ZipArchive task)
+    {
         using(var stream = task.GetEntry("task.json")?.Open())
         using(var textreader = new StreamReader(stream)) {
             var metaData = JsonConvert.DeserializeObject<TaskMetaData>(textreader.ReadToEnd());