dotnet-installer/test/Microsoft.DotNet.Tools.Tests.Utilities/TempFileSystem/TempRoot.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.Runtime.CompilerServices;
 
namespace Microsoft.DotNet.Tools.Test.Utilities
{
    public sealed class TempRoot : IDisposable
    {
        private static readonly bool DoDispose;
        private readonly List<IDisposable> _temps = new List<IDisposable>();
        public static readonly string Root;
 
        static TempRoot()
        {
            var persistedRoot = Environment.GetEnvironmentVariable("TEST_ARTIFACTS");

            if (string.IsNullOrWhiteSpace(persistedRoot))
            {
                Root = Path.Combine(Path.GetTempPath(), "DotnetCLITests");
                DoDispose = true;
            }
            else
            {
                Root = persistedRoot;
                DoDispose = false;
            }

            Directory.CreateDirectory(Root);
        }
 
        public void Dispose()
        {
            if (!DoDispose || _temps == null) return;

            DisposeAll(_temps);
            _temps.Clear();
        }
 
        private static void DisposeAll(IEnumerable<IDisposable> temps)
        {
            foreach (var temp in temps)
            {
                try
                {
                    temp?.Dispose();
                }
                catch
                {
                    // ignore
                }
            }
        }
 
        public DisposableDirectory CreateDirectory()
        {
            var dir = new DisposableDirectory(this);
            _temps.Add(dir);
            return dir;
        }

        public TempFile CreateFile(string prefix = null, string extension = null, string directory = null, [CallerFilePath] string callerSourcePath = null, [CallerLineNumber] int callerLineNumber = 0) => AddFile(new DisposableFile(prefix, extension, directory, callerSourcePath, callerLineNumber));

        public DisposableFile AddFile(DisposableFile file)
        {
            _temps.Add(file);
            return file;
        }
 
        internal static void CreateStream(string fullPath)
        {
            using (var file = new FileStream(fullPath, FileMode.CreateNew)) { }
        }
    }
}