|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
#nullable disable
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
namespace Microsoft.CodeAnalysis.Editing;
/// <summary>
/// An editor for making changes to multiple documents in a solution.
/// </summary>
public class SolutionEditor(Solution solution)
{
private readonly Dictionary<DocumentId, DocumentEditor> _documentEditors = [];
/// <summary>
/// The <see cref="Solution"/> that was specified when the <see cref="SolutionEditor"/> was constructed.
/// </summary>
public Solution OriginalSolution => solution;
/// <summary>
/// Gets the <see cref="DocumentEditor"/> for the corresponding <see cref="DocumentId"/>.
/// </summary>
public async Task<DocumentEditor> GetDocumentEditorAsync(DocumentId id, CancellationToken cancellationToken = default)
{
if (!_documentEditors.TryGetValue(id, out var editor))
{
editor = await DocumentEditor.CreateAsync(solution.GetDocument(id), cancellationToken).ConfigureAwait(false);
_documentEditors.Add(id, editor);
}
return editor;
}
/// <summary>
/// Returns the changed <see cref="Solution"/>.
/// </summary>
public Solution GetChangedSolution()
{
var changedSolution = solution;
foreach (var docEd in _documentEditors.Values)
{
var currentDoc = changedSolution.GetDocument(docEd.OriginalDocument.Id);
var newDoc = currentDoc.WithSyntaxRoot(docEd.GetChangedRoot());
changedSolution = newDoc.Project.Solution;
}
return changedSolution;
}
}
|