feat: add methods to detect default streams and attachment pictures

lib/ffmpeg/media.rb

@@ -195,6 +195,10 @@ def audio_only?
       audio? && !video?
     end
 
+    def audio_with_attached_pic?
+      audio? && streams.any?(&:attached_pic?)
+    end
+
     def silent?
       !audio?
     end

lib/ffmpeg/stream.rb

@@ -101,6 +101,14 @@ def audio?
       codec_type == CodecType::AUDIO
     end
 
+    def default?
+      metadata.dig(:disposition, :default) == 1
+    end
+
+    def attached_pic?
+      metadata.dig(:disposition, :attached_pic) == 1
+    end
+
     def width
       @rotation.nil? || @rotation == 180 ? @width : @height
     end

spec/ffmpeg/media_spec.rb

@@ -247,6 +247,7 @@ module FFMPEG
       end
 
       it 'should return false if the media has video streams' do
+        subject = described_class.new("#{fixture_path}/sounds/napoleon.mp3")
         expect(subject.audio_only?).to be(false)
       end
 
@@ -256,6 +257,22 @@ module FFMPEG
       end
     end
 
+    describe '#audio_with_attached_pic?' do
+      it 'should return true if the media has audio streams with attached pictures' do
+        subject = described_class.new("#{fixture_path}/sounds/napoleon.mp3")
+        expect(subject.audio_with_attached_pic?).to be(true)
+      end
+
+      it 'should return false if the media does not have audio streams with attached pictures' do
+        expect(subject.audio_with_attached_pic?).to be(false)
+      end
+
+      it 'should return false if the media does not have attached pictures' do
+        subject = described_class.new("#{fixture_path}/sounds/hello.wav")
+        expect(subject.audio_with_attached_pic?).to be(false)
+      end
+    end
+
     %i[
       index
       profile

spec/ffmpeg/stream_spec.rb

@@ -62,6 +62,42 @@ module FFMPEG
       end
     end
 
+    describe '#default?' do
+      context 'when marked as default' do
+        let(:metadata) { { disposition: { default: 1 } } }
+
+        it 'should return true' do
+          expect(subject.default?).to be(true)
+        end
+      end
+
+      context 'when not marked as default' do
+        let(:metadata) { { disposition: { default: 0 } } }
+
+        it 'should return false' do
+          expect(subject.default?).to be(false)
+        end
+      end
+    end
+
+    describe '#attached_pic?' do
+      context 'when marked as an attached picture' do
+        let(:metadata) { { disposition: { attached_pic: 1 } } }
+
+        it 'should return true' do
+          expect(subject.attached_pic?).to be(true)
+        end
+      end
+
+      context 'when not marked as an attached picture' do
+        let(:metadata) { { disposition: { attached_pic: 0 } } }
+
+        it 'should return false' do
+          expect(subject.attached_pic?).to be(false)
+        end
+      end
+    end
+
     describe '#width' do
       context 'when the rotation is nil' do
         let(:metadata) { { width: 100, height: 200 } }