using System;
using System.Drawing;
using System.Collections;
using System.ComponentModel;
using System.Windows.Forms;
using System.Data;
using System.IO;
using System.Management;
using System.Globalization;

namespace ExplorerControlLibrary
{
	/// <summary>
	/// Explorer control.
	/// </summary>
	public class ExplorerControl : System.Windows.Forms.UserControl
	{
		private System.Windows.Forms.TreeView tvFolders;
		private System.Windows.Forms.Splitter splitter1;
		private System.Windows.Forms.ListView lvFiles;
		private System.Windows.Forms.ImageList m_imageListTreeView;
		private System.ComponentModel.IContainer components;

		//delegate status event
		public delegate void Status(string e);
		public event Status UpdateStatus;

		public ExplorerControl()
		{
			// This call is required by the Windows.Forms Form Designer.
			InitializeComponent();

			// Populate TreeView with Drive list
			PopulateDriveList();

		}

		/// <summary>
		/// Clean up any resources being used.
		/// </summary>
		protected override void Dispose( bool disposing )
		{
			if( disposing )
			{
				if( components != null )
					components.Dispose();
			}
			base.Dispose( disposing );
		}

		#region Component Designer generated code
		/// <summary>
		/// Required method for Designer support - do not modify 
		/// the contents of this method with the code editor.
		/// </summary>
		private void InitializeComponent()
		{
			this.components = new System.ComponentModel.Container();
			System.Resources.ResourceManager resources = new System.Resources.ResourceManager(typeof(ExplorerControl));
			this.tvFolders = new System.Windows.Forms.TreeView();
			this.m_imageListTreeView = new System.Windows.Forms.ImageList(this.components);
			this.splitter1 = new System.Windows.Forms.Splitter();
			this.lvFiles = new System.Windows.Forms.ListView();
			this.SuspendLayout();
			// 
			// tvFolders
			// 
			this.tvFolders.Dock = System.Windows.Forms.DockStyle.Left;
			this.tvFolders.ImageList = this.m_imageListTreeView;
			this.tvFolders.Name = "tvFolders";
			this.tvFolders.Size = new System.Drawing.Size(136, 416);
			this.tvFolders.TabIndex = 0;
			this.tvFolders.AfterSelect += new System.Windows.Forms.TreeViewEventHandler(this.tvFolders_AfterSelect);
			// 
			// m_imageListTreeView
			// 
			this.m_imageListTreeView.ColorDepth = System.Windows.Forms.ColorDepth.Depth8Bit;
			this.m_imageListTreeView.ImageSize = new System.Drawing.Size(16, 16);
			this.m_imageListTreeView.ImageStream = ((System.Windows.Forms.ImageListStreamer)(resources.GetObject("m_imageListTreeView.ImageStream")));
			this.m_imageListTreeView.TransparentColor = System.Drawing.Color.Transparent;
			// 
			// splitter1
			// 
			this.splitter1.Location = new System.Drawing.Point(136, 0);
			this.splitter1.Name = "splitter1";
			this.splitter1.Size = new System.Drawing.Size(3, 416);
			this.splitter1.TabIndex = 1;
			this.splitter1.TabStop = false;
			// 
			// lvFiles
			// 
			this.lvFiles.Dock = System.Windows.Forms.DockStyle.Fill;
			this.lvFiles.Location = new System.Drawing.Point(139, 0);
			this.lvFiles.Name = "lvFiles";
			this.lvFiles.Size = new System.Drawing.Size(525, 416);
			this.lvFiles.TabIndex = 2;
			this.lvFiles.View = System.Windows.Forms.View.Details;
			// 
			// ExplorerControl
			// 
			this.Controls.AddRange(new System.Windows.Forms.Control[] {
																		  this.lvFiles,
																		  this.splitter1,
																		  this.tvFolders});
			this.Name = "ExplorerControl";
			this.Size = new System.Drawing.Size(664, 416);
			this.ResumeLayout(false);

		}
		#endregion

		/// <summary>
		/// This procedure populate the TreeView with the Drive list
		/// </summary>
		private void PopulateDriveList()
		{
			TreeNode nodeTreeNode;
			int imageIndex = 0;
			int selectIndex = 0;

			const int Removable = 2;
			const int LocalDisk = 3;
			const int Network = 4;
			const int CD = 5;
			//const int RAMDrive = 6;
			
			this.Cursor = Cursors.WaitCursor;
			//clear TreeView
			tvFolders.Nodes.Clear();
			nodeTreeNode = new TreeNode("My Computer",0,0);
			tvFolders.Nodes.Add(nodeTreeNode);

			//set node collection
			TreeNodeCollection nodeCollection = nodeTreeNode.Nodes;

			//Get Drive list
			ManagementObjectCollection queryCollection = getDrives();
			foreach ( ManagementObject mo in queryCollection)
			{
			
				switch (int.Parse( mo["DriveType"].ToString()))
				{
					case Removable:			//removable drives
						imageIndex = 5;
						selectIndex = 5;
						break;
					case LocalDisk:			//Local drives
						imageIndex = 6;
						selectIndex = 6;
						break;
					case CD:				//CD rom drives
						imageIndex = 7;
						selectIndex = 7;
						break;
					case Network:			//Network drives
						imageIndex = 8;
						selectIndex = 8;
						break;
					default:				//defalut to folder
						imageIndex = 2;
						selectIndex = 3;
						break;
				}
				//create new drive node
				nodeTreeNode = new TreeNode(mo["Name"].ToString() + "\\" ,imageIndex,selectIndex);

				//add new node
				nodeCollection.Add(nodeTreeNode);

			}
			

			//Init files ListView
			InitListView();

			this.Cursor = Cursors.Default;

		}

		private void tvFolders_AfterSelect(object sender, System.Windows.Forms.TreeViewEventArgs e)
		{
			//Populate folders and files when a folder is selected
			this.Cursor = Cursors.WaitCursor;

			//get current selected drive or folder
			TreeNode nodeCurrent = e.Node;

			//clear all sub-folders
			nodeCurrent.Nodes.Clear();

			if (nodeCurrent.SelectedImageIndex == 0) 
			{
				//Selected My Computer - repopulate drive list
				updateStatus("Refresh drive list...");
				PopulateDriveList();
				updateStatus("Ready");
			}
			else 
			{
				//populate sub-folders and folder files
				updateStatus("Getting directory information...");
				PopulateDirectory(nodeCurrent, nodeCurrent.Nodes);
				updateStatus("Ready");
			}
			this.Cursor = Cursors.Default;
		}
		
		protected void InitListView()
		{
			//init ListView control
			lvFiles.Clear();		//clear control
			//create column header for ListView
			lvFiles.Columns.Add("Name",150,System.Windows.Forms.HorizontalAlignment.Left);
			lvFiles.Columns.Add("Size",75, System.Windows.Forms.HorizontalAlignment.Right);
			lvFiles.Columns.Add("Created", 140, System.Windows.Forms.HorizontalAlignment.Left);
			lvFiles.Columns.Add("Modified", 140, System.Windows.Forms.HorizontalAlignment.Left);

		}

		protected void PopulateDirectory(TreeNode nodeCurrent, TreeNodeCollection nodeCurrentCollection)
		{
			TreeNode nodeDir;
			int imageIndex = 2;		//unselected image index
			int selectIndex = 3;	//selected image index

			if (nodeCurrent.SelectedImageIndex != 0) 
			{
				//populate treeview with folders
				try
				{
					//check path
					if(Directory.Exists(getFullPath(nodeCurrent.FullPath)) == false)
					{
						MessageBox.Show("Directory or path " + nodeCurrent.ToString() + " does not exist.");
					}
					else
					{
						//populate files
						PopulateFiles(nodeCurrent);

						string[] stringDirectories = Directory.GetDirectories(getFullPath(nodeCurrent.FullPath));
						string stringFullPath = "";
						string stringPathName = "";

						//loop throught all directories
						foreach (string stringDir in stringDirectories)
						{
							stringFullPath = stringDir;
							stringPathName = GetPathName(stringFullPath);
							
							//create node for directories
							nodeDir = new TreeNode(stringPathName.ToString(),imageIndex,selectIndex);
							nodeCurrentCollection.Add(nodeDir);
						}

						//expand tree node
						nodeCurrent.ExpandAll();
					}
				}
				catch (IOException)
				{
					MessageBox.Show("Error: Drive not ready or directory does not exist.");
				}
				catch (UnauthorizedAccessException)
				{
					MessageBox.Show("Error: Drive or directory access denided.");
				}
				catch (Exception e)
				{
					MessageBox.Show("Error: " + e);
				}
			}
		}

		/// <summary>
		/// Get path
		/// </summary>
		/// <param name="stringPath"></param>
		/// <returns></returns>
		
		protected string GetPathName(string stringPath)
		{
			//Get Name of folder
			string[] stringSplit = stringPath.Split('\\');
			int _maxIndex = stringSplit.Length;
			return stringSplit[_maxIndex-1];
		}

		/// <summary>
		/// Populate files info into listview
		/// </summary>
		/// <param name="nodeCurrent"></param>
		protected void PopulateFiles(TreeNode nodeCurrent)
		{
			//Populate listview with files
			string[] lvData =  new string[4];
			
			//update status
			updateStatus("Getting directory files...");

			//clear list
			InitListView();

			if (nodeCurrent.SelectedImageIndex != 0) 
			{
				//check path
				if(Directory.Exists((string) getFullPath(nodeCurrent.FullPath)) == false)
				{
					MessageBox.Show("Directory or path " + nodeCurrent.ToString() + " does not exist.");
				}
				else
				{
					try
					{
						string[] stringFiles = Directory.GetFiles(getFullPath(nodeCurrent.FullPath));
						string stringFileName = "";
						DateTime dtCreateDate, dtModifyDate;
						Int64 lFileSize = 0;

						//loop throught all files
						foreach (string stringFile in stringFiles)
						{
							stringFileName = stringFile;
							FileInfo objFileSize = new FileInfo(stringFileName);
							lFileSize = objFileSize.Length;
							dtCreateDate = objFileSize.CreationTime; //GetCreationTime(stringFileName);
							dtModifyDate = objFileSize.LastWriteTime; //GetLastWriteTime(stringFileName);

							//create listview data
							lvData[0] = GetPathName(stringFileName);
							lvData[1] = formatSize(lFileSize);

							//check if this machine have Day light saving time set
							if (TimeZone.CurrentTimeZone.IsDaylightSavingTime(DateTime.Now) == true)
							{
								//check if file is in local current day light saving time
								if (TimeZone.CurrentTimeZone.IsDaylightSavingTime(dtCreateDate) == false)
								{
									//not in day light saving time adjust time
									lvData[2] = formatDate(dtCreateDate.AddHours(1));
								}
								else
								{
									//is in day light saving time adjust time
									lvData[2] = formatDate(dtCreateDate);
								}

								//check if file is in local current day light saving time
								if (TimeZone.CurrentTimeZone.IsDaylightSavingTime(dtModifyDate) == false)
								{
									//not in day light saving time adjust time
									lvData[3] = formatDate(dtModifyDate.AddHours(1));
								}
								else							
								{
									//not in day light saving time adjust time
									lvData[3] = formatDate(dtModifyDate);
								}
							}
							else
							{
								//machine Day Light Saving Time not set
								lvData[2] = formatDate(dtCreateDate);
								lvData[3] = formatDate(dtModifyDate);
							}
							//Create actual list item
							ListViewItem lvItem = new ListViewItem(lvData,0);
							lvFiles.Items.Add(lvItem);
					
						}
					}
					catch (IOException)
					{
						MessageBox.Show("Error: Drive not ready or directory does not exist.");
					}
					catch (UnauthorizedAccessException)
					{
						MessageBox.Show("Error: Drive or directory access denided.");
					}
					catch (Exception e)
					{
						MessageBox.Show("Error: " + e);
					}
				}
			}

			//update status
			updateStatus("Ready");
		}

		/// <summary>
		/// Get full path
		/// </summary>
		/// <param name="stringPath"></param>
		/// <returns>stringParse</returns>
		protected string getFullPath(string stringPath)
		{
			//Get Full path
			string stringParse = "";
			//remove My Computer from path.
			stringParse = stringPath.Replace("My Computer\\", "");

			return stringParse;
		}
		
		/// <summary>
		/// Get drive collection
		/// </summary>
		/// <returns>queryCollection</returns>
		protected ManagementObjectCollection getDrives()
		{
			//get drive collection
			ManagementObjectSearcher query = new ManagementObjectSearcher("SELECT * From Win32_LogicalDisk ");
			ManagementObjectCollection queryCollection = query.Get();
			
			return queryCollection;
		}

		/// <summary>
		/// Date format
		/// </summary>
		/// <param name="dtDate"></param>
		/// <returns>stringDate</returns>
		protected string formatDate(DateTime dtDate)
		{
			//Get date and time in short format
			string stringDate = "";

			stringDate = dtDate.ToShortDateString().ToString() + " " + dtDate.ToShortTimeString().ToString();

			return stringDate;
		}

		/// <summary>
		/// Format number to KB
		/// </summary>
		/// <param name="lSize"></param>
		/// <returns>stringSize + " KB"</returns>
		protected string formatSize(Int64 lSize)
		{
			//Format number to KB
			string stringSize = "";
			NumberFormatInfo myNfi = new NumberFormatInfo();

			Int64 lKBSize = 0;

			if (lSize < 1024 ) 
			{
				if (lSize == 0) 
				{
					//zero byte
					stringSize = "0";
				}
				else 
				{
					//less than 1K but not zero byte
					stringSize = "1";
				}
			}
			else 
			{
				//convert to KB
				lKBSize = lSize / 1024;
				//format number with default format
				stringSize = lKBSize.ToString("n",myNfi);
				//remove decimal
				stringSize = stringSize.Replace(".00", "");
			}

			return stringSize + " KB";

		}

		/// <summary>
		/// status event
		/// </summary>
		/// <param name="stringStatus"></param>
		private void updateStatus(string stringStatus)
		{
			if (UpdateStatus != null)
				UpdateStatus(stringStatus );
		}

	}
}