Generate metadata needed to populate update feed

Step towards completing 5367

1 files changed, 113 insertions, 0 deletions

diff --git a/src/setup/MetadataTask/GenerateMetadata.cs b/src/setup/MetadataTask/GenerateMetadata.cs
new file mode 100644
index 00000000..4071c271
--- /dev/null
+++ b/src/setup/MetadataTask/GenerateMetadata.cs
@@ -0,0 +1,113 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+namespace WixToolset.Tasks
+{
+    using System;
+    using System.Collections.Generic;
+    using System.IO;
+    using System.Linq;
+    using System.Security.Cryptography;
+    using System.Text.Json;
+    using System.Text.Json.Serialization;
+    using Microsoft.Build.Framework;
+    using Microsoft.Build.Utilities;
+    using WixToolset.Data;
+    using WixToolset.Data.Symbols;
+
+    public class GenerateMetadata : Task
+    {
+        private static readonly JsonSerializerOptions SerializerOptions = new JsonSerializerOptions
+        {
+            PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
+            IgnoreNullValues = true,
+            WriteIndented = true,
+            Converters = { new JsonStringEnumConverter(JsonNamingPolicy.CamelCase) },
+        };
+
+        [Required]
+        public string TargetFile { get; set; }
+
+        [Required]
+        public string WixpdbFile { get; set; }
+
+        public override bool Execute()
+        {
+            var intermediate = Intermediate.Load(this.WixpdbFile);
+
+            var section = intermediate.Sections.Single();
+
+            if (section.Type != SectionType.Bundle)
+            {
+                return false;
+            }
+
+            (var metadata, var sourceLineNumber) = this.GetBundleMetadata(section);
+
+            if (metadata != null)
+            {
+                this.PopulateFileInfo(metadata);
+
+                this.SaveMetadata(metadata, sourceLineNumber);
+            }
+
+            return true;
+        }
+
+        private (Metadata, SourceLineNumber) GetBundleMetadata(IntermediateSection section)
+        {
+            var bundleSymbol = section.Symbols.OfType<WixBundleSymbol>().Single();
+
+            var metadata = new Metadata
+            {
+                Type = MetadataType.Burn,
+                Name = bundleSymbol.Name,
+                Version = bundleSymbol.Version,
+                Publisher = bundleSymbol.Manufacturer,
+                Description = "Installation for " + bundleSymbol.Name,
+                SupportUrl = bundleSymbol.HelpUrl,
+                BundleCode = bundleSymbol.BundleId,
+                UpgradeCode = bundleSymbol.UpgradeCode,
+                AboutUrl = bundleSymbol.AboutUrl,
+                Architecture = PlatformToArchitecture(bundleSymbol.Platform),
+            };
+
+            return (metadata, bundleSymbol.SourceLineNumbers);
+        }
+
+        private void PopulateFileInfo(Metadata metadata)
+        {
+            var fi = new FileInfo(this.TargetFile);
+
+            using (var sha256 = SHA256.Create())
+            using (var stream = fi.OpenRead())
+            {
+                var hash = sha256.ComputeHash(stream);
+
+                metadata.File = fi.Name;
+                metadata.Created = fi.CreationTimeUtc.ToString("O");
+                metadata.Size = fi.Length;
+                metadata.Sha256 = BitConverter.ToString(hash).Replace("-", String.Empty);
+            }
+        }
+
+        private void SaveMetadata(Metadata metadata, SourceLineNumber sourceLineNumber)
+        {
+            var metadataFilePath = Path.ChangeExtension(this.TargetFile, "metadata.json");
+
+            var json = JsonSerializer.Serialize(metadata, SerializerOptions);
+
+            File.WriteAllText(metadataFilePath, json);
+        }
+
+        private static ArchitectureType PlatformToArchitecture(Platform platform)
+        {
+            switch (platform)
+            {
+                case Platform.X86: return ArchitectureType.X86;
+                case Platform.X64: return ArchitectureType.X86;
+                case Platform.ARM64: return ArchitectureType.X86;
+                default: throw new ArgumentException($"Unknown platform {platform}");
+            }
+        }
+    }
+}