using System.Collections.Generic;
using UnityEditor;
using UnityEngine;
using System.IO;
using System.Diagnostics;
using UnityEditor.VersionControl;

namespace Wwise.Wooduan.Tools
{
	internal enum ScriptType
	{
		CSharp,
		JavaScript,
		Lua,
		Python,
		DLL
	}
	
	public class AkWooduanScriptMigration : EditorWindow
	{
		private class FileComparisonData
		{
			public string SourceFilePath;
			public string TargetFilePath;
			public bool WillCopy = true;
			public FileInfo SourceFile;
			public FileInfo TargetFile;
		}

		private string _sourceRootPath;
		private string _targetRootPath;
		private string[] _sourceScripts;
		private ScriptType _scriptType = ScriptType.CSharp;
		private List<FileComparisonData> _changeList = new List<FileComparisonData>();

		#region GUI
		private bool _selectAll = true;
		private bool _selectNew = true;
		private bool _selectLater = true;
		private bool _selectLarger = true;
		private bool _includeExtensions;
		
		private Vector2 _scrollPosition;

		private void OnEnable()
		{
			_targetRootPath = AkWwiseEditorSettings.AkWooduanLibraryFullPath;
		}

		private void OnGUI()
		{
			GUILayout.Label("This tool migrates scripts between games or branches.");			                
			using (new EditorGUILayout.VerticalScope(GUI.skin.box))
			{				
				DisplaySearchPath(ref _sourceRootPath, "Source");
				DisplaySearchPath(ref _targetRootPath, "Target");
				GUILayout.BeginHorizontal();
				EditorGUILayout.LabelField("Script Type", GUILayout.Width(100));
				_scriptType = (ScriptType) EditorGUILayout.EnumPopup(_scriptType, GUILayout.Width(100));
				_includeExtensions = GUILayout.Toggle(_includeExtensions, "Include Extensions");
				if (GUILayout.Button("Search Script Difference")) 
					SearchScriptDifference();
				GUILayout.EndHorizontal();	
			}

			if (_changeList.Count == 0) return;
			
			using (new EditorGUILayout.VerticalScope(GUI.skin.box))
			{				
				GUILayout.BeginHorizontal();
				GUILayout.Label("Script Name", GUILayout.Width(200));
				GUILayout.Label("Source Size", GUILayout.Width(80));
				GUILayout.Label("Source Modified", GUILayout.Width(120));
				GUILayout.Label("Target Size", GUILayout.Width(80));
				GUILayout.Label("Target Modified", GUILayout.Width(120));
				GUILayout.Label("Diff");
				GUILayout.EndHorizontal();
				_scrollPosition =  EditorGUILayout.BeginScrollView(_scrollPosition, GUILayout.Width(680), GUILayout.Height(200));
				foreach (var data in _changeList)
				{
					GUILayout.BeginHorizontal();
					data.WillCopy = GUILayout.Toggle(data.WillCopy,  Path.GetFileName(data.SourceFilePath), GUILayout.Width(200));
					GUILayout.Label(data.SourceFile.Length.ToString(), GUILayout.Width(80));
					GUILayout.Label(data.SourceFile.LastWriteTime.ToShortDateString(), GUILayout.Width(120));					
					if (data.TargetFile != null)
					{
						GUILayout.Label(data.TargetFile.Length.ToString(), GUILayout.Width(80));
						GUILayout.Label(data.TargetFile.LastWriteTime.ToShortDateString(), GUILayout.Width(120));							
					}
					else
						GUILayout.Label("File does not exist at target", GUILayout.Width(204));

					if (data.TargetFile != null && GUILayout.Button("...", GUILayout.Width(30))) 
						Process.Start("p4merge", data.SourceFilePath + " " + data.TargetFilePath);
					GUILayout.EndHorizontal();
				}
				EditorGUILayout.EndScrollView();
			}
			
			GUILayout.BeginHorizontal();
			if (_selectAll != GUILayout.Toggle(_selectAll, "Select All"))
			{
				_selectAll = SelectScripts(!_selectAll, 0);
				_selectNew = _selectLater = _selectLarger = _selectAll;
			}
			if (_selectNew != GUILayout.Toggle(_selectNew, "Select New Scripts")) _selectNew = SelectScripts(!_selectNew, 1);	
			if (_selectLater != GUILayout.Toggle(_selectLater, "Select Later Modified")) _selectLater = SelectScripts(!_selectLater, 2);
			if (_selectLarger != GUILayout.Toggle(_selectLarger, "Select Larger")) _selectLarger = SelectScripts(!_selectLarger, 3);									
			GUILayout.EndHorizontal();
			
			if (GUILayout.Button("Migrate Scripts")) 
				MigrateScripts();
		}

		private void DisplaySearchPath(ref string searchPath, string label)
		{
			EditorGUILayout.BeginHorizontal();
			if (GUILayout.Button(label, GUILayout.Width(60)))
				searchPath = EditorUtility.OpenFolderPanel("Root Folder", searchPath, "");
			searchPath = GUILayout.TextField(searchPath);
			EditorGUILayout.EndHorizontal();
		}

		private bool SelectScripts(bool select, int action)
		{
			foreach (var data in _changeList)
			{
				if (action == 0) data.WillCopy = select;
				if (data.TargetFile == null)
				{
					if (action == 1) data.WillCopy = select;
				}
				else if (action == 2)
				{
					if (data.SourceFile.LastWriteTime > data.TargetFile.LastWriteTime) data.WillCopy = select;
				}	
				else if (action == 3)
				{
					if (data.SourceFile.Length > data.TargetFile.Length) data.WillCopy = select;
				}							
			}
			return select;
		}
		#endregion
		
		private bool ValidateDirectories()
		{
			if (string.IsNullOrEmpty(_sourceRootPath) || string.IsNullOrEmpty(_targetRootPath))
			{
				EditorUtility.DisplayDialog("Error", "Please specify source and target directory!", "OK");
				return false;
			}
			if (!Directory.Exists(_sourceRootPath) || !Directory.Exists(_targetRootPath))
			{
				EditorUtility.DisplayDialog("Error", "Please select valid source and target directory!", "OK");
				return false;
			}
			return true;
		}

		private string GetScriptExtensionName()
		{
			switch (_scriptType)
			{
				case ScriptType.CSharp:
					return "*.cs";
				case ScriptType.JavaScript:
					return "*.js";
				case ScriptType.Lua:
					return "*.lua";
				case ScriptType.Python:
					return "*.py";
				case ScriptType.DLL:
					return "*.dll";
			}
			return "*.*";
		}

		private void SearchScriptDifference()
		{
			if (!ValidateDirectories()) return;
			_sourceScripts = Directory.GetFiles(_sourceRootPath, GetScriptExtensionName(), SearchOption.AllDirectories);
			_changeList.Clear();
			for (var i = 0; i < _sourceScripts.Length; i++)
			{
				var scriptPath = _sourceScripts[i];
				EditorUtility.DisplayProgressBar("Comparing Scripts", scriptPath, (i + 1f) / _sourceScripts.Length);
				if (!_includeExtensions && scriptPath.Contains("Extensions")) continue;
				var cd = new FileComparisonData
				{
					SourceFilePath = scriptPath,
					TargetFilePath = scriptPath.Replace(_sourceRootPath, _targetRootPath),
					SourceFile = new FileInfo(scriptPath)
				};

				if (File.Exists(cd.TargetFilePath))
				{
					cd.TargetFile = new FileInfo(cd.TargetFilePath);
					if (!CompareFiles(cd.SourceFile, cd.TargetFile))
						_changeList.Add(cd);
				}
				else
					_changeList.Add(cd);
			}
			EditorUtility.ClearProgressBar();

			EditorUtility.DisplayDialog("Check Result", _changeList.Count == 0 ? 
				"All files are identical!" : 
				_changeList.Count + " files are different.", "OK");
		}

		private bool CompareFiles(FileInfo source, FileInfo target)
		{
			var text1 = File.ReadAllText(source.FullName);
			var text2 = File.ReadAllText(target.FullName);
			return text1 == text2;
		}

		private void MigrateScripts()
		{
			if (_changeList.Count == 0)
			{
				EditorUtility.DisplayDialog("Invalid Operation", "Please check for difference first!", "OK");
				return;
			}

			var copyCount = 0;
			foreach (var data in _changeList)
			{
				if (!data.WillCopy) continue;
				var folderPath = Path.GetDirectoryName(data.TargetFilePath);
				AkWooduanScriptingTools.CheckDirectoryExist(folderPath);
				if (data.TargetFile != null && data.TargetFile.IsReadOnly)
				{
					if (Provider.isActive)
						Provider.Checkout(data.TargetFile.FullName, CheckoutMode.Asset);					
					data.TargetFile.IsReadOnly = false;
				}
				File.Copy(data.SourceFilePath, data.TargetFilePath, true);
				copyCount++;
			}

			EditorUtility.DisplayDialog("Success!", "Copied " + copyCount + " files to target!", "OK");
			_changeList.Clear();
			AssetDatabase.Refresh();
			Close();
		}
	}
}