dotnet-installer/src/Microsoft.DotNet.Compiler.Common/LibraryExporterExtensions.cs

// Copyright (c) .NET Foundation and contributors. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using Microsoft.DotNet.ProjectModel.Compilation;
using Microsoft.DotNet.ProjectModel.Graph;

namespace Microsoft.DotNet.Cli.Compiler.Common
{
	public static class LibraryExporterExtensions
	{
		public static IEnumerable<LibraryExport> GetAllProjectTypeDependencies(this LibraryExporter exporter)
		{
			return
				exporter.GetDependencies(LibraryType.Project)
					.Concat(exporter.GetDependencies(LibraryType.MSBuildProject));
		}

		public static void CopyTo(this IEnumerable<LibraryAsset> assets, string destinationPath)
		{
			if (!Directory.Exists(destinationPath))
			{
				Directory.CreateDirectory(destinationPath);
			}

			foreach (var asset in assets)
			{
				var file = Path.Combine(destinationPath, Path.GetFileName(asset.ResolvedPath));
				File.Copy(asset.ResolvedPath, file, overwrite: true);
				RemoveFileAttribute(file, FileAttributes.ReadOnly);
			}
		}

		private static void RemoveFileAttribute(String file, FileAttributes attribute)
		{
			if (File.Exists(file))
			{
				var fileAttributes = File.GetAttributes(file);
				if ((fileAttributes & attribute) == attribute)
				{
					File.SetAttributes(file, fileAttributes & ~attribute);
				}
			}
		}

		public static void StructuredCopyTo(this IEnumerable<LibraryAsset> assets, string destinationPath, string tempLocation)
		{
			if (!Directory.Exists(destinationPath))
			{
				Directory.CreateDirectory(destinationPath);
			}

			foreach (var asset in assets)
			{
				var targetName = ResolveTargetName(destinationPath, asset);
				var transformedFile = asset.GetTransformedFile(tempLocation);

				File.Copy(transformedFile, targetName, overwrite: true);
				RemoveFileAttribute(targetName, FileAttributes.ReadOnly);
			}
		}

		private static string ResolveTargetName(string destinationPath, LibraryAsset asset)
		{
			string targetName;
			if (!string.IsNullOrEmpty(asset.RelativePath))
			{
				targetName = Path.Combine(destinationPath, asset.RelativePath);
				var destinationAssetPath = Path.GetDirectoryName(targetName);

				if (!Directory.Exists(destinationAssetPath))
				{
					Directory.CreateDirectory(destinationAssetPath);
				}
			}
			else
			{
				targetName = Path.Combine(destinationPath, Path.GetFileName(asset.ResolvedPath));
			}
			return targetName;
		}
	}
}