/**
 * Copyright (c) Microsoft Corporation.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import * as React from 'react';
import './attachmentsTab.css';
import { ImageDiffView } from '@web/components/imageDiffView';
import type { TestAttachment } from '@web/components/imageDiffView';
import type { ActionTraceEventInContext, MultiTraceModel } from './modelUtil';

export const AttachmentsTab: React.FunctionComponent<{
  model: MultiTraceModel | undefined,
}> = ({ model }) => {
  if (!model)
    return null;
  return <div className='attachments-tab'>
    { model.actions.map((action, index) => <AttachmentsSection key={index} action={action} />) }
  </div>;
};

export const AttachmentsSection: React.FunctionComponent<{
  action: ActionTraceEventInContext | undefined,
}> = ({ action }) => {
  if (!action)
    return null;
  const expected = action.attachments?.find(a => a.name.endsWith('-expected.png') && (a.path || a.sha1)) as TestAttachment | undefined;
  const actual = action.attachments?.find(a => a.name.endsWith('-actual.png') && (a.path || a.sha1)) as TestAttachment | undefined;
  const diff = action.attachments?.find(a => a.name.endsWith('-diff.png') && (a.path || a.sha1)) as TestAttachment | undefined;
  const screenshots = new Set(action.attachments?.filter(a => a.contentType.startsWith('image/')));
  const otherAttachments = new Set(action.attachments || []);
  screenshots.forEach(a => otherAttachments.delete(a));

  const traceUrl = action.context.traceUrl;

  return <>
    {expected && actual && <div className='attachments-section'>Image diff</div>}
    {expected && actual && <ImageDiffView imageDiff={{
      name: 'Image diff',
      expected: { attachment: { ...expected, path: attachmentURL(traceUrl, expected) }, title: 'Expected' },
      actual: { attachment: { ...actual, path: attachmentURL(traceUrl, actual) } },
      diff: diff ? { attachment: { ...diff, path: attachmentURL(traceUrl, diff) } } : undefined,
    }} />}
    {screenshots.size ? <div className='attachments-section'>Screenshots</div> : undefined}
    {[...screenshots].map((a, i) => {
      return <div className='attachment-item' key={`screenshot-${i}`}>
        <div><img draggable='false' src={attachmentURL(traceUrl, a)} /></div>
        <div><a target='_blank' href={attachmentURL(traceUrl, a)}>{a.name}</a></div>
      </div>;
    })}
    {otherAttachments.size ? <div className='attachments-section'>Attachments</div> : undefined}
    {[...otherAttachments].map((a, i) => {
      return <div className='attachment-item' key={`attachment-${i}`}>
        <a target='_blank' href={attachmentURL(traceUrl, a)}>{a.name}</a>
      </div>;
    })}
  </>;
};

function attachmentURL(traceUrl: string, attachment: NonNullable<ActionTraceEventInContext['attachments']>[0]) {
  if (attachment.sha1)
    return 'sha1/' + attachment.sha1 + '?trace=' + encodeURIComponent(traceUrl);
  return 'file?path=' + encodeURIComponent(attachment.path!);
}