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

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

import * as React from 'react';

import { storiesOf } from '@storybook/react';
import { boolean, select, text } from '@storybook/addon-knobs';
import { action } from '@storybook/addon-actions';

import { Avatar, Props } from './Avatar';
import { setup as setupI18n } from '../../js/modules/i18n';
import enMessages from '../../_locales/en/messages.json';
import { Colors, ColorType } from '../types/Colors';

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

const story = storiesOf('Components/Avatar', module);

const colorMap: Record<string, ColorType> = Colors.reduce(
  (m, color) => ({
    ...m,
    [color]: color,
  }),
  {}
);

const conversationTypeMap: Record<string, Props['conversationType']> = {
  direct: 'direct',
  group: 'group',
};

const createProps = (overrideProps: Partial<Props> = {}): Props => ({
  avatarPath: text('avatarPath', overrideProps.avatarPath || ''),
  color: select('color', colorMap, overrideProps.color || 'blue'),
  conversationType: select(
    'conversationType',
    conversationTypeMap,
    overrideProps.conversationType || 'direct'
  ),
  i18n,
  name: text('name', overrideProps.name || ''),
  noteToSelf: boolean('noteToSelf', overrideProps.noteToSelf || false),
  onClick: action('onClick'),
  phoneNumber: text('phoneNumber', overrideProps.phoneNumber || ''),
  size: 80,
  title: '',
});

const sizes: Array<Props['size']> = [112, 80, 52, 32, 28];

story.add('Avatar', () => {
  const props = createProps({
    avatarPath: '/fixtures/giphy-GVNvOUpeYmI7e.gif',
  });

  return sizes.map(size => <Avatar key={size} {...props} size={size} />);
});

story.add('One-word Name', () => {
  const props = createProps({
    name: 'John',
  });

  return sizes.map(size => <Avatar key={size} {...props} size={size} />);
});

story.add('Multi-word Name', () => {
  const props = createProps({
    name: 'John Smith',
  });

  return sizes.map(size => <Avatar key={size} {...props} size={size} />);
});

story.add('Note to Self', () => {
  const props = createProps({
    noteToSelf: true,
  });

  return sizes.map(size => <Avatar key={size} {...props} size={size} />);
});

story.add('Contact Icon', () => {
  const props = createProps();

  return sizes.map(size => <Avatar key={size} {...props} size={size} />);
});

story.add('Group Icon', () => {
  const props = createProps({
    conversationType: 'group',
  });

  return sizes.map(size => <Avatar key={size} {...props} size={size} />);
});

story.add('Colors', () => {
  const props = createProps();

  return Colors.map(color => <Avatar key={color} {...props} color={color} />);
});

story.add('Broken Color', () => {
  const props = createProps({
    color: 'nope' as ColorType,
  });

  return sizes.map(size => <Avatar key={size} {...props} size={size} />);
});

story.add('Broken Avatar', () => {
  const props = createProps({
    avatarPath: 'badimage.png',
  });

  return sizes.map(size => <Avatar key={size} {...props} size={size} />);
});

story.add('Broken Avatar for Group', () => {
  const props = createProps({
    avatarPath: 'badimage.png',
    conversationType: 'group',
  });

  return sizes.map(size => <Avatar key={size} {...props} size={size} />);
});