Umbraco-CMS/src/Umbraco.Web/Install/InstallStatusTracker.cs

using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Newtonsoft.Json;
using Umbraco.Core;
using Umbraco.Core.IO;
using Umbraco.Web.Install.Models;

namespace Umbraco.Web.Install
{
    /// <summary>
    /// An internal in-memory status tracker for the current installation
    /// </summary>
    internal static class InstallStatusTracker
    {

        private static ConcurrentHashSet<InstallTrackingItem> _steps = new ConcurrentHashSet<InstallTrackingItem>();

        private static string GetFile(Guid installId)
        {
            var file = IOHelper.MapPath("~/App_Data/TEMP/Install/"
                                        + "install_"
                                        + installId.ToString("N")
                                        + ".txt");
            return file;
        }

        public static void Reset()
        {
            _steps = new ConcurrentHashSet<InstallTrackingItem>();
            ClearFiles();
        }

        public static void ClearFiles()
        {   
            var dir = IOHelper.MapPath("~/App_Data/TEMP/Install/");
            if (Directory.Exists(dir))
            {
                var files = Directory.GetFiles(dir);
                foreach (var f in files)
                {
                    File.Delete(f);
                }
            }
            else
            {
                Directory.CreateDirectory(dir);
            }
        }

        public static IEnumerable<InstallTrackingItem> InitializeFromFile(Guid installId)
        {
            //check if we have our persisted file and read it
            var file = GetFile(installId);
            if (File.Exists(file))
            {
                var deserialized = JsonConvert.DeserializeObject<IEnumerable<InstallTrackingItem>>(
                    File.ReadAllText(file));
                foreach (var item in deserialized)
                {
                    _steps.Add(item);
                }
            }
            else
            {
                throw new InvalidOperationException("Cannot initialize from file, the installation file with id " + installId + " does not exist");
            }
            return new List<InstallTrackingItem>(_steps);
        }

        public static IEnumerable<InstallTrackingItem> Initialize(Guid installId, IEnumerable<InstallSetupStep> steps)
        {
            //if there are no steps in memory
            if (_steps.Count == 0)
            {
                //check if we have our persisted file and read it
                var file = GetFile(installId);
                if (File.Exists(file))
                {
                    var deserialized = JsonConvert.DeserializeObject<IEnumerable<InstallTrackingItem>>(
                        File.ReadAllText(file));
                    foreach (var item in deserialized)
                    {
                        _steps.Add(item);
                    }
                }
                else
                {
                    ClearFiles();

                    //otherwise just create the steps in memory (brand new install)
                    foreach (var step in steps.OrderBy(x => x.ServerOrder))
                    {
                        _steps.Add(new InstallTrackingItem(step.Name, step.ServerOrder));
                    }
                    //save the file
                    var serialized = JsonConvert.SerializeObject(new List<InstallTrackingItem>(_steps));
                    Directory.CreateDirectory(Path.GetDirectoryName(file));
                    File.WriteAllText(file, serialized);
                }
            }
            else
            {
                //ensure that the file exists with the current install id
                var file = GetFile(installId);
                if (File.Exists(file) == false)
                {
                    ClearFiles();

                    //save the correct file
                    var serialized = JsonConvert.SerializeObject(new List<InstallTrackingItem>(_steps));
                    Directory.CreateDirectory(Path.GetDirectoryName(file));
                    File.WriteAllText(file, serialized);
                }
            }

            return new List<InstallTrackingItem>(_steps);
        }

        public static void SetComplete(Guid installId, string name, IDictionary<string, object> additionalData = null)
        {
            var trackingItem = _steps.Single(x => x.Name == name);
            if (additionalData != null)
            {
                trackingItem.AdditionalData = additionalData;
            }
            trackingItem.IsComplete = true;
            
            //save the file
            var file = GetFile(installId);
            var serialized = JsonConvert.SerializeObject(new List<InstallTrackingItem>(_steps));
            File.WriteAllText(file, serialized);
        }

        public static IEnumerable<InstallTrackingItem> GetStatus()
        {
            return new List<InstallTrackingItem>(_steps).OrderBy(x => x.ServerOrder);
        } 
    }
}