immich/mobile-v2/lib/domain/services/sync.service.dart

import 'package:drift/isolate.dart';
import 'package:immich_mobile/domain/interfaces/remote_asset.interface.dart';
import 'package:immich_mobile/domain/models/asset/remote_asset.model.dart';
import 'package:immich_mobile/domain/models/user.model.dart';
import 'package:immich_mobile/domain/repositories/database.repository.dart';
import 'package:immich_mobile/service_locator.dart';
import 'package:immich_mobile/utils/constants/globals.dart';
import 'package:immich_mobile/utils/immich_api_client.dart';
import 'package:immich_mobile/utils/log_manager.dart';
import 'package:immich_mobile/utils/mixins/log_context.mixin.dart';
import 'package:logging/logging.dart';
import 'package:openapi/api.dart';

class SyncService with LogContext {
  final ImmichApiClient _appClient;
  final DriftDatabaseRepository _db;

  SyncService(this._appClient, this._db);

  Future<bool> doFullSyncForUserDrift(
    User user, {
    DateTime? updatedUtil,
    int? limit,
  }) async {
    final clientData = _appClient.clientData;
    try {
      await _db.computeWithDatabase(
        connect: (connection) => DriftDatabaseRepository(connection),
        computation: (database) async {
          ServiceLocator.configureServicesForIsolate(database: database);
          LogManager.I.init();
          final logger = Logger("SyncService <Isolate>");
          final syncClient =
              ImmichApiClient.clientData(clientData).getSyncApi();

          final chunkSize = limit ?? kFullSyncChunkSize;
          final updatedTill = updatedUtil ?? DateTime.now().toUtc();
          updatedUtil ??= DateTime.now().toUtc();
          String? lastAssetId;

          while (true) {
            logger.info(
              "Requesting more chunks from lastId - ${lastAssetId ?? "<initial_fetch>"}",
            );

            final assets = await syncClient.getFullSyncForUser(AssetFullSyncDto(
              limit: chunkSize,
              updatedUntil: updatedTill,
              lastId: lastAssetId,
              userId: user.id,
            ));
            if (assets == null) {
              break;
            }

            await di<IRemoteAssetRepository>()
                .addAll(assets.map(RemoteAsset.fromDto));

            lastAssetId = assets.lastOrNull?.id;
            if (assets.length != chunkSize) break;
          }

          return true;
        },
      );
    } catch (e, s) {
      log.severe("Error performing full sync for user - ${user.name}", e, s);
    }
    return false;
  }
}