dotnet-installer/src/Microsoft.DotNet.ProjectJsonMigration/MigrationBackupPlan.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.Internal.ProjectModel.Utilities;

namespace Microsoft.DotNet.ProjectJsonMigration
{
    internal class MigrationBackupPlan
    {
        private readonly FileInfo globalJson;

        public MigrationBackupPlan(
            DirectoryInfo projectDirectory,
            DirectoryInfo workspaceDirectory,
            Func<DirectoryInfo, IEnumerable<FileInfo>> getFiles = null)
        {
            if (projectDirectory == null)
            {
                throw new ArgumentNullException(nameof(projectDirectory));
            }
            if (workspaceDirectory == null)
            {
                throw new ArgumentNullException(nameof(workspaceDirectory));
            }

            globalJson = new FileInfo(Path.Combine(
                           workspaceDirectory.FullName,
                           "global.json"));

            projectDirectory = new DirectoryInfo(projectDirectory.FullName.EnsureTrailingSlash());
            workspaceDirectory = new DirectoryInfo(workspaceDirectory.FullName.EnsureTrailingSlash());

            RootBackupDirectory = new DirectoryInfo(
                Path.Combine(
                        workspaceDirectory.Parent.FullName,
                        "backup")
                    .EnsureTrailingSlash());

            ProjectBackupDirectory = new DirectoryInfo(
                Path.Combine(
                        RootBackupDirectory.FullName,
                        projectDirectory.Name)
                    .EnsureTrailingSlash());

            var relativeDirectory = PathUtility.GetRelativePath(
                workspaceDirectory.FullName,
                projectDirectory.FullName);

            getFiles = getFiles ??
                       (dir => dir.EnumerateFiles());

            FilesToMove = getFiles(projectDirectory)
               .Where(f => f.Name == "project.json"
                        || f.Extension == ".xproj"
                        || f.FullName.EndsWith(".xproj.user")
                        || f.FullName.EndsWith(".lock.json"));
        }

        public DirectoryInfo ProjectBackupDirectory { get; }

        public DirectoryInfo RootBackupDirectory { get; }

        public IEnumerable<FileInfo> FilesToMove { get; }

        public void PerformBackup()
        {
            if (globalJson.Exists)
            {
                PathUtility.EnsureDirectoryExists(RootBackupDirectory.FullName);

                globalJson.MoveTo(Path.Combine(
                    ProjectBackupDirectory.Parent.FullName,
                    globalJson.Name));
            }

            PathUtility.EnsureDirectoryExists(ProjectBackupDirectory.FullName);

            foreach (var file in FilesToMove)
            {
                file.MoveTo(
                    Path.Combine(
                        ProjectBackupDirectory.FullName, file.Name));
            }
        }
    }
}