Refactor formatPhoneNumber to accept countryCode

app/soapbox/components/ui/phone-input/country-code-dropdown.tsx

@@ -0,0 +1,34 @@
+import React from 'react';
+
+import DropdownMenu from 'soapbox/containers/dropdown_menu_container';
+import { COUNTRY_CODES, CountryCode } from 'soapbox/utils/phone';
+
+import type { Menu } from 'soapbox/components/dropdown_menu';
+
+interface ICountryCodeDropdown {
+  countryCode: CountryCode,
+  onChange(countryCode: CountryCode): void,
+}
+
+/** Dropdown menu to select a country code. */
+const CountryCodeDropdown: React.FC<ICountryCodeDropdown> = ({ countryCode, onChange }) => {
+
+  const handleMenuItem = (code: CountryCode) => {
+    return () => {
+      onChange(code);
+    };
+  };
+
+  const menu: Menu = COUNTRY_CODES.map(code => ({
+    text: <>{code}</>,
+    action: handleMenuItem(code),
+  }));
+
+  return (
+    <DropdownMenu items={menu}>
+      <>{countryCode}</>
+    </DropdownMenu>
+  );
+};
+
+export default CountryCodeDropdown;

app/soapbox/components/ui/phone-input/phone-input.tsx

@@ -1,33 +1,52 @@
 import React from 'react';
 
-import { formatPhoneNumber } from 'soapbox/utils/phone';
+import { CountryCode, formatPhoneNumber } from 'soapbox/utils/phone';
 
+import HStack from '../hstack/hstack';
 import Input from '../input/input';
 
-interface IPhoneInput extends Pick<React.InputHTMLAttributes<HTMLInputElement>, 'required'> {
+import CountryCodeDropdown from './country-code-dropdown';
+
+interface IPhoneInput extends Pick<React.InputHTMLAttributes<HTMLInputElement>, 'required' | 'autoFocus'> {
   /** Input phone number. */
   value?: string,
+  /** E164 country code. */
+  countryCode?: CountryCode,
   /** Change event handler taking the formatted input. */
   onChange?: (phone: string) => void,
 }
 
 /** Internationalized phone input with country code picker. */
 const PhoneInput: React.FC<IPhoneInput> = (props) => {
-  const { onChange, ...rest } = props;
+  const { countryCode = 1, value = '', onChange, ...rest } = props;
+
+  const handleCountryChange = (code: CountryCode) => {
+    if (onChange) {
+      onChange(formatPhoneNumber(countryCode, value));
+    }
+  };
 
   /** Pass the formatted phone to the handler. */
   const handleChange: React.ChangeEventHandler<HTMLInputElement> = ({ target }) => {
     if (onChange) {
-      onChange(formatPhoneNumber(target.value));
+      onChange(formatPhoneNumber(countryCode, target.value));
     }
   };
 
   return (
-    <Input
-      type='text'
-      onChange={handleChange}
-      {...rest}
-    />
+    <HStack>
+      <CountryCodeDropdown
+        countryCode={countryCode}
+        onChange={handleCountryChange}
+      />
+
+      <Input
+        type='text'
+        onChange={handleChange}
+        value={value}
+        {...rest}
+      />
+    </HStack>
   );
 };
 

app/soapbox/features/ui/components/modals/verify-sms-modal.tsx

@@ -6,11 +6,10 @@ import { verifyCredentials } from 'soapbox/actions/auth';
 import { closeModal } from 'soapbox/actions/modals';
 import snackbar from 'soapbox/actions/snackbar';
 import { reConfirmPhoneVerification, reRequestPhoneVerification } from 'soapbox/actions/verification';
-import { FormGroup, Input, Modal, Stack, Text } from 'soapbox/components/ui';
+import { FormGroup, PhoneInput, Modal, Stack, Text } from 'soapbox/components/ui';
 import { validPhoneNumberRegex } from 'soapbox/features/verification/steps/sms-verification';
 import { useAppDispatch, useAppSelector } from 'soapbox/hooks';
 import { getAccessToken } from 'soapbox/utils/auth';
-import { formatPhoneNumber } from 'soapbox/utils/phone';
 
 interface IVerifySmsModal {
   onClose: (type: string) => void,
@@ -38,10 +37,8 @@ const VerifySmsModal: React.FC<IVerifySmsModal> = ({ onClose }) => {
 
   const isValid = validPhoneNumberRegex.test(phone);
 
-  const onChange = useCallback((event: React.ChangeEvent<HTMLInputElement>) => {
-    const formattedPhone = formatPhoneNumber(event.target.value);
-
-    setPhone(formattedPhone);
+  const onChange = useCallback((phone: string) => {
+    setPhone(phone);
   }, []);
 
   const handleSubmit = (event: React.MouseEvent) => {
@@ -141,8 +138,7 @@ const VerifySmsModal: React.FC<IVerifySmsModal> = ({ onClose }) => {
       case Statuses.READY:
         return (
           <FormGroup labelText='Phone Number'>
-            <Input
-              type='text'
+            <PhoneInput
               value={phone}
               onChange={onChange}
               required

app/soapbox/utils/__tests__/phone.test.ts

@@ -3,27 +3,27 @@ import { formatPhoneNumber } from '../phone';
 describe('Phone unit tests', () => {
   it('Properly formats', () => {
     let number = '';
-    expect(formatPhoneNumber(number)).toEqual('');
+    expect(formatPhoneNumber(1, number)).toEqual('');
 
     number = '5';
-    expect(formatPhoneNumber(number)).toEqual('+1 (5');
+    expect(formatPhoneNumber(1, number)).toEqual('+1 (5');
 
     number = '55';
-    expect(formatPhoneNumber(number)).toEqual('+1 (55');
+    expect(formatPhoneNumber(1, number)).toEqual('+1 (55');
 
     number = '555';
-    expect(formatPhoneNumber(number)).toEqual('+1 (555');
+    expect(formatPhoneNumber(1, number)).toEqual('+1 (555');
 
     number = '55513';
-    expect(formatPhoneNumber(number)).toEqual('+1 (555) 13');
+    expect(formatPhoneNumber(1, number)).toEqual('+1 (555) 13');
 
     number = '555135';
-    expect(formatPhoneNumber(number)).toEqual('+1 (555) 135');
+    expect(formatPhoneNumber(1, number)).toEqual('+1 (555) 135');
 
     number = '5551350';
-    expect(formatPhoneNumber(number)).toEqual('+1 (555) 135-0');
+    expect(formatPhoneNumber(1, number)).toEqual('+1 (555) 135-0');
 
     number = '5551350123';
-    expect(formatPhoneNumber(number)).toEqual('+1 (555) 135-0123');
+    expect(formatPhoneNumber(1, number)).toEqual('+1 (555) 135-0123');
   });
 });

app/soapbox/utils/phone.ts

@@ -1,3 +1,14 @@
+/** List of supported E164 country codes. */
+const COUNTRY_CODES = [
+  1,
+  44,
+] as const;
+
+/** Supported E164 country code. */
+type CountryCode = typeof COUNTRY_CODES[number];
+
+/** Check whether a given value is a country code. */
+const isCountryCode = (value: any): value is CountryCode => COUNTRY_CODES.includes(value);
 
 function removeFormattingFromNumber(number = '') {
   if (number) {
@@ -7,17 +18,14 @@ function removeFormattingFromNumber(number = '') {
   return number;
 }
 
-function formatPhoneNumber(phoneNumber = '') {
+function formatPhoneNumber(countryCode: CountryCode, phoneNumber = '') {
   let formattedPhoneNumber = '';
-  let strippedPhone = removeFormattingFromNumber(phoneNumber);
-  if (strippedPhone.slice(0, 1) === '1') {
-    strippedPhone = strippedPhone.slice(1);
-  }
+  const strippedPhone = removeFormattingFromNumber(phoneNumber);
 
   for (let i = 0; i < strippedPhone.length && i < 10; i++) {
     const character = strippedPhone.charAt(i);
     if (i === 0) {
-      const prefix = '+1 (';
+      const prefix = `+${countryCode} (`;
       formattedPhoneNumber += prefix + character;
     } else if (i === 3) {
       formattedPhoneNumber += `) ${character}`;
@@ -30,4 +38,9 @@ function formatPhoneNumber(phoneNumber = '') {
   return formattedPhoneNumber;
 }
 
-export { formatPhoneNumber };
+export {
+  COUNTRY_CODES,
+  CountryCode,
+  isCountryCode,
+  formatPhoneNumber,
+};