// 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.
 
using System.Collections.Generic;
using Microsoft.CodeAnalysis;
 
namespace Microsoft.VisualStudio.LanguageServices.Implementation.ProjectSystem;
 
internal partial class VisualStudioWorkspaceImpl
{
    private class RemoveAdditionalDocumentUndoUnit : AbstractRemoveDocumentUndoUnit
    {
        public RemoveAdditionalDocumentUndoUnit(
            VisualStudioWorkspaceImpl workspace,
            DocumentId documentId)
            : base(workspace, documentId)
        {
        }
 
        protected override IReadOnlyList<DocumentId> GetDocumentIds(Project fromProject)
            => fromProject.AdditionalDocumentIds;
 
        protected override TextDocument? GetDocument(Solution currentSolution)
            => currentSolution.GetAdditionalDocument(this.DocumentId);
    }
}