Infrastructure,
Simplified.

// <auto-generated> by Kanject DynamoDB Source Generator
#nullable enable
using System.Runtime.CompilerServices;
using Kanject.Core.Api.Abstractions.Models;
using Kanject.Core.NoSqlDatabase.Provider.DynamoDb.Abstractions.Extensions;
using Kanject.Core.NoSqlDatabase.Provider.DynamoDb.Abstractions.Interfaces;
using Kanject.Core.NoSqlDatabase.Provider.DynamoDb.Abstractions.Models;
using Kanject.Core.NoSqlDatabase.Provider.DynamoDbV2;
using Demo.Marketplace.Data.DbContexts;
using Demo.Marketplace.Data.Entities.SellerSettings;

namespace Demo.Marketplace.Data.Repositories;

public record struct SellerSettingRepositoryItemCollection(
    SellerSettingEntity? SellerSetting,
    SellerBundleDiscountSettingEntity? SellerBundleDiscountSetting);

public enum SellerSettingRepositoryItemCollectionType
{
    SellerSetting = 0,
    SellerBundleDiscountSetting = 1,
}

public partial class SellerSettingRepository
    : Repository<SellerSettingEntity>
{
    public SellerSettingRepository() {}

    public SellerSettingRepository(MarketplaceDbContext dbContext)
        : base(dbContext)
    {
        SetUnitOfWork(new UnitOfWork(dbContext));
    }

    protected QueryConfig QueryConfigInstance
    {
        get
        {
            var queryConfig = new QueryConfig { TableName = CurrentTableName };
            if (DbContext is not IDynamoDbContext dbContext)
                return queryConfig;

            if (dbContext.ModelBuilder?.PartitionKey is not null)
                queryConfig.AddPartitionKey(dbContext.ModelBuilder.PartitionKey);
            if (dbContext.ModelBuilder?.SortKey is not null)
                queryConfig.AddSortKey(dbContext.ModelBuilder.SortKey);

            return queryConfig;
        }
    }

    public void BeginTransaction() => UnitOfWork.BeginTransaction();

    public Task CommitAsync(CancellationToken? cancellationToken = null)
        => UnitOfWork.TransactionExpressions.Count > 0
            ? UnitOfWork.CommitAsync(cancellationToken)
            : Task.CompletedTask;

    private SellerSettingRepositorySellerBundleDiscountSettingEntityRepository?
        _sellerBundleDiscountSetting;

    public SellerSettingRepositorySellerBundleDiscountSettingEntityRepository
        SellerBundleDiscountSetting
    {
        get
        {
            _sellerBundleDiscountSetting ??= DbContext is not null
                ? new SellerSettingRepositorySellerBundleDiscountSettingEntityRepository(
                    Unsafe.As<MarketplaceDbContext>(DbContext))
                : new SellerSettingRepositorySellerBundleDiscountSettingEntityRepository();

            if (UnitOfWork is not null)
                _sellerBundleDiscountSetting.SetUnitOfWork(UnitOfWork);

            return _sellerBundleDiscountSetting;
        }
    }

    public Task<SellerSettingEntity?> FindSellerSettingAsync(
        Guid sellerUserId,
        bool isConsistentRead = true,
        QueryConfig? queryConfig = null)
    {
        queryConfig ??= QueryConfigInstance;
        queryConfig.AddPartitionKeyValue(
            SellerSettingEntityKeys.ComposeSellerSettingPartitionKey(sellerUserId));
        queryConfig.AddSortKeyValue(
            DbFunc.Equals(SellerSettingEntityKeys.ComposeSellerSettingSortKey()));

        return GetSingleAsync(
            queryConfig.ReturnEventualConsistentData(!isConsistentRead));
    }

    // + 1,100 lines: typed overloads, update/insert helpers,
    // entity-repository methods, key composers, checks, and DI registration.
}