package vector

import (
	"context"
	"fmt"
	"strings"
)

// 本文件是**向量删除**（索引的清除侧）。动因是一条实测 finding（切片 M1 的
// 读侧时效闸交下来的）：临时记忆到期后**行不在了、向量还在** —— 过期行的向量
// 仍留在 vec_items 里，占掉 idx.Search 的 top-K 预算，于是「语义检索前 K 名」
// 里有一半是永远不会被返回的死条目（读侧闸已挡住那些行），**召回质量静默下降**。
// 到期清扫（internal/runtime/memory/expire.go）在物理删行时同步调这里，把
// 向量一并清掉。
//
// 🔴 **为什么不是 Index 的必需方法**（Deleter 与 Index 并列、由调用方类型断言）：
// Index 是本包对外的稠密索引抽象，加一个必需方法会**立刻打破所有既有实现** ——
// 包括别的包里的测试桩（例如 memory 包的 mockVectorIndex）。清除能力因此按
// **可选扩展**建模：两种生产后端（SQLiteIndex / PGVectorIndex）都实现它，
// 只读的桩可以不实现，调用方断言失败时退化为「不清向量」并留一条日志。
// 这与本仓既有的可选能力手法一致（例如 `a.vindex.(*vector.PGVectorIndex)`）。

// Deleter 是 Index 的**可选扩展**：把已不存在的 id 从索引里清掉。id 不存在时
// 视为成功（删不存在的向量本就是空操作），因此重复调用是幂等的。
//
// 调用方（memory 包）在物理删行之后调用它；**不是**读路径的一部分。
type Deleter interface {
	Delete(ctx context.Context, ids ...string) error
}

// deleteVecItemsSQLTemplate 是两种后端共用的删除语句骨架：`%s` 是占位串
// （SQLite `?` / pgvector `$1,$2,...`）。表名是编译期常量，绝不来自输入。
const deleteVecItemsSQLTemplate = `DELETE FROM vec_items WHERE id IN (%s)`

// vecDeletePlaceholders 为 ids 生成删除语句的占位串与参数表。`dollar` 为 true
// 时用 Postgres 的 `$n` 形式，否则用 SQLite 的 `?`。
//
// 空白 id 被**跳过**而不是报错：删一个不存在的向量是空操作，不该让整批失败
// （Add 拒绝空 id，是因为那会写出一行不可寻址的数据；删除没有这个问题）。
// 全部为空时返回空占位串，调用方据此直接返回。
func vecDeletePlaceholders(ids []string, dollar bool) (string, []any) {
	args := make([]any, 0, len(ids))
	for _, id := range ids {
		if strings.TrimSpace(id) == "" {
			continue
		}
		args = append(args, id)
	}
	if len(args) == 0 {
		return "", nil
	}
	var b strings.Builder
	for i := range args {
		if i > 0 {
			b.WriteByte(',')
		}
		if dollar {
			fmt.Fprintf(&b, "$%d", i+1)
		} else {
			b.WriteByte('?')
		}
	}
	return b.String(), args
}

// Delete 把 ids 对应的向量从 SQLite 索引里删掉。id 不存在时无事发生（幂等）；
// 空 ids 是空操作。ctx 被取消时中止。
func (s *SQLiteIndex) Delete(ctx context.Context, ids ...string) error {
	placeholders, args := vecDeletePlaceholders(ids, false)
	if placeholders == "" {
		return nil
	}
	if _, err := s.db.ExecContext(ctx, fmt.Sprintf(deleteVecItemsSQLTemplate, placeholders), args...); err != nil {
		return fmt.Errorf("memory/vector: delete %d sqlite vectors: %w", len(args), err)
	}
	return nil
}

// Delete 把 ids 对应的向量从 pgvector 索引里删掉。语义与 SQLiteIndex.Delete
// 逐字相同（幂等、空 ids 空操作、ctx 取消即中止），只用 `$n` 占位。
func (p *PGVectorIndex) Delete(ctx context.Context, ids ...string) error {
	placeholders, args := vecDeletePlaceholders(ids, true)
	if placeholders == "" {
		return nil
	}
	if _, err := p.db.ExecContext(ctx, fmt.Sprintf(deleteVecItemsSQLTemplate, placeholders), args...); err != nil {
		return fmt.Errorf("memory/vector: delete %d pg vectors: %w", len(args), err)
	}
	return nil
}