using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Input;
using System.Windows.Media;
using Microsoft.WindowsAPICodePack.Dialogs;

namespace CarConverter {
    /// <summary>
    /// Interaction logic for MainWindow.xaml
    /// </summary>
    public partial class MainWindow {
        private readonly string _optionFilePath = Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData) +
                                                 Path.DirectorySeparatorChar + "CarConverter" +
                                                 Path.DirectorySeparatorChar + "settings.ini";
        private const string TempFolder = "./Temp/Vehicle/";
        private readonly string[] _streamLocations = new[] { "vehicles.rpf", "vehiclemods" };

        private readonly List<FileInfo> _metaFiles = new List<FileInfo>();
        private readonly List<FileInfo> _streamFiles = new List<FileInfo>();

        public MainWindow() {
            InitializeComponent();

            if (File.Exists(_optionFilePath))
                OutPath.Content = File.ReadAllText(_optionFilePath);
            else {
                Directory.CreateDirectory(_optionFilePath.Replace("settings.ini", ""));
                OutPath.Content = Environment.GetFolderPath(Environment.SpecialFolder.Desktop);
            }
        }

        private void OnDrop(object sender, DragEventArgs e) {
            if (!e.Data.GetDataPresent(DataFormats.FileDrop)) return;

            Clear(null, null);
            string dlcFile = ((string[])e.Data.GetData(DataFormats.FileDrop))[0];
            Utils.UnpackDlcFile(dlcFile, TempFolder, Compress.IsChecked.Value);
            UpdateUi(TempFolder);
        }

        private void UpdateUi(string folder) {
            var info = new DirectoryInfo(folder);
            AddSubFiles(info, FolderView);
            UpdateOutLists();
        }

        private void AddSubFiles(DirectoryInfo info, ItemsControl root) {
            foreach (var dir in info.EnumerateDirectories()) {
                var item = new TreeViewItem { Header = dir.Name, DataContext = dir.FullName };
                AddSubFiles(dir, item);
                root.Items.Add(item);
            }
            
            foreach (var file in info.EnumerateFiles()) {
                
                if (file.Extension.Equals(".meta") && _metaFiles.All(meta => meta.Name != file.Name))
                    _metaFiles.Add(file);
                
                if (_streamLocations.Any(url => file.FullName.Contains(url)))
                    _streamFiles.Add(file);

                var item = new TreeViewItem { Header = file.Name, DataContext = file.FullName };
                item.MouseDoubleClick += (sender, args) => ListInteraction(sender, null);
                root.Items.Add(item);
            }
        }

        private void UpdateOutLists() {
            StreamFiles.Items.Clear();
            MetaFiles.Items.Clear();

            foreach (var meta in _metaFiles) {
                var item = new ListViewItem { Content = meta.Name, DataContext = meta.FullName };
                item.KeyDown += ListInteraction;
                MetaFiles.Items.Add(item);
            }

            foreach (var stream in _streamFiles) {
                var item = new ListViewItem { Content = stream.Name, DataContext = stream.FullName };
                item.KeyDown += ListInteraction;
                StreamFiles.Items.Add(item);
            }
        }

        private void Clear(object sender, RoutedEventArgs e) {
            _metaFiles.Clear();
            _streamFiles.Clear();
            MetaFiles.Items.Clear();
            StreamFiles.Items.Clear();
            FolderView.Items.Clear();
            
            if (Directory.Exists(TempFolder))
                Directory.Delete(TempFolder, true);
        }

        private void OnClose(object sender, CancelEventArgs e) => Clear(sender, null);

        private void ListInteraction(object sender, KeyEventArgs e) {
            if (sender is ListViewItem) {
                if (!new[] { Key.Return, Key.Delete }.Contains(e.Key)) return;
                var item = sender as ListViewItem;

                if (item.Parent.Equals(MetaFiles)) {
                    _metaFiles.Remove(_metaFiles.Single(meta => meta.FullName.Equals(item.DataContext)));
                    MetaFiles.Items.Remove(item);
                    return;
                }
                
                if (item.Parent.Equals(StreamFiles)) {
                    _streamFiles.Remove(_streamFiles.Single(stream => stream.FullName.Equals(item.DataContext)));
                    StreamFiles.Items.Remove(item);
                    return;
                }
            }

            if (sender is TreeViewItem) {
                var item = sender as TreeViewItem;

                if ((item.DataContext as string).EndsWith(".meta")) {
                    if (!_metaFiles.Any(file => file.FullName.Equals(item.DataContext as string)))
                        _metaFiles.Add(new FileInfo(item.DataContext as string));
                }
                else {
                    if (!_streamFiles.Any(file => file.FullName.Equals(item.DataContext as string)))
                        _streamFiles.Add(new FileInfo(item.DataContext as string));
                }
                
                UpdateOutLists();
            }
        }

        private void Build(object sender, RoutedEventArgs e) {
            var info = Directory.CreateDirectory((string)OutPath.Content + Path.DirectorySeparatorChar + CarName.Text);
            File.WriteAllText(info.FullName + Path.DirectorySeparatorChar + "fxmanifest.lua", Utils.ManifestContent);

            var streamRoot = info.CreateSubdirectory("stream");

            foreach (var meta in _metaFiles) {
                meta.CopyTo(info.FullName + Path.DirectorySeparatorChar + meta.Name);
            }
            
            foreach (var stream in _streamFiles)
                stream.CopyTo(streamRoot.FullName + Path.DirectorySeparatorChar + stream.Name);

            AnimateBuildButton(sender as Button);
        }

        private async Task AnimateBuildButton(Button button) {
            Brush origColor = button.Foreground;
            object origContent = button.Content;
            
            button.Foreground = Brushes.GreenYellow;
            button.Content = "Done!";

            await Task.Delay(1000);

            button.Content = origContent;
            button.Foreground = origColor;
        }

        private void ChooseOutDir(object sender, RoutedEventArgs e) {
            var dialog = new CommonOpenFileDialog();
            dialog.IsFolderPicker = true;
            dialog.Multiselect = false;
            var result = dialog.ShowDialog();

            if (result == CommonFileDialogResult.Ok) {
                var path = dialog.FileNames.First();
                File.WriteAllText(_optionFilePath, path);
                OutPath.Content = path;
            }
        }
    }
}