DiunaBI/src/Backend/DiunaBI.Plugins.Morska/Processors/T4SingleSourceProcessor.cs

using Microsoft.EntityFrameworkCore;
using DiunaBI.Core.Models;
using DiunaBI.Core.Database.Context;
using Microsoft.Extensions.Logging;
using Google.Apis.Sheets.v4;

namespace DiunaBI.Plugins.Morska.Processors;

public class T4SingleSourceProcessor : MorskaBaseProcessor
{
    public override string ProcessorType => "T4.SingleSource";

    private readonly AppDbContext _db;
    private readonly SpreadsheetsResource.ValuesResource _googleSheetValues;
    private readonly ILogger<T4SingleSourceProcessor> _logger;

    public T4SingleSourceProcessor(
        AppDbContext db,
        SpreadsheetsResource.ValuesResource googleSheetValues,
        ILogger<T4SingleSourceProcessor> logger)
    {
        _db = db;
        _googleSheetValues = googleSheetValues;
        _logger = logger;
    }

    public override void Process(Layer processWorker)
    {
        _logger.LogInformation("T4SingleSource: Starting processing for {ProcessWorkerName} ({ProcessWorkerId})", 
            processWorker.Name, processWorker.Id);

        var year = int.Parse(processWorker.Records?.SingleOrDefault(x => x.Code == "Year")?.Desc1!);
        var month = int.Parse(processWorker.Records?.SingleOrDefault(x => x.Code == "Month")?.Desc1!);
        var sourceLayer = processWorker.Records?.SingleOrDefault(x => x.Code == "SourceLayer")?.Desc1;
        if (sourceLayer == null)
        {
            throw new Exception("SourceLayer record not found");
        }

        var sourceImportWorker = _db.Layers.SingleOrDefault(x => x.Name == sourceLayer && !x.IsDeleted && !x.IsCancelled);
        if (sourceImportWorker == null)
        {
            throw new Exception("SourceImportWorker layer not found");
        }

        var source = processWorker.Records?.SingleOrDefault(x => x.Code == "Source")?.Desc1;
        if (source == null) 
        {
            throw new Exception("Source record not found");
        }

        var processedLayer = _db.Layers
            .Where(x => x.ParentId == processWorker.Id
                        && !x.IsDeleted && !x.IsCancelled)
            .OrderByDescending(x => x.CreatedAt)
            .FirstOrDefault();

        var isNew = false;
        if (processedLayer == null)
        {
            isNew = true;
            processedLayer = new Layer
            {
                Id = Guid.NewGuid(),
                Type = LayerType.Processed,
                ParentId = processWorker.Id,
                Number = _db.Layers.Count() + 1
            };
            processedLayer.Name = $"L{processedLayer.Number}-P-{year}/{month:D2}-{source}-T4";
            processedLayer.CreatedById = Guid.Parse("F392209E-123E-4651-A5A4-0B1D6CF9FF9D");
            processedLayer.ModifiedById = Guid.Parse("F392209E-123E-4651-A5A4-0B1D6CF9FF9D");
            processedLayer.CreatedAt = DateTime.UtcNow;
            processedLayer.ModifiedAt = DateTime.UtcNow;
        }

        processedLayer.ModifiedById = Guid.Parse("F392209E-123E-4651-A5A4-0B1D6CF9FF9D");
        processedLayer.ModifiedAt = DateTime.UtcNow;

        var dataSource = _db.Layers
            .Include(x => x.Records)
            .Where(x => x.ParentId == sourceImportWorker.Id
                        && !x.IsDeleted && !x.IsCancelled)
            .OrderByDescending(x => x.CreatedAt)
            .AsNoTracking()
            .FirstOrDefault();

        if (dataSource == null)
        {
            throw new Exception($"DataSource not found, {sourceImportWorker.Name}");
        }

        var newRecords = dataSource.Records!.Select(record => new Record
        {
            Id = Guid.NewGuid(),
            Code = record.Code,
            Desc1 = record.Desc1,
            Value1 = record.Value1,
            CreatedAt = DateTime.UtcNow,
            ModifiedAt = DateTime.UtcNow
        }).ToList();

        if (isNew)
        {
            _db.Layers.Add(processedLayer);
        }
        else
        {
            _db.Layers.Update(processedLayer);
        }

        SaveRecords(processedLayer.Id, newRecords);
        _db.SaveChanges();

        _logger.LogInformation("T4SingleSource: Successfully completed processing for {ProcessWorkerName} ({ProcessWorkerId})", 
            processWorker.Name, processWorker.Id);
    }

    private void SaveRecords(Guid layerId, ICollection<Record> records)
    {
        var toDelete = _db.Records.Where(x => x.LayerId == layerId).ToList();
        if (toDelete.Count > 0)
        {
            _db.Records.RemoveRange(toDelete);
        }

        foreach (var record in records)
        {
            record.CreatedById = Guid.Parse("F392209E-123E-4651-A5A4-0B1D6CF9FF9D");
            record.CreatedAt = DateTime.UtcNow;
            record.ModifiedById = Guid.Parse("F392209E-123E-4651-A5A4-0B1D6CF9FF9D");
            record.ModifiedAt = DateTime.UtcNow;
            record.LayerId = layerId;
            _db.Records.Add(record);
        }

        _logger.LogDebug("T4SingleSource: Saved {RecordCount} records for layer {LayerId}", records.Count, layerId);
    }
}