-- CỰC KỲ NGUY HIỂM: Phản mẫu chết người trong hệ thống tài chính
BEGIN;
  UPDATE accounts SET balance = balance - 500000 WHERE id = 'alice_acc';
  UPDATE accounts SET balance = balance + 500000 WHERE id = 'bob_acc';
COMMIT;

-- 1. Bảng Quản Lý Tài Khoản (Metadata và cấu hình tỷ lệ tiền tệ)
CREATE TABLE accounts (
    id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
    account_number VARCHAR(34) NOT NULL UNIQUE,
    holder_id UUID NOT NULL,
    currency CHAR(3) NOT NULL, -- ISO 4217 (ví dụ: 'VND', 'USD')
    scale SMALLINT NOT NULL DEFAULT 0, -- 0 cho VND, 2 cho USD/EUR
    account_category VARCHAR(16) NOT NULL CHECK (account_category IN ('ASSET', 'LIABILITY', 'EQUITY', 'REVENUE', 'EXPENSE')),
    is_active BOOLEAN NOT NULL DEFAULT TRUE,
    created_at TIMESTAMPTZ NOT NULL DEFAULT clock_timestamp()
);

-- 2. Bảng Đầu Bút Toán Giao Dịch (Header Table)
CREATE TABLE journal_transactions (
    id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
    idempotency_key VARCHAR(128) NOT NULL UNIQUE,
    reference_id VARCHAR(64) NOT NULL,
    description TEXT NOT NULL,
    posted_at TIMESTAMPTZ NOT NULL DEFAULT clock_timestamp()
);

-- 3. Bảng Chân Bút Toán Bất Biến (Journal Postings - Lines Table)
CREATE TABLE journal_postings (
    id BIGSERIAL PRIMARY KEY,
    transaction_id UUID NOT NULL REFERENCES journal_transactions(id) ON DELETE RESTRICT,
    account_id UUID NOT NULL REFERENCES accounts(id) ON DELETE RESTRICT,
    amount BIGINT NOT NULL, -- Đơn vị minor (Dương là NỢ, Âm là CÓ)
    sequence_num SMALLINT NOT NULL,
    created_at TIMESTAMPTZ NOT NULL DEFAULT clock_timestamp()
);

-- Chặn hoàn toàn thao tác UPDATE và DELETE trên tầng sổ cái
CREATE OR REPLACE RULE no_update_postings AS ON UPDATE TO journal_postings DO INSTEAD NOTHING;
CREATE OR REPLACE RULE no_delete_postings AS ON DELETE TO journal_postings DO INSTEAD NOTHING;

-- 4. Trigger Kiểm Tra Cân Bằng Bút Toán Nguyên Tử (Deferred Trigger)
CREATE OR REPLACE FUNCTION verify_transaction_balance() RETURNS TRIGGER AS $$
DECLARE
    net_sum BIGINT;
BEGIN
    SELECT COALESCE(SUM(amount), 0) INTO net_sum
    FROM journal_postings
    WHERE transaction_id = NEW.transaction_id;

    IF net_sum <> 0 THEN
        RAISE EXCEPTION 'Vi phạm bất biến kế toán: Giao dịch % có tổng lệch % (yêu cầu = 0)', NEW.transaction_id, net_sum;
    END IF;
    RETURN NEW;
END;
$$ LANGUAGE plpgsql;

CREATE CONSTRAINT TRIGGER trg_verify_journal_balance
AFTER INSERT ON journal_postings
DEFERRABLE INITIALLY DEFERRED
FOR EACH ROW
EXECUTE FUNCTION verify_transaction_balance();

// Package ledger hiện thực bộ xử lý giao dịch sổ cái kế toán kép chuẩn SOTA 2027.
// Sử dụng Go 1.25: Range-over-func iterators, custom error types, và atomic ring-buffer batching.
package main

import (
	"context"
	"errors"
	"fmt"
	"iter"
	"log/slog"
	"os"
	"sync"
	"time"

	tb "github.com/tigerbeetle/tigerbeetle-go"
	tb_types "github.com/tigerbeetle/tigerbeetle-go/pkg/types"
)

// Khai báo các lỗi miền tài chính chuẩn mực
var (
	ErrUnbalancedJournal   = errors.New("bút toán sổ cái không cân bằng (tổng Nợ khác tổng Có)")
	ErrInsufficientBalance = errors.New("số dư khả dụng không đủ để thực hiện giao dịch")
	ErrTransferTimeout     = errors.New("giao dịch pending đã hết hạn phong tỏa")
	ErrDuplicateTransfer   = errors.New("idempotency key đã tồn tại trong sổ cái")
)

// JournalLeg định nghĩa một chân giao dịch nợ hoặc có trong sổ cái
type JournalLeg struct {
	AccountID tb_types.Uint128
	Amount    int64 // Dương: Nợ (Debit), Âm: Có (Credit)
}

// JournalEntry chứa tập hợp các chân giao dịch đối ứng tạo nên một bút toán hoàn chỉnh
type JournalEntry struct {
	TransactionID tb_types.Uint128
	LedgerID      uint32
	Legs          []JournalLeg
	Timestamp     time.Time
}

// ValidateBalance kiểm tra bất biến toán học: Tổng Nợ + Tổng Có == 0
// Ứng dụng Go 1.25 range-over-func iterator để duyệt zero-alloc
func (entry *JournalEntry) ValidateBalance() error {
	var netSum int64
	for leg := range entry.AllLegs() {
		netSum += leg.Amount
	}
	if netSum != 0 {
		return fmt.Errorf("%w: sai lệch %d minor units", ErrUnbalancedJournal, netSum)
	}
	return nil
}

// AllLegs trả về một iter.Seq[JournalLeg] theo chuẩn iterator Go 1.25
func (entry *JournalEntry) AllLegs() iter.Seq[JournalLeg] {
	return func(yield func(JournalLeg) bool) {
		for _, leg := range entry.Legs {
			if !yield(leg) {
				return
			}
		}
	}
}

// LedgerEngine quản lý kết nối và thực thi giao dịch sổ cái với TigerBeetle
type LedgerEngine struct {
	client tb.Client
	logger *slog.Logger
	mu     sync.RWMutex
}

// NewLedgerEngine khởi tạo engine kết nối cụm TigerBeetle
func NewLedgerEngine(clusterID tb_types.Uint128, addresses []string, logger *slog.Logger) (*LedgerEngine, error) {
	client, err := tb.NewClient(clusterID, addresses)
	if err != nil {
		return nil, fmt.Errorf("không thể kết nối cụm TigerBeetle: %w", err)
	}
	return &LedgerEngine{
		client: client,
		logger: logger,
	}, nil
}

// ExecuteTwoPhaseTransfer khởi tạo giao dịch phong tỏa số dư (Pending Transfer)
func (e *LedgerEngine) ExecuteTwoPhaseTransfer(
	ctx context.Context,
	txID tb_types.Uint128,
	debitAcc tb_types.Uint128,
	creditAcc tb_types.Uint128,
	amount uint64,
	timeoutSec uint32,
) error {
	e.logger.Info("Bắt đầu khởi tạo giao dịch phong tỏa (Pending)",
		"tx_id", txID.String(),
		"amount", amount,
		"timeout_sec", timeoutSec,
	)

	transfers := []tb_types.Transfer{
		{
			ID:              txID,
			DebitAccountID:  debitAcc,
			CreditAccountID: creditAcc,
			Amount:          tb_types.ToUint128(amount),
			Ledger:          1,    // Sổ Cái Tiền Gửi VND
			Code:            1001, // Mã nghiệp vụ Chuyển Khoản Trực Tuyến
			Flags:           tb_types.TransferFlags{Pending: true}.ToUint16(),
			Timeout:         timeoutSec,
		},
	}

	results, err := e.client.CreateTransfers(transfers)
	if err != nil {
		e.logger.Error("Lỗi I/O khi gửi yêu cầu tới TigerBeetle", "err", err)
		return err
	}

	for _, res := range results {
		if res.Result != tb_types.TransferOK {
			e.logger.Warn("TigerBeetle từ chối giao dịch", "code", res.Result)
			return fmt.Errorf("giao dịch thất bại: mã lỗi %d", res.Result)
		}
	}

	e.logger.Info("Phong tỏa số dư thành công", "tx_id", txID.String())
	return nil
}

// PostPendingTransfer quyết toán chính thức giao dịch đã phong tỏa (Post Pending)
func (e *LedgerEngine) PostPendingTransfer(ctx context.Context, postID, pendingTxID tb_types.Uint128) error {
	e.logger.Info("Quyết toán giao dịch phong tỏa (Post Pending)", "pending_tx_id", pendingTxID.String())

	transfers := []tb_types.Transfer{
		{
			ID:        postID,
			PendingID: pendingTxID,
			Flags:     tb_types.TransferFlags{PostPendingTransfer: true}.ToUint16(),
		},
	}

	results, err := e.client.CreateTransfers(transfers)
	if err != nil {
		return err
	}

	for _, res := range results {
		if res.Result != tb_types.TransferOK {
			return fmt.Errorf("lỗi quyết toán giao dịch %d: mã lỗi %d", pendingTxID, res.Result)
		}
	}

	e.logger.Info("Giao dịch đã quyết toán thành công vào sổ cái", "tx_id", postID.String())
	return nil
}

// VoidPendingTransfer giải phóng phong tỏa và hoàn tiền về tài khoản nguồn (Void Pending)
func (e *LedgerEngine) VoidPendingTransfer(ctx context.Context, voidID, pendingTxID tb_types.Uint128) error {
	e.logger.Info("Hủy phong tỏa giao dịch (Void Pending)", "pending_tx_id", pendingTxID.String())

	transfers := []tb_types.Transfer{
		{
			ID:        voidID,
			PendingID: pendingTxID,
			Flags:     tb_types.TransferFlags{VoidPendingTransfer: true}.ToUint16(),
		},
	}

	results, err := e.client.CreateTransfers(transfers)
	if err != nil {
		return err
	}

	for _, res := range results {
		if res.Result != tb_types.TransferOK {
			return fmt.Errorf("lỗi hủy phong tỏa giao dịch %d: mã lỗi %d", pendingTxID, res.Result)
		}
	}

	e.logger.Info("Đã giải phóng phong tỏa thành công", "void_id", voidID.String())
	return nil
}

func main() {
	logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelInfo}))
	logger.Info("Khởi động mô-đun Sổ Cái Bất Biến Core Banking Go 1.25")

	// Ví dụ kiểm tra tính cân bằng của Bút toán sổ cái
	entry := JournalEntry{
		TransactionID: tb_types.ToUint128(9901),
		LedgerID:      1,
		Timestamp:     time.Now(),
		Legs: []JournalLeg{
			{AccountID: tb_types.ToUint128(101), Amount: 500000},  // Nợ Alice: 500,000 VND
			{AccountID: tb_types.ToUint128(202), Amount: -500000}, // Có Bob: 500,000 VND
		},
	}

	if err := entry.ValidateBalance(); err != nil {
		logger.Error("Phát hiện bất biến bị vi phạm", "err", err)
	} else {
		logger.Info("Bút toán hợp lệ: Tổng Nợ bằng Tổng Có tuyệt đối")
	}
}