DEV-MERGE-LEGACY-BACKEND-2026/internal/academic/repository/divide_repo.go

// 分账树仓储（t_user_divide_into / t_user_divide_into_log）——AC-09。
//
// 依据：unified/03-migration/academic-domain-design.md §3.3 #22~#25（分账 4 SP）
// / §1.7「全局无作用域表·例外二」/ §11.1 拍板 11-12。
//
// **本仓储不继承 BaseRepo**（设计 §11.1 拍板：全局推广体系无 institution_id/campus_id）：
//   - 两表无租户列，`Scoped` 无从下手（叠 institution_id 会 Unknown column）；
//   - 越权防线改为 **Handler 角色白名单（super_admin/academic）+ Service 显式数据范围参数**，
//     本文件的数据访问方法因此**不接受**作用域参数，评审时必须显式标注"平台级/全局数据"；
//   - 禁止把本仓储用于任何机构/校区维度的报表裁剪（需要裁剪时按 user_id 精确取数）。
//
// 表语义（Nested Set）：
//
//	t_user_divide_into     一人一节点（PK user_id）；root_user_id 标记所属树；
//	                       left_val/right_val 为嵌套集区间（恒有 left < right）；layer 根=1。
//	t_user_divide_into_log 分成明细（只增；type=1 同 user+student / type=2 同 user+student+order 幂等）。
//
// 事务纪律：写方法一律接收 `tx *gorm.DB`（仓储不开事务，事务边界在 DivideService）。
package repository

import (
	"context"
	"errors"
	"strings"
	"time"

	"gorm.io/gorm"

	"yuansiwei/unified-backend/internal/academic/model"
	"yuansiwei/unified-backend/internal/academic/money"
	infraerrors "yuansiwei/unified-backend/internal/infra/errors"
)

// DivideRepo 分账树仓储（全局数据，**不继承 BaseRepo**，见文件头）。
type DivideRepo struct {
	// DB 数据库句柄（连接池由 infra/db 统一建立；nil 时任何查询返回 ErrInternal）。
	DB *gorm.DB
}

// NewDivideRepo 构造分账树仓储。
//
// 入参：db *gorm.DB——连接池句柄（必填）。
// 出参：*DivideRepo。
func NewDivideRepo(db *gorm.DB) *DivideRepo { return &DivideRepo{DB: db} }

// db 取查询句柄（ctx 内含 tx 时复用同一事务，保证事务内读一致）。
//
// 入参：ctx context.Context（可 nil）。出参：(*gorm.DB, error)；未装配 DB → ErrInternal。
func (r *DivideRepo) db(ctx context.Context) (*gorm.DB, error) {
	if r == nil || r.DB == nil {
		return nil, infraerrors.WithCause(infraerrors.ErrInternal, errString("repository: DivideRepo.DB 为 nil"))
	}
	if tx := TxFromContext(ctx); tx != nil {
		return tx, nil
	}
	if ctx == nil {
		return r.DB, nil
	}
	return r.DB.WithContext(ctx), nil
}

// GetByUserID 取分账树节点（主键 user_id）。
//
// 入参：ctx 请求上下文；userID 节点用户（ULID；空 → ErrBadParam）。
// 出参：
//   - *model.UserDivideInto：节点（left_val/right_val/layer/quota 原样返回，金额为字符串）；
//   - ErrNotFound（3001）：该用户尚未进入分账体系（**不是**错误路径，调用方按业务分支处理）；
//   - ErrBadParam / ErrDBError。
func (r *DivideRepo) GetByUserID(ctx context.Context, userID string) (*model.UserDivideInto, error) {
	if strings.TrimSpace(userID) == "" {
		return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: GetByUserID 需要 user_id"))
	}
	db, err := r.db(ctx)
	if err != nil {
		return nil, err
	}
	var out model.UserDivideInto
	if err := db.Where("user_id = ?", userID).First(&out).Error; err != nil {
		return nil, wrapQueryErr(err, "GetByUserID(t_user_divide_into)")
	}
	return &out, nil
}

// GetSubtree 取节点子树（严格位于 (leftVal, rightVal) 区间内的后代）。
//
// 入参：
//   - ctx 请求上下文；
//   - leftVal/rightVal int32：父节点的左右值（Nested Set 区间；rightVal <= leftVal → ErrBadParam）；
//   - divideType int8：分成类型过滤（model.DivideTypeOneOff / DivideTypePerLesson；0 = 不筛类型）。
//
// 出参：
//   - []model.UserDivideInto：后代节点，按 left_val 升序（= 树的深度优先前序，便于逐层遍历）；
//   - ErrBadParam / ErrDBError。
//
// 用途：额度级联下降（update_quota）/ 子树统计。**不含**父节点自身。
func (r *DivideRepo) GetSubtree(ctx context.Context, leftVal, rightVal int32, divideType int8) ([]model.UserDivideInto, error) {
	if leftVal < 0 || rightVal <= leftVal {
		return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: GetSubtree 左右值区间非法"))
	}
	db, err := r.db(ctx)
	if err != nil {
		return nil, err
	}
	q := db.Model(&model.UserDivideInto{}).
		Where("left_val > ? AND right_val < ?", leftVal, rightVal)
	if divideType > 0 {
		q = q.Where("type = ?", divideType)
	}
	var rows []model.UserDivideInto
	if err := q.Order("left_val asc").Find(&rows).Error; err != nil {
		return nil, infraerrors.WithCause(infraerrors.ErrDBError, err)
	}
	return rows, nil
}

// ShiftRight 右值扩张：`right_val = right_val + delta WHERE root_user_id = ? AND right_val >= threshold`。
//
// 入参：
//   - ctx 请求上下文；
//   - tx 事务句柄（**必传**）；
//   - rootUserID **所属树根**（ULID；**必填**，见下方红线）；
//   - threshold int32：扩张阈值（新节点插入点 = 父节点右值；父与全部祖先的 right 均 >= 它，一次覆盖）；
//   - delta int32：增量（插入子节点固定 +2；撤销固定 -2）。
//
// 出参：int64 受影响行数；ErrBadParam / ErrDBError。
//
// 红线（audit V-02，dump `yuansiwei_finance.sql:763707-763708`）：
//
//	dump 原文的位移语句带 `WHERE root_userid = rootUserid AND left_val >= parentRight`。
//	**漏掉 root 限定的后果**：多棵树共用一张表（各自 root 的左右值区间可能重叠/相邻），
//	同输入会产出不同的 left/right → 双跑快照 diff 必然不一致，且会污染别的树的区间。
//	（同一 dump 的另一处 left 位移同理，行号 763708。）
//
// 与 ShiftLeft **必须同一事务、同一 rootUserID 与 threshold**（两步各自只动一列，
// 合成标准嵌套集插入算法；拆事务会让树在中间态不可用）。
func (r *DivideRepo) ShiftRight(ctx context.Context, tx *gorm.DB, rootUserID string, threshold int32, delta int32) (int64, error) {
	return r.shift(ctx, tx, "right_val", rootUserID, threshold, delta)
}

// ShiftLeft 左值扩张：`left_val = left_val + delta WHERE root_user_id = ? AND left_val >= threshold`。
//
// 入参：语义同 ShiftRight（只作用于 left_val 列——即"位于插入点右侧的兄弟及其后代"）。
// 出参：int64 受影响行数；ErrBadParam / ErrDBError。
func (r *DivideRepo) ShiftLeft(ctx context.Context, tx *gorm.DB, rootUserID string, threshold int32, delta int32) (int64, error) {
	return r.shift(ctx, tx, "left_val", rootUserID, threshold, delta)
}

// shift 左右值扩张的公共实现（列名由内部常量传入，**不接受外部字符串**，防注入）。
//
// 入参：col 列名（"left_val"/"right_val"）；rootUserID 树根（必填）；
// threshold/delta 语义见 ShiftRight。
// 出参：int64 受影响行数；ErrBadParam / ErrDBError。
func (r *DivideRepo) shift(ctx context.Context, tx *gorm.DB, col, rootUserID string, threshold int32, delta int32) (int64, error) {
	if err := requireTx(tx, "shift("+col+")"); err != nil {
		return 0, err
	}
	if col != "left_val" && col != "right_val" {
		return 0, infraerrors.WithCause(infraerrors.ErrInternal, errString("repository: shift 列名非法"))
	}
	if strings.TrimSpace(rootUserID) == "" {
		// root 限定不可省：漏了会串改其它树的区间（audit V-02）
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam,
			errString("repository: shift 需要 root_user_id（多树隔离，dump 763707）"))
	}
	if threshold < 0 {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: shift 阈值不能为负"))
	}
	if delta == 0 {
		return 0, nil
	}
	res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
		Where("root_user_id = ?", strings.TrimSpace(rootUserID)).
		Where(col+" >= ?", threshold).
		UpdateColumn(col, gorm.Expr(col+" + ?", delta))
	if res.Error != nil {
		return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
	}
	return res.RowsAffected, nil
}

// InsertNode 插入分账树节点（新节点左右值由 Service 用嵌套集算法算出）。
//
// 入参：
//   - ctx 请求上下文；
//   - tx 事务句柄（**必传**）；
//   - node 节点（UserID 非空；必须满足 LeftVal < RightVal，否则 ErrBadParam）。
//
// 出参：nil 成功；ErrBadParam / ErrDBError。
//
// 红线：与 ShiftRight/ShiftLeft **必须同事务**；已存在同 user_id 节点 → 唯一键冲突（ErrDBError，
// 调用方按"已存在分支"重入，不得重试插入）。
func (r *DivideRepo) InsertNode(ctx context.Context, tx *gorm.DB, node *model.UserDivideInto) error {
	if err := requireTx(tx, "InsertNode"); err != nil {
		return err
	}
	if node == nil || strings.TrimSpace(node.UserID) == "" {
		return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: InsertNode 需要 user_id"))
	}
	if node.LeftVal >= node.RightVal {
		return infraerrors.WithCause(infraerrors.ErrBadParam,
			errString("repository: InsertNode 违反 Nested Set 不变式 left_val < right_val"))
	}
	if err := tx.WithContext(ctxOrBackground(ctx)).Create(node).Error; err != nil {
		return infraerrors.WithCause(infraerrors.ErrDBError, err)
	}
	return nil
}

// InsertLog 写入分成明细（**只增**）。
//
// 入参：
//   - ctx 请求上下文；
//   - tx 事务句柄（**必传**）；
//   - log 明细（Type∈{1,2}、UserID/StudentID 非空、Money 为元字符串）。
//
// 出参：nil 成功；ErrBadParam / ErrDBError。
//
// 红线：与节点 total_income 累加（IncTotalIncome）必须同事务（否则"明细有、收入没加"）。
func (r *DivideRepo) InsertLog(ctx context.Context, tx *gorm.DB, log *model.UserDivideIntoLog) error {
	if err := requireTx(tx, "InsertLog"); err != nil {
		return err
	}
	if log == nil || strings.TrimSpace(log.UserID) == "" || strings.TrimSpace(log.StudentID) == "" {
		return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: InsertLog 需要 user_id/student_id"))
	}
	if log.Type != model.DivideTypeOneOff && log.Type != model.DivideTypePerLesson {
		return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: InsertLog 分成类型非法"))
	}
	if err := tx.WithContext(ctxOrBackground(ctx)).Create(log).Error; err != nil {
		return infraerrors.WithCause(infraerrors.ErrDBError, err)
	}
	return nil
}

// DeleteLog 物理删除分成明细（撤销分支二：回滚 total_income 后删行）。
//
// 入参：
//   - ctx 请求上下文；
//   - tx 事务句柄（**必传**）；
//   - id 明细 ID（<=0 → ErrBadParam）。
//
// 出参：nil 成功/行不存在（幂等）；ErrBadParam / ErrDBError。
//
// 红线（设计 §3.3 #23）：旧语义允许物理删除本表行（与"流水只增"纪律的唯一例外），
// 但必须与 total_income 回滚同事务；删除前须判定"是否已有首次扣费上课记录"（改绑分支优先）。
func (r *DivideRepo) DeleteLog(ctx context.Context, tx *gorm.DB, id int64) error {
	if err := requireTx(tx, "DeleteLog"); err != nil {
		return err
	}
	if id <= 0 {
		return infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: DeleteLog 需要 id"))
	}
	if err := tx.WithContext(ctxOrBackground(ctx)).Where("id = ?", id).
		Delete(&model.UserDivideIntoLog{}).Error; err != nil {
		return infraerrors.WithCause(infraerrors.ErrDBError, err)
	}
	return nil
}

// FindLog 按**幂等键**查分成明细（两级幂等判定唯一入口）。
//
// 入参：
//   - ctx 请求上下文；
//   - userID 获得分成的用户（ULID）；studentID 报课学生（ULID）；divideType 分成类型；
//   - orderNo 订单号（type=2 的上课记录 id 字符串；**type=1 时忽略**——一次付清不区分订单）。
//
// 出参：
//   - *model.UserDivideIntoLog：命中行；
//   - ErrNotFound：未命中（调用方据此判定"未生成"，非异常）；
//   - ErrBadParam / ErrDBError。
//
// 幂等键（dump divide_into_log_add 口径，**不可改**）：
//   - type=1（一次付清）：(user_id, student_id, type)；
//   - type=2（按课次）  ：(user_id, student_id, type, order_no)。
func (r *DivideRepo) FindLog(ctx context.Context, userID, studentID string, divideType int8, orderNo string) (*model.UserDivideIntoLog, error) {
	if strings.TrimSpace(userID) == "" || strings.TrimSpace(studentID) == "" {
		return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: FindLog 需要 user_id/student_id"))
	}
	db, err := r.db(ctx)
	if err != nil {
		return nil, err
	}
	q := db.Model(&model.UserDivideIntoLog{}).
		Where("user_id = ? AND student_id = ? AND type = ?", userID, studentID, divideType)
	if divideType == model.DivideTypePerLesson {
		q = q.Where("order_no = ?", strings.TrimSpace(orderNo))
	}
	var out model.UserDivideIntoLog
	if err := q.Order("id asc").First(&out).Error; err != nil {
		return nil, wrapQueryErr(err, "FindLog(t_user_divide_into_log)")
	}
	return &out, nil
}

// ListLogsByStudentOrder 取某「学生 + 订单」产生的**全部**分成明细（撤销的两参入口）。
//
// 入参：ctx 请求上下文；studentID 报课学生（ULID）；orderNo 订单号（上课记录 id 字符串）。
// 出参：
//   - []model.UserDivideIntoLog：按 id 升序（生成先后；空切片表示无记录，幂等空操作）；
//   - ErrBadParam / ErrDBError。
//
// 语义（dump `divide_into_log_delete(studentid, orderid)` 两参）：同一订单的分成会沿上行链
// 生成多行（本人 + 各级上线），撤销时必须**整组处理**，故按 (student_id, order_no) 取全量，
// 而非只取一行。
func (r *DivideRepo) ListLogsByStudentOrder(ctx context.Context, studentID, orderNo string) ([]model.UserDivideIntoLog, error) {
	if strings.TrimSpace(studentID) == "" || strings.TrimSpace(orderNo) == "" {
		return nil, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ListLogsByStudentOrder 需要 student_id/order_no"))
	}
	db, err := r.db(ctx)
	if err != nil {
		return nil, err
	}
	var rows []model.UserDivideIntoLog
	if err := db.Model(&model.UserDivideIntoLog{}).
		Where("student_id = ? AND order_no = ?", studentID, orderNo).
		Order("id asc").
		Find(&rows).Error; err != nil {
		return nil, infraerrors.WithCause(infraerrors.ErrDBError, err)
	}
	return rows, nil
}

// UpdateNodeParent 改挂节点父链（只改父/根/层级，**不重排左右值**）。
//
// ⚠ **当前无业务调用方**（audit D4：dump `:763695` 的 "subid 已存在" 分支是**完全 no-op**，
// 分支体内没有任何写动作）。本方法因此**不属于 dump 流程**，仅保留给后台数据修复；
// 若将来要用于业务链路，必须先评审——它不重排左右值，改挂自带子树的节点会产生区间交叠。
//
// 入参：
//   - ctx 请求上下文；tx 事务句柄（**必传**）；
//   - userID 被改挂节点；parentUserID 新上级；rootUserID 所属树根；layer 新层级（根=1）。
//
// 出参：int64 受影响行数；ErrBadParam / ErrDBError。
//
// 红线（修正 dump 缺陷，模型注释 §UserDivideInto.root_user_id 已标注）：
// 旧 SP 在该分支**漏写 root_user_id**（留 NULL，导致子树查询/级联失效）；本方法强制显式写根。
// 遗留：不重排左右值（与原 SP 一致）——若被改挂节点自带子树，区间可能出现交叠，
// 已在交付说明中登记为遗留项（AC-09 范围内不引入区间重排，避免与迁移期数据冲突）。
func (r *DivideRepo) UpdateNodeParent(ctx context.Context, tx *gorm.DB, userID, parentUserID, rootUserID string, layer int32) (int64, error) {
	if err := requireTx(tx, "UpdateNodeParent"); err != nil {
		return 0, err
	}
	if strings.TrimSpace(userID) == "" || strings.TrimSpace(parentUserID) == "" || strings.TrimSpace(rootUserID) == "" {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateNodeParent 需要 user/parent/root"))
	}
	if layer < 1 {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateNodeParent layer 必须 >= 1"))
	}
	res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
		Where("user_id = ?", userID).
		Updates(map[string]any{
			"parent_user_id": parentUserID,
			"root_user_id":   rootUserID,
			"layer":          layer,
		})
	if res.Error != nil {
		return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
	}
	return res.RowsAffected, nil
}

// UpdateNodeQuota 更新节点分成额度（不触碰左右值/收入）。
//
// 入参：ctx；tx（**必传**）；userID 节点；quota 新额度（单位：分；负 → ErrBadParam）。
// 出参：int64 受影响行数；ErrBadParam / ErrDBError。
//
// 红线：**必须**与 ClampSubtreeQuota 同事务（否则出现"父降了、子仍高于父"的越权额度窗口）。
func (r *DivideRepo) UpdateNodeQuota(ctx context.Context, tx *gorm.DB, userID string, quota money.Amount) (int64, error) {
	if err := requireTx(tx, "UpdateNodeQuota"); err != nil {
		return 0, err
	}
	if strings.TrimSpace(userID) == "" {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateNodeQuota 需要 user_id"))
	}
	if err := quota.Validate(); err != nil {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
	}
	if quota.IsNegative() {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: 分成额度不能为负"))
	}
	res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
		Where("user_id = ?", userID).
		UpdateColumn("quota", quota.String())
	if res.Error != nil {
		return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
	}
	return res.RowsAffected, nil
}

// ClampSubtreeQuota 额度级联下降：把子树内**同分成类型**且**超过**上限的节点统一压到上限。
//
// 入参：
//   - ctx；tx（**必传**）；
//   - leftVal/rightVal int32 父节点区间（rightVal <= leftVal → ErrBadParam）；
//   - divideType 分成类型过滤（model.DivideTypeOneOff/PerLesson；**0 = 不筛**，仅 admin 全量场景）；
//   - quota 上限（单位：分；负 → ErrBadParam）。
//
// 出参：int64 受影响行数；ErrBadParam / ErrDBError。
//
// SQL（audit A6 / dump `yuansiwei_finance.sql:763872-763873` 原文条件，**缺一不可**）：
//
//	UPDATE t_user_divide_into SET quota=updateQuota
//	 WHERE left_val>uLeft AND right_val<uRight AND quota>updateQuota AND type=uType;
//
// 语义（"子不得超父"）：只降不升——低于上限的节点保持原额度（分别配置的阶梯额度不被抹平）。
// 红线：`type` 过滤不可省 —— 否则一次付清(1)与按课次(2)两条线的额度会被跨类型抹平（漏实现 = 缺陷）。
func (r *DivideRepo) ClampSubtreeQuota(ctx context.Context, tx *gorm.DB, rootUserID string, leftVal, rightVal int32, divideType int8, quota money.Amount) (int64, error) {
	if err := requireTx(tx, "ClampSubtreeQuota"); err != nil {
		return 0, err
	}
	if strings.TrimSpace(rootUserID) == "" {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ClampSubtreeQuota 需要 root_user_id"))
	}
	if leftVal < 0 || rightVal <= leftVal {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ClampSubtreeQuota 左右值区间非法"))
	}
	if divideType != 0 && divideType != model.DivideTypeOneOff && divideType != model.DivideTypePerLesson {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: ClampSubtreeQuota 分成类型非法"))
	}
	if err := quota.Validate(); err != nil {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
	}
	if quota.IsNegative() {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: 分成额度不能为负"))
	}
	q := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
		// root_user_id 过滤**必须**在：`left_val`/`right_val` 只在**同一棵树内**唯一。
		// 每棵树的 left 都从 1 起 ⇒ 多个树根的区间在数值上必然互相重叠
		//（真实数据 26 棵树、跨树重叠 325 对 = C(26,2) 满重叠）。
		// 缺这一条 ⇒ 调 A 树的额会压低 B 树里区间数值重叠的节点（台账 F25，P0）。
		Where("root_user_id = ?", strings.TrimSpace(rootUserID)).
		Where("left_val > ? AND right_val < ?", leftVal, rightVal).
		Where("quota > ?", quota.String())
	if divideType != 0 {
		q = q.Where("type = ?", divideType)
	}
	res := q.UpdateColumn("quota", quota.String())
	if res.Error != nil {
		return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
	}
	return res.RowsAffected, nil
}

// IncTotalIncome 累加节点分成总收入（上行分成 +，撤销回滚 −）。
//
// 入参：ctx；tx（**必传**）；userID 节点；delta 增量（单位：分；0 → 空操作）。
// 出参：int64 受影响行数；ErrBadParam / ErrDBError。
//
// 红线：与 InsertLog/DeleteLog 同事务；金额字符串入场（decimal 精确，零 float64）。
func (r *DivideRepo) IncTotalIncome(ctx context.Context, tx *gorm.DB, userID string, delta money.Amount) (int64, error) {
	if err := requireTx(tx, "IncTotalIncome"); err != nil {
		return 0, err
	}
	if strings.TrimSpace(userID) == "" {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: IncTotalIncome 需要 user_id"))
	}
	if delta.IsZero() {
		return 0, nil
	}
	if err := delta.Validate(); err != nil {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, err)
	}
	res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideInto{}).
		Where("user_id = ?", userID).
		UpdateColumn("total_income", gorm.Expr("total_income + ?", delta.String()))
	if res.Error != nil {
		return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
	}
	return res.RowsAffected, nil
}

// UpdateLogOrderNo 改绑分成明细的订单号（撤销分支一：保留分成，改指首次扣费上课记录）。
//
// 入参：ctx；tx（**必传**）；id 明细 ID；orderNo 新订单号（上课记录 id 字符串）；remark 备注（可空）。
// 出参：int64 受影响行数；ErrBadParam / ErrDBError。
//
// 红线（dump 口径）：改绑**不改 money**、**不回滚 total_income**（分成保留）；
// order_no 是 type=2 的幂等键组成列，改绑即改幂等语义 → 必须写审计备注（默认"推广分成"）。
func (r *DivideRepo) UpdateLogOrderNo(ctx context.Context, tx *gorm.DB, id int64, orderNo, remark string) (int64, error) {
	if err := requireTx(tx, "UpdateLogOrderNo"); err != nil {
		return 0, err
	}
	if id <= 0 || strings.TrimSpace(orderNo) == "" {
		return 0, infraerrors.WithCause(infraerrors.ErrBadParam, errString("repository: UpdateLogOrderNo 需要 id/order_no"))
	}
	// create_time 重置为 DB 当前时间（dump 改绑分支：`create_time=now()`，让改绑后的明细排到最新）
	fields := map[string]any{"order_no": orderNo, "create_time": gorm.Expr("CURRENT_TIMESTAMP")}
	if strings.TrimSpace(remark) != "" {
		fields["remark"] = remark
	}
	res := tx.WithContext(ctxOrBackground(ctx)).Model(&model.UserDivideIntoLog{}).
		Where("id = ?", id).
		Updates(fields)
	if res.Error != nil {
		return 0, infraerrors.WithCause(infraerrors.ErrDBError, res.Error)
	}
	return res.RowsAffected, nil
}

// IsNotFound 判断错误是否为仓储 ErrNotFound（供 Service 分支判断，避免各处抄 errors.Is）。
//
// 入参：err error（可 nil）。出参：true 表示"记录不存在"。
func IsNotFound(err error) bool { return err != nil && errors.Is(err, infraerrors.ErrNotFound) }

// ---------------------------------------------------------------- AC-13 查询原语（端点 38 / 43）

// ListAllNodes 取全部分账节点（Nested Set，按 `root_user_id, left_val` 排序）。
//
// 入参：ctx。
// 出参：
//   - []model.UserDivideInto：稳定顺序（同 root 内按 left 升序 ⇒ 天然是树的先序）；
//   - ErrDBError。
//
// 作用域：`t_user_divide_into` **无** institution_id / campus_id（全局推广体系，设计 §1.7 例外一），
// 故不叠 Scoped；可见性由端点角色白名单（super_admin/finance/market）控制，见 handler 注释。
//
// 为什么不在此拼树：拼树 + 树不变式校验属业务规则（layer = parent.layer+1、
// 区间嵌套或相离），放 Service 才能单测（`service.divide_tree.go`）。
func (r *DivideRepo) ListAllNodes(ctx context.Context, rootUserID string) ([]model.UserDivideInto, error) {
	db, err := r.db(ctx)
	if err != nil {
		return nil, err
	}
	q := db.Model(&model.UserDivideInto{})
	if strings.TrimSpace(rootUserID) != "" {
		// 端点 38 的 `rootUserId?`：只取某一棵树（多树共存时前端按需拉取，避免全量森林）
		q = q.Where("root_user_id = ?", strings.TrimSpace(rootUserID))
	}
	rows := make([]model.UserDivideInto, 0, 64)
	if qErr := q.Order("root_user_id ASC").Order("left_val ASC").Find(&rows).Error; qErr != nil {
		return nil, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
	}
	return rows, nil
}

// ListAncestorIDs 取某节点的**祖先链**（端点 39 的 `{ancestorIds:[]}`，设计 §3.3 #21 末尾原语）。
//
// 入参：ctx；userID 新加入的下线；leftVal/rightVal/rootUserID 其 Nested Set 区间与树根。
// 出参：[]string 祖先 user_id（**不含自身**；由内向外或由外向内按 `left_val ASC` = 由外向内）；
//
//	ErrDBError。
//
// dump 原文：`SELECT userid ... WHERE left<userLeft AND right>userRight AND root_userid=root`
// ⇒ 天然含"区间包含该节点"的所有祖先；**root 限定不可省**（D5：跨树区间数值可能重叠，
// 缺 root 会把别的树的节点算成本节点祖先）。
func (r *DivideRepo) ListAncestorIDs(
	ctx context.Context, leftVal, rightVal int32, rootUserID string,
) ([]string, error) {
	db, err := r.db(ctx)
	if err != nil {
		return nil, err
	}
	ids := make([]string, 0, 8)
	q := db.Model(&model.UserDivideInto{}).
		Where("left_val < ? AND right_val > ?", leftVal, rightVal)
	if strings.TrimSpace(rootUserID) != "" {
		q = q.Where("root_user_id = ?", strings.TrimSpace(rootUserID))
	}
	if qErr := q.Order("left_val ASC").Pluck("user_id", &ids).Error; qErr != nil {
		return nil, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
	}
	return ids, nil
}

// ListSubtreeUserIDsWithQuotaAtLeast 取子树内 `quota >= 阈值` 的 user_id（端点 40 的 `{affectedUserIds:[]}`）。
//
// 入参：ctx；leftVal/rightVal 子树区间（含自身）；quota 新额度（调用方已转为分）。
// 出参：[]string 受影响 user_id（`left_val ASC` 由外向内）；ErrDBError。
//
// dump 原文：`SELECT userid ... WHERE left>=uLeft AND right<=uRight AND quota>=newQuota`（含自身）。
// 该查询在 `update_quota` **末尾**执行 ⇒ 本方法应在 UpdateQuota 之后调用（读已提交状态）。
func (r *DivideRepo) ListSubtreeUserIDsWithQuotaAtLeast(
	ctx context.Context, leftVal, rightVal int32, quota money.Amount,
) ([]string, error) {
	db, err := r.db(ctx)
	if err != nil {
		return nil, err
	}
	ids := make([]string, 0, 8)
	if qErr := db.Model(&model.UserDivideInto{}).
		Where("left_val >= ? AND right_val <= ? AND quota >= ?", leftVal, rightVal, quota.String()).
		Order("left_val ASC").Pluck("user_id", &ids).Error; qErr != nil {
		return nil, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
	}
	return ids, nil
}

// DivideLogFilter 分账流水筛选（端点 43）。
//
// 字段语义（nil/空 = 不筛）：
//   - FindUserID 分成获得者（**表列是 `user_id`**；契约响应字段名才叫 `findUserId`）；StudentID 学生；
//   - From/To：`create_time >= from AND create_time < to`（**右开**，与契约 §1.3 一致）。
type DivideLogFilter struct {
	// FindUserID 分成获得者（ULID；空 = 不筛）。
	FindUserID string
	// StudentID 学生（ULID；空 = 不筛）。
	StudentID string
	// From 起始时间（含；nil = 不筛）。
	From *time.Time
	// To 结束时间（**不含**；nil = 不筛）。
	To *time.Time
}

// ListLogsFiltered 分页查分账流水（端点 43，带用户/学生/时间窗筛选，附总条数）。
//
// 入参：ctx；f DivideLogFilter；page PageQuery（内部钳制 pageSize ∈ [1,200]）。
// 出参：
//   - []model.UserDivideIntoLog：`create_time DESC, id DESC`；空结果非 nil 切片；
//   - int64：同条件总条数（**与列表逐字同 WHERE**，同快照）；
//   - ErrDBError。
//
// ⚠ 该端点**无 dump SP 对照**（dump 只有 add/delete/update_quota 三个 SP）⇒ 排序与窗口由 Go 定，
// 已登记为「Go 新增查询」（契约 v1.4 端点 43 注）。
func (r *DivideRepo) ListLogsFiltered(
	ctx context.Context, f DivideLogFilter, page PageQuery,
) ([]model.UserDivideIntoLog, int64, error) {
	db, err := r.db(ctx)
	if err != nil {
		return nil, 0, err
	}
	base := db.Model(&model.UserDivideIntoLog{})
	if f.FindUserID != "" {
		base = base.Where("user_id = ?", f.FindUserID)
	}
	if f.StudentID != "" {
		base = base.Where("student_id = ?", f.StudentID)
	}
	if f.From != nil {
		base = base.Where("create_time >= ?", *f.From)
	}
	if f.To != nil {
		base = base.Where("create_time < ?", *f.To)
	}
	var total int64
	if cErr := base.Count(&total).Error; cErr != nil {
		return nil, 0, infraerrors.WithCause(infraerrors.ErrDBError, cErr)
	}
	pageSize := page.Limit()
	rows := make([]model.UserDivideIntoLog, 0, pageSize)
	if qErr := base.Order("create_time DESC").Order("id DESC").
		Limit(pageSize).Offset(page.Offset()).Find(&rows).Error; qErr != nil {
		return nil, 0, infraerrors.WithCause(infraerrors.ErrDBError, qErr)
	}
	if rows == nil {
		rows = make([]model.UserDivideIntoLog, 0)
	}
	return rows, total, nil
}