using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Data;
using System.Windows.Documents;
using System.Windows.Input;
using System.Windows.Media;
using System.Windows.Media.Imaging;
using System.Windows.Navigation;
using System.Diagnostics;
using System.Windows.Shapes;
using System.IO;
using Ookii.Dialogs.Wpf;
using System.Threading;
using System.Windows.Threading;
using LibCPK;

namespace CriPakGUI
{
    /// <summary>
    /// MainWindow.xaml 的交互逻辑
    /// </summary>
    public partial class MainWindow : Window
    {
        public MainWindow()
        {
            InitializeComponent();
            SetBasicPrefs();
        }

        private void SetBasicPrefs()
        {
            menu_savefiles.IsEnabled = false;
            menu_importAssets.IsEnabled = false;
            progressbar0.Maximum = 100;
            myPackage.basePath = @"C:/";
        }
        private void menu_openfile_Click(object sender, RoutedEventArgs e)
        {
            Debug.WriteLine("载入 cpk");
            string fName;
            string baseName;
            VistaOpenFileDialog openFileDialog = new VistaOpenFileDialog();
            openFileDialog.InitialDirectory = "";
            openFileDialog.Filter = "Criware CPK|*.cpk";
            openFileDialog.RestoreDirectory = true;
            openFileDialog.FilterIndex = 1;
            if (openFileDialog.ShowDialog().Value)
            {
                fName = openFileDialog.FileName;
                baseName = System.IO.Path.GetFileName(fName);
                status_cpkname.Content = baseName;
                beginLoadCPK(fName);
                button_extract.IsEnabled = true;
                button_importassets.IsEnabled = true;
                
            }
        }

        private void beginLoadCPK(string fName)
        {
            ThreadPool.QueueUserWorkItem(o =>
            {
                Application.Current.Dispatcher.BeginInvoke(
                    DispatcherPriority.SystemIdle,
                    new Action(() =>
                    {
                        cpkwrapper cpk = new cpkwrapper(fName);
                        status_cpkmsg.Content = string.Format("{0} 文件已注入.", cpk.nums);
                        datagrid_cpk.ItemsSource = cpk.table;

                        menu_importAssets.IsEnabled = true;
                        menu_savefiles.IsEnabled = true;
                        myPackage.basePath = System.IO.Path.GetDirectoryName(fName);
                        myPackage.baseName = System.IO.Path.GetFileName(fName);
                        myPackage.fileName = fName;
                    } )
               );
            });  
        }


        private void menu_importAssets_Click(object sender, RoutedEventArgs e)
        {
            CpkPatcher patcherWindow = new CpkPatcher(this.Top, this.Left);
            patcherWindow.ShowDialog();
        }


        private delegate void progressbarDelegate(float no);

        private delegate void datagridDelegate(bool value);

        private void updateDatagrid(bool value)
        {
            datagrid_cpk.IsEnabled = value;
            button_extract.IsEnabled = value;
            button_importassets.IsEnabled = value;
        }

        private void updateprogressbar(float no)
        {
            progressbar0.Value = no;
        }

        private void menu_savefiles_Click(object sender, RoutedEventArgs e)
        {
            VistaFolderBrowserDialog saveFilesDialog = new VistaFolderBrowserDialog();
            saveFilesDialog.SelectedPath = myPackage.basePath;
            if (saveFilesDialog.ShowDialog().Value)
            {
                Debug.Print(saveFilesDialog.SelectedPath + "/" + myPackage.baseName + "_unpacked");
                ThreadPool.QueueUserWorkItem(new WaitCallback(beginExtractCPK), saveFilesDialog.SelectedPath);
                
            }

        }

        private void beginExtractCPK(object foutDir)
        {
            string outDir;
            outDir = (string)(foutDir + "/" + myPackage.baseName + "_unpacked");
            if (myPackage.cpk != null)
            {
                if (!Directory.Exists(outDir))
                {
                    Directory.CreateDirectory(outDir);
                }
                BinaryReader oldFile = new BinaryReader(File.OpenRead(myPackage.cpk_name));
                List<FileEntry> entries = null;

                entries = myPackage.cpk.FileTable.Where(x => x.FileType == "FILE").ToList();

                if (entries.Count == 0)
                {
                    MessageBox.Show("释放文件出错,类目为空.");
                    oldFile.Close();
                    return;
                }

                int i = 0;
                int id;
                string currentName;
                bool bFileRepeated = Tools.CheckListRedundant(entries);
                this.Dispatcher.Invoke(new datagridDelegate(updateDatagrid), new object[] { (bool)false });

                while (i < entries.Count)
                {
                    this.Dispatcher.Invoke(new progressbarDelegate(updateprogressbar), new object[] { (float)i / (float)entries.Count * 100f });//异步委托

                    if (!String.IsNullOrEmpty((string)entries[i].DirName))
                    {
                        Directory.CreateDirectory(outDir + "/" + entries[i].DirName.ToString());
                    }

                    id = Convert.ToInt32(entries[i].ID);
                    if (id > 0 &&　bFileRepeated)
                    {
                        currentName = (((entries[i].DirName != null) ?
                                        entries[i].DirName + "/" : "") + string.Format("[{0}]", id.ToString()) + entries[i].FileName);
                        currentName = currentName.TrimStart('/');
                    }
                    else
                    {
                        currentName = ((entries[i].DirName != null) ? entries[i].DirName + "/" : "") + entries[i].FileName;
                        currentName = currentName.TrimStart('/');
                    }

                    oldFile.BaseStream.Seek((long)entries[i].FileOffset, SeekOrigin.Begin);

                    string isComp = Encoding.ASCII.GetString(oldFile.ReadBytes(8));
                    oldFile.BaseStream.Seek((long)entries[i].FileOffset, SeekOrigin.Begin);

                    byte[] chunk = oldFile.ReadBytes(Int32.Parse(entries[i].FileSize.ToString()));

                    if (isComp == "CRILAYLA")
                    {
                        int size = Int32.Parse((entries[i].ExtractSize ?? entries[i].FileSize).ToString());

                        if (size != 0)
                        {
                            chunk = myPackage.cpk.DecompressLegacyCRI(chunk, size);
                        }
                    }

                    Debug.WriteLine(" 文件名 :{0}\n    文件偏移值:{1:x8}    释放大小:{2:x8}   块大小:{3:x8} {4}",
                                                                entries[i].FileName.ToString(),
                                                                (long)entries[i].FileOffset,
                                                                entries[i].ExtractSize,
                                                                entries[i].FileSize,
                                                                ((float)i / (float)entries.Count) * 100f);
                    string dstpath = outDir + "/" + currentName;
                    dstpath = Tools.GetSafePath(dstpath);
                    string dstdir = System.IO.Path.GetDirectoryName(dstpath);
                    if (!Directory.Exists(dstdir))
                    {
                        Directory.CreateDirectory(dstdir);
                    }
                    File.WriteAllBytes(dstpath, chunk);
                    i += 1;
                }
                oldFile.Close();
                this.Dispatcher.Invoke(new progressbarDelegate(updateprogressbar), new object[] { 100f });
                this.Dispatcher.Invoke(new datagridDelegate(updateDatagrid), new object[] { (bool)true });
                MessageBox.Show("释放完成.");

            }

        }

        private void button_extract_Click(object sender, RoutedEventArgs e)
        {
            VistaFolderBrowserDialog saveFilesDialog = new VistaFolderBrowserDialog();
            saveFilesDialog.SelectedPath = myPackage.basePath + "/";
            if (saveFilesDialog.ShowDialog().Value)
            {
                Debug.Print(saveFilesDialog.SelectedPath + "/" + myPackage.baseName + "_unpacked");
                ThreadPool.QueueUserWorkItem(new WaitCallback(beginExtractCPK), saveFilesDialog.SelectedPath);

            }
        }

        private void button_importassets_Click(object sender, RoutedEventArgs e)
        {
            CpkPatcher patcherWindow = new CpkPatcher(this.Top, this.Left);
            patcherWindow.ShowDialog();


        }

        private void menu_aboutgui_Click(object sender, RoutedEventArgs e)
        {
            WindowAboutGUI aboutwindow = new WindowAboutGUI(this.Top, this.Left);
            aboutwindow.ShowDialog();
        }

        private void dgmenu1_Cilck(object sender, MouseButtonEventArgs e)
        {
            Point p = e.GetPosition(this.datagrid_cpk);
            HitTestResult htr = VisualTreeHelper.HitTest(this.datagrid_cpk, p);
            TextBlock o = htr.VisualHit as TextBlock;
            if (o != null)
            {
                DataGridRow dgr = VisualTreeHelper.GetParent(o) as DataGridRow;

                dgr.Focus();
                dgr.IsSelected = true;
            }
        }
        private void dgitem1_Click(object sender, RoutedEventArgs e)
        {
            
            CPKTable t = this.datagrid_cpk.SelectedItem as CPKTable;
            if (t != null)
            {
                if (t.FileSize > 0 && t.FileType == "FILE")
                {
                    VistaSaveFileDialog saveFilesDialog = new VistaSaveFileDialog();
                    saveFilesDialog.InitialDirectory = myPackage.basePath ;
                    saveFilesDialog.FileName = myPackage.basePath + "/" + t._localName;
                    if (saveFilesDialog.ShowDialog().Value)
                    {
                        byte[] chunk = ExtractItem(t);

                        File.WriteAllBytes(saveFilesDialog.FileName, chunk);
                        MessageBox.Show(String.Format("解压至 :{0}", saveFilesDialog.FileName));
                    }
                    
                } 
            }
            
        }

        private void dgitem2_Click(object sender, RoutedEventArgs e)
        {
            MessageBox.Show("当前不支持该功能");
        }

        private byte[] ExtractItem(CPKTable t)
        {
            CPKTable entries = t as CPKTable;
            BinaryReader oldFile = new BinaryReader(File.OpenRead(myPackage.cpk_name));
            oldFile.BaseStream.Seek((long)entries.FileOffset, SeekOrigin.Begin);

            string isComp = Encoding.ASCII.GetString(oldFile.ReadBytes(8));
            oldFile.BaseStream.Seek((long)entries.FileOffset, SeekOrigin.Begin);

            byte[] chunk = oldFile.ReadBytes(Int32.Parse(entries.FileSize.ToString()));

            if (isComp == "CRILAYLA")
            {
                int size;
                if (entries.ExtractSize == 0)
                {
                    size = entries.FileSize;
                }
                else
                {
                    size = entries.ExtractSize;
                }

                if (size != 0)
                {
                    chunk = myPackage.cpk.DecompressLegacyCRI(chunk, size);
                }
            }
            oldFile.Close();
            return chunk;




        }

        private void menu_makeCSV_Click(object sender, RoutedEventArgs e)
        {
            MessageBox.Show("当前不支持该功能");
        }

        private void comboBox_encodings_SelectionChanged(object sender, SelectionChangedEventArgs e)
        {
            int cur = comboBox_encodings.SelectedIndex;
            Encoding current_codepage;
            switch (cur)
            {
                case 0:
                    current_codepage = Encoding.GetEncoding(65001);
                    break;
                case 1:
                    current_codepage = Encoding.GetEncoding(932);
                    break;
                default:
                    current_codepage = Encoding.GetEncoding(65001);
                    break;

            }
            if (current_codepage != myPackage.encoding)
            {
                myPackage.encoding = current_codepage;
                if (myPackage.fileName != null)
                {

                    beginLoadCPK(myPackage.fileName);
                }

            }
            
            

        }
    }
}