packages/storybook8/stories/Components/MessageThread/TappingInlineImages/TappingInlineImages.story.tsx

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

import {
  FluentThemeProvider,
  MessageThread,
  Message,
  ImageOverlay,
  ChatMessage,
  InlineImage
} from '@azure/communication-react';
import { Persona, PersonaSize } from '@fluentui/react';
import React, { useState } from 'react';
import { ArgsFrom } from '../../../controlsUtils';

const storyControls = {
  tapMethod: { control: 'select', options: ['overlay', 'alert'] }
};

const TappingInlineImageStory = (args: ArgsFrom<typeof storyControls>): JSX.Element => {
  const [overlayImageItem, setOverlayImageItem] = useState<{
    imageSrc: string;
    title: string;
    titleIcon: JSX.Element;
    downloadImagename: string;
  }>();

  const onInlineImageClicked = (attachmentId: string, messageId: string): Promise<void> => {
    if (args.tapMethod === 'alert') {
      alert(`Image clicked with attachmentId: ${attachmentId}`);
      return Promise.resolve();
    }

    const filteredMessages = messages?.filter((message) => {
      return message.messageId === messageId;
    });
    if (!filteredMessages || filteredMessages.length <= 0) {
      return Promise.reject(`Message not found with messageId ${messageId}`);
    }
    const chatMessage = filteredMessages[0] as ChatMessage;
    const title = 'Image';
    const titleIcon = (
      <Persona text={chatMessage.senderDisplayName} size={PersonaSize.size32} hidePersonaDetails={true} />
    );
    const document = new DOMParser().parseFromString(chatMessage.content ?? '', 'text/html');
    let imgSrc = '';
    document.querySelectorAll('img').forEach((img) => {
      if (img.id === attachmentId) {
        imgSrc = img.src;
      }
    });
    const overlayImage = {
      title,
      titleIcon,
      downloadImagename: attachmentId,
      imageSrc: imgSrc
    };
    setOverlayImageItem(overlayImage);
    return Promise.resolve();
  };

  const inlineImageOptions = {
    onRenderInlineImage: (
      inlineImage: InlineImage,
      defaultOnRender: (inlineImage: InlineImage) => JSX.Element
    ): JSX.Element => {
      return (
        <span
          onClick={() => onInlineImageClicked(inlineImage.imageAttributes.id || '', inlineImage.messageId)}
          tabIndex={0}
          role="button"
          onKeyDown={(e) => {
            if (e.key === 'Enter') {
              onInlineImageClicked(inlineImage.imageAttributes.id || '', inlineImage.messageId);
            }
          }}
        >
          {defaultOnRender(inlineImage)}
        </span>
      );
    }
  };

  const messages: Message[] = [
    {
      messageType: 'chat',
      senderId: 'user3',
      content:
        '<p>How should I design my new house?</p><p><img alt="image" src="images/inlineImageExample1.png" itemscope="png" width="166" height="250" id="SomeImageId1" style="vertical-align:bottom; aspect-ratio: 166 / 250"></p><p><img alt="image" src="images/inlineImageExample2.png" itemscope="png" width="374" height="250" id="SomeImageId2" style="vertical-align:bottom; aspect-ratio: 374 / 250"></p><p>&nbsp;</p>',
      senderDisplayName: 'Miguel Garcia',
      messageId: Math.random().toString(),
      createdOn: new Date('2019-04-13T00:00:00.000+08:09'),
      mine: false,
      attached: false,
      contentType: 'html'
    },
    {
      messageType: 'chat',
      senderId: 'user2',
      senderDisplayName: 'Robert Tolbert',
      messageId: Math.random().toString(),
      content: 'Cool, I love the second one!',
      createdOn: new Date('2019-04-13T00:00:00.000+08:10'),
      mine: true,
      attached: false,
      contentType: 'text'
    }
  ];
  return (
    <FluentThemeProvider>
      <MessageThread userId={'1'} messages={messages} inlineImageOptions={inlineImageOptions} />
      {
        <ImageOverlay
          isOpen={overlayImageItem !== undefined}
          imageSrc={overlayImageItem?.imageSrc || ''}
          title="Image"
          onDismiss={() => {
            setOverlayImageItem(undefined);
          }}
          onDownloadButtonClicked={() => {
            alert('Download button clicked');
          }}
        />
      }
    </FluentThemeProvider>
  );
};

export const MessageThreadTappingInlineImage = TappingInlineImageStory.bind({});