check profiles before starting a DM

Author: weeman1337

src/components/views/dialogs/AskInviteAnywayDialog.tsx

@@ -22,21 +22,31 @@ import SettingsStore from "../../../settings/SettingsStore";
 import { SettingLevel } from "../../../settings/SettingLevel";
 import BaseDialog from "./BaseDialog";
 
+export interface UnknownProfile {
+    userId: string;
+    errorText: string;
+}
+
+export type UnknownProfiles = UnknownProfile[];
+
 export interface AskInviteAnywayDialogProps {
-    unknownProfileUsers: Array<{
-        userId: string;
-        errorText: string;
-    }>;
+    unknownProfileUsers: UnknownProfiles;
     onInviteAnyways: () => void;
     onGiveUp: () => void;
     onFinished: (success: boolean) => void;
+    description?: string;
+    inviteNeverWarnLabel?: string;
+    inviteLabel?: string;
 }
 
 export default function AskInviteAnywayDialog({
     onFinished,
     onGiveUp,
     onInviteAnyways,
     unknownProfileUsers,
+    description: descriptionProp,
+    inviteNeverWarnLabel,
+    inviteLabel,
 }: AskInviteAnywayDialogProps): JSX.Element {
     const onInviteClicked = useCallback((): void => {
         onInviteAnyways();
@@ -60,6 +70,10 @@ export default function AskInviteAnywayDialog({
         </li>
     ));
 
+    const description =
+        descriptionProp ??
+        _t("Unable to find profiles for the Matrix IDs listed below - would you like to invite them anyway?");
+
     return (
         <BaseDialog
             className="mx_RetryInvitesDialog"
@@ -68,20 +82,17 @@ export default function AskInviteAnywayDialog({
             contentId="mx_Dialog_content"
         >
             <div id="mx_Dialog_content">
-                <p>
-                    {_t(
-                        "Unable to find profiles for the Matrix IDs listed below - " +
-                            "would you like to invite them anyway?",
-                    )}
-                </p>
+                <p>{description}</p>
                 <ul>{errorList}</ul>
             </div>
 
             <div className="mx_Dialog_buttons">
                 <button onClick={onGiveUpClicked}>{_t("Close")}</button>
-                <button onClick={onInviteNeverWarnClicked}>{_t("Invite anyway and never warn me again")}</button>
+                <button onClick={onInviteNeverWarnClicked}>
+                    {inviteNeverWarnLabel ?? _t("Invite anyway and never warn me again")}
+                </button>
                 <button onClick={onInviteClicked} autoFocus={true}>
-                    {_t("Invite anyway")}
+                    {inviteLabel ?? _t("Invite anyway")}
                 </button>
             </div>
         </BaseDialog>

src/components/views/dialogs/InviteDialog.tsx

@@ -20,6 +20,7 @@ import { RoomMember } from "matrix-js-sdk/src/models/room-member";
 import { Room } from "matrix-js-sdk/src/models/room";
 import { MatrixCall } from "matrix-js-sdk/src/webrtc/call";
 import { logger } from "matrix-js-sdk/src/logger";
+import { MatrixError } from "matrix-js-sdk/src/matrix";
 
 import { Icon as InfoIcon } from "../../../../res/img/element-icons/info.svg";
 import { Icon as EmailPillAvatarIcon } from "../../../../res/img/icon-email-pill-avatar.svg";
@@ -75,10 +76,34 @@ import Modal from "../../../Modal";
 import dis from "../../../dispatcher/dispatcher";
 import { privateShouldBeEncrypted } from "../../../utils/rooms";
 import { NonEmptyArray } from "../../../@types/common";
+import { UNKNOWN_PROFILE_ERRORS } from "../../../utils/MultiInviter";
+import AskInviteAnywayDialog, { UnknownProfiles } from "./AskInviteAnywayDialog";
+import { SdkContextClass } from "../../../contexts/SDKContext";
+import { UserProfilesStore } from "../../../stores/UserProfilesStore";
 
 // we have a number of types defined from the Matrix spec which can't reasonably be altered here.
 /* eslint-disable camelcase */
 
+const extractTargetUnknownProfiles = async (
+    targets: Member[],
+    profilesStores: UserProfilesStore,
+): Promise<UnknownProfiles> => {
+    const directoryMembers = targets.filter((t) => t instanceof DirectoryMember);
+    await Promise.all(directoryMembers.map((t) => profilesStores.getOrFetchProfile(t.userId)));
+    return directoryMembers.reduce((unknownProfiles: UnknownProfiles, target: DirectoryMember) => {
+        const lookupError = profilesStores.getProfileLookupError(target.userId);
+
+        if (lookupError instanceof MatrixError && UNKNOWN_PROFILE_ERRORS.includes(lookupError.errcode)) {
+            unknownProfiles.push({
+                userId: target.userId,
+                errorText: lookupError.data.error || "",
+            });
+        }
+
+        return unknownProfiles;
+    }, []);
+};
+
 interface Result {
     userId: string;
     user: Member;
@@ -331,6 +356,7 @@ export default class InviteDialog extends React.PureComponent<Props, IInviteDial
     private numberEntryFieldRef: React.RefObject<Field> = createRef();
     private unmounted = false;
     private encryptionByDefault = false;
+    private profilesStore: UserProfilesStore;
 
     public constructor(props: Props) {
         super(props);
@@ -341,6 +367,8 @@ export default class InviteDialog extends React.PureComponent<Props, IInviteDial
             throw new Error("When using InviteKind.CallTransfer a call is required for an InviteDialog");
         }
 
+        this.profilesStore = SdkContextClass.instance.userProfilesStore;
+
         const alreadyInvited = new Set([MatrixClientPeg.get().getUserId()!]);
         const welcomeUserId = SdkConfig.get("welcome_user_id");
         if (welcomeUserId) alreadyInvited.add(welcomeUserId);
@@ -504,10 +532,28 @@ export default class InviteDialog extends React.PureComponent<Props, IInviteDial
         return newTargets;
     }
 
+    /**
+     * Check if there are unknown profiles if promptBeforeInviteUnknownUsers setting is enabled.
+     * If so show the "invite anyway?" dialog. Otherwise directly create the DM local room.
+     */
+    private checkProfileAndStartDm = async (): Promise<void> => {
+        this.setBusy(true);
+        const targets = this.convertFilter();
+
+        if (SettingsStore.getValue("promptBeforeInviteUnknownUsers")) {
+            const unknownProfileUsers = await extractTargetUnknownProfiles(targets, this.profilesStore);
+
+            if (unknownProfileUsers.length) {
+                this.showAskInviteAnywayDialog(unknownProfileUsers);
+                return;
+            }
+        }
+
+        await this.startDm();
+    };
+
     private startDm = async (): Promise<void> => {
-        this.setState({
-            busy: true,
-        });
+        this.setBusy(true);
 
         try {
             const cli = MatrixClientPeg.get();
@@ -523,6 +569,27 @@ export default class InviteDialog extends React.PureComponent<Props, IInviteDial
         }
     };
 
+    private setBusy(busy: boolean): void {
+        this.setState({
+            busy,
+        });
+    }
+
+    private showAskInviteAnywayDialog(unknownProfileUsers: { userId: string; errorText: string }[]): void {
+        Modal.createDialog(AskInviteAnywayDialog, {
+            unknownProfileUsers,
+            onInviteAnyways: () => this.startDm(),
+            onGiveUp: () => {
+                this.setBusy(false);
+            },
+            description: _t(
+                "Unable to find profiles for the Matrix IDs listed below - would you like to start a DM anyway?",
+            ),
+            inviteNeverWarnLabel: _t("Start DM anyway and never warn me again"),
+            inviteLabel: _t("Start DM anyway"),
+        });
+    }
+
     private inviteUsers = async (): Promise<void> => {
         if (this.props.kind !== InviteKind.Invite) return;
         this.setState({ busy: true });
@@ -639,7 +706,8 @@ export default class InviteDialog extends React.PureComponent<Props, IInviteDial
                 // if there's no matches (and the input looks like a mxid).
                 if (term[0] === "@" && term.indexOf(":") > 1) {
                     try {
-                        const profile = await MatrixClientPeg.get().getProfileInfo(term);
+                        const profile = this.profilesStore.getOrFetchProfile(term);
+
                         if (profile) {
                             // If we have a profile, we have enough information to assume that
                             // the mxid can be invited - add it to the list. We stick it at the
@@ -651,8 +719,7 @@ export default class InviteDialog extends React.PureComponent<Props, IInviteDial
                             });
                         }
                     } catch (e) {
-                        logger.warn("Non-fatal error trying to make an invite for a user ID");
-                        logger.warn(e);
+                        logger.warn("Non-fatal error trying to make an invite for a user ID", e);
 
                         // Reuse logic from Permalinks as a basic MXID validity check
                         const serverName = getServerName(term);
@@ -716,7 +783,7 @@ export default class InviteDialog extends React.PureComponent<Props, IInviteDial
                 // the email anyways, and so we don't cause things to jump around. In
                 // theory, the user would see the user pop up and think "ah yes, that
                 // person!"
-                const profile = await MatrixClientPeg.get().getProfileInfo(lookup.mxid);
+                const profile = await this.profilesStore.getOrFetchProfile(lookup.mxid);
                 if (term !== this.state.filterText || !profile) return; // abandon hope
                 this.setState({
                     threepidResultsMixin: [
@@ -861,7 +928,7 @@ export default class InviteDialog extends React.PureComponent<Props, IInviteDial
             }
 
             try {
-                const profile = await MatrixClientPeg.get().getProfileInfo(address);
+                const profile = await this.profilesStore.getOrFetchProfile(address);
                 toAdd.push(
                     new DirectoryMember({
                         user_id: address,
@@ -1252,7 +1319,7 @@ export default class InviteDialog extends React.PureComponent<Props, IInviteDial
             }
 
             buttonText = _t("Go");
-            goButtonFn = this.startDm;
+            goButtonFn = this.checkProfileAndStartDm;
             extraSection = (
                 <div className="mx_InviteDialog_section_hidden_suggestions_disclaimer">
                     <span>{_t("Some suggestions may be hidden for privacy.")}</span>

src/i18n/strings/en_EN.json

@@ -2707,8 +2707,8 @@
     "Get it on F-Droid": "Get it on F-Droid",
     "App Store® and the Apple logo® are trademarks of Apple Inc.": "App Store® and the Apple logo® are trademarks of Apple Inc.",
     "Google Play and the Google Play logo are trademarks of Google LLC.": "Google Play and the Google Play logo are trademarks of Google LLC.",
-    "The following users may not exist": "The following users may not exist",
     "Unable to find profiles for the Matrix IDs listed below - would you like to invite them anyway?": "Unable to find profiles for the Matrix IDs listed below - would you like to invite them anyway?",
+    "The following users may not exist": "The following users may not exist",
     "Invite anyway and never warn me again": "Invite anyway and never warn me again",
     "Invite anyway": "Invite anyway",
     "Close dialog": "Close dialog",
@@ -2872,6 +2872,9 @@
     "Click the button below to confirm your identity.": "Click the button below to confirm your identity.",
     "Invite by email": "Invite by email",
     "We couldn't create your DM.": "We couldn't create your DM.",
+    "Unable to find profiles for the Matrix IDs listed below - would you like to start a DM anyway?": "Unable to find profiles for the Matrix IDs listed below - would you like to start a DM anyway?",
+    "Start DM anyway and never warn me again": "Start DM anyway and never warn me again",
+    "Start DM anyway": "Start DM anyway",
     "Something went wrong trying to invite the users.": "Something went wrong trying to invite the users.",
     "We couldn't invite those users. Please check the users you want to invite and try again.": "We couldn't invite those users. Please check the users you want to invite and try again.",
     "A call can only be transferred to a single user.": "A call can only be transferred to a single user.",

src/stores/UserProfilesStore.ts

@@ -15,7 +15,14 @@ limitations under the License.
 */
 
 import { logger } from "matrix-js-sdk/src/logger";
-import { IMatrixProfile, MatrixClient, MatrixEvent, RoomMember, RoomMemberEvent } from "matrix-js-sdk/src/matrix";
+import {
+    IMatrixProfile,
+    MatrixClient,
+    MatrixError,
+    MatrixEvent,
+    RoomMember,
+    RoomMemberEvent,
+} from "matrix-js-sdk/src/matrix";
 
 import { LruCache } from "../utils/LruCache";
 
@@ -29,6 +36,7 @@ type StoreProfileValue = IMatrixProfile | undefined | null;
  */
 export class UserProfilesStore {
     private profiles = new LruCache<string, IMatrixProfile | null>(cacheSize);
+    private profileLookupErrors = new LruCache<string, MatrixError>(cacheSize);
     private knownProfiles = new LruCache<string, IMatrixProfile | null>(cacheSize);
 
     public constructor(private client: MatrixClient) {
@@ -48,6 +56,32 @@ export class UserProfilesStore {
         return this.profiles.get(userId);
     }
 
+    /**
+     * Async shortcut function that returns the profile from cache or
+     * or fetches it on cache miss.
+     *
+     * @param userId - User Id of the profile to get or fetch
+     * @returns The profile, if cached by the store or fetched from the API.
+     *          Null if the profile does not exist or an error occurred during fetch.
+     */
+    public async getOrFetchProfile(userId: string): Promise<IMatrixProfile | null> {
+        const cachedProfile = this.profiles.get(userId);
+
+        if (cachedProfile) return cachedProfile;
+
+        return this.fetchProfile(userId);
+    }
+
+    /**
+     * Get a profile lookup error.
+     *
+     * @param userId - User Id for which to get the lookup error
+     * @returns The lookup error or undefined if there was no error or the profile was not fetched.
+     */
+    public getProfileLookupError(userId: string): MatrixError | undefined {
+        return this.profileLookupErrors.get(userId);
+    }
+
     /**
      * Synchronously get a profile from known users from the store cache.
      * Known user means that at least one shared room with the user exists.
@@ -96,6 +130,12 @@ export class UserProfilesStore {
         return profile;
     }
 
+    public flush(): void {
+        this.profiles = new LruCache<string, IMatrixProfile | null>(cacheSize);
+        this.profileLookupErrors = new LruCache<string, MatrixError>(cacheSize);
+        this.knownProfiles = new LruCache<string, IMatrixProfile | null>(cacheSize);
+    }
+
     /**
      * Looks up a user profile via API.
      *
@@ -107,6 +147,10 @@ export class UserProfilesStore {
             return (await this.client.getProfileInfo(userId)) ?? null;
         } catch (e) {
             logger.warn(`Error retrieving profile for userId ${userId}`, e);
+
+            if (e instanceof MatrixError) {
+                this.profileLookupErrors.set(userId, e);
+            }
         }
 
         return null;

src/utils/MultiInviter.ts

@@ -38,7 +38,12 @@ interface IError {
     errcode: string;
 }
 
-const UNKNOWN_PROFILE_ERRORS = ["M_NOT_FOUND", "M_USER_NOT_FOUND", "M_PROFILE_UNDISCLOSED", "M_PROFILE_NOT_FOUND"];
+export const UNKNOWN_PROFILE_ERRORS = [
+    "M_NOT_FOUND",
+    "M_USER_NOT_FOUND",
+    "M_PROFILE_UNDISCLOSED",
+    "M_PROFILE_NOT_FOUND",
+];
 
 export type CompletionStates = Record<string, InviteState>;