/*
    Copyright (C) 2014-2019 de4dot@gmail.com

    This file is part of dnSpy

    dnSpy is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    dnSpy is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with dnSpy.  If not, see <http://www.gnu.org/licenses/>.
*/

using System;
using System.Collections.Generic;
using System.ComponentModel.Composition;
using System.Linq;
using dnSpy.AsmEditor.Assembly;
using dnSpy.AsmEditor.SaveModule;
using dnSpy.AsmEditor.UndoRedo;
using dnSpy.Contracts.Documents;
using dnSpy.Contracts.Documents.TreeView;
using dnSpy.Contracts.Menus;

namespace dnSpy.AsmEditor.Commands {
	[ExportMenuItem(OwnerGuid = MenuConstants.APP_MENU_FILE_GUID, Header = "res:CloseOldInMemoryModules", Group = MenuConstants.GROUP_APP_MENU_FILE_OPEN, Order = 50)]
	sealed class CloseOldInMemoryModulesCommand : MenuItemBase {
		readonly IDocumentTreeView documentTreeView;
		readonly Lazy<IUndoCommandService> undoCommandService;
		readonly Lazy<IDocumentSaver> documentSaver;
		readonly IAppService appService;

		[ImportingConstructor]
		CloseOldInMemoryModulesCommand(IDocumentTreeView documentTreeView, Lazy<IUndoCommandService> undoCommandService, Lazy<IDocumentSaver> documentSaver, IAppService appService) {
			this.documentTreeView = documentTreeView;
			this.undoCommandService = undoCommandService;
			this.documentSaver = documentSaver;
			this.appService = appService;
		}

		AssemblyDocumentNode[]? GetNodes() {
			var nodes = new List<AssemblyDocumentNode>();
			foreach (var node in documentTreeView.TreeView.Root.DataChildren.OfType<AssemblyDocumentNode>()) {
				var doc = node.Document;
				if ((doc as IInMemoryDocument)?.IsActive == false)
					nodes.Add(node);
			}
			return nodes.Count == 0 ? null : nodes.ToArray();
		}

		public override bool IsEnabled(IMenuItemContext context) => GetNodes() is not null;
		public override void Execute(IMenuItemContext context) {
			var nodes = GetNodes();
			if (nodes is not null)
				RemoveAssemblyCommand.Execute(undoCommandService, documentSaver, appService, nodes);
		}
	}
}