require 'spec_helper'

describe M3u8::Playlist do
  it 'should generate codecs string' do
    codecs = M3u8::Playlist.codecs
    expect(codecs).to be_nil

    codecs = M3u8::Playlist.codecs audio: 'aac-lc'
    expect(codecs).to eq 'mp4a.40.2'

    codecs = M3u8::Playlist.codecs audio: 'AAC-LC'
    expect(codecs).to eq 'mp4a.40.2'

    codecs = M3u8::Playlist.codecs audio: 'he-aac'
    expect(codecs).to eq 'mp4a.40.5'

    codecs = M3u8::Playlist.codecs audio: 'HE-AAC'
    expect(codecs).to eq 'mp4a.40.5'

    codecs = M3u8::Playlist.codecs audio: 'he-acc1'
    expect(codecs).to be_nil

    codecs = M3u8::Playlist.codecs audio: 'mp3'
    expect(codecs).to eq 'mp4a.40.34'

    codecs = M3u8::Playlist.codecs audio: 'MP3'
    expect(codecs).to eq 'mp4a.40.34'

    options = { profile: 'baseline', level: 3.0 }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.66.30'

    options = { profile: 'baseline', level: 3.0, audio: 'aac-lc' }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.66.30,mp4a.40.2'

    options = { profile: 'baseline', level: 3.0, audio: 'mp3' }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.66.30,mp4a.40.34'

    options = { profile: 'baseline', level: 3.1 }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.42001f'

    options = { profile: 'baseline', level: 3.1, audio: 'he-aac' }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.42001f,mp4a.40.5'

    options = { profile: 'main', level: 3.0 }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.77.30'

    options = { profile: 'main', level: 3.0, audio: 'aac-lc' }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.77.30,mp4a.40.2'

    options = { profile: 'main', level: 3.1 }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.4d001f'

    options = { profile: 'main', level: 4.0 }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.4d0028'

    options = { profile: 'high', level: 3.1 }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.64001f'

    options = { profile: 'high', level: 4.0 }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.640028'

    options = { profile: 'high', level: 4.1 }
    codecs = M3u8::Playlist.codecs options
    expect(codecs).to eq 'avc1.640028'
  end

  it 'should render master playlist' do
    playlist = M3u8::Playlist.new
    playlist.add_playlist '1', 'playlist_url', 6400, audio: 'mp3'

    output = "#EXTM3U\n" +
             %(#EXT-X-STREAM-INF:PROGRAM-ID=1,CODECS="mp4a.40.34") +
             ",BANDWIDTH=6400\nplaylist_url\n"

    expect(playlist.to_s).to eq output

    playlist = M3u8::Playlist.new
    options = { width: 1920, height: 1080, profile: 'high', level: 4.1,
                audio: 'aac-lc' }
    playlist.add_playlist '2', 'playlist_url', 50_000, options

    output = "#EXTM3U\n" \
             '#EXT-X-STREAM-INF:PROGRAM-ID=2,RESOLUTION=1920x1080,' +
             %(CODECS="avc1.640028,mp4a.40.2",BANDWIDTH=50000\n) +
             "playlist_url\n"

    expect(playlist.to_s).to eq output

    playlist = M3u8::Playlist.new
    playlist.add_playlist '1', 'playlist_url', 6400, audio: 'mp3'
    options = { width: 1920, height: 1080, profile: 'high', level: 4.1,
                audio: 'aac-lc' }
    playlist.add_playlist '2', 'playlist_url', 50_000, options

    output = "#EXTM3U\n" +
             %(#EXT-X-STREAM-INF:PROGRAM-ID=1,CODECS="mp4a.40.34") +
             ",BANDWIDTH=6400\nplaylist_url\n#EXT-X-STREAM-INF:PROGRAM-ID=2," +
             %(RESOLUTION=1920x1080,CODECS="avc1.640028,mp4a.40.2") +
             ",BANDWIDTH=50000\nplaylist_url\n"

    expect(playlist.to_s).to eq output
  end

  it 'should render playlist' do
    playlist = M3u8::Playlist.new
    playlist.add_segment 11.344644, '1080-7mbps00000.ts'

    output = "#EXTM3U\n" \
      "#EXT-X-VERSION:3\n" \
      "#EXT-X-MEDIA-SEQUENCE:0\n" \
      "#EXT-X-ALLOW-CACHE:YES\n" \
      "#EXT-X-TARGETDURATION:10\n" \
      "#EXTINF:11.344644,\n" \
      "1080-7mbps00000.ts\n" \
      "#EXT-X-ENDLIST\n"

    expect(playlist.to_s).to eq output

    playlist.add_segment 11.261233, '1080-7mbps00001.ts'

    output = "#EXTM3U\n" \
      "#EXT-X-VERSION:3\n" \
      "#EXT-X-MEDIA-SEQUENCE:0\n" \
      "#EXT-X-ALLOW-CACHE:YES\n" \
      "#EXT-X-TARGETDURATION:10\n" \
      "#EXTINF:11.344644,\n" \
      "1080-7mbps00000.ts\n" \
      "#EXTINF:11.261233,\n" \
      "1080-7mbps00001.ts\n" \
      "#EXT-X-ENDLIST\n"

    expect(playlist.to_s).to eq output

    options = { version: 1, cache: false, target: 12, sequence: 1 }
    playlist = M3u8::Playlist.new options
    playlist.add_segment 11.344644, '1080-7mbps00000.ts'

    output = "#EXTM3U\n" \
      "#EXT-X-VERSION:1\n" \
      "#EXT-X-MEDIA-SEQUENCE:1\n" \
      "#EXT-X-ALLOW-CACHE:NO\n" \
      "#EXT-X-TARGETDURATION:12\n" \
      "#EXTINF:11.344644,\n" \
      "1080-7mbps00000.ts\n" \
      "#EXT-X-ENDLIST\n"

    expect(playlist.to_s).to eq output
  end

  it 'should write playlist to io' do
    test_io = StringIO.new
    playlist = M3u8::Playlist.new
    playlist.add_playlist '1', 'playlist_url', 6400, audio: 'mp3'
    playlist.write test_io

    output = "#EXTM3U\n" +
             %(#EXT-X-STREAM-INF:PROGRAM-ID=1,CODECS="mp4a.40.34",) +
             "BANDWIDTH=6400\nplaylist_url\n"

    expect(test_io.string).to eq output

    test_io = StringIO.new
    playlist.write test_io

    output = "#EXTM3U\n" +
             %(#EXT-X-STREAM-INF:PROGRAM-ID=1,CODECS="mp4a.40.34",) +
             "BANDWIDTH=6400\nplaylist_url\n"

    expect(test_io.string).to eq output
  end

  it 'should report if it is a master playlist' do
    playlist = M3u8::Playlist.new
    expect(playlist.master?).to be false

    playlist.add_playlist '1', 'playlist_url', 6400, audio: 'mp3'
    expect(playlist.master?).to be true
  end

  it 'should raise error if type of playlist is changed' do
    playlist = M3u8::Playlist.new
    playlist.add_playlist '1', 'playlist_url', 6400, audio: 'mp3'

    message = 'Playlist is a master playlist, segment can not be added.'
    expect { playlist.add_segment 11.344644, '1080-7mbps00000.ts' }
      .to raise_error(M3u8::PlaylistTypeError, message)

    playlist = M3u8::Playlist.new
    playlist.add_segment 11.344644, '1080-7mbps00000.ts'
    message = 'Playlist is not a master playlist, playlist can not be added.'
    expect { playlist.add_playlist '1', 'playlist_url', 6400 }
      .to raise_error(M3u8::PlaylistTypeError, message)
  end

  it 'should raise error on write if item types are mixed' do
    playlist = M3u8::Playlist.new

    hash = { program_id: 1, width: 1920, height: 1080, codecs: 'avc',
             bitrate: 540, playlist: 'test.url' }
    item = M3u8::PlaylistItem.new(hash)
    playlist.items.push item

    hash = { duration: 10.991, segment: 'test.ts' }
    item = M3u8::SegmentItem.new(hash)
    playlist.items.push item

    message = 'Playlist contains mixed types of items'
    io = StringIO.new
    expect { playlist.write io }
      .to raise_error(M3u8::PlaylistTypeError, message)
  end

  it 'should return valid status' do
    playlist = M3u8::Playlist.new
    expect(playlist.valid?).to be true

    hash = { program_id: 1, width: 1920, height: 1080, codecs: 'avc',
             bitrate: 540, playlist: 'test.url' }
    item = M3u8::PlaylistItem.new(hash)
    playlist.items.push item
    expect(playlist.valid?).to be true

    hash = { program_id: 1, width: 1920, height: 1080, codecs: 'avc',
             bitrate: 540, playlist: 'test.url' }
    item = M3u8::PlaylistItem.new(hash)
    playlist.items.push item
    expect(playlist.valid?).to be true

    hash = { duration: 10.991, segment: 'test.ts' }
    item = M3u8::SegmentItem.new(hash)
    playlist.items.push item

    expect(playlist.valid?).to be false
  end

  it 'should raise error if codecs are missing' do
    playlist = M3u8::Playlist.new
    message = 'An audio or video codec should be provided.'
    expect { playlist.add_playlist '1', 'playlist_url', 6400 }
      .to raise_error(M3u8::MissingCodecError, message)
  end

  it 'should expose options as attributes' do
    options = { version: 1, cache: false, target: 12, sequence: 1 }
    playlist = M3u8::Playlist.new options
    expect(playlist.version).to be 1
    expect(playlist.cache).to be false
    expect(playlist.target).to be 12
    expect(playlist.sequence).to be 1
  end
end