require 'spec_helper'

describe "Transcoder" do
  before do
    class ContentDatastream < ActiveFedora::Datastream
      include Hydra::Derivatives::ExtractMetadata
    end
    class GenericFile < ActiveFedora::Base
      include Hydra::Derivatives
      has_metadata 'characterization', type: ActiveFedora::SimpleDatastream do |m|
          m.field "mime_type", :string
      end

      delegate :mime_type, :to => :characterization, :unique => true
      has_file_datastream 'content', type: ContentDatastream

      makes_derivatives do |obj| 
        case obj.mime_type
        when 'application/pdf'
          obj.transform_datastream :content, { :thumb => "100x100>" }
        when 'audio/wav'
          obj.transform_datastream :content, { :mp3 => {format: 'mp3'}, :ogg => {format: 'ogg'} }, processor: :audio
        when 'video/avi'
          obj.transform_datastream :content, { :mp4 => {format: 'mp4'}, :webm => {format: 'webm'} }, processor: :video
        when 'image/png', 'image/jpg'
          obj.transform_datastream :content, { :medium => "300x300>", :thumb => "100x100>" }
        when 'application/vnd.ms-powerpoint'
          obj.transform_datastream :content, { :access => { :format=>'pdf' } }, processor: 'document'
        when 'text/rtf'
          obj.transform_datastream :content, { :preservation=> {:format => 'odf' }, :access => { :format=>'pdf' } }, processor: 'document'
        end
      end
      
      makes_derivatives :generate_special_derivatives
      
      def generate_special_derivatives
        if label == "special" && mime_type == 'image/png'
          transform_datastream :content, { :medium => {size: "200x300>", datastream: 'special_ds'} }
        end
      end

    end
  end
  
  describe "with an attached image" do
    let(:attachment) { File.open(File.expand_path('../../fixtures/world.png', __FILE__))}
    let(:file) { GenericFile.new(mime_type: 'image/png').tap { |t| t.content.content = attachment; t.save } }

    it "should transcode" do
      file.datastreams.key?('content_medium').should be_false
      file.create_derivatives
      file.datastreams['content_medium'].should have_content
      file.datastreams['content_medium'].mimeType.should == 'image/png'
      file.datastreams['content_thumb'].should have_content 
      file.datastreams['content_thumb'].mimeType.should == 'image/png'
      file.datastreams.key?('content_text').should be_false 
    end
  end

  describe "with an attached pdf" do
    let(:attachment) { File.open(File.expand_path('../../fixtures/test.pdf', __FILE__))}
    let(:file) { GenericFile.new(mime_type: 'application/pdf').tap { |t| t.content.content = attachment; t.save } }

    it "should transcode" do
      file.datastreams.key?('content_thumb').should be_false
      file.create_derivatives
      file.datastreams['content_thumb'].should have_content 
      file.datastreams['content_thumb'].mimeType.should == 'image/png'
    end
  end

  describe "with an attached audio", unless: ENV['TRAVIS'] == 'true' do
    let(:attachment) { File.open(File.expand_path('../../fixtures/piano_note.wav', __FILE__))}
    let(:file) { GenericFile.new(mime_type: 'audio/wav').tap { |t| t.content.content = attachment; t.save } }

    it "should transcode" do
      file.create_derivatives
      file.datastreams['content_mp3'].should have_content
      file.datastreams['content_mp3'].mimeType.should == 'audio/mpeg'
      file.datastreams['content_ogg'].should have_content 
      file.datastreams['content_ogg'].mimeType.should == 'audio/ogg'
    end
  end

  describe "with an attached video", unless: ENV['TRAVIS'] == 'true' do
    let(:attachment) { File.open(File.expand_path('../../fixtures/countdown.avi', __FILE__))}
    let(:file) { GenericFile.new(mime_type: 'video/avi').tap { |t| t.content.content = attachment; t.save } }

    it "should transcode" do
      file.create_derivatives
      file.datastreams['content_mp4'].should have_content
      file.datastreams['content_mp4'].mimeType.should == 'video/mp4'
      file.datastreams['content_webm'].should have_content 
      file.datastreams['content_webm'].mimeType.should == 'video/webm'
    end
  end
  
  describe "using callback methods" do
    let(:attachment) { File.open(File.expand_path('../../fixtures/world.png', __FILE__))}
    let(:file) { GenericFile.new(mime_type: 'image/png', label: "special").tap { |t| t.content.content = attachment; t.save } }

    it "should transcode" do
      file.datastreams.key?('special_ds').should be_false
      file.create_derivatives
      file.datastreams['special_ds'].should have_content      
      file.datastreams['special_ds'].mimeType.should == 'image/png'
      file.datastreams['special_ds'].should have_content 
    end
  end

  describe "with an attached Powerpoint", unless: ENV['TRAVIS'] == 'true' do
    let(:attachment) { File.open(File.expand_path('../../fixtures/FlashPix.ppt', __FILE__))}
    let(:file) { GenericFile.new(mime_type: 'application/vnd.ms-powerpoint').tap { |t| t.content.content = attachment; t.save } }

    it "should transcode" do
      file.create_derivatives
      file.datastreams['content_access'].should have_content
      file.datastreams['content_access'].mimeType.should == 'application/pdf'
    end
  end

  describe "with an attached rich text format", unless: ENV['TRAVIS'] == 'true' do
    let(:attachment) { File.open(File.expand_path('../../fixtures/sample.rtf', __FILE__))}
    let(:file) { GenericFile.new(mime_type: 'text/rtf').tap { |t| t.content.content = attachment; t.save } }

    it "should transcode" do
      file.create_derivatives
      file.datastreams['content_access'].should have_content
      file.datastreams['content_access'].mimeType.should == 'application/pdf'
      file.datastreams['content_preservation'].should have_content
      file.datastreams['content_preservation'].mimeType.should == 'application/vnd.oasis.opendocument.text'
    end
  end
end