files_db.dart 51 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598
  1. import 'dart:developer' as dev;
  2. import 'dart:io' as io;
  3. import 'package:flutter/foundation.dart';
  4. import 'package:logging/logging.dart';
  5. import 'package:path/path.dart';
  6. import 'package:path_provider/path_provider.dart';
  7. import 'package:photos/models/backup_status.dart';
  8. import 'package:photos/models/file.dart';
  9. import 'package:photos/models/file_load_result.dart';
  10. import 'package:photos/models/file_type.dart';
  11. import 'package:photos/models/location/location.dart';
  12. import 'package:photos/models/magic_metadata.dart';
  13. import 'package:photos/utils/file_uploader_util.dart';
  14. import 'package:sqflite/sqflite.dart';
  15. import 'package:sqflite_migration/sqflite_migration.dart';
  16. class FilesDB {
  17. /*
  18. Note: columnUploadedFileID and columnCollectionID have to be compared against
  19. both NULL and -1 because older clients might have entries where the DEFAULT
  20. was unset, and a migration script to set the DEFAULT would break in case of
  21. duplicate entries for un-uploaded files that were created due to a collision
  22. in background and foreground syncs.
  23. */
  24. static const _databaseName = "ente.files.db";
  25. static final Logger _logger = Logger("FilesDB");
  26. static const filesTable = 'files';
  27. static const tempTable = 'temp_files';
  28. static const columnGeneratedID = '_id';
  29. static const columnUploadedFileID = 'uploaded_file_id';
  30. static const columnOwnerID = 'owner_id';
  31. static const columnCollectionID = 'collection_id';
  32. static const columnLocalID = 'local_id';
  33. static const columnTitle = 'title';
  34. static const columnDeviceFolder = 'device_folder';
  35. static const columnLatitude = 'latitude';
  36. static const columnLongitude = 'longitude';
  37. static const columnFileType = 'file_type';
  38. static const columnFileSubType = 'file_sub_type';
  39. static const columnDuration = 'duration';
  40. static const columnExif = 'exif';
  41. static const columnHash = 'hash';
  42. static const columnMetadataVersion = 'metadata_version';
  43. static const columnIsDeleted = 'is_deleted';
  44. static const columnCreationTime = 'creation_time';
  45. static const columnModificationTime = 'modification_time';
  46. static const columnUpdationTime = 'updation_time';
  47. static const columnEncryptedKey = 'encrypted_key';
  48. static const columnKeyDecryptionNonce = 'key_decryption_nonce';
  49. static const columnFileDecryptionHeader = 'file_decryption_header';
  50. static const columnThumbnailDecryptionHeader = 'thumbnail_decryption_header';
  51. static const columnMetadataDecryptionHeader = 'metadata_decryption_header';
  52. static const columnFileSize = 'file_size';
  53. // MMD -> Magic Metadata
  54. static const columnMMdEncodedJson = 'mmd_encoded_json';
  55. static const columnMMdVersion = 'mmd_ver';
  56. static const columnPubMMdEncodedJson = 'pub_mmd_encoded_json';
  57. static const columnPubMMdVersion = 'pub_mmd_ver';
  58. // part of magic metadata
  59. // Only parse & store selected fields from JSON in separate columns if
  60. // we need to write query based on that field
  61. static const columnMMdVisibility = 'mmd_visibility';
  62. static final initializationScript = [...createTable(filesTable)];
  63. static final migrationScripts = [
  64. ...alterDeviceFolderToAllowNULL(),
  65. ...alterTimestampColumnTypes(),
  66. ...addIndices(),
  67. ...addMetadataColumns(),
  68. ...addMagicMetadataColumns(),
  69. ...addUniqueConstraintOnCollectionFiles(),
  70. ...addPubMagicMetadataColumns(),
  71. ...createOnDeviceFilesAndPathCollection(),
  72. ...addFileSizeColumn(),
  73. ...updateIndexes(),
  74. ...createEntityDataTable(),
  75. ];
  76. final dbConfig = MigrationConfig(
  77. initializationScript: initializationScript,
  78. migrationScripts: migrationScripts,
  79. );
  80. // make this a singleton class
  81. FilesDB._privateConstructor();
  82. static final FilesDB instance = FilesDB._privateConstructor();
  83. // only have a single app-wide reference to the database
  84. static Future<Database>? _dbFuture;
  85. Future<Database> get database async {
  86. // lazily instantiate the db the first time it is accessed
  87. _dbFuture ??= _initDatabase();
  88. return _dbFuture!;
  89. }
  90. // this opens the database (and creates it if it doesn't exist)
  91. Future<Database> _initDatabase() async {
  92. final io.Directory documentsDirectory =
  93. await getApplicationDocumentsDirectory();
  94. final String path = join(documentsDirectory.path, _databaseName);
  95. _logger.info("DB path " + path);
  96. return await openDatabaseWithMigration(path, dbConfig);
  97. }
  98. // SQL code to create the database table
  99. static List<String> createTable(String tableName) {
  100. return [
  101. '''
  102. CREATE TABLE $tableName (
  103. $columnGeneratedID INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
  104. $columnLocalID TEXT,
  105. $columnUploadedFileID INTEGER DEFAULT -1,
  106. $columnOwnerID INTEGER,
  107. $columnCollectionID INTEGER DEFAULT -1,
  108. $columnTitle TEXT NOT NULL,
  109. $columnDeviceFolder TEXT,
  110. $columnLatitude REAL,
  111. $columnLongitude REAL,
  112. $columnFileType INTEGER,
  113. $columnModificationTime TEXT NOT NULL,
  114. $columnEncryptedKey TEXT,
  115. $columnKeyDecryptionNonce TEXT,
  116. $columnFileDecryptionHeader TEXT,
  117. $columnThumbnailDecryptionHeader TEXT,
  118. $columnMetadataDecryptionHeader TEXT,
  119. $columnIsDeleted INTEGER DEFAULT 0,
  120. $columnCreationTime TEXT NOT NULL,
  121. $columnUpdationTime TEXT,
  122. UNIQUE($columnLocalID, $columnUploadedFileID, $columnCollectionID)
  123. );
  124. ''',
  125. ];
  126. }
  127. static List<String> addIndices() {
  128. return [
  129. '''
  130. CREATE INDEX IF NOT EXISTS collection_id_index ON $filesTable($columnCollectionID);
  131. ''',
  132. '''
  133. CREATE INDEX IF NOT EXISTS device_folder_index ON $filesTable($columnDeviceFolder);
  134. ''',
  135. '''
  136. CREATE INDEX IF NOT EXISTS creation_time_index ON $filesTable($columnCreationTime);
  137. ''',
  138. '''
  139. CREATE INDEX IF NOT EXISTS updation_time_index ON $filesTable($columnUpdationTime);
  140. '''
  141. ];
  142. }
  143. static List<String> alterDeviceFolderToAllowNULL() {
  144. return [
  145. ...createTable(tempTable),
  146. '''
  147. INSERT INTO $tempTable
  148. SELECT *
  149. FROM $filesTable;
  150. DROP TABLE $filesTable;
  151. ALTER TABLE $tempTable
  152. RENAME TO $filesTable;
  153. '''
  154. ];
  155. }
  156. static List<String> alterTimestampColumnTypes() {
  157. return [
  158. '''
  159. DROP TABLE IF EXISTS $tempTable;
  160. ''',
  161. '''
  162. CREATE TABLE $tempTable (
  163. $columnGeneratedID INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
  164. $columnLocalID TEXT,
  165. $columnUploadedFileID INTEGER DEFAULT -1,
  166. $columnOwnerID INTEGER,
  167. $columnCollectionID INTEGER DEFAULT -1,
  168. $columnTitle TEXT NOT NULL,
  169. $columnDeviceFolder TEXT,
  170. $columnLatitude REAL,
  171. $columnLongitude REAL,
  172. $columnFileType INTEGER,
  173. $columnModificationTime INTEGER NOT NULL,
  174. $columnEncryptedKey TEXT,
  175. $columnKeyDecryptionNonce TEXT,
  176. $columnFileDecryptionHeader TEXT,
  177. $columnThumbnailDecryptionHeader TEXT,
  178. $columnMetadataDecryptionHeader TEXT,
  179. $columnCreationTime INTEGER NOT NULL,
  180. $columnUpdationTime INTEGER,
  181. UNIQUE($columnLocalID, $columnUploadedFileID, $columnCollectionID)
  182. );
  183. ''',
  184. '''
  185. INSERT INTO $tempTable
  186. SELECT
  187. $columnGeneratedID,
  188. $columnLocalID,
  189. $columnUploadedFileID,
  190. $columnOwnerID,
  191. $columnCollectionID,
  192. $columnTitle,
  193. $columnDeviceFolder,
  194. $columnLatitude,
  195. $columnLongitude,
  196. $columnFileType,
  197. CAST($columnModificationTime AS INTEGER),
  198. $columnEncryptedKey,
  199. $columnKeyDecryptionNonce,
  200. $columnFileDecryptionHeader,
  201. $columnThumbnailDecryptionHeader,
  202. $columnMetadataDecryptionHeader,
  203. CAST($columnCreationTime AS INTEGER),
  204. CAST($columnUpdationTime AS INTEGER)
  205. FROM $filesTable;
  206. ''',
  207. '''
  208. DROP TABLE $filesTable;
  209. ''',
  210. '''
  211. ALTER TABLE $tempTable
  212. RENAME TO $filesTable;
  213. ''',
  214. ];
  215. }
  216. static List<String> addMetadataColumns() {
  217. return [
  218. '''
  219. ALTER TABLE $filesTable ADD COLUMN $columnFileSubType INTEGER;
  220. ''',
  221. '''
  222. ALTER TABLE $filesTable ADD COLUMN $columnDuration INTEGER;
  223. ''',
  224. '''
  225. ALTER TABLE $filesTable ADD COLUMN $columnExif TEXT;
  226. ''',
  227. '''
  228. ALTER TABLE $filesTable ADD COLUMN $columnHash TEXT;
  229. ''',
  230. '''
  231. ALTER TABLE $filesTable ADD COLUMN $columnMetadataVersion INTEGER;
  232. ''',
  233. ];
  234. }
  235. static List<String> addMagicMetadataColumns() {
  236. return [
  237. '''
  238. ALTER TABLE $filesTable ADD COLUMN $columnMMdEncodedJson TEXT DEFAULT '{}';
  239. ''',
  240. '''
  241. ALTER TABLE $filesTable ADD COLUMN $columnMMdVersion INTEGER DEFAULT 0;
  242. ''',
  243. '''
  244. ALTER TABLE $filesTable ADD COLUMN $columnMMdVisibility INTEGER DEFAULT $visibilityVisible;
  245. '''
  246. ];
  247. }
  248. static List<String> addUniqueConstraintOnCollectionFiles() {
  249. return [
  250. '''
  251. DELETE from $filesTable where $columnCollectionID || '-' || $columnUploadedFileID IN
  252. (SELECT $columnCollectionID || '-' || $columnUploadedFileID from $filesTable WHERE
  253. $columnCollectionID is not NULL AND $columnUploadedFileID is NOT NULL
  254. AND $columnCollectionID != -1 AND $columnUploadedFileID != -1
  255. GROUP BY ($columnCollectionID || '-' || $columnUploadedFileID) HAVING count(*) > 1)
  256. AND ($columnCollectionID || '-' || $columnUploadedFileID || '-' || $columnGeneratedID) NOT IN
  257. (SELECT $columnCollectionID || '-' || $columnUploadedFileID || '-' || max($columnGeneratedID)
  258. from $filesTable WHERE
  259. $columnCollectionID is not NULL AND $columnUploadedFileID is NOT NULL
  260. AND $columnCollectionID != -1 AND $columnUploadedFileID != -1 GROUP BY
  261. ($columnCollectionID || '-' || $columnUploadedFileID) HAVING count(*) > 1);
  262. ''',
  263. '''
  264. CREATE UNIQUE INDEX IF NOT EXISTS cid_uid ON $filesTable ($columnCollectionID, $columnUploadedFileID)
  265. WHERE $columnCollectionID is not NULL AND $columnUploadedFileID is not NULL
  266. AND $columnCollectionID != -1 AND $columnUploadedFileID != -1;
  267. '''
  268. ];
  269. }
  270. static List<String> addPubMagicMetadataColumns() {
  271. return [
  272. '''
  273. ALTER TABLE $filesTable ADD COLUMN $columnPubMMdEncodedJson TEXT DEFAULT '{}';
  274. ''',
  275. '''
  276. ALTER TABLE $filesTable ADD COLUMN $columnPubMMdVersion INTEGER DEFAULT 0;
  277. '''
  278. ];
  279. }
  280. static List<String> createOnDeviceFilesAndPathCollection() {
  281. return [
  282. '''
  283. CREATE TABLE IF NOT EXISTS device_files (
  284. id TEXT NOT NULL,
  285. path_id TEXT NOT NULL,
  286. UNIQUE(id, path_id)
  287. );
  288. ''',
  289. '''
  290. CREATE TABLE IF NOT EXISTS device_collections (
  291. id TEXT PRIMARY KEY NOT NULL,
  292. name TEXT,
  293. modified_at INTEGER NOT NULL DEFAULT 0,
  294. should_backup INTEGER NOT NULL DEFAULT 0,
  295. count INTEGER NOT NULL DEFAULT 0,
  296. collection_id INTEGER DEFAULT -1,
  297. upload_strategy INTEGER DEFAULT 0,
  298. cover_id TEXT
  299. );
  300. ''',
  301. '''
  302. CREATE INDEX IF NOT EXISTS df_id_idx ON device_files (id);
  303. ''',
  304. '''
  305. CREATE INDEX IF NOT EXISTS df_path_id_idx ON device_files (path_id);
  306. ''',
  307. ];
  308. }
  309. static List<String> createEntityDataTable() {
  310. return [
  311. '''
  312. CREATE TABLE IF NOT EXISTS entities (
  313. id TEXT PRIMARY KEY NOT NULL,
  314. type TEXT NOT NULL,
  315. ownerID INTEGER NOT NULL,
  316. data TEXT NOT NULL DEFAULT '{}',
  317. updatedAt INTEGER NOT NULL
  318. );
  319. '''
  320. ];
  321. }
  322. static List<String> addFileSizeColumn() {
  323. return [
  324. '''
  325. ALTER TABLE $filesTable ADD COLUMN $columnFileSize INTEGER;
  326. ''',
  327. ];
  328. }
  329. static List<String> updateIndexes() {
  330. return [
  331. '''
  332. DROP INDEX IF EXISTS device_folder_index;
  333. ''',
  334. '''
  335. CREATE INDEX IF NOT EXISTS file_hash_index ON $filesTable($columnHash);
  336. ''',
  337. ];
  338. }
  339. Future<void> clearTable() async {
  340. final db = await instance.database;
  341. await db.delete(filesTable);
  342. await db.delete("device_files");
  343. await db.delete("device_collections");
  344. await db.delete("entities");
  345. }
  346. Future<void> deleteDB() async {
  347. if (kDebugMode) {
  348. debugPrint("Deleting files db");
  349. final io.Directory documentsDirectory =
  350. await getApplicationDocumentsDirectory();
  351. final String path = join(documentsDirectory.path, _databaseName);
  352. io.File(path).deleteSync(recursive: true);
  353. _dbFuture = null;
  354. }
  355. }
  356. Future<void> insertMultiple(
  357. List<File> files, {
  358. ConflictAlgorithm conflictAlgorithm = ConflictAlgorithm.replace,
  359. }) async {
  360. final startTime = DateTime.now();
  361. final db = await instance.database;
  362. var batch = db.batch();
  363. int batchCounter = 0;
  364. for (File file in files) {
  365. if (batchCounter == 400) {
  366. await batch.commit(noResult: true);
  367. batch = db.batch();
  368. batchCounter = 0;
  369. }
  370. batch.insert(
  371. filesTable,
  372. _getRowForFile(file),
  373. conflictAlgorithm: conflictAlgorithm,
  374. );
  375. batchCounter++;
  376. }
  377. await batch.commit(noResult: true);
  378. final endTime = DateTime.now();
  379. final duration = Duration(
  380. microseconds:
  381. endTime.microsecondsSinceEpoch - startTime.microsecondsSinceEpoch,
  382. );
  383. _logger.info(
  384. "Batch insert of " +
  385. files.length.toString() +
  386. " took " +
  387. duration.inMilliseconds.toString() +
  388. "ms.",
  389. );
  390. }
  391. Future<int> insert(File file) async {
  392. final db = await instance.database;
  393. return db.insert(
  394. filesTable,
  395. _getRowForFile(file),
  396. conflictAlgorithm: ConflictAlgorithm.replace,
  397. );
  398. }
  399. Future<File?> getFile(int generatedID) async {
  400. final db = await instance.database;
  401. final results = await db.query(
  402. filesTable,
  403. where: '$columnGeneratedID = ?',
  404. whereArgs: [generatedID],
  405. );
  406. if (results.isEmpty) {
  407. return null;
  408. }
  409. return convertToFiles(results)[0];
  410. }
  411. Future<File?> getUploadedFile(int uploadedID, int collectionID) async {
  412. final db = await instance.database;
  413. final results = await db.query(
  414. filesTable,
  415. where: '$columnUploadedFileID = ? AND $columnCollectionID = ?',
  416. whereArgs: [
  417. uploadedID,
  418. collectionID,
  419. ],
  420. );
  421. if (results.isEmpty) {
  422. return null;
  423. }
  424. return convertToFiles(results)[0];
  425. }
  426. Future<Set<int>> getUploadedFileIDs(int collectionID) async {
  427. final db = await instance.database;
  428. final results = await db.query(
  429. filesTable,
  430. columns: [columnUploadedFileID],
  431. where:
  432. '$columnCollectionID = ? AND ($columnUploadedFileID IS NOT NULL AND $columnUploadedFileID IS NOT -1)',
  433. whereArgs: [
  434. collectionID,
  435. ],
  436. );
  437. final ids = <int>{};
  438. for (final result in results) {
  439. ids.add(result[columnUploadedFileID] as int);
  440. }
  441. return ids;
  442. }
  443. Future<BackedUpFileIDs> getBackedUpIDs() async {
  444. final db = await instance.database;
  445. final results = await db.query(
  446. filesTable,
  447. columns: [columnLocalID, columnUploadedFileID],
  448. where:
  449. '$columnLocalID IS NOT NULL AND ($columnUploadedFileID IS NOT NULL AND $columnUploadedFileID IS NOT -1)',
  450. );
  451. final localIDs = <String>{};
  452. final uploadedIDs = <int>{};
  453. for (final result in results) {
  454. localIDs.add(result[columnLocalID] as String);
  455. uploadedIDs.add(result[columnUploadedFileID] as int);
  456. }
  457. return BackedUpFileIDs(localIDs.toList(), uploadedIDs.toList());
  458. }
  459. Future<FileLoadResult> getAllPendingOrUploadedFiles(
  460. int startTime,
  461. int endTime,
  462. int ownerID, {
  463. int? limit,
  464. bool? asc,
  465. int visibility = visibilityVisible,
  466. Set<int>? ignoredCollectionIDs,
  467. }) async {
  468. final stopWatch = Stopwatch()..start();
  469. final db = await instance.database;
  470. final order = (asc ?? false ? 'ASC' : 'DESC');
  471. final results = await db.query(
  472. filesTable,
  473. where:
  474. '$columnCreationTime >= ? AND $columnCreationTime <= ? AND ($columnOwnerID IS NULL OR $columnOwnerID = ?) AND ($columnCollectionID IS NOT NULL AND $columnCollectionID IS NOT -1)'
  475. ' AND $columnMMdVisibility = ?',
  476. whereArgs: [startTime, endTime, ownerID, visibility],
  477. orderBy:
  478. '$columnCreationTime ' + order + ', $columnModificationTime ' + order,
  479. limit: limit,
  480. );
  481. final files = convertToFiles(results);
  482. final List<File> deduplicatedFiles =
  483. _deduplicatedAndFilterIgnoredFiles(files, ignoredCollectionIDs);
  484. dev.log(
  485. "getAllPendingOrUploadedFiles time taken: ${stopWatch.elapsedMilliseconds} ms",
  486. );
  487. stopWatch.stop();
  488. return FileLoadResult(deduplicatedFiles, files.length == limit);
  489. }
  490. Future<FileLoadResult> getAllLocalAndUploadedFiles(
  491. int startTime,
  492. int endTime,
  493. int ownerID, {
  494. int? limit,
  495. bool? asc,
  496. Set<int>? ignoredCollectionIDs,
  497. }) async {
  498. final db = await instance.database;
  499. final order = (asc ?? false ? 'ASC' : 'DESC');
  500. final results = await db.query(
  501. filesTable,
  502. where:
  503. '$columnCreationTime >= ? AND $columnCreationTime <= ? AND ($columnOwnerID IS NULL OR $columnOwnerID = ?) AND ($columnMMdVisibility IS NULL OR $columnMMdVisibility = ?)'
  504. ' AND ($columnLocalID IS NOT NULL OR ($columnCollectionID IS NOT NULL AND $columnCollectionID IS NOT -1))',
  505. whereArgs: [startTime, endTime, ownerID, visibilityVisible],
  506. orderBy:
  507. '$columnCreationTime ' + order + ', $columnModificationTime ' + order,
  508. limit: limit,
  509. );
  510. final files = convertToFiles(results);
  511. final List<File> deduplicatedFiles =
  512. _deduplicatedAndFilterIgnoredFiles(files, ignoredCollectionIDs);
  513. return FileLoadResult(deduplicatedFiles, files.length == limit);
  514. }
  515. List<File> deduplicateByLocalID(List<File> files) {
  516. final localIDs = <String>{};
  517. final List<File> deduplicatedFiles = [];
  518. for (final file in files) {
  519. final id = file.localID;
  520. if (id == null) {
  521. continue;
  522. }
  523. if (localIDs.contains(id)) {
  524. continue;
  525. }
  526. localIDs.add(id);
  527. deduplicatedFiles.add(file);
  528. }
  529. return deduplicatedFiles;
  530. }
  531. List<File> _deduplicatedAndFilterIgnoredFiles(
  532. List<File> files,
  533. Set<int>? ignoredCollectionIDs,
  534. ) {
  535. final Set<int> uploadedFileIDs = <int>{};
  536. // ignoredFileUploadIDs is to keep a track of files which are part of
  537. // archived collection
  538. final Set<int> ignoredFileUploadIDs = <int>{};
  539. final List<File> deduplicatedFiles = [];
  540. for (final file in files) {
  541. final id = file.uploadedFileID;
  542. final bool isFileUploaded = id != null && id != -1;
  543. final bool isCollectionIgnored = ignoredCollectionIDs != null &&
  544. ignoredCollectionIDs.contains(file.collectionID);
  545. if (isCollectionIgnored || ignoredFileUploadIDs.contains(id)) {
  546. if (isFileUploaded) {
  547. ignoredFileUploadIDs.add(id);
  548. // remove the file from the list of deduplicated files
  549. if (uploadedFileIDs.contains(id)) {
  550. deduplicatedFiles
  551. .removeWhere((element) => element.uploadedFileID == id);
  552. uploadedFileIDs.remove(id);
  553. }
  554. }
  555. continue;
  556. }
  557. if (isFileUploaded && uploadedFileIDs.contains(id)) {
  558. continue;
  559. }
  560. if (isFileUploaded) {
  561. uploadedFileIDs.add(id);
  562. }
  563. deduplicatedFiles.add(file);
  564. }
  565. return deduplicatedFiles;
  566. }
  567. Future<FileLoadResult> getFilesInCollection(
  568. int collectionID,
  569. int startTime,
  570. int endTime, {
  571. int? limit,
  572. bool? asc,
  573. int visibility = visibilityVisible,
  574. }) async {
  575. final db = await instance.database;
  576. final order = (asc ?? false ? 'ASC' : 'DESC');
  577. const String whereClause =
  578. '$columnCollectionID = ? AND $columnCreationTime >= ? AND $columnCreationTime <= ?';
  579. final List<Object> whereArgs = [collectionID, startTime, endTime];
  580. final results = await db.query(
  581. filesTable,
  582. where: whereClause,
  583. whereArgs: whereArgs,
  584. orderBy:
  585. '$columnCreationTime ' + order + ', $columnModificationTime ' + order,
  586. limit: limit,
  587. );
  588. final files = convertToFiles(results);
  589. return FileLoadResult(files, files.length == limit);
  590. }
  591. Future<List<File>> getAllFilesCollection(int collectionID) async {
  592. final db = await instance.database;
  593. const String whereClause = '$columnCollectionID = ?';
  594. final List<Object> whereArgs = [collectionID];
  595. final results = await db.query(
  596. filesTable,
  597. where: whereClause,
  598. whereArgs: whereArgs,
  599. );
  600. final files = convertToFiles(results);
  601. return files;
  602. }
  603. Future<FileLoadResult> getFilesInCollections(
  604. List<int> collectionIDs,
  605. int startTime,
  606. int endTime,
  607. int userID, {
  608. int? limit,
  609. bool? asc,
  610. }) async {
  611. if (collectionIDs.isEmpty) {
  612. return FileLoadResult(<File>[], false);
  613. }
  614. String inParam = "";
  615. for (final id in collectionIDs) {
  616. inParam += "'" + id.toString() + "',";
  617. }
  618. inParam = inParam.substring(0, inParam.length - 1);
  619. final db = await instance.database;
  620. final order = (asc ?? false ? 'ASC' : 'DESC');
  621. final String whereClause =
  622. '$columnCollectionID IN ($inParam) AND $columnCreationTime >= ? AND '
  623. '$columnCreationTime <= ? AND $columnOwnerID = ?';
  624. final List<Object> whereArgs = [startTime, endTime, userID];
  625. final results = await db.query(
  626. filesTable,
  627. where: whereClause,
  628. whereArgs: whereArgs,
  629. orderBy:
  630. '$columnCreationTime ' + order + ', $columnModificationTime ' + order,
  631. limit: limit,
  632. );
  633. final files = convertToFiles(results);
  634. final dedupeResult = _deduplicatedAndFilterIgnoredFiles(files, {});
  635. _logger.info("Fetched " + dedupeResult.length.toString() + " files");
  636. return FileLoadResult(files, files.length == limit);
  637. }
  638. Future<List<File>> getFilesCreatedWithinDurations(
  639. List<List<int>> durations,
  640. Set<int> ignoredCollectionIDs, {
  641. String order = 'ASC',
  642. }) async {
  643. if (durations.isEmpty) {
  644. return <File>[];
  645. }
  646. final db = await instance.database;
  647. String whereClause = "( ";
  648. for (int index = 0; index < durations.length; index++) {
  649. whereClause += "($columnCreationTime >= " +
  650. durations[index][0].toString() +
  651. " AND $columnCreationTime < " +
  652. durations[index][1].toString() +
  653. ")";
  654. if (index != durations.length - 1) {
  655. whereClause += " OR ";
  656. }
  657. }
  658. whereClause += ") AND $columnMMdVisibility = $visibilityVisible";
  659. final results = await db.query(
  660. filesTable,
  661. where: whereClause,
  662. orderBy: '$columnCreationTime ' + order,
  663. );
  664. final files = convertToFiles(results);
  665. return _deduplicatedAndFilterIgnoredFiles(files, ignoredCollectionIDs);
  666. }
  667. // Files which user added to a collection manually but they are not
  668. // uploaded yet or files belonging to a collection which is marked for backup
  669. Future<List<File>> getFilesPendingForUpload() async {
  670. final db = await instance.database;
  671. final results = await db.query(
  672. filesTable,
  673. where:
  674. '($columnUploadedFileID IS NULL OR $columnUploadedFileID IS -1) AND '
  675. '$columnCollectionID IS NOT NULL AND $columnCollectionID IS NOT -1 AND '
  676. '$columnLocalID IS NOT NULL AND $columnLocalID IS NOT -1',
  677. orderBy: '$columnCreationTime DESC',
  678. groupBy: columnLocalID,
  679. );
  680. final files = convertToFiles(results);
  681. // future-safe filter just to ensure that the query doesn't end up returning files
  682. // which should not be backed up
  683. files.removeWhere(
  684. (e) =>
  685. e.collectionID == null ||
  686. e.localID == null ||
  687. e.uploadedFileID != null,
  688. );
  689. return files;
  690. }
  691. Future<List<File>> getUnUploadedLocalFiles() async {
  692. final db = await instance.database;
  693. final results = await db.query(
  694. filesTable,
  695. where:
  696. '($columnUploadedFileID IS NULL OR $columnUploadedFileID IS -1) AND $columnLocalID IS NOT NULL',
  697. orderBy: '$columnCreationTime DESC',
  698. groupBy: columnLocalID,
  699. );
  700. return convertToFiles(results);
  701. }
  702. Future<List<int>> getUploadedFileIDsToBeUpdated(int ownerID) async {
  703. final db = await instance.database;
  704. final rows = await db.query(
  705. filesTable,
  706. columns: [columnUploadedFileID],
  707. where: '($columnLocalID IS NOT NULL AND $columnOwnerID = ? AND '
  708. '($columnUploadedFileID '
  709. 'IS NOT '
  710. 'NULL AND $columnUploadedFileID IS NOT -1) AND $columnUpdationTime IS NULL)',
  711. whereArgs: [ownerID],
  712. orderBy: '$columnCreationTime DESC',
  713. distinct: true,
  714. );
  715. final uploadedFileIDs = <int>[];
  716. for (final row in rows) {
  717. uploadedFileIDs.add(row[columnUploadedFileID] as int);
  718. }
  719. return uploadedFileIDs;
  720. }
  721. Future<File?> getUploadedFileInAnyCollection(int uploadedFileID) async {
  722. final db = await instance.database;
  723. final results = await db.query(
  724. filesTable,
  725. where: '$columnUploadedFileID = ?',
  726. whereArgs: [
  727. uploadedFileID,
  728. ],
  729. limit: 1,
  730. );
  731. if (results.isEmpty) {
  732. return null;
  733. }
  734. return convertToFiles(results)[0];
  735. }
  736. Future<Set<String>> getExistingLocalFileIDs(int ownerID) async {
  737. final db = await instance.database;
  738. final rows = await db.query(
  739. filesTable,
  740. columns: [columnLocalID],
  741. distinct: true,
  742. where: '$columnLocalID IS NOT NULL AND ($columnOwnerID IS NULL OR '
  743. '$columnOwnerID = ?)',
  744. whereArgs: [ownerID],
  745. );
  746. final result = <String>{};
  747. for (final row in rows) {
  748. result.add(row[columnLocalID] as String);
  749. }
  750. return result;
  751. }
  752. Future<Set<String>> getLocalIDsMarkedForOrAlreadyUploaded(int ownerID) async {
  753. final db = await instance.database;
  754. final rows = await db.query(
  755. filesTable,
  756. columns: [columnLocalID],
  757. distinct: true,
  758. where: '$columnLocalID IS NOT NULL AND ($columnCollectionID IS NOT NULL '
  759. 'AND '
  760. '$columnCollectionID != -1) AND ($columnOwnerID = ? OR '
  761. '$columnOwnerID IS NULL)',
  762. whereArgs: [ownerID],
  763. );
  764. final result = <String>{};
  765. for (final row in rows) {
  766. result.add(row[columnLocalID] as String);
  767. }
  768. return result;
  769. }
  770. Future<Set<String>> getLocalFileIDsForCollection(int collectionID) async {
  771. final db = await instance.database;
  772. final rows = await db.query(
  773. filesTable,
  774. columns: [columnLocalID],
  775. where: '$columnLocalID IS NOT NULL AND $columnCollectionID = ?',
  776. whereArgs: [collectionID],
  777. );
  778. final result = <String>{};
  779. for (final row in rows) {
  780. result.add(row[columnLocalID] as String);
  781. }
  782. return result;
  783. }
  784. // Sets the collectionID for the files with given LocalIDs if the
  785. // corresponding file entries are not already mapped to some other collection
  786. Future<int> setCollectionIDForUnMappedLocalFiles(
  787. int collectionID,
  788. Set<String> localIDs,
  789. ) async {
  790. final db = await instance.database;
  791. String inParam = "";
  792. for (final localID in localIDs) {
  793. inParam += "'" + localID + "',";
  794. }
  795. inParam = inParam.substring(0, inParam.length - 1);
  796. return await db.rawUpdate(
  797. '''
  798. UPDATE $filesTable
  799. SET $columnCollectionID = $collectionID
  800. WHERE $columnLocalID IN ($inParam) AND ($columnCollectionID IS NULL OR
  801. $columnCollectionID = -1);
  802. ''',
  803. );
  804. }
  805. Future<int> updateUploadedFile(
  806. String localID,
  807. String? title,
  808. Location? location,
  809. int creationTime,
  810. int modificationTime,
  811. int? updationTime,
  812. ) async {
  813. final db = await instance.database;
  814. return await db.update(
  815. filesTable,
  816. {
  817. columnTitle: title,
  818. columnLatitude: location?.latitude,
  819. columnLongitude: location?.longitude,
  820. columnCreationTime: creationTime,
  821. columnModificationTime: modificationTime,
  822. columnUpdationTime: updationTime,
  823. },
  824. where: '$columnLocalID = ?',
  825. whereArgs: [localID],
  826. );
  827. }
  828. /*
  829. This method should only return localIDs which are not uploaded yet
  830. and can be mapped to incoming remote entry
  831. */
  832. Future<List<File>> getUnlinkedLocalMatchesForRemoteFile(
  833. int ownerID,
  834. String localID,
  835. FileType fileType, {
  836. required String title,
  837. required String deviceFolder,
  838. }) async {
  839. final db = await instance.database;
  840. // on iOS, match using localID and fileType. title can either match or
  841. // might be null based on how the file was imported
  842. String whereClause = ''' ($columnOwnerID = ? OR $columnOwnerID IS NULL) AND
  843. $columnLocalID = ? AND $columnFileType = ? AND
  844. ($columnTitle=? OR $columnTitle IS NULL) ''';
  845. List<Object> whereArgs = [
  846. ownerID,
  847. localID,
  848. getInt(fileType),
  849. title,
  850. ];
  851. if (io.Platform.isAndroid) {
  852. whereClause = ''' ($columnOwnerID = ? OR $columnOwnerID IS NULL) AND
  853. $columnLocalID = ? AND $columnFileType = ? AND $columnTitle=? AND $columnDeviceFolder= ?
  854. ''';
  855. whereArgs = [
  856. ownerID,
  857. localID,
  858. getInt(fileType),
  859. title,
  860. deviceFolder,
  861. ];
  862. }
  863. final rows = await db.query(
  864. filesTable,
  865. where: whereClause,
  866. whereArgs: whereArgs,
  867. );
  868. return convertToFiles(rows);
  869. }
  870. Future<List<File>> getUploadedFilesWithHashes(
  871. FileHashData hashData,
  872. FileType fileType,
  873. int ownerID,
  874. ) async {
  875. String inParam = "'${hashData.fileHash}'";
  876. if (fileType == FileType.livePhoto && hashData.zipHash != null) {
  877. inParam += ",'${hashData.zipHash}'";
  878. }
  879. final db = await instance.database;
  880. final rows = await db.query(
  881. filesTable,
  882. where: '($columnUploadedFileID != NULL OR $columnUploadedFileID != -1) '
  883. 'AND $columnOwnerID = ? AND $columnFileType ='
  884. ' ? '
  885. 'AND $columnHash IN ($inParam)',
  886. whereArgs: [
  887. ownerID,
  888. getInt(fileType),
  889. ],
  890. );
  891. return convertToFiles(rows);
  892. }
  893. Future<int> update(File file) async {
  894. final db = await instance.database;
  895. return await db.update(
  896. filesTable,
  897. _getRowForFile(file),
  898. where: '$columnGeneratedID = ?',
  899. whereArgs: [file.generatedID],
  900. );
  901. }
  902. Future<int> updateUploadedFileAcrossCollections(File file) async {
  903. final db = await instance.database;
  904. return await db.update(
  905. filesTable,
  906. _getRowForFileWithoutCollection(file),
  907. where: '$columnUploadedFileID = ?',
  908. whereArgs: [file.uploadedFileID],
  909. );
  910. }
  911. Future<int> updateLocalIDForUploaded(int uploadedID, String localID) async {
  912. final db = await instance.database;
  913. return await db.update(
  914. filesTable,
  915. {columnLocalID: localID},
  916. where: '$columnUploadedFileID = ? AND $columnLocalID IS NULL',
  917. whereArgs: [uploadedID],
  918. );
  919. }
  920. Future<int> delete(int uploadedFileID) async {
  921. final db = await instance.database;
  922. return db.delete(
  923. filesTable,
  924. where: '$columnUploadedFileID =?',
  925. whereArgs: [uploadedFileID],
  926. );
  927. }
  928. Future<int> deleteByGeneratedID(int genID) async {
  929. final db = await instance.database;
  930. return db.delete(
  931. filesTable,
  932. where: '$columnGeneratedID =?',
  933. whereArgs: [genID],
  934. );
  935. }
  936. Future<int> deleteMultipleUploadedFiles(List<int> uploadedFileIDs) async {
  937. final db = await instance.database;
  938. return await db.delete(
  939. filesTable,
  940. where: '$columnUploadedFileID IN (${uploadedFileIDs.join(', ')})',
  941. );
  942. }
  943. Future<int> deleteMultipleByGeneratedIDs(List<int> generatedIDs) async {
  944. if (generatedIDs.isEmpty) {
  945. return 0;
  946. }
  947. final db = await instance.database;
  948. return await db.delete(
  949. filesTable,
  950. where: '$columnGeneratedID IN (${generatedIDs.join(', ')})',
  951. );
  952. }
  953. Future<int> deleteLocalFile(File file) async {
  954. final db = await instance.database;
  955. if (file.localID != null) {
  956. // delete all files with same local ID
  957. return db.delete(
  958. filesTable,
  959. where: '$columnLocalID =?',
  960. whereArgs: [file.localID],
  961. );
  962. } else {
  963. return db.delete(
  964. filesTable,
  965. where: '$columnGeneratedID =?',
  966. whereArgs: [file.generatedID],
  967. );
  968. }
  969. }
  970. Future<void> deleteLocalFiles(List<String> localIDs) async {
  971. String inParam = "";
  972. for (final localID in localIDs) {
  973. inParam += "'" + localID + "',";
  974. }
  975. inParam = inParam.substring(0, inParam.length - 1);
  976. final db = await instance.database;
  977. await db.rawQuery(
  978. '''
  979. UPDATE $filesTable
  980. SET $columnLocalID = NULL
  981. WHERE $columnLocalID IN ($inParam);
  982. ''',
  983. );
  984. }
  985. Future<List<File>> getLocalFiles(List<String> localIDs) async {
  986. String inParam = "";
  987. for (final localID in localIDs) {
  988. inParam += "'" + localID + "',";
  989. }
  990. inParam = inParam.substring(0, inParam.length - 1);
  991. final db = await instance.database;
  992. final results = await db.query(
  993. filesTable,
  994. where: '$columnLocalID IN ($inParam)',
  995. );
  996. return convertToFiles(results);
  997. }
  998. Future<int> deleteUnSyncedLocalFiles(List<String> localIDs) async {
  999. String inParam = "";
  1000. for (final localID in localIDs) {
  1001. inParam += "'" + localID + "',";
  1002. }
  1003. inParam = inParam.substring(0, inParam.length - 1);
  1004. final db = await instance.database;
  1005. return db.delete(
  1006. filesTable,
  1007. where:
  1008. '($columnUploadedFileID is NULL OR $columnUploadedFileID = -1 ) AND $columnLocalID IN ($inParam)',
  1009. );
  1010. }
  1011. Future<int> deleteFromCollection(int uploadedFileID, int collectionID) async {
  1012. final db = await instance.database;
  1013. return db.delete(
  1014. filesTable,
  1015. where: '$columnUploadedFileID = ? AND $columnCollectionID = ?',
  1016. whereArgs: [uploadedFileID, collectionID],
  1017. );
  1018. }
  1019. Future<int> deleteFilesFromCollection(
  1020. int collectionID,
  1021. List<int> uploadedFileIDs,
  1022. ) async {
  1023. final db = await instance.database;
  1024. return db.delete(
  1025. filesTable,
  1026. where:
  1027. '$columnCollectionID = ? AND $columnUploadedFileID IN (${uploadedFileIDs.join(', ')})',
  1028. whereArgs: [collectionID],
  1029. );
  1030. }
  1031. Future<int> collectionFileCount(int collectionID) async {
  1032. final db = await instance.database;
  1033. final count = Sqflite.firstIntValue(
  1034. await db.rawQuery(
  1035. 'SELECT COUNT(*) FROM $filesTable where $columnCollectionID = '
  1036. '$collectionID AND $columnUploadedFileID IS NOT -1',
  1037. ),
  1038. );
  1039. return count ?? 0;
  1040. }
  1041. Future<int> archivedFilesCount(
  1042. int visibility,
  1043. int ownerID,
  1044. Set<int> hiddenCollections,
  1045. ) async {
  1046. final db = await instance.database;
  1047. final count = Sqflite.firstIntValue(
  1048. await db.rawQuery(
  1049. 'SELECT COUNT(distinct($columnUploadedFileID)) FROM $filesTable where '
  1050. '$columnMMdVisibility'
  1051. ' = $visibility AND $columnOwnerID = $ownerID AND $columnCollectionID NOT IN (${hiddenCollections.join(', ')})',
  1052. ),
  1053. );
  1054. return count ?? 0;
  1055. }
  1056. Future<int> deleteCollection(int collectionID) async {
  1057. final db = await instance.database;
  1058. return db.delete(
  1059. filesTable,
  1060. where: '$columnCollectionID = ?',
  1061. whereArgs: [collectionID],
  1062. );
  1063. }
  1064. Future<int> removeFromCollection(int collectionID, List<int> fileIDs) async {
  1065. final db = await instance.database;
  1066. return db.delete(
  1067. filesTable,
  1068. where:
  1069. '$columnCollectionID =? AND $columnUploadedFileID IN (${fileIDs.join(', ')})',
  1070. whereArgs: [collectionID],
  1071. );
  1072. }
  1073. Future<List<File>> getPendingUploadForCollection(int collectionID) async {
  1074. final db = await instance.database;
  1075. final results = await db.query(
  1076. filesTable,
  1077. where: '$columnCollectionID = ? AND ($columnUploadedFileID IS NULL OR '
  1078. '$columnUploadedFileID = -1)',
  1079. whereArgs: [collectionID],
  1080. );
  1081. return convertToFiles(results);
  1082. }
  1083. Future<Set<String>> getLocalIDsPresentInEntries(
  1084. List<File> existingFiles,
  1085. int collectionID,
  1086. ) async {
  1087. String inParam = "";
  1088. for (final existingFile in existingFiles) {
  1089. if (existingFile.localID != null) {
  1090. inParam += "'" + existingFile.localID! + "',";
  1091. }
  1092. }
  1093. inParam = inParam.substring(0, inParam.length - 1);
  1094. final db = await instance.database;
  1095. final rows = await db.rawQuery(
  1096. '''
  1097. SELECT $columnLocalID
  1098. FROM $filesTable
  1099. WHERE $columnLocalID IN ($inParam) AND $columnCollectionID !=
  1100. $collectionID AND $columnLocalID IS NOT NULL;
  1101. ''',
  1102. );
  1103. final result = <String>{};
  1104. for (final row in rows) {
  1105. result.add(row[columnLocalID] as String);
  1106. }
  1107. return result;
  1108. }
  1109. Future<List<File>> getLatestCollectionFiles() async {
  1110. debugPrint("Fetching latestCollectionFiles from db");
  1111. const String query = '''
  1112. SELECT $filesTable.*
  1113. FROM $filesTable
  1114. INNER JOIN
  1115. (
  1116. SELECT $columnCollectionID, MAX($columnCreationTime) AS max_creation_time
  1117. FROM $filesTable
  1118. WHERE
  1119. ($columnCollectionID IS NOT NULL AND $columnCollectionID IS NOT -1
  1120. AND $columnUploadedFileID IS NOT NULL AND $columnUploadedFileID IS
  1121. NOT -1)
  1122. GROUP BY $columnCollectionID
  1123. ) latest_files
  1124. ON $filesTable.$columnCollectionID = latest_files.$columnCollectionID
  1125. AND $filesTable.$columnCreationTime = latest_files.max_creation_time;
  1126. ''';
  1127. final db = await instance.database;
  1128. final rows = await db.rawQuery(
  1129. query,
  1130. );
  1131. final files = convertToFiles(rows);
  1132. // TODO: Do this de-duplication within the SQL Query
  1133. final collectionMap = <int, File>{};
  1134. for (final file in files) {
  1135. if (collectionMap.containsKey(file.collectionID)) {
  1136. if (collectionMap[file.collectionID]!.updationTime! <
  1137. file.updationTime!) {
  1138. continue;
  1139. }
  1140. }
  1141. collectionMap[file.collectionID!] = file;
  1142. }
  1143. return collectionMap.values.toList();
  1144. }
  1145. Future<void> markForReUploadIfLocationMissing(List<String> localIDs) async {
  1146. if (localIDs.isEmpty) {
  1147. return;
  1148. }
  1149. String inParam = "";
  1150. for (final localID in localIDs) {
  1151. inParam += "'" + localID + "',";
  1152. }
  1153. inParam = inParam.substring(0, inParam.length - 1);
  1154. final db = await instance.database;
  1155. await db.rawUpdate(
  1156. '''
  1157. UPDATE $filesTable
  1158. SET $columnUpdationTime = NULL
  1159. WHERE $columnLocalID IN ($inParam)
  1160. AND ($columnLatitude IS NULL OR $columnLongitude IS NULL OR $columnLongitude = 0.0 or $columnLongitude = 0.0);
  1161. ''',
  1162. );
  1163. }
  1164. Future<bool> doesFileExistInCollection(
  1165. int uploadedFileID,
  1166. int collectionID,
  1167. ) async {
  1168. final db = await instance.database;
  1169. final rows = await db.query(
  1170. filesTable,
  1171. where: '$columnUploadedFileID = ? AND $columnCollectionID = ?',
  1172. whereArgs: [uploadedFileID, collectionID],
  1173. limit: 1,
  1174. );
  1175. return rows.isNotEmpty;
  1176. }
  1177. Future<Map<int, File>> getFilesFromIDs(List<int> ids) async {
  1178. final result = <int, File>{};
  1179. if (ids.isEmpty) {
  1180. return result;
  1181. }
  1182. String inParam = "";
  1183. for (final id in ids) {
  1184. inParam += "'" + id.toString() + "',";
  1185. }
  1186. inParam = inParam.substring(0, inParam.length - 1);
  1187. final db = await instance.database;
  1188. final results = await db.query(
  1189. filesTable,
  1190. where: '$columnUploadedFileID IN ($inParam)',
  1191. );
  1192. final files = convertToFiles(results);
  1193. for (final file in files) {
  1194. result[file.uploadedFileID!] = file;
  1195. }
  1196. return result;
  1197. }
  1198. Future<Map<int, File>> getFilesFromGeneratedIDs(List<int> ids) async {
  1199. final result = <int, File>{};
  1200. if (ids.isEmpty) {
  1201. return result;
  1202. }
  1203. String inParam = "";
  1204. for (final id in ids) {
  1205. inParam += "'" + id.toString() + "',";
  1206. }
  1207. inParam = inParam.substring(0, inParam.length - 1);
  1208. final db = await instance.database;
  1209. final results = await db.query(
  1210. filesTable,
  1211. where: '$columnGeneratedID IN ($inParam)',
  1212. );
  1213. final files = convertToFiles(results);
  1214. for (final file in files) {
  1215. result[file.generatedID as int] = file;
  1216. }
  1217. return result;
  1218. }
  1219. Future<Map<int, List<File>>> getAllFilesGroupByCollectionID(
  1220. List<int> ids,
  1221. ) async {
  1222. final result = <int, List<File>>{};
  1223. if (ids.isEmpty) {
  1224. return result;
  1225. }
  1226. String inParam = "";
  1227. for (final id in ids) {
  1228. inParam += "'" + id.toString() + "',";
  1229. }
  1230. inParam = inParam.substring(0, inParam.length - 1);
  1231. final db = await instance.database;
  1232. final results = await db.query(
  1233. filesTable,
  1234. where: '$columnUploadedFileID IN ($inParam)',
  1235. );
  1236. final files = convertToFiles(results);
  1237. for (File eachFile in files) {
  1238. if (!result.containsKey(eachFile.collectionID)) {
  1239. result[eachFile.collectionID as int] = <File>[];
  1240. }
  1241. result[eachFile.collectionID]!.add(eachFile);
  1242. }
  1243. return result;
  1244. }
  1245. Future<Set<int>> getAllCollectionIDsOfFile(
  1246. int uploadedFileID,
  1247. ) async {
  1248. final db = await instance.database;
  1249. final results = await db.query(
  1250. filesTable,
  1251. where: '$columnUploadedFileID = ? AND $columnCollectionID != -1',
  1252. columns: [columnCollectionID],
  1253. whereArgs: [uploadedFileID],
  1254. distinct: true,
  1255. );
  1256. final collectionIDsOfFile = <int>{};
  1257. for (var result in results) {
  1258. collectionIDsOfFile.add(result['collection_id'] as int);
  1259. }
  1260. return collectionIDsOfFile;
  1261. }
  1262. List<File> convertToFiles(List<Map<String, dynamic>> results) {
  1263. final List<File> files = [];
  1264. for (final result in results) {
  1265. files.add(_getFileFromRow(result));
  1266. }
  1267. return files;
  1268. }
  1269. Future<List<String>> getGeneratedIDForFilesOlderThan(
  1270. int cutOffTime,
  1271. int ownerID,
  1272. ) async {
  1273. final db = await instance.database;
  1274. final rows = await db.query(
  1275. filesTable,
  1276. columns: [columnGeneratedID],
  1277. distinct: true,
  1278. where:
  1279. '$columnCreationTime <= ? AND ($columnOwnerID IS NULL OR $columnOwnerID = ?)',
  1280. whereArgs: [cutOffTime, ownerID],
  1281. );
  1282. final result = <String>[];
  1283. for (final row in rows) {
  1284. result.add(row[columnGeneratedID].toString());
  1285. }
  1286. return result;
  1287. }
  1288. // For givenUserID, get List of unique LocalIDs for files which are
  1289. // uploaded by the given user and location is missing
  1290. Future<List<String>> getLocalIDsForFilesWithoutLocation(int ownerID) async {
  1291. final db = await instance.database;
  1292. final rows = await db.query(
  1293. filesTable,
  1294. columns: [columnLocalID],
  1295. distinct: true,
  1296. where: '$columnOwnerID = ? AND $columnLocalID IS NOT NULL AND '
  1297. '($columnLatitude IS NULL OR '
  1298. '$columnLongitude IS NULL OR $columnLongitude = 0.0 or $columnLongitude = 0.0)',
  1299. whereArgs: [ownerID],
  1300. );
  1301. final result = <String>[];
  1302. for (final row in rows) {
  1303. result.add(row[columnLocalID].toString());
  1304. }
  1305. return result;
  1306. }
  1307. // For given list of localIDs and ownerID, get a list of uploaded files
  1308. // owned by given user
  1309. Future<List<File>> getFilesForLocalIDs(
  1310. List<String> localIDs,
  1311. int ownerID,
  1312. ) async {
  1313. final db = await instance.database;
  1314. final rows = await db.query(
  1315. filesTable,
  1316. where:
  1317. '$columnLocalID IN (${localIDs.map((e) => "'$e'").join(',')}) AND $columnOwnerID = ?',
  1318. whereArgs: [ownerID],
  1319. );
  1320. return _deduplicatedAndFilterIgnoredFiles(convertToFiles(rows), {});
  1321. }
  1322. Future<List<File>> getAllFilesFromDB(Set<int> collectionsToIgnore) async {
  1323. final db = await instance.database;
  1324. final List<Map<String, dynamic>> result =
  1325. await db.query(filesTable, orderBy: '$columnCreationTime DESC');
  1326. final List<File> files = convertToFiles(result);
  1327. final List<File> deduplicatedFiles =
  1328. _deduplicatedAndFilterIgnoredFiles(files, collectionsToIgnore);
  1329. return deduplicatedFiles;
  1330. }
  1331. Future<Map<FileType, int>> fetchFilesCountbyType(int userID) async {
  1332. final db = await instance.database;
  1333. final result = await db.rawQuery(
  1334. "SELECT $columnFileType, COUNT(DISTINCT $columnUploadedFileID) FROM $filesTable WHERE $columnUploadedFileID != -1 AND $columnOwnerID == $userID GROUP BY $columnFileType",
  1335. );
  1336. final filesCount = <FileType, int>{};
  1337. for (var e in result) {
  1338. filesCount.addAll(
  1339. {getFileType(e[columnFileType] as int): e.values.last as int},
  1340. );
  1341. }
  1342. return filesCount;
  1343. }
  1344. Future<FileLoadResult> fetchAllUploadedAndSharedFilesWithLocation(
  1345. int startTime,
  1346. int endTime, {
  1347. int? limit,
  1348. bool? asc,
  1349. Set<int>? ignoredCollectionIDs,
  1350. }) async {
  1351. final db = await instance.database;
  1352. final order = (asc ?? false ? 'ASC' : 'DESC');
  1353. final results = await db.query(
  1354. filesTable,
  1355. where:
  1356. '$columnLatitude IS NOT NULL AND $columnLongitude IS NOT NULL AND ($columnLatitude IS NOT 0 OR $columnLongitude IS NOT 0)'
  1357. ' AND $columnCreationTime >= ? AND $columnCreationTime <= ? AND '
  1358. '($columnMMdVisibility IS NULL OR $columnMMdVisibility = ?)'
  1359. ' AND ($columnLocalID IS NOT NULL OR ($columnCollectionID IS NOT NULL AND $columnCollectionID IS NOT -1))',
  1360. whereArgs: [startTime, endTime, visibilityVisible],
  1361. orderBy:
  1362. '$columnCreationTime ' + order + ', $columnModificationTime ' + order,
  1363. limit: limit,
  1364. );
  1365. final files = convertToFiles(results);
  1366. final List<File> deduplicatedFiles =
  1367. _deduplicatedAndFilterIgnoredFiles(files, ignoredCollectionIDs);
  1368. return FileLoadResult(deduplicatedFiles, files.length == limit);
  1369. }
  1370. Map<String, dynamic> _getRowForFile(File file) {
  1371. final row = <String, dynamic>{};
  1372. if (file.generatedID != null) {
  1373. row[columnGeneratedID] = file.generatedID;
  1374. }
  1375. row[columnLocalID] = file.localID;
  1376. row[columnUploadedFileID] = file.uploadedFileID ?? -1;
  1377. row[columnOwnerID] = file.ownerID;
  1378. row[columnCollectionID] = file.collectionID ?? -1;
  1379. row[columnTitle] = file.title;
  1380. row[columnDeviceFolder] = file.deviceFolder;
  1381. // if (file.location == null ||
  1382. // (file.location!.latitude == null && file.location!.longitude == null)) {
  1383. // file.location = Location.randomLocation();
  1384. // }
  1385. if (file.location != null) {
  1386. row[columnLatitude] = file.location!.latitude;
  1387. row[columnLongitude] = file.location!.longitude;
  1388. }
  1389. row[columnFileType] = getInt(file.fileType);
  1390. row[columnCreationTime] = file.creationTime;
  1391. row[columnModificationTime] = file.modificationTime;
  1392. row[columnUpdationTime] = file.updationTime;
  1393. row[columnEncryptedKey] = file.encryptedKey;
  1394. row[columnKeyDecryptionNonce] = file.keyDecryptionNonce;
  1395. row[columnFileDecryptionHeader] = file.fileDecryptionHeader;
  1396. row[columnThumbnailDecryptionHeader] = file.thumbnailDecryptionHeader;
  1397. row[columnMetadataDecryptionHeader] = file.metadataDecryptionHeader;
  1398. row[columnFileSubType] = file.fileSubType ?? -1;
  1399. row[columnDuration] = file.duration ?? 0;
  1400. row[columnExif] = file.exif;
  1401. row[columnHash] = file.hash;
  1402. row[columnMetadataVersion] = file.metadataVersion;
  1403. row[columnFileSize] = file.fileSize;
  1404. row[columnMMdVersion] = file.mMdVersion;
  1405. row[columnMMdEncodedJson] = file.mMdEncodedJson ?? '{}';
  1406. row[columnMMdVisibility] = file.magicMetadata.visibility;
  1407. row[columnPubMMdVersion] = file.pubMmdVersion;
  1408. row[columnPubMMdEncodedJson] = file.pubMmdEncodedJson ?? '{}';
  1409. // override existing fields to avoid re-writing all queries and logic
  1410. if (file.pubMagicMetadata != null) {
  1411. if (file.pubMagicMetadata!.editedTime != null) {
  1412. row[columnCreationTime] = file.pubMagicMetadata!.editedTime;
  1413. }
  1414. if (file.pubMagicMetadata!.lat != null &&
  1415. file.pubMagicMetadata!.long != null) {
  1416. row[columnLatitude] = file.pubMagicMetadata!.lat;
  1417. row[columnLongitude] = file.pubMagicMetadata!.long;
  1418. }
  1419. }
  1420. return row;
  1421. }
  1422. Map<String, dynamic> _getRowForFileWithoutCollection(File file) {
  1423. final row = <String, dynamic>{};
  1424. row[columnLocalID] = file.localID;
  1425. row[columnUploadedFileID] = file.uploadedFileID ?? -1;
  1426. row[columnOwnerID] = file.ownerID;
  1427. row[columnTitle] = file.title;
  1428. row[columnDeviceFolder] = file.deviceFolder;
  1429. if (file.location != null) {
  1430. row[columnLatitude] = file.location!.latitude;
  1431. row[columnLongitude] = file.location!.longitude;
  1432. }
  1433. row[columnFileType] = getInt(file.fileType);
  1434. row[columnCreationTime] = file.creationTime;
  1435. row[columnModificationTime] = file.modificationTime;
  1436. row[columnUpdationTime] = file.updationTime;
  1437. row[columnFileDecryptionHeader] = file.fileDecryptionHeader;
  1438. row[columnThumbnailDecryptionHeader] = file.thumbnailDecryptionHeader;
  1439. row[columnMetadataDecryptionHeader] = file.metadataDecryptionHeader;
  1440. row[columnFileSubType] = file.fileSubType ?? -1;
  1441. row[columnDuration] = file.duration ?? 0;
  1442. row[columnExif] = file.exif;
  1443. row[columnHash] = file.hash;
  1444. row[columnMetadataVersion] = file.metadataVersion;
  1445. row[columnMMdVersion] = file.mMdVersion;
  1446. row[columnMMdEncodedJson] = file.mMdEncodedJson ?? '{}';
  1447. row[columnMMdVisibility] = file.magicMetadata.visibility;
  1448. row[columnPubMMdVersion] = file.pubMmdVersion;
  1449. row[columnPubMMdEncodedJson] = file.pubMmdEncodedJson ?? '{}';
  1450. if (file.pubMagicMetadata != null &&
  1451. file.pubMagicMetadata!.editedTime != null) {
  1452. // override existing creationTime to avoid re-writing all queries related
  1453. // to loading the gallery
  1454. row[columnCreationTime] = file.pubMagicMetadata!.editedTime!;
  1455. }
  1456. return row;
  1457. }
  1458. File _getFileFromRow(Map<String, dynamic> row) {
  1459. final file = File();
  1460. file.generatedID = row[columnGeneratedID];
  1461. file.localID = row[columnLocalID];
  1462. file.uploadedFileID =
  1463. row[columnUploadedFileID] == -1 ? null : row[columnUploadedFileID];
  1464. file.ownerID = row[columnOwnerID];
  1465. file.collectionID =
  1466. row[columnCollectionID] == -1 ? null : row[columnCollectionID];
  1467. file.title = row[columnTitle];
  1468. file.deviceFolder = row[columnDeviceFolder];
  1469. if (row[columnLatitude] != null && row[columnLongitude] != null) {
  1470. file.location = Location(
  1471. latitude: row[columnLatitude],
  1472. longitude: row[columnLongitude],
  1473. );
  1474. }
  1475. file.fileType = getFileType(row[columnFileType]);
  1476. file.creationTime = row[columnCreationTime];
  1477. file.modificationTime = row[columnModificationTime];
  1478. file.updationTime = row[columnUpdationTime] ?? -1;
  1479. file.encryptedKey = row[columnEncryptedKey];
  1480. file.keyDecryptionNonce = row[columnKeyDecryptionNonce];
  1481. file.fileDecryptionHeader = row[columnFileDecryptionHeader];
  1482. file.thumbnailDecryptionHeader = row[columnThumbnailDecryptionHeader];
  1483. file.metadataDecryptionHeader = row[columnMetadataDecryptionHeader];
  1484. file.fileSubType = row[columnFileSubType] ?? -1;
  1485. file.duration = row[columnDuration] ?? 0;
  1486. file.exif = row[columnExif];
  1487. file.hash = row[columnHash];
  1488. file.metadataVersion = row[columnMetadataVersion] ?? 0;
  1489. file.fileSize = row[columnFileSize];
  1490. file.mMdVersion = row[columnMMdVersion] ?? 0;
  1491. file.mMdEncodedJson = row[columnMMdEncodedJson] ?? '{}';
  1492. file.pubMmdVersion = row[columnPubMMdVersion] ?? 0;
  1493. file.pubMmdEncodedJson = row[columnPubMMdEncodedJson] ?? '{}';
  1494. return file;
  1495. }
  1496. }