ente/lib/db/embeddings_db.dart

import "dart:io";

import "package:isar/isar.dart";
import 'package:path_provider/path_provider.dart';
import "package:photos/core/event_bus.dart";
import "package:photos/events/embedding_updated_event.dart";
import "package:photos/models/embedding.dart";

class EmbeddingsDB {
  late final Isar _isar;

  EmbeddingsDB._privateConstructor();

  static final EmbeddingsDB instance = EmbeddingsDB._privateConstructor();

  Future<void> init() async {
    final dir = await getApplicationDocumentsDirectory();
    _isar = await Isar.open(
      [EmbeddingSchema],
      directory: dir.path,
    );
    await _clearDeprecatedStore(dir);
  }

  Future<void> clearTable() async {
    await _isar.clear();
  }

  Future<List<Embedding>> getAll(Model model) async {
    return _isar.embeddings.filter().modelEqualTo(model).findAll();
  }

  Future<int> put(Embedding embedding) {
    return _isar.writeTxn(() async {
      final id = await _isar.embeddings.put(embedding);
      Bus.instance.fire(EmbeddingUpdatedEvent());
      return id;
    });
  }

  Future<void> putMany(List<Embedding> embeddings) {
    return _isar.writeTxn(() async {
      await _isar.embeddings.putAll(embeddings);
      Bus.instance.fire(EmbeddingUpdatedEvent());
    });
  }

  Future<List<Embedding>> getUnsyncedEmbeddings() async {
    return await _isar.embeddings.filter().updationTimeEqualTo(null).findAll();
  }

  Future<void> deleteAllForModel(Model model) async {
    await _isar.writeTxn(() async {
      final embeddings =
          await _isar.embeddings.filter().modelEqualTo(model).findAll();
      await _isar.embeddings.deleteAll(embeddings.map((e) => e.id).toList());
      Bus.instance.fire(EmbeddingUpdatedEvent());
    });
  }

  Future<void> _clearDeprecatedStore(Directory dir) async {
    final deprecatedStore = Directory(dir.path + "/object-box-store");
    if (await deprecatedStore.exists()) {
      await deprecatedStore.delete(recursive: true);
    }
  }
}