136 lines
4.2 KiB
TypeScript
136 lines
4.2 KiB
TypeScript
// Copyright 2022 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
import type { CDSResponseType } from '../textsecure/cds/Types.d';
|
|
import type { WebAPIType } from '../textsecure/WebAPI';
|
|
import type { AciString } from '../types/ServiceId';
|
|
import * as log from '../logging/log';
|
|
import { isDirectConversation, isMe } from './whatTypeOfConversation';
|
|
import { parseNumber } from './libphonenumberUtil';
|
|
|
|
type PhoneNumberTransformation = {
|
|
oldPattern: RegExp;
|
|
newPattern: RegExp;
|
|
oldToNew: (e164: string) => string;
|
|
newToOld: (e164: string) => string;
|
|
};
|
|
|
|
const PHONE_TRANSFORMATIONS: Record<string, PhoneNumberTransformation> = {
|
|
'229': {
|
|
// Benin
|
|
oldPattern: /^\+229\d{8}$/,
|
|
newPattern: /^\+22901\d{8}$/,
|
|
oldToNew: (e164: string) => e164.replace(/^\+229(\d{8})$/, '+22901$1'),
|
|
newToOld: (e164: string) => e164.replace(/^\+22901(\d{8})$/, '+229$1'),
|
|
},
|
|
'52': {
|
|
// Mexico
|
|
oldPattern: /^\+521\d{10}$/,
|
|
newPattern: /^\+52\d{10}$/,
|
|
oldToNew: (e164: string) => e164.replace(/^\+521(\d{10})$/, '+52$1'),
|
|
newToOld: (e164: string) => e164.replace(/^\+52(\d{10})$/, '+521$1'),
|
|
},
|
|
'54': {
|
|
// Argentina
|
|
oldPattern: /^\+54\d{10}$/,
|
|
newPattern: /^\+549\d{10}$/,
|
|
oldToNew: (e164: string) => e164.replace(/^\+54(\d{10})$/, '+549$1'),
|
|
newToOld: (e164: string) => e164.replace(/^\+549(\d{10})$/, '+54$1'),
|
|
},
|
|
};
|
|
|
|
type ReturnType = CDSResponseType & {
|
|
// Maps from provided E164 phone numbers to their alternate representations
|
|
// found in CDSI. If a E164 appears as a key in this map, you should use the
|
|
// corresponding E164 value for any subsequent operations, as that's
|
|
// the format stored in CDSI's database.
|
|
transformedE164s: Map<string, string>;
|
|
};
|
|
|
|
export async function getServiceIdsForE164s(
|
|
server: Pick<WebAPIType, 'cdsLookup'>,
|
|
e164s: ReadonlyArray<string>
|
|
): Promise<ReturnType> {
|
|
const expandedE164s = new Set(e164s);
|
|
|
|
const transformationMap = new Map<string, string>();
|
|
|
|
for (const e164 of e164s) {
|
|
const parsedNumber = parseNumber(e164);
|
|
|
|
if (parsedNumber.isValidNumber && parsedNumber.countryCode) {
|
|
const transform = PHONE_TRANSFORMATIONS[parsedNumber.countryCode];
|
|
if (transform) {
|
|
if (transform.oldPattern.test(e164)) {
|
|
const newFormat = transform.oldToNew(e164);
|
|
expandedE164s.add(newFormat);
|
|
transformationMap.set(e164, newFormat);
|
|
} else if (transform.newPattern.test(e164)) {
|
|
const oldFormat = transform.newToOld(e164);
|
|
expandedE164s.add(oldFormat);
|
|
transformationMap.set(e164, oldFormat);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// Note: these have no relationship to supplied e164s. We just provide
|
|
// all available information to the server so that it could return as many
|
|
// ACI+PNI+E164 matches as possible.
|
|
const acisAndAccessKeys = new Array<{ aci: AciString; accessKey: string }>();
|
|
|
|
for (const convo of window.ConversationController.getAll()) {
|
|
if (!isDirectConversation(convo.attributes) || isMe(convo.attributes)) {
|
|
continue;
|
|
}
|
|
|
|
const aci = convo.getAci();
|
|
if (!aci) {
|
|
continue;
|
|
}
|
|
|
|
convo.deriveAccessKeyIfNeeded();
|
|
const accessKey = convo.get('accessKey');
|
|
if (!accessKey) {
|
|
continue;
|
|
}
|
|
|
|
acisAndAccessKeys.push({ aci, accessKey });
|
|
}
|
|
|
|
const expandedE164sArray = Array.from(expandedE164s);
|
|
|
|
log.info(
|
|
`getServiceIdsForE164s(${expandedE164sArray}): acis=${acisAndAccessKeys.length} ` +
|
|
`accessKeys=${acisAndAccessKeys.length}`
|
|
);
|
|
const response = await server.cdsLookup({
|
|
e164s: expandedE164sArray,
|
|
acisAndAccessKeys,
|
|
returnAcisWithoutUaks: false,
|
|
useLibsignal: window.Signal.RemoteConfig.isEnabled(
|
|
'desktop.cdsiViaLibsignal'
|
|
),
|
|
});
|
|
|
|
const e164sWithVariantsInCdsi = new Map(
|
|
Array.from(transformationMap).filter(([providedE164, alternateE164]) => {
|
|
if (
|
|
response.entries.has(providedE164) &&
|
|
response.entries.has(alternateE164)
|
|
) {
|
|
log.warn(`both ${providedE164} and ${alternateE164} are in CDSI`);
|
|
return false;
|
|
}
|
|
if (response.entries.has(alternateE164)) {
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
})
|
|
);
|
|
|
|
return {
|
|
...response,
|
|
transformedE164s: e164sWithVariantsInCdsi,
|
|
};
|
|
}
|