/*
 * This file is part of the Meeds project (https://meeds.io/).
 * Copyright (C) 2022 Meeds Association contact@meeds.io
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */
package org.exoplatform.services.thumbnail;

import org.junit.Before;
import org.junit.Test;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.nio.file.Files;

import static org.junit.Assert.*;

public class ImageResizeServiceImplTest {

  private ImageResizeService imageResizeService;

  private byte[]             testImageContent;
  private byte[]             testVerticalImageContent;
  @Before
  public void setUp() throws Exception {
    this.imageResizeService = new ImageResizeServiceImpl();
    File file = new File(getClass().getClassLoader().getResource("images/testresize.jpg").getFile());
    testImageContent = Files.readAllBytes(file.toPath());

    File fileVertical = new File(getClass().getClassLoader().getResource("images/testVerticalresize.jpg").getFile());
    testVerticalImageContent = Files.readAllBytes(fileVertical.toPath());
  }

  @Test
  public void scaleHorizontaleImage() throws Exception {
    byte[] resizedImage = imageResizeService.scaleImage(testImageContent, 0, 0, false, false);
    BufferedImage bufferedImage = toBufferedImage(resizedImage);
    assertTrue(resizedImage.length > 0);
    assertEquals(427, bufferedImage.getHeight());
    assertEquals(640, bufferedImage.getWidth());

    resizedImage = imageResizeService.scaleImage(testImageContent, 15, 15, false, false);
    assertTrue(resizedImage.length > 0);
    bufferedImage = toBufferedImage(resizedImage);
    assertEquals(15, bufferedImage.getHeight());
    assertNotEquals(15, bufferedImage.getWidth());
    assertTrue(bufferedImage.getWidth() > bufferedImage.getHeight());


    resizedImage = imageResizeService.scaleImage(testImageContent, 15, 15, true, true);
    assertTrue(resizedImage.length > 0);
    bufferedImage = toBufferedImage(resizedImage);
    assertEquals(15, bufferedImage.getHeight());
    assertEquals(15, bufferedImage.getWidth());

    resizedImage = imageResizeService.scaleImage(testImageContent, 0, 15, true, false);
    assertTrue(resizedImage.length > 0);
    bufferedImage = toBufferedImage(resizedImage);
    assertEquals(15, bufferedImage.getHeight());
    assertNotEquals(15, bufferedImage.getWidth());


    resizedImage = imageResizeService.scaleImage(testImageContent, 15, 0, true, false);
    assertTrue(resizedImage.length > 0);
    bufferedImage = toBufferedImage(resizedImage);
    assertEquals(15, bufferedImage.getWidth());
    assertNotEquals(15, bufferedImage.getHeight());

    resizedImage = imageResizeService.scaleImage(testImageContent, 15, 0, true, false);
    assertTrue(resizedImage.length > 0);
    bufferedImage = toBufferedImage(resizedImage);
    assertEquals(15, bufferedImage.getWidth());
    assertNotEquals(15, bufferedImage.getHeight());
  }

  @Test
  public void scaleLandscapeToLandscapeSizeImage() throws Exception {

    //original image is landscape
    //target size is landscape
    //We must fit in width
    byte[] resizedImage = imageResizeService.scaleImage(testImageContent, 150, 80, false, true);
    assertTrue(resizedImage.length > 0);
    BufferedImage bufferedImage = toBufferedImage(resizedImage);
    assertNotEquals(80, bufferedImage.getHeight());
    assertEquals(150, bufferedImage.getWidth());

  }

  @Test
  public void scaleLandscapeToPortraitSizeImage() throws Exception {

    //original image is landscape
    //target size is portrait
    //We must fit in height
    byte[] resizedImage = imageResizeService.scaleImage(testImageContent, 10, 50, false, true);
    assertTrue(resizedImage.length > 0);
    BufferedImage bufferedImage = toBufferedImage(resizedImage);
    assertEquals(50, bufferedImage.getHeight());
    assertNotEquals(10, bufferedImage.getWidth());

  }

  @Test
  public void scalePortraitToLandscapeSizeImage() throws Exception {

    //original image is portrait
    //target size is landscape
    //We must fit in width
    byte[] resizedImage = imageResizeService.scaleImage(testVerticalImageContent, 150, 80, false, true);
    assertTrue(resizedImage.length > 0);
    BufferedImage bufferedImage = toBufferedImage(resizedImage);
    assertNotEquals(80, bufferedImage.getHeight());
    assertEquals(150, bufferedImage.getWidth());

  }

  @Test
  public void scalePortraitToPortraitSizeImage() throws Exception {

    //original image is portrait
    //target size is portrait
    //We must fit in height
    byte[] resizedImage = imageResizeService.scaleImage(testVerticalImageContent, 10, 50, false, true);
    assertTrue(resizedImage.length > 0);
    BufferedImage bufferedImage = toBufferedImage(resizedImage);
    assertEquals(50, bufferedImage.getHeight());
    assertNotEquals(10, bufferedImage.getWidth());

  }

  @Test
  public void testShouldNotIncreaseSizeImage() throws Exception {


    byte[] resizedImage = imageResizeService.scaleImage(testImageContent, 2000, 1000, false, true);
    assertTrue(resizedImage.length > 0);
    BufferedImage bufferedImage = toBufferedImage(resizedImage);
    assertEquals(resizedImage.length, testImageContent.length);
    assertEquals(427, bufferedImage.getHeight());
    assertEquals(640, bufferedImage.getWidth());

  }
  private BufferedImage toBufferedImage(byte[] content) throws IOException {
    ByteArrayInputStream bis = new ByteArrayInputStream(content);
    return ImageIO.read(bis);
  }

  @Test
  public void testShouldReturnOriginalImageIfImageTypeNotSupported() {
    try {
      File fileWebPImage = new File(getClass().getClassLoader().getResource("images/meeds.webp").getFile());
      byte[] testFileWebPImage = Files.readAllBytes(fileWebPImage.toPath());
      byte[] resizedImage = imageResizeService.scaleImage(testFileWebPImage, 2000, 1000, false, true);
      assertEquals(resizedImage.length, testFileWebPImage.length);
    }catch (Exception e) {
      fail();
    }
  }

  @Test
  public void testShouldReturnOriginalImageIfImageHaveError() {
    try {
      File fileImage = new File(getClass().getClassLoader().getResource("images/image-with-error.png").getFile());
      byte[] testFileImage = Files.readAllBytes(fileImage.toPath());
      byte[] resizedImage = imageResizeService.scaleImage(testFileImage, 2000, 1000, false, true);
      assertEquals(resizedImage.length, testFileImage.length);
    }catch (Exception e) {
      fail();
    }
  }

  @Test
  public void testShouldFitHeight() {
    try {
      File fileImage = new File(getClass().getClassLoader().getResource("images/large-banner.png").getFile());
      byte[] testFileImage = Files.readAllBytes(fileImage.toPath());
      byte[] resizedImage = imageResizeService.scaleImage(testFileImage, 250, 150, false, true);
      BufferedImage bufferedImage = toBufferedImage(resizedImage);
      assertEquals(150, bufferedImage.getHeight());
      assertTrue(bufferedImage.getWidth() > 250);
    }catch (Exception e) {
      fail();
    }
  }
}
