import 'package:flutter/material.dart'; import 'package:logging/logging.dart'; import 'package:photos/core/configuration.dart'; import "package:photos/core/errors.dart"; import 'package:photos/db/files_db.dart'; import 'package:photos/ente_theme_data.dart'; import "package:photos/generated/l10n.dart"; import 'package:photos/models/api/collection/create_request.dart'; import "package:photos/models/api/collection/user.dart"; import 'package:photos/models/collection.dart'; import 'package:photos/models/file.dart'; import 'package:photos/models/files_split.dart'; import "package:photos/models/metadata/collection_magic.dart"; import "package:photos/models/metadata/common_keys.dart"; import 'package:photos/services/collections_service.dart'; import 'package:photos/services/hidden_service.dart'; import 'package:photos/services/user_service.dart'; import 'package:photos/theme/colors.dart'; import 'package:photos/theme/ente_theme.dart'; import 'package:photos/ui/common/progress_dialog.dart'; import 'package:photos/ui/components/action_sheet_widget.dart'; import 'package:photos/ui/components/buttons/button_widget.dart'; import 'package:photos/ui/components/dialog_widget.dart'; import 'package:photos/ui/components/models/button_type.dart'; import 'package:photos/ui/payment/subscription.dart'; import 'package:photos/utils/date_time_util.dart'; import 'package:photos/utils/dialog_util.dart'; import 'package:photos/utils/email_util.dart'; import 'package:photos/utils/share_util.dart'; import 'package:photos/utils/toast_util.dart'; import "package:styled_text/styled_text.dart"; class CollectionActions { final Logger logger = Logger((CollectionActions).toString()); final CollectionsService collectionsService; CollectionActions(this.collectionsService); Future enableUrl( BuildContext context, Collection collection, { bool enableCollect = false, }) async { try { await CollectionsService.instance.createShareUrl( collection, enableCollect: enableCollect, ); return true; } catch (e) { if (e is SharingNotPermittedForFreeAccountsError) { _showUnSupportedAlert(context); } else { logger.severe("Failed to update shareUrl collection", e); showGenericErrorDialog(context: context); } return false; } } Future disableUrl(BuildContext context, Collection collection) async { final actionResult = await showActionSheet( context: context, buttons: [ ButtonWidget( buttonType: ButtonType.critical, isInAlert: true, shouldStickToDarkTheme: true, buttonAction: ButtonAction.first, shouldSurfaceExecutionStates: true, labelText: S.of(context).yesRemove, onTap: () async { await CollectionsService.instance.disableShareUrl(collection); }, ), ButtonWidget( buttonType: ButtonType.secondary, buttonAction: ButtonAction.cancel, isInAlert: true, shouldStickToDarkTheme: true, labelText: S.of(context).cancel, ) ], title: S.of(context).removePublicLink, body: //'This will remove the public link for accessing "${collection.name}".', S.of(context).disableLinkMessage(collection.displayName), ); if (actionResult?.action != null) { if (actionResult!.action == ButtonAction.error) { showGenericErrorDialog(context: context); } return actionResult.action == ButtonAction.first; } else { return false; } } Future createSharedCollectionLink( BuildContext context, List files, ) async { final dialog = createProgressDialog( context, S.of(context).creatingLink, isDismissible: true, ); dialog.show(); try { // create album with emptyName, use collectionCreationTime on UI to // show name logger.finest("creating album for sharing files"); final File fileWithMinCreationTime = files.reduce( (a, b) => (a.creationTime ?? 0) < (b.creationTime ?? 0) ? a : b, ); final File fileWithMaxCreationTime = files.reduce( (a, b) => (a.creationTime ?? 0) > (b.creationTime ?? 0) ? a : b, ); final String dummyName = getNameForDateRange( fileWithMinCreationTime.creationTime!, fileWithMaxCreationTime.creationTime!, ); final CreateRequest req = await collectionsService.buildCollectionCreateRequest( dummyName, visibility: visibleVisibility, subType: subTypeSharedFilesCollection, ); final collection = await collectionsService.createAndCacheCollection( req, ); logger.finest("adding files to share to new album"); await collectionsService.addToCollection(collection.id, files); logger.finest("creating public link for the newly created album"); await CollectionsService.instance.createShareUrl(collection); dialog.hide(); return collection; } catch (e, s) { dialog.hide(); showGenericErrorDialog(context: context); logger.severe("Failing to create link for selected files", e, s); } return null; } // removeParticipant remove the user from a share album Future removeParticipant( BuildContext context, Collection collection, User user, ) async { final actionResult = await showActionSheet( context: context, buttons: [ ButtonWidget( buttonType: ButtonType.critical, isInAlert: true, shouldStickToDarkTheme: true, buttonAction: ButtonAction.first, shouldSurfaceExecutionStates: true, labelText: S.of(context).yesRemove, onTap: () async { final newSharees = await CollectionsService.instance .unshare(collection.id, user.email); collection.updateSharees(newSharees); }, ), ButtonWidget( buttonType: ButtonType.secondary, buttonAction: ButtonAction.cancel, isInAlert: true, shouldStickToDarkTheme: true, labelText: S.of(context).cancel, ) ], title: S.of(context).removeWithQuestionMark, body: S.of(context).removeParticipantBody(user.email), ); if (actionResult?.action != null) { if (actionResult!.action == ButtonAction.error) { showGenericErrorDialog(context: context); } return actionResult.action == ButtonAction.first; } return false; } // addEmailToCollection returns true if add operation was successful Future addEmailToCollection( BuildContext context, Collection collection, String email, CollectionParticipantRole role, { bool showProgress = false, }) async { if (!isValidEmail(email)) { await showErrorDialog( context, S.of(context).invalidEmailAddress, S.of(context).enterValidEmail, ); return false; } else if (email.trim() == Configuration.instance.getEmail()) { await showErrorDialog( context, S.of(context).oops, S.of(context).youCannotShareWithYourself, ); return false; } ProgressDialog? dialog; String? publicKey; if (showProgress) { dialog = createProgressDialog( context, S.of(context).sharing, isDismissible: true, ); await dialog.show(); } try { publicKey = await UserService.instance.getPublicKey(email); } catch (e) { await dialog?.hide(); logger.severe("Failed to get public key", e); showGenericErrorDialog(context: context); return false; } // getPublicKey can return null when no user is associated with given // email id if (publicKey == null || publicKey == '') { // todo: neeraj replace this as per the design where a new screen // is used for error. Do this change along with handling of network errors await showDialogWidget( context: context, title: S.of(context).inviteToEnte, icon: Icons.info_outline, body: S.of(context).emailNoEnteAccount(email), isDismissible: true, buttons: [ ButtonWidget( buttonType: ButtonType.neutral, icon: Icons.adaptive.share, labelText: S.of(context).sendInvite, isInAlert: true, onTap: () async { shareText( S.of(context).shareTextRecommendUsingEnte, ); }, ), ], ); return false; } else { try { final newSharees = await CollectionsService.instance .share(collection.id, email, publicKey, role); await dialog?.hide(); collection.updateSharees(newSharees); return true; } catch (e) { await dialog?.hide(); if (e is SharingNotPermittedForFreeAccountsError) { _showUnSupportedAlert(context); } else { logger.severe("failed to share collection", e); showGenericErrorDialog(context: context); } return false; } } } // deleteCollectionSheet returns true if the album is successfully deleted Future deleteCollectionSheet( BuildContext bContext, Collection collection, ) async { final textTheme = getEnteTextTheme(bContext); final currentUserID = Configuration.instance.getUserID()!; if (collection.owner!.id != currentUserID) { throw AssertionError("Can not delete album owned by others"); } if (collection.hasSharees) { final bool confirmDelete = await _confirmSharedAlbumDeletion(bContext, collection); if (!confirmDelete) { return false; } } final actionResult = await showActionSheet( context: bContext, buttons: [ ButtonWidget( labelText: S.of(bContext).keepPhotos, buttonType: ButtonType.neutral, buttonSize: ButtonSize.large, buttonAction: ButtonAction.first, shouldStickToDarkTheme: true, isInAlert: true, onTap: () async { try { final List files = await FilesDB.instance.getAllFilesCollection(collection.id); await moveFilesFromCurrentCollection(bContext, collection, files); // collection should be empty on server now await collectionsService.trashEmptyCollection(collection); } catch (e, s) { logger.severe("Failed to keep photos & delete collection", e, s); rethrow; } }, ), ButtonWidget( labelText: S.of(bContext).deletePhotos, buttonType: ButtonType.critical, buttonSize: ButtonSize.large, buttonAction: ButtonAction.second, shouldStickToDarkTheme: true, isInAlert: true, onTap: () async { try { await collectionsService.trashNonEmptyCollection(collection); } catch (e) { logger.severe("Failed to delete collection", e); rethrow; } }, ), ButtonWidget( labelText: S.of(bContext).cancel, buttonType: ButtonType.secondary, buttonSize: ButtonSize.large, buttonAction: ButtonAction.third, shouldStickToDarkTheme: true, isInAlert: true, ), ], bodyWidget: StyledText( text: S.of(bContext).deleteAlbumDialog, style: textTheme.body.copyWith(color: textMutedDark), tags: { 'bold': StyledTextTag( style: textTheme.body.copyWith(color: textBaseDark), ), }, ), actionSheetType: ActionSheetType.defaultActionSheet, ); if (actionResult?.action != null && actionResult!.action == ButtonAction.error) { showGenericErrorDialog(context: bContext); return false; } if ((actionResult?.action != null) && (actionResult!.action == ButtonAction.first || actionResult.action == ButtonAction.second)) { return true; } return false; } // _confirmSharedAlbumDeletion should be shown when user tries to delete an // album shared with other ente users. Future _confirmSharedAlbumDeletion( BuildContext context, Collection collection, ) async { final actionResult = await showChoiceActionSheet( context, isCritical: true, title: S.of(context).deleteSharedAlbum, firstButtonLabel: S.of(context).deleteAlbum, body: S.of(context).deleteSharedAlbumDialogBody, ); return actionResult?.action != null && actionResult!.action == ButtonAction.first; } /* _moveFilesFromCurrentCollection removes the file from the current collection. Based on the file and collection ownership, files will be either moved to different collection (Case A). or will just get removed from current collection (Case B). ------------------------------- Case A: Files and collection belong to the same user. Such files will be moved to a collection which belongs to the user and removed from the current collection as part of move operation. Note: Even files are present in the destination collection, we need to make move API call on the server so that the files are removed from current collection and are actually moved to a collection owned by the user. ------------------------------- Case B: Owner of files and collections are different. In such cases, we will just remove (not move) the files from the given collection. */ Future moveFilesFromCurrentCollection( BuildContext context, Collection collection, Iterable files, ) async { final int currentUserID = Configuration.instance.getUserID()!; final isCollectionOwner = collection.owner!.id == currentUserID; final FilesSplit split = FilesSplit.split( files, Configuration.instance.getUserID()!, ); if (isCollectionOwner && split.ownedByOtherUsers.isNotEmpty) { await collectionsService.removeFromCollection( collection.id, split.ownedByOtherUsers, ); } else if (!isCollectionOwner && split.ownedByCurrentUser.isNotEmpty) { // collection is not owned by the user, just remove files owned // by current user and return await collectionsService.removeFromCollection( collection.id, split.ownedByCurrentUser, ); return; } if (!isCollectionOwner && split.ownedByOtherUsers.isNotEmpty) { showShortToast(context, S.of(context).canOnlyRemoveFilesOwnedByYou); return; } // pendingAssignMap keeps a track of files which are yet to be assigned to // to destination collection. final Map pendingAssignMap = {}; // destCollectionToFilesMap contains the destination collection and // files entry which needs to be moved in destination. // After the end of mapping logic, the number of files entries in // pendingAssignMap should be equal to files in destCollectionToFilesMap final Map> destCollectionToFilesMap = {}; final List uploadedIDs = []; for (File f in split.ownedByCurrentUser) { if (f.uploadedFileID != null) { pendingAssignMap[f.uploadedFileID!] = f; uploadedIDs.add(f.uploadedFileID!); } } final Map> collectionToFilesMap = await FilesDB.instance.getAllFilesGroupByCollectionID(uploadedIDs); // Find and map the files from current collection to to entries in other // collections. This mapping is done to avoid moving all the files to // uncategorized during remove from album. for (MapEntry> entry in collectionToFilesMap.entries) { if (!_isAutoMoveCandidate(collection.id, entry.key, currentUserID)) { continue; } final targetCollection = collectionsService.getCollectionByID(entry.key)!; // for each file which already exist in the destination collection // add entries in the moveDestCollectionToFiles map for (File file in entry.value) { // Check if the uploaded file is still waiting to be mapped if (pendingAssignMap.containsKey(file.uploadedFileID)) { if (!destCollectionToFilesMap.containsKey(targetCollection.id)) { destCollectionToFilesMap[targetCollection.id] = []; } destCollectionToFilesMap[targetCollection.id]! .add(pendingAssignMap[file.uploadedFileID!]!); pendingAssignMap.remove(file.uploadedFileID); } } } // Move the remaining files to uncategorized collection if (pendingAssignMap.isNotEmpty) { final Collection uncategorizedCollection = await collectionsService.getUncategorizedCollection(); final int toCollectionID = uncategorizedCollection.id; for (MapEntry entry in pendingAssignMap.entries) { final file = entry.value; if (pendingAssignMap.containsKey(file.uploadedFileID)) { if (!destCollectionToFilesMap.containsKey(toCollectionID)) { destCollectionToFilesMap[toCollectionID] = []; } destCollectionToFilesMap[toCollectionID]! .add(pendingAssignMap[file.uploadedFileID!]!); } } } // Verify that all files are mapped. int mappedFilesCount = 0; destCollectionToFilesMap.forEach((key, value) { mappedFilesCount += value.length; }); if (mappedFilesCount != uploadedIDs.length) { throw AssertionError( "Failed to map all files toMap: ${uploadedIDs.length} and mapped " "$mappedFilesCount", ); } for (MapEntry> entry in destCollectionToFilesMap.entries) { await collectionsService.move(entry.key, collection.id, entry.value); } } // This method returns true if the given destination collection is a good // target to moving files during file remove or delete collection but keey // photos action. Uncategorized or favorite type of collections are not // good auto-move candidates. Uncategorized will be fall back for all files // which could not be mapped to a potential target collection bool _isAutoMoveCandidate(int fromCollectionID, toCollectionID, int userID) { if (fromCollectionID == toCollectionID) { return false; } final Collection? targetCollection = collectionsService.getCollectionByID(toCollectionID); // ignore non-cached collections, uncategorized and favorite // collections and collections ignored by others if (targetCollection == null || (CollectionType.uncategorized == targetCollection.type || targetCollection.type == CollectionType.favorites) || targetCollection.owner!.id != userID) { return false; } return true; } void _showUnSupportedAlert(BuildContext context) { final AlertDialog alert = AlertDialog( title: Text(S.of(context).sorry), content: Text( S.of(context).subscribeToEnableSharing, ), actions: [ TextButton( child: Text( S.of(context).subscribe, style: TextStyle( color: Theme.of(context).colorScheme.greenAlternative, ), ), onPressed: () { Navigator.of(context, rootNavigator: true).pop(); Navigator.of(context).pushReplacement( MaterialPageRoute( builder: (BuildContext context) { return getSubscriptionPage(); }, ), ); }, ), TextButton( child: Text( S.of(context).ok, style: TextStyle( color: Theme.of(context).colorScheme.onSurface, ), ), onPressed: () { Navigator.of(context, rootNavigator: true).pop(); }, ), ], ); showDialog( context: context, builder: (BuildContext context) { return alert; }, ); } }