Add AI batch reconciliation for accessibility bills
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
import 'package:flutter/material.dart';
|
||||
import 'package:go_router/go_router.dart';
|
||||
import 'package:miaoji_zhang/features/home/pages/transaction_edit_page.dart';
|
||||
import 'package:miaoji_zhang/shared/api/api_client.dart';
|
||||
import 'package:miaoji_zhang/shared/api/business_api.dart';
|
||||
@@ -10,14 +11,20 @@ import 'package:miaoji_zhang/shared/services/shanghai_time.dart';
|
||||
import 'package:miaoji_zhang/shared/services/transaction_events.dart';
|
||||
import 'package:miaoji_zhang/shared/theme/app_theme.dart';
|
||||
import 'package:miaoji_zhang/shared/widgets/app_controls.dart';
|
||||
import 'package:shared_preferences/shared_preferences.dart';
|
||||
|
||||
class RecognitionImportService {
|
||||
RecognitionImportService._();
|
||||
|
||||
static bool _processing = false;
|
||||
static Future<void>? _activeImport;
|
||||
static bool _rerunRequested = false;
|
||||
|
||||
static Future<void> configureNativeContext() async {
|
||||
final session = SessionStore.instance;
|
||||
final preferences = await SharedPreferences.getInstance();
|
||||
if (preferences.getBool('ai_batch_consent_v2') != true) {
|
||||
await ScreenshotChannel.setRecognitionToggle('ai_screenshot', false);
|
||||
}
|
||||
await ScreenshotChannel.configureRecognitionContext(
|
||||
hasAccount: session.isAccount,
|
||||
aiAllowed: session.aiEnabled,
|
||||
@@ -26,17 +33,46 @@ class RecognitionImportService {
|
||||
);
|
||||
}
|
||||
|
||||
static Future<void> importAutomatic() async {
|
||||
if (_processing || !SessionStore.instance.hasSession) return;
|
||||
_processing = true;
|
||||
static Future<void> importAutomatic() {
|
||||
if (!SessionStore.instance.hasSession) return Future.value();
|
||||
final active = _activeImport;
|
||||
if (active != null) {
|
||||
_rerunRequested = true;
|
||||
return active;
|
||||
}
|
||||
final operation = _runAutomaticImports();
|
||||
_activeImport = operation;
|
||||
return operation.whenComplete(() {
|
||||
if (identical(_activeImport, operation)) _activeImport = null;
|
||||
});
|
||||
}
|
||||
|
||||
static Future<void> _runAutomaticImports() async {
|
||||
try {
|
||||
await CurrentLedgerStore.instance.ensureLoaded();
|
||||
final candidates = await ScreenshotChannel.drainRecognitionCandidates();
|
||||
for (final candidate in candidates.where((item) => item.canAutoImport)) {
|
||||
await _import(candidate);
|
||||
}
|
||||
do {
|
||||
_rerunRequested = false;
|
||||
await CurrentLedgerStore.instance.ensureLoaded();
|
||||
final candidates = await ScreenshotChannel.drainRecognitionCandidates();
|
||||
final automatic = candidates
|
||||
.where((item) => item.canAutoImport)
|
||||
.toList();
|
||||
final batches = <String, List<RecognitionCandidate>>{};
|
||||
for (final candidate in automatic) {
|
||||
final batchId = candidate.batchId;
|
||||
if (batchId == null || batchId.isEmpty) continue;
|
||||
batches.putIfAbsent(batchId, () => []).add(candidate);
|
||||
}
|
||||
for (final entry in batches.entries) {
|
||||
await _importBatch(entry.key, entry.value);
|
||||
}
|
||||
for (final candidate in automatic.where(
|
||||
(item) => item.batchId == null || item.batchId!.isEmpty,
|
||||
)) {
|
||||
await _import(candidate);
|
||||
}
|
||||
} while (_rerunRequested);
|
||||
} finally {
|
||||
_processing = false;
|
||||
_rerunRequested = false;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -49,6 +85,20 @@ class RecognitionImportService {
|
||||
await importAutomatic();
|
||||
return;
|
||||
}
|
||||
if (kind == 'recognition_batch_review') {
|
||||
await importAutomatic();
|
||||
if (!context.mounted) return;
|
||||
final batchId = action['batchId']?.toString();
|
||||
context.push(
|
||||
Uri(
|
||||
path: '/recognition-batches',
|
||||
queryParameters: batchId == null || batchId.isEmpty
|
||||
? null
|
||||
: {'batchId': batchId},
|
||||
).toString(),
|
||||
);
|
||||
return;
|
||||
}
|
||||
if (kind == 'recognition_undo') {
|
||||
final transactionId = (action['transactionId'] as num?)?.toInt();
|
||||
final candidateId = action['candidateId']?.toString();
|
||||
@@ -130,17 +180,7 @@ class RecognitionImportService {
|
||||
}
|
||||
|
||||
static Future<TxItem> _import(RecognitionCandidate candidate) async {
|
||||
if (candidate.type != 'income' && candidate.type != 'expense') {
|
||||
throw StateError('识别结果缺少明确的收支类型');
|
||||
}
|
||||
final categories = await TxApi.categories(candidate.type);
|
||||
if (categories.isEmpty) throw StateError('当前账本没有可用分类');
|
||||
final category =
|
||||
categories
|
||||
.where((item) => item.name == candidate.categoryHint)
|
||||
.firstOrNull ??
|
||||
categories.where((item) => item.name == '其他').firstOrNull ??
|
||||
categories.first;
|
||||
final category = await _resolveCategory(candidate, {});
|
||||
final occurredUtc = validOccurredAtUtc(candidate.occurredAtEpochMs);
|
||||
final transaction = await TxApi.create(
|
||||
categoryId: category.id,
|
||||
@@ -164,6 +204,73 @@ class RecognitionImportService {
|
||||
return transaction;
|
||||
}
|
||||
|
||||
static Future<void> _importBatch(
|
||||
String batchId,
|
||||
List<RecognitionCandidate> candidates,
|
||||
) async {
|
||||
final categoryCache = <String, List<CategoryItem>>{};
|
||||
final drafts = <RecognitionBatchDraft>[];
|
||||
for (final candidate in candidates) {
|
||||
final category = await _resolveCategory(candidate, categoryCache);
|
||||
final occurredUtc = validOccurredAtUtc(candidate.occurredAtEpochMs);
|
||||
drafts.add(
|
||||
RecognitionBatchDraft(
|
||||
candidateId: candidate.id,
|
||||
clientRequestId: candidate.clientRequestId,
|
||||
categoryId: category.id,
|
||||
type: candidate.type,
|
||||
amount: candidate.amount,
|
||||
note: candidate.merchant?.trim().isNotEmpty == true
|
||||
? candidate.merchant!.trim()
|
||||
: (candidate.note ?? '智能识别'),
|
||||
paymentMethod: candidate.appName,
|
||||
source: candidate.source,
|
||||
sourceText: candidate.sourceText,
|
||||
occurredAt: ShanghaiTime.toCivil(occurredUtc),
|
||||
),
|
||||
);
|
||||
}
|
||||
final transactions = await TxApi.createRecognitionBatch(batchId, drafts);
|
||||
if (transactions.length != drafts.length) {
|
||||
throw StateError('批次入账结果不完整,请稍后重试');
|
||||
}
|
||||
for (final candidate in candidates) {
|
||||
final transaction = transactions[candidate.id];
|
||||
if (transaction == null) {
|
||||
throw StateError('批次入账缺少候选 ${candidate.id}');
|
||||
}
|
||||
final acknowledged =
|
||||
await ScreenshotChannel.acknowledgeRecognitionCandidate(
|
||||
candidate.id,
|
||||
'imported',
|
||||
transactionId: transaction.id,
|
||||
);
|
||||
if (!acknowledged) throw StateError('批次状态确认失败,请稍后重试');
|
||||
}
|
||||
TransactionEvents.notifyChanged();
|
||||
}
|
||||
|
||||
static Future<CategoryItem> _resolveCategory(
|
||||
RecognitionCandidate candidate,
|
||||
Map<String, List<CategoryItem>> cache,
|
||||
) async {
|
||||
if (candidate.type != 'income' && candidate.type != 'expense') {
|
||||
throw StateError('识别结果缺少明确的收支类型');
|
||||
}
|
||||
final categories = cache[candidate.type] ??= await TxApi.categories(
|
||||
candidate.type,
|
||||
);
|
||||
if (categories.isEmpty) throw StateError('当前账本没有可用分类');
|
||||
return categories
|
||||
.where((item) => item.id == candidate.categoryId)
|
||||
.firstOrNull ??
|
||||
categories
|
||||
.where((item) => item.name == candidate.categoryHint)
|
||||
.firstOrNull ??
|
||||
categories.where((item) => item.name == '其他').firstOrNull ??
|
||||
categories.first;
|
||||
}
|
||||
|
||||
static DateTime validOccurredAtUtc(int epochMs, {DateTime? now}) {
|
||||
final current = (now ?? DateTime.now()).toUtc();
|
||||
final parsed = DateTime.fromMillisecondsSinceEpoch(epochMs, isUtc: true);
|
||||
|
||||
Reference in New Issue
Block a user