ente/lib/utils/trash_diff_fetcher.dart
Neeraj Gupta e3f4092a11
rename
2022-05-08 11:43:43 +05:30

122 lines
4.8 KiB
Dart

import 'dart:convert';
import 'dart:math';
import 'package:dio/dio.dart';
import 'package:flutter_sodium/flutter_sodium.dart';
import 'package:logging/logging.dart';
import 'package:photos/core/configuration.dart';
import 'package:photos/core/network.dart';
import 'package:photos/models/magic_metadata.dart';
import 'package:photos/models/trash_file.dart';
import 'package:photos/utils/crypto_util.dart';
import 'package:photos/utils/file_download_util.dart';
class TrashDiffFetcher {
final _logger = Logger("TrashDiffFetcher");
final _dio = Network.instance.getDio();
Future<Diff> getTrashFilesDiff(int sinceTime) async {
try {
final response = await _dio.get(
Configuration.instance.getHttpEndpoint() + "/trash/v2/diff",
options: Options(
headers: {"X-Auth-Token": Configuration.instance.getToken()}),
queryParameters: {
"sinceTime": sinceTime,
},
);
int latestUpdatedAtTime = 0;
final trashedFiles = <TrashFile>[];
final deletedUploadIDs = <int>[];
final restoredFiles = <TrashFile>[];
if (response != null) {
final diff = response.data["diff"] as List;
final bool hasMore = response.data["hasMore"] as bool;
final startTime = DateTime.now();
for (final item in diff) {
if (item["isDeleted"]) {
deletedUploadIDs.add(item["file"]["id"]);
continue;
}
final trash = TrashFile();
trash.createdAt = item['createdAt'];
trash.updateAt = item['updatedAt'];
latestUpdatedAtTime = max(latestUpdatedAtTime, trash.updateAt);
trash.deleteBy = item['deleteBy'];
trash.uploadedFileID = item["file"]["id"];
trash.collectionID = item["file"]["collectionID"];
trash.updationTime = item["file"]["updationTime"];
trash.ownerID = item["file"]["ownerID"];
trash.encryptedKey = item["file"]["encryptedKey"];
trash.keyDecryptionNonce = item["file"]["keyDecryptionNonce"];
trash.fileDecryptionHeader = item["file"]["file"]["decryptionHeader"];
trash.thumbnailDecryptionHeader =
item["file"]["thumbnail"]["decryptionHeader"];
trash.metadataDecryptionHeader =
item["file"]["metadata"]["decryptionHeader"];
final fileDecryptionKey = decryptFileKey(trash);
final encodedMetadata = await CryptoUtil.decryptChaCha(
Sodium.base642bin(item["file"]["metadata"]["encryptedData"]),
fileDecryptionKey,
Sodium.base642bin(trash.metadataDecryptionHeader),
);
Map<String, dynamic> metadata =
jsonDecode(utf8.decode(encodedMetadata));
trash.applyMetadata(metadata);
if (item["file"]['magicMetadata'] != null) {
final utfEncodedMmd = await CryptoUtil.decryptChaCha(
Sodium.base642bin(item["file"]['magicMetadata']['data']),
fileDecryptionKey,
Sodium.base642bin(item["file"]['magicMetadata']['header']));
trash.mMdEncodedJson = utf8.decode(utfEncodedMmd);
trash.mMdVersion = item["file"]['magicMetadata']['version'];
}
if (item["file"]['pubMagicMetadata'] != null) {
final utfEncodedMmd = await CryptoUtil.decryptChaCha(
Sodium.base642bin(item["file"]['pubMagicMetadata']['data']),
fileDecryptionKey,
Sodium.base642bin(item["file"]['pubMagicMetadata']['header']));
trash.pubMmdEncodedJson = utf8.decode(utfEncodedMmd);
trash.pubMmdVersion = item["file"]['pubMagicMetadata']['version'];
trash.pubMagicMetadata =
PubMagicMetadata.fromEncodedJson(trash.pubMmdEncodedJson);
}
if (item['isRestored']) {
restoredFiles.add(trash);
continue;
}
trashedFiles.add(trash);
}
final endTime = DateTime.now();
_logger.info("time for parsing " +
diff.length.toString() +
": " +
Duration(
microseconds: (endTime.microsecondsSinceEpoch -
startTime.microsecondsSinceEpoch))
.inMilliseconds
.toString());
return Diff(trashedFiles, restoredFiles, deletedUploadIDs, hasMore,
latestUpdatedAtTime);
} else {
return Diff(<TrashFile>[], <TrashFile>[], <int>[], false, 0);
}
} catch (e, s) {
_logger.severe(e, s);
rethrow;
}
}
}
class Diff {
final List<TrashFile> trashedFiles;
final List<TrashFile> restoredFiles;
final List<int> deletedUploadIDs;
final bool hasMore;
final int lastSyncedTimeStamp;
Diff(this.trashedFiles, this.restoredFiles, this.deletedUploadIDs,
this.hasMore, this.lastSyncedTimeStamp);
}