/*
 * Copyright (c) 2010, Geoff Norton <gnorton@novell.com>
 * Copyright (c) 2010, JB Evain <jbevain@novell.com>
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
using System;
using System.Collections.Generic;
using System.IO;
using System.Text;

using Mono.Cecil;
using Mono.Linker;

namespace Xamarin.Bundler {
	public partial class MonoMacResolver : CoreResolver {
		public List <string> CommandLineAssemblies { get; set; }
		public List<Exception> Exceptions = new List<Exception> ();
		public string GlobalAssemblyCache;
		public string[] SystemFrameworkDirectories;

		public AssemblyDefinition GetAssembly (string fileName)
		{
			return Resolve (new AssemblyNameReference (Path.GetFileNameWithoutExtension (fileName), null), new ReaderParameters { AssemblyResolver = this });
		}

		public override AssemblyDefinition Resolve (AssemblyNameReference reference, ReaderParameters parameters)
		{
			var name = reference.Name;

			AssemblyDefinition assembly;
			if (cache.TryGetValue (name, out assembly))
				return assembly;
			
			if (CommandLineAssemblies != null && CommandLineAssemblies.Count > 0) {
				string cmdasm = CommandLineAssemblies.Find (t => {
					if (String.IsNullOrEmpty (t))
						return false;
					return String.Compare (name, Path.GetFileNameWithoutExtension (t), StringComparison.Ordinal) == 0;
				});
				assembly = String.IsNullOrEmpty (cmdasm) ? null : Load (cmdasm);
				if (assembly != null)
					return assembly;
			}

			if (ArchDirectory != null) {
				assembly = SearchDirectory (name, ArchDirectory);
				if (assembly != null)
					return assembly;
			}
			
			assembly = SearchDirectory (name, FrameworkDirectory);
			if (assembly != null)
				return assembly;

			if (FrameworkDirectory != null) {
				var pclPath = Path.Combine (FrameworkDirectory, "Facades");
				assembly = SearchDirectory (name, pclPath);
				if (assembly != null)
					return assembly;
			}

			assembly = SearchDirectory (name, RootDirectory, ".exe");
			if (assembly != null)
				return assembly;

			if (!string.IsNullOrEmpty (GlobalAssemblyCache)) {
				var gac_folder = new StringBuilder ()
					.Append (reference.Version)
					.Append ("__");

				for (int i = 0; i < reference.PublicKeyToken.Length; i++)
					gac_folder.Append (reference.PublicKeyToken [i].ToString ("x2"));

				var gac_path = Path.Combine (GlobalAssemblyCache, reference.Name, gac_folder.ToString (), reference.Name + ".dll");
				if (File.Exists (gac_path)) {
					if (Driver.IsUnifiedFullXamMacFramework)
						ErrorHelper.Warning (176, Errors.MX0176, reference.ToString (), gac_path);
					return Load (gac_path);
				}
			}

			if (SystemFrameworkDirectories?.Length > 0) {
				foreach (var dir in SystemFrameworkDirectories) {
					assembly = SearchDirectory (reference.Name, dir);
					if (assembly != null) {
						if (Driver.IsUnifiedFullXamMacFramework)
							ErrorHelper.Warning (176, Errors.MX0176, reference.ToString (), assembly.MainModule.FileName);
						return assembly;
					}
				}
			}

			return null;
		}

		public override void Configure ()
		{
			base.Configure ();

			if (!Driver.UseLegacyAssemblyResolution && (Driver.IsUnifiedFullSystemFramework || Driver.IsUnifiedFullXamMacFramework)) {
				// We need to look in the GAC/System mono for both FullSystem and FullXamMac, because that's
				// how we've been resolving assemblies in the past (Cecil has a fall-back mode where it looks
				// in the GAC, and we never disabled that, meaning that we always looked in the GAC if failing
				// to resolve from somewhere else). This makes it explicit that we look in the GAC, and we
				// now also warn when using FullXamMac and finding assemblies in the GAC.
				GlobalAssemblyCache = Path.Combine (Driver.SystemMonoDirectory, "lib", "mono", "gac");
				var framework_dir = Path.GetDirectoryName (typeof (object).Module.FullyQualifiedName);
				SystemFrameworkDirectories = new [] {
						framework_dir,
						Path.Combine (framework_dir, "Facades")
					};
			}
		}
	}

	public class MonoMacAssemblyResolver : AssemblyResolver {
		public MonoMacResolver Resolver;

		public MonoMacAssemblyResolver (MonoMacResolver resolver)
			: base (resolver.cache ?? new Dictionary<string, AssemblyDefinition> ())
		{
			this.Resolver = resolver;
		}

		public override AssemblyDefinition Resolve (AssemblyNameReference name, ReaderParameters parameters)
		{
			return Resolver.Resolve (name, parameters);
		}
	}
}