oi-metrology/Server/Repositories/WaferCounterRepository.cs

using OI.Metrology.Server.Models;
using OI.Metrology.Shared.DataModels;
using OI.Metrology.Shared.Models;
using OI.Metrology.Shared.Models.Stateless;
using OI.Metrology.Shared.Repositories;
using System.Globalization;

namespace OI.Metrology.Server.Repository;

public class WaferCounterRepository : IWaferCounterRepository
{

    private record Record(int Check,
                          int Total,
                          string? SlotMap);

    private readonly string _MockRoot;
    private readonly string _RepositoryName;
    private readonly AppSettings _AppSettings;
    private readonly IHttpClientFactory _HttpClientFactory;
    private readonly IDbConnectionFactory _DBConnectionFactory;
    private readonly IFileShareRepository _FileShareRepository;
    private readonly string _OpenInsightApplicationProgrammingInterface;

    public WaferCounterRepository(AppSettings appSettings, IDbConnectionFactory dbConnectionFactory, IHttpClientFactory httpClientFactory, IFileShareRepository fileShareRepository)
    {
        _AppSettings = appSettings;
        _MockRoot = appSettings.MockRoot;
        _HttpClientFactory = httpClientFactory;
        _DBConnectionFactory = dbConnectionFactory;
        _FileShareRepository = fileShareRepository;
        _RepositoryName = nameof(WaferCounterRepository)[..^10];
        _OpenInsightApplicationProgrammingInterface = appSettings.OpenInsightApplicationProgrammingInterface;
    }

    private void MoveFile(string waferSizeDirectory, NginxFileSystemSortable nginxFileSystemSortable)
    {
        Calendar calendar = new CultureInfo("en-US").Calendar;
        string from = Path.Combine(waferSizeDirectory, nginxFileSystemSortable.Name);
        string weekOfYear = $"{nginxFileSystemSortable.DateTime:yyyy}_Week_{calendar.GetWeekOfYear(nginxFileSystemSortable.DateTime, CalendarWeekRule.FirstDay, DayOfWeek.Sunday):00}";
        string to = Path.Combine(waferSizeDirectory, "Archive", weekOfYear, nginxFileSystemSortable.Name);
        _FileShareRepository.MoveFile(from, to);
    }

    private static Record GetRecord(string line1, string line2)
    {
        Record result;
        string? waferMap = string.IsNullOrEmpty(line2) || line2.Length != 8 ? null : line2.Substring(1, 1);
        int check = waferMap == "1" ? 1 : 0;
        int total = int.Parse(line1[1..]);
        // string wafers = Array.from(line2[2..]);
        foreach (char item in line2[2..])
        {
            switch (item)
            {
                case '0':
                    check += 0;
                    waferMap += "0000";
                    break;
                case '1':
                    check += 1;
                    waferMap += "0001";
                    break;
                case '2':
                    check += 1;
                    waferMap += "0010";
                    break;
                case '3':
                    check += 2;
                    waferMap += "0011";
                    break;
                case '4':
                    check += 1;
                    waferMap += "0100";
                    break;
                case '5':
                    check += 2;
                    waferMap += "0101";
                    break;
                case '6':
                    check += 2;
                    waferMap += "0110";
                    break;
                case '7':
                    check += 3;
                    waferMap += "0111";
                    break;
                case '8':
                    check += 1;
                    waferMap += "1000";
                    break;
                case '9':
                    check += 2;
                    waferMap += "1001";
                    break;
                case 'A':
                    check += 2;
                    waferMap += "1010";
                    break;
                case 'B':
                    check += 3;
                    waferMap += "1011";
                    break;
                case 'C':
                    check += 2;
                    waferMap += "1100";
                    break;
                case 'D':
                    check += 3;
                    waferMap += "1101";
                    break;
                case 'E':
                    check += 3;
                    waferMap += "1110";
                    break;
                case 'F':
                    check += 4;
                    waferMap += "1111";
                    break;
                default:
                    break;
            }
        }
        result = new(check, total, waferMap);
        return result;
    }

    private Uri GetWaferSizeUri(string area, string waferSize) =>
        _FileShareRepository.Append(new Uri(_AppSettings.EcMesaFileShareCharacterizationSi), "WaferCounter", area, waferSize);

    private string GetWaferSizeDirectory(string area, string waferSize, bool destination) =>
        destination ? Path.Combine(_AppSettings.WaferCounterDestinationDirectory, area, waferSize) : Path.Combine(_AppSettings.EcCharacterizationSi, "WaferCounter", area, waferSize);

    string? IWaferCounterRepository.GetSlotMap(string line1, string line2) =>
        GetRecord(line1, line2).SlotMap;

    private List<NginxFileSystemSortable> GetNginxFileSystemSortableCollection(HttpClient httpClient, Uri waferSizeUri, string waferSizeDirectory)
    {
        List<NginxFileSystemSortable> results = new();
        DateTime dateTime = DateTime.Now;
        long ticks = dateTime.AddSeconds(_AppSettings.WaferCounterTwoFileSecondsWait).Ticks;
        for (int i = 0; i < int.MaxValue; i++)
        {
            results = _FileShareRepository.GetNginxFileSystemSortableCollection(httpClient, waferSizeUri, ".wc");
            if (results.Count > 1 || DateTime.Now.Ticks > ticks)
                break;
            Thread.Sleep(250);
        }
        for (int i = 1; i < results.Count; i++)
            MoveFile(waferSizeDirectory, results[i]);
        return results;
    }

    private static WaferCounter GetLastQuantityAndSlotMapWithText(string waferSize, string text, HttpClient httpClient, NginxFileSystemSortable nginxFileSystemSortable)
    {
        WaferCounter result;
        Task<string> value = httpClient.GetStringAsync(nginxFileSystemSortable.Uri);
        value.Wait();
        string[] lines = value.Result.Split("\r\n");
        if (lines.Length <= 1)
            throw new Exception("Incomplete file length!");
        string[] segments = nginxFileSystemSortable.Name.Split('-');
        Record record = GetRecord(lines[0], lines[1]);
        string equipmentId = segments.Length <= 1 ? nginxFileSystemSortable.Name : segments[1].Split('.')[0];
        if (string.IsNullOrEmpty(record.SlotMap) || record.SlotMap.Length != 25)
            throw new Exception("Wrong length!");
        if (record.Total != record.Check)
            throw new Exception("Invalid!");
        result = new(nginxFileSystemSortable.DateTime, nginxFileSystemSortable.DateTime.ToString("yyyy-MM-dd hh:mm tt"), $"WC{waferSize}{equipmentId}", text, record.Total, record.SlotMap);
        return result;
    }

    WaferCounter IWaferCounterRepository.GetLastQuantityAndSlotMap(string area, string waferSize)
    {
        WaferCounter result;
        Uri waferSizeUri = GetWaferSizeUri(area, waferSize);
        HttpClient httpClient = _HttpClientFactory.CreateClient();
        string waferSizeDirectory = GetWaferSizeDirectory(area, waferSize, destination: false);
        List<NginxFileSystemSortable> nginxFileSystemSortableCollection = GetNginxFileSystemSortableCollection(httpClient, waferSizeUri, waferSizeDirectory);
        if (nginxFileSystemSortableCollection.Count < 1)
            throw new Exception("No files!");
        string text = string.Empty;
        result = GetLastQuantityAndSlotMapWithText(waferSize, text, httpClient, nginxFileSystemSortableCollection[0]);
        return result;
    }

    private void Save(string waferSizeDestinationDirectory, string area, string waferSize, string text, NginxFileSystemSortable nginxFileSystemSortable, WaferCounter result) =>
        _FileShareRepository.FileWrite(Path.Combine(waferSizeDestinationDirectory, $"{nginxFileSystemSortable.Name}.csv"), $"100,{waferSize},{area},{nginxFileSystemSortable.DateTime},{text},{result.Total:00},{result.SlotMap} ");

    WaferCounter IWaferCounterRepository.GetLastQuantityAndSlotMapWithText(string area, string waferSize, string text)
    {
        WaferCounter result;
        Uri waferSizeUri = GetWaferSizeUri(area, waferSize);
        HttpClient httpClient = _HttpClientFactory.CreateClient();
        string waferSizeDirectory = GetWaferSizeDirectory(area, waferSize, destination: false);
        List<NginxFileSystemSortable> nginxFileSystemSortableCollection = GetNginxFileSystemSortableCollection(httpClient, waferSizeUri, waferSizeDirectory);
        if (nginxFileSystemSortableCollection.Count < 1)
            throw new Exception("No files!");
        result = GetLastQuantityAndSlotMapWithText(waferSize, text, httpClient, nginxFileSystemSortableCollection[0]);
        if (result is not null)
        {
            string waferSizeDestinationDirectory = _AppSettings.WaferCounterDestinationDirectory;
            // string waferSizeDestinationDirectory = GetWaferSizeUri(area, waferSize, destination: true);
            Save(waferSizeDestinationDirectory, area, waferSize, text, nginxFileSystemSortableCollection[0], result);
        }
        if (result is null)
            throw new NullReferenceException(nameof(result));
        return result;
    }

}