file_uploader.dart 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147
  1. import 'dart:convert';
  2. import 'dart:io' as io;
  3. import 'package:dio/dio.dart';
  4. import 'package:flutter_sodium/flutter_sodium.dart';
  5. import 'package:logging/logging.dart';
  6. import 'package:photos/core/configuration.dart';
  7. import 'package:photos/core/constants.dart';
  8. import 'package:photos/models/file.dart';
  9. import 'package:photos/models/upload_url.dart';
  10. import 'package:photos/utils/crypto_util.dart';
  11. import 'package:photos/utils/file_name_util.dart';
  12. import 'package:photos/utils/file_util.dart';
  13. class FileUploader {
  14. final _logger = Logger("FileUploader");
  15. final _dio = Dio();
  16. Future<UploadURL> getUploadURL() {
  17. return Dio()
  18. .get(
  19. Configuration.instance.getHttpEndpoint() +
  20. "/encrypted-files/upload-url",
  21. options: Options(
  22. headers: {"X-Auth-Token": Configuration.instance.getToken()}),
  23. )
  24. .then((response) => UploadURL.fromMap(response.data));
  25. }
  26. Future<String> putFile(UploadURL uploadURL, io.File file) async {
  27. _logger.info("Putting file to " + uploadURL.url);
  28. return Dio()
  29. .put(uploadURL.url,
  30. data: file.openRead(),
  31. options: Options(headers: {
  32. Headers.contentLengthHeader: await file.length(),
  33. }))
  34. .catchError((e) {
  35. _logger.severe(e);
  36. }).then((value) {
  37. return uploadURL.objectKey;
  38. });
  39. }
  40. Future<File> encryptAndUploadFile(File file) async {
  41. _logger.info("Uploading " + file.toString());
  42. final encryptedFileName = file.generatedID.toString() + ".encrypted";
  43. final tempDirectory = Configuration.instance.getTempDirectory();
  44. final encryptedFilePath = tempDirectory + encryptedFileName;
  45. final sourceFile = (await (await file.getAsset()).originFile);
  46. final encryptedFile = io.File(encryptedFilePath);
  47. final fileAttributes =
  48. await CryptoUtil.encryptFile(sourceFile.path, encryptedFilePath);
  49. final fileUploadURL = await getUploadURL();
  50. String fileObjectKey = await putFile(fileUploadURL, encryptedFile);
  51. final thumbnailData = (await (await file.getAsset()).thumbDataWithSize(
  52. THUMBNAIL_LARGE_SIZE,
  53. THUMBNAIL_LARGE_SIZE,
  54. quality: 50,
  55. ));
  56. final encryptedThumbnailName =
  57. file.generatedID.toString() + "_thumbnail.encrypted";
  58. final encryptedThumbnailPath = tempDirectory + encryptedThumbnailName;
  59. final encryptedThumbnail =
  60. CryptoUtil.encryptStream(thumbnailData, fileAttributes.key.bytes);
  61. io.File(encryptedThumbnailPath)
  62. .writeAsBytesSync(encryptedThumbnail.encryptedData);
  63. final thumbnailUploadURL = await getUploadURL();
  64. String thumbnailObjectKey =
  65. await putFile(thumbnailUploadURL, io.File(encryptedThumbnailPath));
  66. final encryptedMetadata = CryptoUtil.encryptStream(
  67. utf8.encode(jsonEncode(file.getMetadata())), fileAttributes.key.bytes);
  68. final encryptedFileKey = await CryptoUtil.encrypt(
  69. fileAttributes.key.bytes,
  70. key: Configuration.instance.getKey(),
  71. );
  72. final data = {
  73. "encryptedKey": encryptedFileKey.encryptedData.base64,
  74. "keyDecryptionNonce": encryptedFileKey.nonce.base64,
  75. "file": {
  76. "objectKey": fileObjectKey,
  77. "header": fileAttributes.header.base64,
  78. },
  79. "thumbnail": {
  80. "objectKey": thumbnailObjectKey,
  81. "header": Sodium.bin2base64(encryptedThumbnail.header),
  82. },
  83. "metadata": {
  84. "encryptedData": Sodium.bin2base64(encryptedMetadata.encryptedData),
  85. "header": Sodium.bin2base64(encryptedMetadata.header),
  86. }
  87. };
  88. return _dio
  89. .post(
  90. Configuration.instance.getHttpEndpoint() + "/encrypted-files",
  91. options:
  92. Options(headers: {"X-Auth-Token": Configuration.instance.getToken()}),
  93. data: data,
  94. )
  95. .then((response) {
  96. encryptedFile.deleteSync();
  97. io.File(encryptedThumbnailPath).deleteSync();
  98. final data = response.data;
  99. file.uploadedFileID = data["id"];
  100. file.updationTime = data["updationTime"];
  101. file.ownerID = data["ownerID"];
  102. file.encryptedKey = encryptedFileKey.encryptedData.base64;
  103. file.keyDecryptionNonce = encryptedFileKey.nonce.base64;
  104. file.fileDecryptionHeader = fileAttributes.header.base64;
  105. file.thumbnailDecryptionHeader =
  106. Sodium.bin2base64(encryptedThumbnail.header);
  107. file.metadataDecryptionHeader =
  108. Sodium.bin2base64(encryptedMetadata.header);
  109. return file;
  110. });
  111. }
  112. Future<File> uploadFile(File localPhoto) async {
  113. final title = getJPGFileNameForHEIC(localPhoto);
  114. final formData = FormData.fromMap({
  115. "file": MultipartFile.fromBytes(await getBytesFromDisk(localPhoto),
  116. filename: title),
  117. "deviceFileID": localPhoto.localID,
  118. "deviceFolder": localPhoto.deviceFolder,
  119. "title": title,
  120. "creationTime": localPhoto.creationTime,
  121. "modificationTime": localPhoto.modificationTime,
  122. });
  123. return _dio
  124. .post(
  125. Configuration.instance.getHttpEndpoint() + "/files",
  126. options:
  127. Options(headers: {"X-Auth-Token": Configuration.instance.getToken()}),
  128. data: formData,
  129. )
  130. .then((response) {
  131. return File.fromJson(response.data);
  132. });
  133. }
  134. }