From 05174814f6f884b837b46e2611a23d61648c0dd0 Mon Sep 17 00:00:00 2001
From: Nate McMaster <natemcmaster@users.noreply.github.com>
Date: Tue, 30 Jan 2018 15:22:31 -0800
Subject: [PATCH] Ensure the LZMA includes CLI global tools .nupkg,
 nupkg.sha512, and .nuspec files (#829)

---
 build/PackageArchive.targets           | 11 +--
 build/tasks/CreateNuGetV3FolderFeed.cs | 95 ++++++++++++++++++++++++++
 build/tasks/RepoTasks.tasks            |  1 +
 3 files changed, 99 insertions(+), 8 deletions(-)
 create mode 100644 build/tasks/CreateNuGetV3FolderFeed.cs

diff --git a/build/PackageArchive.targets b/build/PackageArchive.targets
index 4970e4bb95..5c3f4775de 100644
--- a/build/PackageArchive.targets
+++ b/build/PackageArchive.targets
@@ -55,19 +55,14 @@
     <RepoTasks.JoinItems
          Left="@(ArtifactInfo)" LeftKey="PackageId" LeftMetadata="*" LeftItemSpec="PackageId"
          Right="@(PackageArtifact->WithMetadataValue('LZMA', 'true')->WithMetadataValue('PackageType', 'DotnetTool'))">
-       <Output TaskParameter="JoinResult" ItemName="_DotnetToolPackages" />
+       <Output TaskParameter="JoinResult" ItemName="_DotnetToolPackageId" />
     </RepoTasks.JoinItems>
 
     <ItemGroup>
-      <_DotnetToolPackages Update="@(_DotnetToolPackages)">
-        <DestFileName>$([System.String]::new('%(PackageId).%(Version)').ToLowerInvariant())</DestFileName>
-        <DestDirName>$([System.String]::new('%(PackageId)/%(Version)').ToLowerInvariant())</DestDirName>
-      </_DotnetToolPackages>
+      <_DotnetToolPackages Include="$(MetapackageRestoreSource)%(_DotnetToolPackageId.Identity).%(_DotnetToolPackageId.Version).nupkg" />
     </ItemGroup>
 
-    <!-- No need to extract global CLI tool packages. They are extracted from the CliFallbackFolder into $HOME/.dotnet/tools by dotnet-install-tool. -->
-    <MakeDir Directories="$(FallbackStagingDir)\%(_DotnetToolPackages.DestDirName)\" Condition="'%(Identity)'!= ''" />
-    <Copy SourceFiles="$(MetapackageRestoreSource)%(_DotnetToolPackages.Identity).%(Version).nupkg" DestinationFiles="$(FallbackStagingDir)\%(DestDirName)\%(DestFileName).nupkg" Condition="'%(Identity)'!= ''" />
+    <RepoTasks.CreateNuGetV3FolderFeed Packages="@(_DotnetToolPackages)" DestinationFolder="$(FallbackStagingDir)" Overwrite="true" />
 
     <!-- Create the archive -->
     <Exec Command="$(ArchiverPath) -a $(FallbackOutputPath) $(FallbackStagingDir)" />
diff --git a/build/tasks/CreateNuGetV3FolderFeed.cs b/build/tasks/CreateNuGetV3FolderFeed.cs
new file mode 100644
index 0000000000..8d76e3f165
--- /dev/null
+++ b/build/tasks/CreateNuGetV3FolderFeed.cs
@@ -0,0 +1,95 @@
+// Copyright (c) .NET Foundation. All rights reserved.
+// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
+
+using System;
+using System.IO;
+using System.Security.Cryptography;
+using System.Text;
+using Microsoft.Build.Framework;
+using Microsoft.Build.Utilities;
+using NuGet.Packaging;
+using NuGet.Packaging.Core;
+using NuGet.Versioning;
+
+namespace RepoTasks
+{
+    /// <summary>
+    /// Layouts nuget packages as a V3 feed
+    /// </summary>
+    public class CreateNuGetV3FolderFeed : Task
+    {
+        [Required]
+        public ITaskItem[] Packages { get; set; }
+
+        [Required]
+        public string DestinationFolder { get; set; }
+
+        public bool Overwrite { get; set; }
+
+        public override bool Execute()
+        {
+            Directory.CreateDirectory(DestinationFolder);
+
+            foreach (var file in Packages)
+            {
+                PackageIdentity identity;
+                using (var reader = new PackageArchiveReader(file.ItemSpec))
+                {
+                    identity = reader.GetIdentity();
+                }
+
+                Log.LogMessage(MessageImportance.High, "Adding {0} to feed '{1}'", identity, DestinationFolder);
+
+                var packageFolder = Path.Combine(DestinationFolder, identity.Id.ToLowerInvariant(), identity.Version.ToNormalizedString());
+                var nuspecFile = Path.Combine(packageFolder, $"{identity.Id.ToLowerInvariant()}.{identity.Version.ToNormalizedString()}.nuspec");
+                var nupkgFile = Path.Combine(packageFolder, $"{identity.Id.ToLowerInvariant()}.{identity.Version.ToNormalizedString()}.nupkg");
+                var sha512File = Path.Combine(packageFolder,  $"{identity.Id.ToLowerInvariant()}.{identity.Version.ToNormalizedString()}.nupkg.sha512");
+
+                if (!Overwrite && File.Exists(nuspecFile))
+                {
+                    Log.LogError("File already exists: {0}", nuspecFile);
+                    continue;
+                }
+                if (!Overwrite && File.Exists(nupkgFile))
+                {
+                    Log.LogError("File already exists: {0}", nupkgFile);
+                    continue;
+                }
+                if (!Overwrite && File.Exists(sha512File))
+                {
+                    Log.LogError("File already exists: {0}", sha512File);
+                    continue;
+                }
+
+                Directory.CreateDirectory(packageFolder);
+                using (var reader = new PackageArchiveReader(file.ItemSpec))
+                using (var nuspec = File.Create(nuspecFile))
+                using (var metadata = reader.GetNuspec())
+                {
+                    Log.LogMessage("Creating {0}", nuspecFile);
+                    metadata.CopyTo(nuspec);
+                }
+
+                Log.LogMessage("Copying {0}", nupkgFile);
+                File.Copy(file.ItemSpec, nupkgFile);
+                Log.LogMessage("Creating {0}", sha512File);
+                File.WriteAllText(sha512File, GetFileHash(file.ItemSpec));
+            }
+
+            return !Log.HasLoggedErrors;
+        }
+
+        private static string GetFileHash(string filePath)
+        {
+            byte[] hash;
+
+            using (var algorithm = new SHA512Managed())
+            using (var stream = File.OpenRead(filePath))
+            {
+                hash = algorithm.ComputeHash(stream);
+            }
+
+            return Convert.ToBase64String(hash);
+        }
+    }
+}
diff --git a/build/tasks/RepoTasks.tasks b/build/tasks/RepoTasks.tasks
index 51d38ac01d..389a0f71f2 100644
--- a/build/tasks/RepoTasks.tasks
+++ b/build/tasks/RepoTasks.tasks
@@ -11,6 +11,7 @@
   <UsingTask TaskName="RepoTasks.ConsolidateManifests" AssemblyFile="$(_RepoTaskAssembly)" />
   <UsingTask TaskName="RepoTasks.CopyPackagesToSplitFolders" AssemblyFile="$(_RepoTaskAssembly)" />
   <UsingTask TaskName="RepoTasks.CreateCommonManifest" AssemblyFile="$(_RepoTaskAssembly)" />
+  <UsingTask TaskName="RepoTasks.CreateNuGetV3FolderFeed" AssemblyFile="$(_RepoTaskAssembly)" />
   <UsingTask TaskName="RepoTasks.GenerateRestoreSourcesPropsFile" AssemblyFile="$(_RepoTaskAssembly)" />
   <UsingTask TaskName="RepoTasks.JoinItems" AssemblyFile="$(_RepoTaskAssembly)" />
   <UsingTask TaskName="RepoTasks.OrderBy" AssemblyFile="$(_RepoTaskAssembly)" />