fluffychat/lib/utils/matrix_sdk_extensions.dart/flutter_matrix_hive_database.dart

112 lines
3.4 KiB
Dart
Raw Normal View History

2021-06-11 11:40:38 +02:00
import 'dart:convert';
import 'dart:io';
import 'dart:typed_data';
import 'package:flutter/foundation.dart';
import 'package:flutter/services.dart';
2021-10-26 18:50:34 +02:00
2021-06-11 11:40:38 +02:00
import 'package:flutter_secure_storage/flutter_secure_storage.dart';
import 'package:hive/hive.dart';
import 'package:hive_flutter/hive_flutter.dart';
2021-10-26 18:50:34 +02:00
import 'package:matrix/matrix.dart';
2021-06-11 11:40:38 +02:00
import 'package:path_provider/path_provider.dart';
import '../platform_infos.dart';
class FlutterMatrixHiveStore extends FamedlySdkHiveDatabase {
FlutterMatrixHiveStore(String name, {HiveCipher encryptionCipher})
2021-06-11 11:40:38 +02:00
: super(
name,
encryptionCipher: encryptionCipher,
2021-06-14 08:18:55 +02:00
);
2021-06-11 11:40:38 +02:00
static bool _hiveInitialized = false;
static const String _hiveCipherStorageKey = 'hive_encryption_key';
static Future<FamedlySdkHiveDatabase> hiveDatabaseBuilder(
Client client) async {
if (!kIsWeb && !_hiveInitialized) {
_hiveInitialized = true;
}
HiveCipher hiverCipher;
try {
2021-06-12 13:41:22 +02:00
// Workaround for secure storage is calling Platform.operatingSystem on web
if (kIsWeb) throw MissingPluginException();
2021-10-14 18:09:30 +02:00
const secureStorage = FlutterSecureStorage();
2021-06-11 11:40:38 +02:00
final containsEncryptionKey =
await secureStorage.containsKey(key: _hiveCipherStorageKey);
if (!containsEncryptionKey) {
final key = Hive.generateSecureKey();
await secureStorage.write(
key: _hiveCipherStorageKey,
value: base64UrlEncode(key),
);
}
// workaround for if we just wrote to the key and it still doesn't exist
final rawEncryptionKey =
await secureStorage.read(key: _hiveCipherStorageKey);
if (rawEncryptionKey == null) throw MissingPluginException();
final encryptionKey = base64Url.decode(rawEncryptionKey);
2021-06-11 11:40:38 +02:00
hiverCipher = HiveAesCipher(encryptionKey);
} on MissingPluginException catch (_) {
2021-06-14 08:18:55 +02:00
Logs().i('Hive encryption is not supported on this platform');
2021-06-11 11:40:38 +02:00
}
final db = FlutterMatrixHiveStore(
2021-06-11 11:40:38 +02:00
client.clientName,
encryptionCipher: hiverCipher,
);
try {
await db.open();
} catch (e, s) {
Logs().e('Unable to open Hive. Delete and try again...', e, s);
2021-10-25 10:46:58 +02:00
await db.clear();
await db.open();
}
2021-06-11 11:40:38 +02:00
return db;
}
@override
2021-06-14 08:18:55 +02:00
int get maxFileSize => supportsFileStoring ? 100 * 1024 * 1024 : 0;
2021-06-11 11:40:38 +02:00
@override
bool get supportsFileStoring => (PlatformInfos.isIOS ||
PlatformInfos.isAndroid ||
PlatformInfos.isDesktop);
2021-06-11 11:40:38 +02:00
2021-06-14 08:18:55 +02:00
Future<String> _getFileStoreDirectory() async {
try {
try {
return (await getApplicationSupportDirectory()).path;
} catch (_) {
return (await getApplicationDocumentsDirectory()).path;
}
} catch (_) {
2021-06-14 08:18:55 +02:00
return (await getDownloadsDirectory()).path;
}
2021-06-11 11:40:38 +02:00
}
@override
2021-08-26 19:03:08 +02:00
Future<Uint8List> getFile(Uri mxcUri) async {
2021-06-14 08:18:55 +02:00
if (!supportsFileStoring) return null;
final tempDirectory = await _getFileStoreDirectory();
2021-08-26 19:03:08 +02:00
final file =
File('$tempDirectory/${Uri.encodeComponent(mxcUri.toString())}');
2021-06-11 11:40:38 +02:00
if (await file.exists() == false) return null;
final bytes = await file.readAsBytes();
2021-06-14 08:18:55 +02:00
return bytes;
2021-06-11 11:40:38 +02:00
}
@override
2021-08-26 19:03:08 +02:00
Future storeFile(Uri mxcUri, Uint8List bytes, int time) async {
2021-06-14 08:18:55 +02:00
if (!supportsFileStoring) return null;
final tempDirectory = await _getFileStoreDirectory();
2021-08-26 19:03:08 +02:00
final file =
File('$tempDirectory/${Uri.encodeComponent(mxcUri.toString())}');
2021-06-11 11:40:38 +02:00
if (await file.exists()) return;
2021-06-14 08:18:55 +02:00
await file.writeAsBytes(bytes);
2021-06-11 11:40:38 +02:00
return;
}
}