ente/lib/utils/file_download_util.dart
2022-12-29 19:06:35 +05:30

74 lines
2.5 KiB
Dart

import 'dart:io' as io;
import 'dart:typed_data';
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/file.dart' as ente;
import 'package:photos/services/collections_service.dart';
import 'package:photos/utils/crypto_util.dart';
final _logger = Logger("file_download_util");
Future<io.File?> downloadAndDecrypt(
ente.File file, {
ProgressCallback? progressCallback,
}) {
_logger.info("Downloading file " + file.uploadedFileID.toString());
final encryptedFilePath = Configuration.instance.getTempDirectory() +
file.generatedID.toString() +
".encrypted";
final encryptedFile = io.File(encryptedFilePath);
final startTime = DateTime.now().millisecondsSinceEpoch;
return Network.instance
.getDio()
.download(
file.downloadUrl,
encryptedFilePath,
options: Options(
headers: {"X-Auth-Token": Configuration.instance.getToken()},
),
onReceiveProgress: progressCallback,
)
.then((response) async {
if (response.statusCode != 200) {
_logger.warning("Could not download file: ", response.toString());
return null;
} else if (!encryptedFile.existsSync()) {
_logger.warning("File was not downloaded correctly.");
return null;
}
_logger.info("File downloaded: " + file.uploadedFileID.toString());
_logger.info(
"Download speed: " +
(await io.File(encryptedFilePath).length() /
(DateTime.now().millisecondsSinceEpoch - startTime))
.toString() +
"kBps",
);
final decryptedFilePath = Configuration.instance.getTempDirectory() +
file.generatedID.toString() +
".decrypted";
final decryptedFile = io.File(decryptedFilePath);
await CryptoUtil.decryptFile(
encryptedFilePath,
decryptedFilePath,
Sodium.base642bin(file.fileDecryptionHeader!),
decryptFileKey(file),
);
_logger.info("File decrypted: " + file.uploadedFileID.toString());
await encryptedFile.delete();
return decryptedFile;
});
}
Uint8List decryptFileKey(ente.File file) {
final encryptedKey = Sodium.base642bin(file.encryptedKey!);
final nonce = Sodium.base642bin(file.keyDecryptionNonce!);
final collectionKey =
CollectionsService.instance.getCollectionKey(file.collectionID!);
return CryptoUtil.decryptSync(encryptedKey, collectionKey, nonce);
}