WorkingDirControl.cs 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Text;
  5. using System.Threading.Tasks;
  6. namespace crusherScanner
  7. {
  8. internal class WorkingDirControl
  9. {
  10. /// <summary>
  11. /// Update sample as scanned in the working directory.
  12. /// </summary>
  13. /// <param name="scanData">Sample data containing Job and Barcode.</param>
  14. public static void UpdateWorkingDir(System.ComponentModel.DoWorkEventArgs e)
  15. {
  16. if (e.Argument == null)
  17. {
  18. return;
  19. }
  20. Scan scanData = (Scan)e.Argument;
  21. string[] DataDirs = new string[2] { Properties.Settings.Default.OreDefWorkFile, Properties.Settings.Default.OreDefInFile };
  22. string? JobFile = GetJobFilePath(scanData, DataDirs);
  23. if (JobFile != null)
  24. {
  25. if (JobFile.Contains(DataDirs[0]))
  26. {
  27. scanData = updateSampleInJobFile(scanData, JobFile);
  28. }
  29. else
  30. {
  31. try
  32. {
  33. File.Move($"{DataDirs[1]}\\{scanData.sampleData.Job}.csv", $"{DataDirs[0]}\\{scanData.sampleData.Job}.csv");
  34. scanData = updateSampleInJobFile(scanData, $"{DataDirs[0]}\\{scanData.sampleData.Job}.csv");
  35. }
  36. catch (Exception)
  37. {
  38. }
  39. }
  40. e.Result = scanData;
  41. }
  42. else
  43. {
  44. e.Result = scanData; // This shouldn't happen.
  45. }
  46. }
  47. /// <summary>
  48. /// Is the currently scanned sample's job in the working directory.
  49. /// </summary>
  50. /// <param name="scanData">Sample data containing Job and Barcode.</param>
  51. private static string? GetJobFilePath(Scan scanData,string[] DataDirs)
  52. {
  53. try
  54. {
  55. foreach (var DataDir in DataDirs)
  56. {
  57. if (Directory.Exists(DataDir))
  58. {
  59. var jobFiles = Directory.EnumerateFiles(DataDir, "ML??????.csv", SearchOption.TopDirectoryOnly);
  60. foreach (var jobFile in jobFiles)
  61. {
  62. if (jobFile.Contains(scanData.sampleData.Job))
  63. {
  64. return jobFile;
  65. }
  66. }
  67. }
  68. }
  69. }
  70. catch (Exception ex)
  71. {
  72. MessageBox.Show("An error occurred." + Environment.NewLine + ex.Message, "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
  73. }
  74. return null;
  75. }
  76. private static Scan updateSampleInJobFile(Scan scanData, string DataFile)
  77. {
  78. List<string> fileLines = new();
  79. using FileStream OreDefWorkingFile = new(DataFile, FileMode.Open, System.IO.FileAccess.ReadWrite, FileShare.None);
  80. {
  81. using StreamReader Reader = new(OreDefWorkingFile);
  82. {
  83. var line = Reader.ReadLine();
  84. while (line != null)
  85. {
  86. if (line != null && line.Contains('\u002C'))
  87. {
  88. scanData.JobTotal++;
  89. string[] CSV = line.Split('\u002C');
  90. if(CSV[0] == scanData.sampleData.Sid)
  91. {
  92. CSV[8] = scanData.sampleData.Sid;
  93. fileLines.Add(string.Join('\u002C', CSV));
  94. }
  95. else
  96. {
  97. fileLines.Add(line);
  98. }
  99. if(CSV[8] != "")
  100. {
  101. scanData.JobCount++;
  102. }
  103. }
  104. line = Reader.ReadLine();
  105. }
  106. }
  107. OreDefWorkingFile.Position = 0;
  108. using StreamWriter writer = new(OreDefWorkingFile);
  109. {
  110. foreach (var line in fileLines)
  111. {
  112. writer.WriteLine(line);
  113. }
  114. }
  115. writer.Close();
  116. }
  117. scanData.JobCount--; // Compensate for the file header
  118. scanData.JobTotal--; // Compensate for the file header
  119. return scanData;
  120. }
  121. }
  122. }