# frozen_string_literal: true

require 'spec_helper'

# rubocop:disable Metrics/BlockLength
describe Assembly::Image do
  it 'does not run if no input file is passed in' do
    @ai = Assembly::Image.new('')
    expect{ @ai.create_jp2 }.to raise_error
  end

  it 'indicates the default jp2 filename' do
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    expect(@ai.jp2_filename).to eq TEST_TIF_INPUT_FILE.gsub('.tif', '.jp2')
  end

  it 'indicates the default jp2 filename' do
    @ai = Assembly::Image.new('/path/to/a/file_with_no_extension')
    expect(@ai.jp2_filename).to eq '/path/to/a/file_with_no_extension.jp2'
  end

  it 'indicates the default DPG jp2 filename' do
    @ai = Assembly::Image.new(TEST_DPG_TIF_INPUT_FILE)
    expect(@ai.dpg_jp2_filename).to eq TEST_DPG_TIF_INPUT_FILE.gsub('.tif', '.jp2').gsub('_00_', '_05_')
  end

  it 'indicates the default jp2 filename' do
    @ai = Assembly::Image.new('/path/to/a/file_with_no_00_extension')
    expect(@ai.dpg_jp2_filename).to eq '/path/to/a/file_with_no_05_extension.jp2'
  end

  it 'creates the jp2 without a temp file when given an uncompressed RGB tif' do
    generate_test_image(TEST_TIF_INPUT_FILE)
    expect(File).to exist TEST_TIF_INPUT_FILE
    expect(File).to_not exist TEST_JP2_OUTPUT_FILE
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    result = @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE)
    # Indicates a temp tiff was not created.
    expect(@ai.tmp_path).to be_nil
    expect(result).to be_a_kind_of Assembly::Image
    expect(result.path).to eq TEST_JP2_OUTPUT_FILE
    expect(TEST_JP2_OUTPUT_FILE).to be_a_jp2
    expect(result.exif.colorspace).to eq 'sRGB'
    @jp2 = Assembly::Image.new(TEST_JP2_OUTPUT_FILE)
    expect(@jp2.height).to eq 100
    expect(@jp2.width).to eq 100
  end

  it 'creates the jp2 with a temp file when given a compressed RGB tif' do
    generate_test_image(TEST_TIF_INPUT_FILE, compress: true)
    expect(File).to exist TEST_TIF_INPUT_FILE
    expect(File).to_not exist TEST_JP2_OUTPUT_FILE
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    result = @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE)
    # Indicates a temp tiff was not created.
    expect(@ai.tmp_path).to_not be_nil
    expect(result).to be_a_kind_of Assembly::Image
    expect(result.path).to eq TEST_JP2_OUTPUT_FILE
    expect(TEST_JP2_OUTPUT_FILE).to be_a_jp2
    expect(result.exif.colorspace).to eq 'sRGB'
    @jp2 = Assembly::Image.new(TEST_JP2_OUTPUT_FILE)
    expect(@jp2.height).to eq 100
    expect(@jp2.width).to eq 100
  end

  it 'creates grayscale jp2 when given a bitonal tif' do
    skip 'The latest version of Kakadu may require some changes for this work correctly'
    # error message is
    #  JP2 creation command failed: kdu_compress   -precise -no_weights -quiet Creversible=no Cmodes=BYPASS
    # Corder=RPCL Cblk=\{64,64\} Cprecincts=\{256,256\},\{256,256\},\{128,128\} ORGgen_plt=yes -rate 1.5 Clevels=5
    # Clayers=2 -i '/tmp/408d3740-e25f-4c1b-889f-3f138d088fe4.tif' -o '/home/travis/build/sul-dlss/assembly-image/spec/test_data/output/test.jp2'
    # with result Kakadu Error:
    #  The number of colours associated with the colour space identified by the source
    #  file (possible from an embedded ICC profile) is not consistent with the number
    #  of supplied image components and/or colour palette.  You can address this
    #  problem by supplying a `-jp2_space' or `-jpx_space' argument to explicitly
    #  identify a colour space that has anywhere from 1 to 1 colour components.
    generate_test_image(TEST_TIF_INPUT_FILE, image_type: 'Bilevel')
    expect(File).to exist TEST_TIF_INPUT_FILE
    expect(File).to_not exist TEST_JP2_OUTPUT_FILE
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    expect(@ai).to have_color_profile
    result = @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE)
    # Indicates a temp tiff was not created.
    expect(@ai.tmp_path).to be_nil
    expect(TEST_JP2_OUTPUT_FILE).to be_a_jp2
    expect(result.exif.colorspace).to eq 'Grayscale'
  end

  it 'creates color jp2 when given a color tif but bitonal image data (1 channels and 1 bits per pixel)' do
    generate_test_image(TEST_TIF_INPUT_FILE, color: 'Bilevel', image_type: 'TrueColor', profile: '')
    expect(File).to exist TEST_TIF_INPUT_FILE
    expect(File).to_not exist TEST_JP2_OUTPUT_FILE
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    expect(@ai).to_not have_color_profile
    result = @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE)
    # Indicates a temp tiff was not created.
    expect(@ai.tmp_path).to be_nil
    expect(TEST_JP2_OUTPUT_FILE).to be_a_jp2
    expect(result.exif.colorspace).to eq 'sRGB'
  end

  it 'creates grayscale jp2 when given a graycale tif but with bitonal image data (1 channel and 1 bits per pixel)' do
    generate_test_image(TEST_TIF_INPUT_FILE, color: 'Bilevel', image_type: 'Grayscale', profile: '')
    expect(File).to exist TEST_TIF_INPUT_FILE
    expect(File).to_not exist TEST_JP2_OUTPUT_FILE
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    # Indicates a temp tiff was not created.
    expect(@ai.tmp_path).to be_nil
    expect(@ai).to_not have_color_profile
    result = @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE)
    expect(TEST_JP2_OUTPUT_FILE).to be_a_jp2
    expect(result.exif.colorspace).to eq 'Grayscale'
  end

  it 'creates color jp2 when given a color tif but with greyscale image data (1 channel and 8 bits per pixel)' do
    generate_test_image(TEST_TIF_INPUT_FILE, color: 'Grayscale', image_type: 'TrueColor', profile: '')
    expect(File).to exist TEST_TIF_INPUT_FILE
    expect(File).to_not exist TEST_JP2_OUTPUT_FILE
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    # Indicates a temp tiff was not created.
    expect(@ai.tmp_path).to be_nil
    expect(@ai).to_not have_color_profile
    result = @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE)
    expect(TEST_JP2_OUTPUT_FILE).to be_a_jp2
    expect(result.exif.colorspace).to eq 'sRGB'
  end

  it 'creates a jp2 when the source image has no profile' do
    generate_test_image(TEST_TIF_INPUT_FILE, profile: '') # generate a test input with no profile
    expect(File).to exist TEST_TIF_INPUT_FILE
    expect(File).to_not exist TEST_JP2_OUTPUT_FILE
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    # Indicates a temp tiff was not created.
    expect(@ai.tmp_path).to be_nil
    expect(@ai).to_not have_color_profile
    expect(@ai).to be_a_valid_image
    expect(@ai).to be_jp2able
    @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE)
    expect(TEST_JP2_OUTPUT_FILE).to be_a_jp2
  end

  it "does not run if the output file exists and you don't allow overwriting" do
    generate_test_image(TEST_TIF_INPUT_FILE)
    generate_test_image(TEST_JP2_OUTPUT_FILE)
    expect(File).to exist TEST_TIF_INPUT_FILE
    expect(File).to exist TEST_JP2_OUTPUT_FILE
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    expect{ @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE) }.to raise_error(SecurityError)
  end

  it 'gets the correct image height and width' do
    generate_test_image(TEST_TIF_INPUT_FILE)
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    expect(@ai.height).to eq 100
    expect(@ai.width).to eq 100
  end

  it 'does not run if the input file is a jp2' do
    generate_test_image(TEST_TIF_INPUT_FILE, profile: '') # generate a test input with no profile
    expect(File).to exist TEST_TIF_INPUT_FILE
    expect(File).to_not exist TEST_JP2_OUTPUT_FILE
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    # Indicates a temp tiff was not created.
    expect(@ai.tmp_path).to be_nil
    expect(@ai).to_not have_color_profile
    expect(@ai).to be_a_valid_image
    expect(@ai).to be_jp2able
    @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE)
    expect(TEST_JP2_OUTPUT_FILE).to be_a_jp2
    @ai = Assembly::Image.new(TEST_JP2_OUTPUT_FILE)
    expect(@ai).to be_valid_image
    expect(@ai).to_not be_jp2able
    expect { @ai.create_jp2 }.to raise_error
  end

  it 'runs if you specify a bogus output profile, because this is not currently an option' do
    generate_test_image(TEST_TIF_INPUT_FILE)
    expect(File).to exist TEST_TIF_INPUT_FILE
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    result = @ai.create_jp2(output_profile: 'bogusness')
    expect(result).to be_a_kind_of Assembly::Image
    expect(result.path).to eq TEST_JP2_INPUT_FILE
    expect(TEST_JP2_INPUT_FILE).to be_a_jp2
    expect(result.exif.colorspace).to eq 'sRGB'
  end

  it 'creates jp2 when given a JPEG' do
    generate_test_image(TEST_JPEG_INPUT_FILE)
    expect(File).to exist TEST_JPEG_INPUT_FILE
    expect(File).to_not exist TEST_JP2_OUTPUT_FILE
    @ai = Assembly::Image.new(TEST_JPEG_INPUT_FILE)
    result = @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE)
    # Indicates a temp tiff was created.
    expect(@ai.tmp_path).not_to be_nil
    expect(File).not_to exist @ai.tmp_path
    expect(result).to be_a_kind_of Assembly::Image
    expect(result.path).to eq TEST_JP2_OUTPUT_FILE
    expect(TEST_JP2_OUTPUT_FILE).to be_a_jp2
  end

  it 'does not run if you specify a bogus tmp folder' do
    generate_test_image(TEST_JPEG_INPUT_FILE)
    bogus_folder = '/crapsticks'
    expect(File).to exist TEST_JPEG_INPUT_FILE
    expect(File).to_not exist bogus_folder
    @ai = Assembly::Image.new(TEST_JPEG_INPUT_FILE)
    expect { @ai.create_jp2(tmp_folder: bogus_folder) }.to raise_error
  end

  it 'creates a jp2 and preserve the temporary file if specified' do
    generate_test_image(TEST_JPEG_INPUT_FILE)
    expect(File).to exist TEST_JPEG_INPUT_FILE
    @ai = Assembly::Image.new(TEST_JPEG_INPUT_FILE)
    result = @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE, preserve_tmp_source: true)
    # Indicates a temp tiff was created.
    expect(@ai.tmp_path).not_to be_nil
    expect(File).to exist @ai.tmp_path
    expect(result).to be_a_kind_of Assembly::Image
    expect(result.path).to eq TEST_JP2_OUTPUT_FILE
    expect(TEST_JP2_OUTPUT_FILE).to be_a_jp2
    expect(File.exist?(@ai.tmp_path)).to be true
  end

  it 'creates a jp2 of the same filename and in the same location as the input if no output file is specified, and should cleanup tmp file' do
    generate_test_image(TEST_TIF_INPUT_FILE)
    expect(File).to exist TEST_TIF_INPUT_FILE
    expect(File.exist?(TEST_JP2_INPUT_FILE)).to be false
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    result = @ai.create_jp2
    expect(result).to be_a_kind_of Assembly::Image
    expect(result.path).to eq TEST_JP2_INPUT_FILE
    expect(TEST_JP2_INPUT_FILE).to be_a_jp2
    expect(result.exif.colorspace).to eq 'sRGB'
    expect(@ai.tmp_path).to be_nil
  end

  it 'recreates jp2 if the output file exists and if you allow overwriting' do
    generate_test_image(TEST_TIF_INPUT_FILE)
    generate_test_image(TEST_JP2_OUTPUT_FILE)
    expect(File).to exist TEST_TIF_INPUT_FILE
    expect(File).to exist TEST_JP2_OUTPUT_FILE
    @ai = Assembly::Image.new(TEST_TIF_INPUT_FILE)
    result = @ai.create_jp2(output: TEST_JP2_OUTPUT_FILE, overwrite: true)
    expect(result).to be_a_kind_of Assembly::Image
    expect(result.path).to eq TEST_JP2_OUTPUT_FILE
    expect(TEST_JP2_OUTPUT_FILE).to be_a_jp2
    expect(result.exif.colorspace).to eq 'sRGB'
  end

  after(:each) do
    # after each test, empty out the input and output test directories
    remove_files(TEST_INPUT_DIR)
    remove_files(TEST_OUTPUT_DIR)
  end
end
# rubocop:enable Metrics/BlockLength