import type { Attachment } from 'stream-chat';

import {
  generateImageAttachment,
  generateVideoAttachment,
} from '../../../mock-builders/generator/attachment';
import { buildGallery } from '../utils/buildGallery/buildGallery';

describe('buildGallery', () => {
  const defaultSizeConfig = {
    gridHeight: 192,
    gridWidth: 256,
    maxHeight: 192,
    maxWidth: 256,
    minHeight: 120,
    minWidth: 120,
  };

  it('gallery size should not exceed max sizes provided by size config', () => {
    const imageSizeTestCases = [
      { original_height: 1000, original_width: 2000 },
      { original_height: 100, original_width: 200 },
      { original_height: 10, original_width: 20 },
      { original_height: 200, original_width: 300 },
    ];

    imageSizeTestCases.forEach((size) => {
      const attachments: Attachment[] = [];
      for (let numOfImages = 0; numOfImages < 4; numOfImages++) {
        const a1 = generateImageAttachment({
          ...size,
        });

        attachments.push(a1);

        const { height, thumbnailGrid, width } = buildGallery({
          images: attachments,
          sizeConfig: defaultSizeConfig,
        });

        expect(height).toBeLessThanOrEqual(defaultSizeConfig.maxHeight);
        expect(height).toBeGreaterThanOrEqual(defaultSizeConfig.minHeight);

        expect(width).toBeLessThanOrEqual(defaultSizeConfig.maxWidth);
        expect(width).toBeGreaterThanOrEqual(defaultSizeConfig.minWidth);

        expect(thumbnailGrid[0][0].flex).toBe(1);
      }
    });
  });

  it('aspect ratio of gallery container should be same as that of original image', () => {
    const sizeConfig = {
      ...defaultSizeConfig,
      maxHeight: 10000,
      maxWidth: 10000,
      minHeight: 0,
      minWidth: 0,
    };

    const imageSizeTestCases = [
      { original_height: 1000, original_width: 2000 },
      { original_height: 100, original_width: 200 },
      { original_height: 10, original_width: 20 },
      { original_height: 200, original_width: 300 },
    ];

    imageSizeTestCases.forEach((size) => {
      const a1 = generateImageAttachment({
        ...size,
      });

      const { height, width } = buildGallery({
        images: [a1],
        sizeConfig,
      });

      const originalAspectRatio = size.original_width / size.original_height;
      const resultingAspectRatio = width / height;
      expect(originalAspectRatio).toBe(resultingAspectRatio);
    });
  });

  it('gallery size should default to gridHeight and gridWidth if original image size is unavailable', () => {
    const attachments: Attachment[] = [];
    for (let numOfImages = 0; numOfImages < 4; numOfImages++) {
      // During each iteration, size of attachments goes up.
      attachments.push(generateImageAttachment());
      const { height, width } = buildGallery({
        images: attachments,
        sizeConfig: defaultSizeConfig,
      });

      expect(height).toBe(defaultSizeConfig.gridHeight);
      expect(width).toBe(defaultSizeConfig.gridWidth);
    }
  });

  it('thumbnail size should be smaller than the limits set by sizeConfig', () => {
    const bigImage = generateImageAttachment({
      image_url:
        'https://us-east.stream-io-cdn.com/23kn4k2j3n4k2n3k4n23?sig=34k23n4k23nk423&oh=200&ow=300&h=*&w=*&resize=*',
      original_height: 1200,
      original_width: 900,
    });

    const { thumbnailGrid: tg1 } = buildGallery({
      images: [bigImage],
      sizeConfig: defaultSizeConfig,
    });
    const t1 = tg1[0][0];
    expect(t1.url.includes('&resize=clip')).toBe(true);
    expect(t1.flex).toBe(1);

    const smallImage = generateImageAttachment({
      image_url:
        'https://us-east.stream-io-cdn.com/23kn4k2j3n4k2n3k4n23?sig=34k23n4k23nk423&h=*&w=*&resize=*&oh=200&ow=300',
      original_height: 30,
      original_width: 20,
    });

    const { thumbnailGrid: tg2 } = buildGallery({
      images: [smallImage],
      sizeConfig: defaultSizeConfig,
    });

    const t2 = tg2[0][0];
    expect(t2.url.includes('&h=*')).toBe(true);
    expect(t2.url.includes('&w=*')).toBe(true);
    expect(t2.url.includes('&resize=*')).toBe(true);
  });

  it('resizes the thumb_url of a video attachment', () => {
    const video = generateVideoAttachment({
      image_url:
        'https://us-east.stream-io-cdn.com/23kn4k2j3n4k2n3k4n23?sig=34k23n4k23nk423&oh=2532&ow=1170&s=v',
      thumb_url:
        'https://us-east.stream-io-cdn.com/23kn4k2j3n4k2n3k4n23?sig=34k23n4k23nk423&oh=2532&ow=1170&s=v',
    });

    const { thumbnailGrid } = buildGallery({
      images: [video],
      sizeConfig: defaultSizeConfig,
    });

    const thumbnail = thumbnailGrid[0][0];
    expect(thumbnail.thumb_url?.includes('&h=')).toBe(true);
    expect(thumbnail.thumb_url?.includes('&w=')).toBe(true);
    expect(thumbnail.thumb_url?.includes('&resize=clip')).toBe(true);
  });

  it('leaves a non-resizable thumb_url untouched', () => {
    const thumbUrl = 'https://not-a-stream-cdn.example.com/thumb.jpg';
    const video = generateVideoAttachment({ thumb_url: thumbUrl });

    const { thumbnailGrid } = buildGallery({
      images: [video],
      sizeConfig: defaultSizeConfig,
    });

    expect(thumbnailGrid[0][0].thumb_url).toBe(thumbUrl);
  });
});