signal-desktop/ts/components/SafetyNumberViewer.stories.tsx

// Copyright 2020 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only

import * as React from 'react';
import { action } from '@storybook/addon-actions';
import { boolean, text } from '@storybook/addon-knobs';

import type { PropsType } from './SafetyNumberViewer';
import { SafetyNumberViewer } from './SafetyNumberViewer';
import { setupI18n } from '../util/setupI18n';
import {
  SafetyNumberIdentifierType,
  SafetyNumberMode,
} from '../types/safetyNumber';
import enMessages from '../../_locales/en/messages.json';
import { getDefaultConversation } from '../test-both/helpers/getDefaultConversation';

function generateQRData() {
  const data = new Uint8Array(128);
  for (let i = 0; i < data.length; i += 1) {
    data[i] = Math.floor(Math.random() * 256);
  }
  return data;
}

function generateNumberBlocks() {
  const result = new Array<string>();
  for (let i = 0; i < 12; i += 1) {
    let digits = '';
    for (let j = 0; j < 5; j += 1) {
      digits += Math.floor(Math.random() * 10);
    }
    result.push(digits);
  }
  return result;
}

const i18n = setupI18n('en', enMessages);

const contactWithAllData = getDefaultConversation({
  title: 'Summer Smith',
  name: 'Summer Smith',
  phoneNumber: '(305) 123-4567',
  isVerified: true,
});

const contactWithJustProfile = getDefaultConversation({
  avatarPath: undefined,
  title: '-*Smartest Dude*-',
  profileName: '-*Smartest Dude*-',
  name: undefined,
  phoneNumber: '(305) 123-4567',
});

const contactWithJustNumber = getDefaultConversation({
  avatarPath: undefined,
  profileName: undefined,
  name: undefined,
  title: '(305) 123-4567',
  phoneNumber: '(305) 123-4567',
});

const contactWithNothing = getDefaultConversation({
  id: 'some-guid',
  avatarPath: undefined,
  profileName: undefined,
  title: 'Unknown contact',
  name: undefined,
  phoneNumber: undefined,
});

const createProps = (overrideProps: Partial<PropsType> = {}): PropsType => ({
  contact: overrideProps.contact || contactWithAllData,
  generateSafetyNumber: action('generate-safety-number'),
  i18n,
  safetyNumberMode:
    overrideProps.safetyNumberMode ?? SafetyNumberMode.DefaultE164AndThenACI,
  safetyNumbers: overrideProps.safetyNumbers ?? [
    {
      identifierType: SafetyNumberIdentifierType.ACIIdentifier,
      numberBlocks: text(
        'safetyNumber',
        generateNumberBlocks().join(' ')
      ).split(' '),
      qrData: generateQRData(),
    },
  ],
  toggleVerified: action('toggle-verified'),
  verificationDisabled: boolean(
    'verificationDisabled',
    overrideProps.verificationDisabled !== undefined
      ? overrideProps.verificationDisabled
      : false
  ),
  onClose: action('onClose'),
});

export default {
  title: 'Components/SafetyNumberViewer',
};

export function SafetyNumber(): JSX.Element {
  return <SafetyNumberViewer {...createProps({})} />;
}

export function SafetyNumberBeforeE164Transition(): JSX.Element {
  return (
    <SafetyNumberViewer
      {...createProps({
        safetyNumberMode: SafetyNumberMode.JustE164,
        safetyNumbers: [
          {
            identifierType: SafetyNumberIdentifierType.E164Identifier,
            numberBlocks: text(
              'safetyNumber',
              generateNumberBlocks().join(' ')
            ).split(' '),
            qrData: generateQRData(),
          },
        ],
      })}
    />
  );
}

SafetyNumberBeforeE164Transition.story = {
  name: 'Safety Number (before e164 transition)',
};

export function SafetyNumberE164Transition(): JSX.Element {
  return (
    <SafetyNumberViewer
      {...createProps({
        safetyNumberMode: SafetyNumberMode.DefaultE164AndThenACI,
        safetyNumbers: [
          {
            identifierType: SafetyNumberIdentifierType.E164Identifier,
            numberBlocks: generateNumberBlocks(),
            qrData: generateQRData(),
          },
          {
            identifierType: SafetyNumberIdentifierType.ACIIdentifier,
            numberBlocks: generateNumberBlocks(),
            qrData: generateQRData(),
          },
        ],
      })}
    />
  );
}

SafetyNumberE164Transition.story = {
  name: 'Safety Number (e164 transition)',
};

export function SafetyNumberNotVerified(): JSX.Element {
  return (
    <SafetyNumberViewer
      {...createProps({
        contact: {
          ...contactWithAllData,
          isVerified: false,
        },
      })}
    />
  );
}

SafetyNumberNotVerified.story = {
  name: 'Safety Number (not verified)',
};

export function VerificationDisabled(): JSX.Element {
  return (
    <SafetyNumberViewer
      {...createProps({
        verificationDisabled: true,
      })}
    />
  );
}

export function SafetyNumberDialogClose(): JSX.Element {
  return (
    <SafetyNumberViewer
      {...createProps({
        onClose: action('close'),
      })}
    />
  );
}

SafetyNumberDialogClose.story = {
  name: 'Safety Number (dialog close)',
};

export function JustProfileAndNumber(): JSX.Element {
  return (
    <SafetyNumberViewer
      {...createProps({
        contact: contactWithJustProfile,
      })}
    />
  );
}

JustProfileAndNumber.story = {
  name: 'Just Profile and Number',
};

export function JustNumber(): JSX.Element {
  return (
    <SafetyNumberViewer
      {...createProps({
        contact: contactWithJustNumber,
      })}
    />
  );
}

export function NoPhoneNumberCannotVerify(): JSX.Element {
  return (
    <SafetyNumberViewer
      {...createProps({
        contact: contactWithNothing,
      })}
    />
  );
}

NoPhoneNumberCannotVerify.story = {
  name: 'No Phone Number (cannot verify)',
};