/**
 * 迭代 07 回归测试 v2：交易记录本地存储 + 手动归属（R-009 二次定稿）
 * 独立数据目录（ODL_TEST_DATA_DIR），不触碰真实数据（技术约束-011）。
 *
 * 覆盖：
 *  1. 建表：trade_orders 含 strategy_id/holding_id 列（冗余，Q1）
 *  2. UPSERT 幂等 + 不覆盖归属列（Q2）
 *  3. 手动设置归属（set-attribution，Q4 可改）
 *  4. 归属候选列表（该 code 当前持仓策略）
 *  5. 历史查询按持久化归属过滤（含未关联）
 */
import { mkdtempSync, rmSync } from 'node:fs';
import { tmpdir } from 'node:os';
import { join } from 'node:path';
import { SqliteStore } from '../src/component/SqliteStore.js';

const dir = mkdtempSync(join(tmpdir(), 'odl-test-r009-v2-'));
process.env.ODL_TEST_DATA_DIR = dir;

let pass = 0;
let fail = 0;
function assert(cond, msg) {
  if (cond) { pass++; console.log('  ✅ ' + msg); }
  else { fail++; console.log('  ❌ ' + msg); }
}

const store = new SqliteStore({ dataDir: dir });
store.init();

// 1. 建表：trade_orders 含归属列
console.log('1. 建表');
const orderCols = store.db.prepare('PRAGMA table_info(trade_orders)').all().map(c => c.name);
assert(orderCols.includes('strategy_id'), 'trade_orders 有 strategy_id 列（冗余 Q1）');
assert(orderCols.includes('holding_id'), 'trade_orders 有 holding_id 列（冗余 Q1）');

// 2. UPSERT 幂等 + 不覆盖归属列
console.log('2. UPSERT 幂等 + 不覆盖归属');
const order = {
  orderId: 'O-001', tradeDate: '20260901', code: '300057.SZ', name: '万顺新材',
  exchange: 'SZ', direction: 'sell', directionCode: 49, optName: '限价卖出',
  status: 50, orderVolume: 100, tradedVolume: 0, limitPrice: 7, tradedPrice: 0,
  amount: 0, insertDate: '20260901', insertTime: '093000',
  cancelInfo: '', errorMsg: '',
};
store.upsertOrders([order]);
// 设置归属（用户手动）
store.openHolding('manual-t', '300057.SZ', 1000);
const cands = store.getAttributionCandidates('300057.SZ');
assert(cands.length === 1 && cands[0].strategyId === 'manual-t', '归属候选：当前持仓 manual-t（300057.SZ）');
const holding = store.db.prepare('SELECT holding_id FROM strategy_holdings WHERE strategy_id=? AND code=? AND closed_at IS NULL').get('manual-t', '300057.SZ');
const ok = store.setOrderAttribution('O-001', 'manual-t', holding.holding_id);
assert(ok, '手动设置归属成功');
// 再次 UPSERT（模拟下次同步，状态流转）→ 归属列保留
store.upsertOrders([{ ...order, status: 56, tradedVolume: 100, tradedPrice: 7, amount: 700 }]);
const row = store.db.prepare('SELECT status, strategy_id, holding_id FROM trade_orders WHERE order_id=?').get('O-001');
assert(row.status === 56, 'UPSERT 覆盖状态（已报→已成）');
assert(row.strategy_id === 'manual-t' && row.holding_id === holding.holding_id, 'UPSERT 不覆盖归属列（strategy_id/holding_id 保留）');

// 3. 手动设置归属可改（Q4）
console.log('3. 归属可修改');
store.openHolding('grid-supermarket', '300057.SZ', 800);
const gHold = store.db.prepare('SELECT holding_id FROM strategy_holdings WHERE strategy_id=? AND code=? AND closed_at IS NULL').get('grid-supermarket', '300057.SZ');
store.setOrderAttribution('O-001', 'grid-supermarket', gHold.holding_id);
const row2 = store.db.prepare('SELECT strategy_id, holding_id FROM trade_orders WHERE order_id=?').get('O-001');
assert(row2.strategy_id === 'grid-supermarket', '归属可改为 grid-supermarket');

// 清除归属（改回未关联）
store.setOrderAttribution('O-001', null, null);
const row3 = store.db.prepare('SELECT strategy_id, holding_id FROM trade_orders WHERE order_id=?').get('O-001');
assert(row3.strategy_id === null && row3.holding_id === null, '归属可清除（未关联）');

// 4. 候选列表（一票多策略）
console.log('4. 候选列表（一票多策略）');
const cands2 = store.getAttributionCandidates('300057.SZ');
assert(cands2.length === 2, '一票多策略候选 2 个（grid 800 / manual 1000）');
assert(cands2[0].shares === 1000 && cands2[0].strategyId === 'manual-t', '候选按份额降序（manual-t 1000 在前）');

// 5. 历史查询按持久化归属过滤
console.log('5. 历史查询按持久化归属过滤');
store.upsertOrders([{
  orderId: 'O-002', tradeDate: '20260901', code: '600519.SH', name: '贵州茅台',
  exchange: 'SH', direction: 'buy', directionCode: 48, optName: '限价买入',
  status: 56, orderVolume: 100, tradedVolume: 100, limitPrice: 1700, tradedPrice: 1700,
  amount: 170000, insertDate: '20260901', insertTime: '100000',
  cancelInfo: '', errorMsg: '',
}]);
// O-002 不设归属 → 未关联；O-001 设回 manual-t
store.setOrderAttribution('O-001', 'manual-t', holding.holding_id);
const byManual = store.getOrderHistory({ start: '20260901', end: '20260901', strategyId: 'manual-t' });
assert(byManual.length === 1 && byManual[0].orderId === 'O-001', '按持久化归属 manual-t 过滤命中 O-001');
const byGrid = store.getOrderHistory({ start: '20260901', end: '20260901', strategyId: 'grid-supermarket' });
assert(byGrid.length === 0, '按 grid-supermarket 过滤为空（O-001 已设 manual-t）');
const unassigned = store.getOrderHistory({ start: '20260901', end: '20260901', strategyId: '__unassigned__' });
assert(unassigned.length === 1 && unassigned[0].orderId === 'O-002', '未关联过滤命中 O-002（未设置归属）');
const all = store.getOrderHistory({ start: '20260901', end: '20260901' });
assert(all.length === 2 && all.find(o => o.orderId === 'O-001').strategyId === 'manual-t', '查询返回持久化归属字段');

store.close();
rmSync(dir, { recursive: true, force: true });
console.log('\n结果: ' + pass + ' 通过, ' + fail + ' 失败');
process.exit(fail > 0 ? 1 : 0);