using System;
using System.Collections.Generic;
using System.Data;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
using System.Windows.Forms;

namespace CyberforumPhoto
{
    public partial class Form1 : Form
    {
        /// <summary>
        /// Базовый класс для создание узлов в элементе TreeView
        /// </summary>
        public class BaseNode : TreeNode
        {
            /// <summary>
            /// Возвращает или задает значение. Может-ли данный узел загружать массив изображений.
            /// </summary>
            public bool IsloadArrayPicture { get; set; }
            /// <summary>
            /// Возвращает или задает значение. Может-ли данный узел загружать массив изображений.
            /// </summary>
            public int CountPicture { get; set; }
            List<BaseNode> nodeList = new List<BaseNode>();
            public IEnumerable<List<BaseNode>> GetChildNodes(int Count = 5)
            {
                return ToChunks(this.Nodes, Count);
            }

            private IEnumerable<List<BaseNode>> ToChunks(TreeNodeCollection items, int chunkSize)
            {
                List<BaseNode> chunk = new List<BaseNode>(chunkSize);
                foreach (BaseNode item in items)
                {
                    chunk.Add(item);
                    if (chunk.Count == chunkSize)
                    {
                        yield return chunk;
                        chunk = new List<BaseNode>(chunkSize);
                    }
                }
                if (chunk.Any())
                    yield return chunk;
            }

            public BaseNode(string Text) : base(Text) { }
        }

        public Form1()
        {
            InitializeComponent();
        }

        Dictionary<string, string> dirs = new Dictionary<string, string>();
        public IEnumerable<FileInfo> GetFilesByExtensions(DirectoryInfo dir, params string[] extensions)
        {
            IEnumerable<FileInfo> files = dir.GetFiles("*.*", SearchOption.AllDirectories);
            return files.Where(f => extensions.Contains(f.Extension));
        }

        DriveInfo[] driveInfo = DriveInfo.GetDrives();
        DirectoryInfo directory;
        BaseNode NodeDriver = null;
        BaseNode NodeDirectory = null;
        BaseNode NodePicture;
        int stepProgress;
        private void FindImageDriver()
        {
            directory = new DirectoryInfo("C:\\");
            NodeDriver = new BaseNode(directory.Name);

            var h = directory.GetDirectories().Where(x =>
            !x.Attributes.ToString().Replace(" ", "").Split(',')[0].Contains("Hidden") &
            !x.Attributes.ToString().Replace(" ", "").Split(',')[0].Contains("ReadOnly")).Where(x => x.Name != "Windows");

            foreach (DirectoryInfo dir in h)
            {
                IEnumerable<FileInfo> fileInfos = GetFilesByExtensions(dir, ".jpg", ".png");
                if (fileInfos.Count() != 0)
                {
                    this.Invoke(new Action(() =>
                    {
                        progress.ProgressBar.Maximum = fileInfos.Count();
                    }));
                    NodeDirectory = new BaseNode(dir.Name);
                    NodeDirectory.ImageIndex = 0;

                    NodeDriver.Nodes.Add(NodeDirectory);
                    NodeDriver.ImageIndex = 3;
                    NodeDriver.Tag = false;
                }
                foreach (FileInfo file in fileInfos)
                {
                    this.Invoke(new Action(() =>
                    {
                        progress.ProgressBar.Value = stepProgress++;
                    }));

                    NodePicture = new BaseNode(file.FullName);
                    NodePicture.Tag = false;
                    NodePicture.ImageIndex = file.Extension.Contains(".jpg") ? 1 : 2;

                    NodeDirectory.Nodes.Add(NodePicture);
                    NodeDirectory.IsloadArrayPicture = true;

                }
                stepProgress = 0;
            }

            this.Invoke(new Action(() =>
            {
                treeView1.Nodes.Add(NodeDriver);
            }));

            this.Invoke(new Action(() =>
            {
                lbStatus.Text = "";
                progress.ProgressBar.Value = 0;
            }));
        }

        private void Form1_LoadAsync(object sender, EventArgs e)
        {   
           new Task(new Action(FindImageDriver)).Start();
        }

        public void ViewPhoto(List<BaseNode> baseNodes)
        {
            panel1.Controls.Clear();
            foreach (BaseNode path in baseNodes)
            {
                int Top = 5;
                PictureBox image = new PictureBox();
                image.Image = Image.FromFile(path.Text);
                image.SizeMode = PictureBoxSizeMode.StretchImage;
                image.Width = 200;
                image.Height = 160;

                image.Margin = new Padding(0, ++Top, 0, 0);

                panel1.Controls.Add(image);
            }
        }

        IEnumerable<List<BaseNode>> array;
        private void treeView1_AfterSelect(object sender, TreeViewEventArgs e)
        {
            BaseNode node = (treeView1.SelectedNode as TreeNode) as BaseNode;
            
            
            if (node.IsloadArrayPicture)
            {
                array = node.GetChildNodes(30);
                ViewPhoto(array.FirstOrDefault());
                panel1.VerticalScroll.Maximum = array.Count();
                panel1.VerticalScroll.SmallChange = 1;
            }
            
        }

        private void panel1_Scroll(object sender, ScrollEventArgs e)
        {
            
            try
            {
                ViewPhoto(array.ToList()[e.NewValue]);
            }
            catch (Exception)
            {

            }
        }

        private void tableLayoutPanel1_Paint(object sender, PaintEventArgs e)
        {

        }

        private void pictureBox1_Click(object sender, EventArgs e)
        {
            
        }

        private void panel1_Paint(object sender, PaintEventArgs e)
        {

        }
    }
}