/*
    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.ComponentModel.Composition;
using dnlib.DotNet;
using dnSpy.Contracts.Metadata;

namespace dnSpy.Debugger.DotNet.Metadata {
	[ExportModuleIdFactoryProvider(ModuleIdFactoryProviderConstants.OrderDebugger)]
	sealed class ModuleIdFactoryProviderImpl : IModuleIdFactoryProvider {
		readonly DsDocumentProvider documentProvider;

		[ImportingConstructor]
		ModuleIdFactoryProviderImpl(DsDocumentProvider documentProvider) => this.documentProvider = documentProvider;

		public IModuleIdFactory? Create() => new ModuleIdFactory(documentProvider);

		sealed class ModuleIdFactory : IModuleIdFactory {
			readonly DsDocumentProvider documentProvider;

			public ModuleIdFactory(DsDocumentProvider documentProvider) => this.documentProvider = documentProvider;

			public ModuleId? Create(ModuleDef module) {
				foreach (var info in documentProvider.DocumentInfos) {
					if (info.Document is IModuleIdHolder midHolder && info.Document.ModuleDef == module)
						return midHolder.ModuleId;
				}
				return null;
			}
		}
	}
}