ANTOOZA-BACKEND/Infrastructure/StockData/SeibroSupplyImport.cs

using Application.Abstractions.Data;
using Domain.Entities.Stocks;
using Microsoft.EntityFrameworkCore;

namespace Infrastructure.StockData;

/// <summary>
/// SeibroSupplySyncService 의 DB 반영 로직 분리 (테스트 가능) — HTTP 없이 파서 출력(fixture)으로 검증한다
/// (SeibroDividendImport 패턴 미러링). 다섯 엔티티(대차·보호예수·증감·유통변경·비상장) upsert / window delete+insert 담당.
/// </summary>
public static class SeibroSupplyImport
{
    /// <summary>
    /// getSlbDealingByIsin 결과를 SecuritiesLending 로 upsert — UQ (Isin, StdDt). 응답엔 Isin/StdDt 가 없어 요청값을 스탬핑한다.
    /// rows 는 per-ISIN×일 1콜 결과(보통 0~1행). 빈 결과면 no-op.
    /// </summary>
    public static async Task<(int Inserted, int Updated)> UpsertLendingAsync(IAppDbContext db, string isin, DateOnly stdDt, IReadOnlyList<SeibroSlbDealingParser.Row> rows, CancellationToken ct)
    {
        if (rows.Count == 0)
        {
            return (0, 0);
        }

        var row = rows[0]; // 한 종목·일에 1행 (KSD 중개거래)
        var entity = await db.SecuritiesLending.FirstOrDefaultAsync(c => c.Isin == isin && c.StdDt == stdDt, ct);
        var inserted = 0;
        var updated = 0;

        if (entity is not null)
        {
            var was = entity.UpdatedAt;
            entity.Update(row.TotIssuStkqty, row.TrQty, row.MatcQty, row.RedQty, row.SlbTrRemQty, row.FrinerLendRemAmt, row.FrinerLendRemRatio, row.NativeLendRemAmt, row.NativeLendRemRatio, row.FrinerBrwRemAmt, row.FrinerBrwRemRatio, row.NativeBrwRemAmt, row.NativeBrwRemRatio);
            if (entity.UpdatedAt != was)
            {
                updated++;
            }
        }
        else
        {
            var created = SecuritiesLending.Create(isin, stdDt, row.TotIssuStkqty, row.TrQty, row.MatcQty, row.RedQty, row.SlbTrRemQty, row.FrinerLendRemAmt, row.FrinerLendRemRatio, row.NativeLendRemAmt, row.NativeLendRemRatio, row.FrinerBrwRemAmt, row.FrinerBrwRemRatio, row.NativeBrwRemAmt, row.NativeBrwRemRatio);
            await db.SecuritiesLending.AddAsync(created, ct);
            inserted++;
        }

        await db.SaveChangesAsync(ct);
        return (inserted, updated);
    }

    /// <summary>
    /// getSafeDpDutyDepoStatus 결과를 LockupEvent 로 window(예수일 begin~end 범위) delete+insert 반영.
    /// 날짜 스윕(SAFEDP_DT 기준)이라 자연키 idempotent 대신 창 재구성으로 삭제분 반영. begin/end 는 이번 스윕 창.
    /// </summary>
    public static async Task<(int Inserted, int Deleted)> ReplaceLockupWindowAsync(IAppDbContext db, DateOnly begin, DateOnly end, IReadOnlyList<SeibroLockupParser.Row> rows, CancellationToken ct)
    {
        var deleted = await db.LockupEvent.Where(c => c.SafedpDt >= begin && c.SafedpDt <= end).ExecuteDeleteAsync(ct);

        // 창 내 중복 자연키(같은 회사·종목·예수일·업무·사유) 방어 — 첫 행만 채택
        var seen = new HashSet<(int, string, DateOnly, string, string)>();
        var inserted = 0;

        foreach (var row in rows)
        {
            if (row.SafedpDt < begin || row.SafedpDt > end)
            {
                continue; // 스윕 창 밖(회사번호 지정 시 3년 반환 등) — 다른 창이 담당
            }

            var key = (row.IssucoCustno, row.ShotnIsin, row.SafedpDt, row.OccrSeq, row.DutySafedpRacd);
            if (!seen.Add(key))
            {
                continue;
            }

            var created = LockupEvent.Create(row.IssucoCustno, row.KorSecnNm, row.SecnKacd, row.ShotnIsin, row.MartTpcd, row.OccrSeq, row.SafedpDt, row.SafedpQty, row.DutySafedpRacd, row.ReturnDt, row.ReturnQty, row.TotalStkCnt);
            await db.LockupEvent.AddAsync(created, ct);
            inserted++;
        }

        await db.SaveChangesAsync(ct);
        return (inserted, deleted);
    }

    /// <summary>
    /// getStkIncdecDetails 결과를 StockIssuanceChange 로 upsert — UQ (Isin, SecnIssuNtimes, IssuDt, SecnIssuRacd).
    /// rows 는 rolling per-회사(SHOTN_ISIN) 1콜 결과(한 종목 전 발행이력).
    /// </summary>
    public static async Task<(int Inserted, int Updated)> UpsertIssuanceChangesAsync(IAppDbContext db, IReadOnlyList<SeibroStkIncdecParser.Row> rows, CancellationToken ct)
    {
        if (rows.Count == 0)
        {
            return (0, 0);
        }

        var isins = rows.Select(c => c.Isin).Distinct().ToList();
        var existing = await db.StockIssuanceChange.Where(c => isins.Contains(c.Isin)).ToListAsync(ct);
        var byKey = existing.ToDictionary(c => (c.Isin, c.SecnIssuNtimes, c.IssuDt, c.SecnIssuRacd));
        var inserted = 0;
        var updated = 0;

        foreach (var row in rows)
        {
            var key = (row.Isin, row.SecnIssuNtimes, row.IssuDt, row.SecnIssuRacd);

            if (byKey.TryGetValue(key, out var entity))
            {
                var was = entity.UpdatedAt;
                entity.Update(row.IssucoCustno, row.SecnKacd, row.Issuprc, row.IssuQty, row.SecnIssuNm, row.ListDt);
                if (entity.UpdatedAt != was)
                {
                    updated++;
                }
            }
            else
            {
                var created = StockIssuanceChange.Create(row.IssucoCustno, row.Isin, row.SecnKacd, row.SecnIssuNtimes, row.IssuDt, row.Issuprc, row.IssuQty, row.SecnIssuRacd, row.SecnIssuNm, row.ListDt);
                await db.StockIssuanceChange.AddAsync(created, ct);
                byKey[key] = created;
                inserted++;
            }
        }

        await db.SaveChangesAsync(ct);
        return (inserted, updated);
    }

    /// <summary>
    /// getStkListInfo 결과를 ListingChangeEvent 로 window(적용일 begin~end 범위) delete+insert 반영.
    /// 월 윈도우 스윕(ALT_BEGIN_DT~ALT_EXPRY_DT)이라 창 재구성으로 삭제분 반영.
    /// </summary>
    public static async Task<(int Inserted, int Deleted)> ReplaceListingChangeWindowAsync(IAppDbContext db, DateOnly begin, DateOnly end, IReadOnlyList<SeibroStkListParser.Row> rows, CancellationToken ct)
    {
        var deleted = await db.ListingChangeEvent.Where(c => c.ApliDt >= begin && c.ApliDt <= end).ExecuteDeleteAsync(ct);

        var seen = new HashSet<(string, DateOnly, string?, string?)>();
        var inserted = 0;

        foreach (var row in rows)
        {
            if (row.ApliDt < begin || row.ApliDt > end)
            {
                continue;
            }

            var key = (row.Isin, row.ApliDt, row.BfaltCirclForm, row.AfaltCirclForm);
            if (!seen.Add(key))
            {
                continue;
            }

            var created = ListingChangeEvent.Create(row.IssucoCustno, row.Isin, row.ShotnIsin, row.KorSecnNm, row.BfaltCirclForm, row.AfaltCirclForm, row.ApliDt);
            await db.ListingChangeEvent.AddAsync(created, ct);
            inserted++;
        }

        await db.SaveChangesAsync(ct);
        return (inserted, deleted);
    }

    /// <summary>
    /// getUnlistCirclInfo 결과를 UnlistedCirculation 로 upsert — UQ (Isin, StdDt). 응답엔 StdDt 가 없어 요청값을 스탬핑한다.
    /// rows 는 STD_DT 스윕 1콜 결과(그날 비상장 유통 종목들).
    /// </summary>
    public static async Task<(int Inserted, int Updated)> UpsertUnlistedAsync(IAppDbContext db, DateOnly stdDt, IReadOnlyList<SeibroUnlistCirclParser.Row> rows, CancellationToken ct)
    {
        if (rows.Count == 0)
        {
            return (0, 0);
        }

        var isins = rows.Select(c => c.Isin).Distinct().ToList();
        var existing = await db.UnlistedCirculation.Where(c => c.StdDt == stdDt && isins.Contains(c.Isin)).ToListAsync(ct);
        var byKey = existing.ToDictionary(c => c.Isin);
        var seen = new HashSet<string>();
        var inserted = 0;
        var updated = 0;

        foreach (var row in rows)
        {
            if (!seen.Add(row.Isin))
            {
                continue; // 같은 콜 내 ISIN 중복 방어
            }

            if (byKey.TryGetValue(row.Isin, out var entity))
            {
                var was = entity.UpdatedAt;
                entity.Update(row.ShotnIsin, row.RepSecnNm, row.SecnKacdNm, row.GenDepoQty, row.RthingReturnStkqty, row.AcnttrQty, row.IsElectronicSecurity);
                if (entity.UpdatedAt != was)
                {
                    updated++;
                }
            }
            else
            {
                var created = UnlistedCirculation.Create(row.Isin, stdDt, row.ShotnIsin, row.RepSecnNm, row.SecnKacdNm, row.GenDepoQty, row.RthingReturnStkqty, row.AcnttrQty, row.IsElectronicSecurity);
                await db.UnlistedCirculation.AddAsync(created, ct);
                inserted++;
            }
        }

        await db.SaveChangesAsync(ct);
        return (inserted, updated);
    }
}