ehbrs-tools 0.3.0

data/lib/ehbrs/videos/file.rb

@@ -0,0 +1,70 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/core_ext'
+require 'ehbrs/videos/track'
+
+module Ehbrs
+  module Videos
+    class File
+      enable_simple_cache
+
+      TIME_PATTERN = /(\d+):(\d{2}):(\d{2})(?:\.(\d+))/.freeze
+
+      class << self
+        def seconds_to_time(seconds)
+          t = seconds.floor
+          hmsf_to_time((t / 3600), ((t / 60) % 60), (t % 60), (seconds - t).round(3))
+        end
+
+        def time_to_seconds(time)
+          m = TIME_PATTERN.match(time)
+          raise "Time pattern not find in \"#{time}\"" unless m
+
+          hmsf_to_seconds(m[1], m[2], m[3], m[4])
+        end
+
+        private
+
+        def hmsf_to_time(hour, minute, second, float_part)
+          r = [hour, minute, second].map { |y| y.to_s.rjust(2, '0') }
+          r += float_part > 0.0 ? ".#{float_part.to_s.gsub(/\A(0|[^\d])+/, '')}" : '.0'
+          r
+        end
+
+        def hmsf_to_seconds(hour, minute, second, float_part)
+          r = hour.to_f * 3600 + minute.to_f * 60 + second.to_f
+          r += float_part.to_f / (10**float_part.length) if float_part
+          r
+        end
+      end
+
+      common_constructor :path
+
+      private
+
+      def tracks_uncached
+        r = []
+        content.each_line do |line|
+          t = ::Ehbrs::Videos::Track.create_from_string(line)
+          r << t if t && t.type != 'Data'
+        end
+        r
+      end
+
+      def content_uncached
+        EacRubyUtils::Envs.local.command('ffprobe', path).execute!(output: :stderr).scrub
+      end
+
+      def duration_uncached
+        m = /Duration\:\s*(#{TIME_PATTERN})/.match(content)
+        raise 'Duration pattern not find in content' unless m
+
+        self.class.time_to_seconds(m[1])
+      end
+
+      def duration_s_uncached
+        self.class.seconds_to_time(duration)
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/profiles/base.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module Ehbrs
+  module Videos
+    module Profiles
+      module Base
+        extend ::ActiveSupport::Concern
+
+        attr_reader :convert_job
+
+        included do
+          include ActiveSupport::Callbacks
+          define_callbacks :convert, :swap
+        end
+
+        def ffmpeg_args
+          []
+        end
+
+        def on_convert_job
+          old_value = convert_job
+          begin
+            yield
+          ensure
+            self.convert_job = old_value
+          end
+        end
+
+        private
+
+        attr_writer :convert_job
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/profiles/same_quality.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/profiles/base'
+
+module Ehbrs
+  module Videos
+    module Profiles
+      module SameQuality
+        extend ::ActiveSupport::Concern
+
+        included do
+          include ::Ehbrs::Videos::Profiles::Base
+        end
+
+        def ffmpeg_args
+          super + ['-q:a', '0', '-q:v', '0']
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/track.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/core_ext'
+
+module Ehbrs
+  module Videos
+    class Track
+      FFPROBE_PATTERN = /\A\s*Stream\s\#(\d+:\d+)(?:\(([^\)]+)\))?:\s*([^:]+):\s*([a-z0-9]+)(.*)/
+                        .freeze
+      class << self
+        def create_from_string(string)
+          m = FFPROBE_PATTERN.match(string)
+          return nil unless m
+
+          new(m[1].to_i, m[3], m[2], m[4], m[5].strip)
+        end
+      end
+
+      common_constructor :number, :type, :language, :codec, :extra
+
+      def to_s
+        "[#{type}(#{number}): #{codec}/#{language || '-'}" +
+          extra.if_present('') { |v| " | #{v}" } + ']'
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/check_result.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/core_ext'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      class CheckResult
+        enable_simple_cache
+        common_constructor :source, :check
+
+        def passed?
+          message.blank?
+        end
+
+        private
+
+        def message_uncached
+          check.check(source)
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/check_set.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/unsupported/checks'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      class CheckSet
+        enable_simple_cache
+        common_constructor :checks
+
+        class << self
+          # type: "file" or "track"
+          def build(profiles, type)
+            r = {}
+            profiles.each do |profile|
+              profile.send("#{type}_checks").each do |check|
+                r[check] ||= CheckWithProfiles.new(check)
+                r[check].add_profile(profile)
+              end
+            end
+            new(r.values)
+          end
+        end
+
+        class CheckWithProfiles < ::SimpleDelegator
+          def initialize(check)
+            super(check)
+            @profiles = []
+          end
+
+          def check_name
+            __getobj__.class.name.demodulize
+          end
+
+          def add_profile(profile)
+            @profiles << profile
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/check_support.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/unsupported/check_result'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module CheckSupport
+        extend ::ActiveSupport::Concern
+
+        included do
+          include ::EacRubyUtils::Console::Speaker
+          include ::EacRubyUtils::SimpleCache
+        end
+
+        def aggressions_banner(title)
+          return if passed?
+
+          info title
+          pad_speaker do
+            unpassed_checks.each do |u|
+              info "* #{u.message}"
+            end
+          end
+        end
+
+        def ffmpeg_fix_args
+          unpassed_checks.flat_map do |check|
+            check.check.fix.ffmpeg_args(self)
+          end
+        end
+
+        def passed?
+          unpassed_checks.none?
+        end
+
+        private
+
+        def unpassed_checks_uncached
+          checks.reject(&:passed?)
+        end
+
+        def checks_uncached
+          options.fetch(check_set_key).checks.map do |check|
+            ::Ehbrs::Videos::Unsupported::CheckResult.new(self, check)
+          end
+        rescue StandardError => e
+          raise "#{e.message} (Source: #{self})"
+        end
+
+        def fix_blocks_uncached
+          checks.reject(&:passed?).select { |c| c.check.fix.blank? }
+        end
+
+        def fixes_uncached
+          checks.reject(&:passed?).map { |c| c.check.fix }.reject(&:blank?)
+        end
+
+        def pad_speaker
+          on_speaker_node do |node|
+            node.stderr_line_prefix = node.stderr_line_prefix.to_s + '  '
+            yield
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/checks.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/core_ext'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module Checks
+        require_sub __FILE__
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/checks/codec_extra_unlisted.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module Checks
+        class CodecExtraUnlisted
+          common_constructor :codec, :listed_extras
+
+          def check(track)
+            return nil unless track.codec == codec
+            return nil if listed_extras.any? do |extra|
+              track.extra.downcase.include?(extra.downcase)
+            end
+
+            "Not listed extra for codec \"#{codec}\" (Track: #{track}" \
+              ", listed extra: #{listed_extras})"
+          end
+
+          def fix
+            nil
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/checks/codec_extra_unsupported.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/unsupported/fixes/supported_codec'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module Checks
+        class CodecExtraUnsupported
+          common_constructor :codec, :extra
+
+          def check(track)
+            return nil unless track.codec == codec
+            return nil unless track.extra.downcase.include?(extra.downcase)
+
+            "Unsupported extra \"#{extra}\" for codec \"#{codec}\" and track #{track}"
+          end
+
+          def fix
+            ::Ehbrs::Videos::Unsupported::Fixes::SupportedCodec.new
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/checks/codec_unlisted.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module Checks
+        class CodecUnlisted
+          common_constructor :listed_codecs
+
+          def check(track)
+            return nil if listed_codecs.include?(track.codec)
+
+            "Not listed codec \"#{track.codec}\" (Track: #{track}, listed: #{listed_codecs})"
+          end
+
+          def fix
+            nil
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/checks/codec_unsupported.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/unsupported/fixes/supported_codec'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module Checks
+        class CodecUnsupported
+          common_constructor :codec
+
+          def check(track)
+            return nil unless track.codec == codec
+
+            "Unsupported codec \"#{codec}\" for track #{track}"
+          end
+
+          def fix
+            ::Ehbrs::Videos::Unsupported::Fixes::SupportedCodec.new
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/checks/invalid_extension.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/unsupported/fixes/supported_container'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module Checks
+        class InvalidExtension
+          common_constructor :extension
+
+          def check(video)
+            return nil unless ::File.extname(video.path) == extension
+
+            "File has invalid extension: #{extension}"
+          end
+
+          def fix
+            ::Ehbrs::Videos::Unsupported::Fixes::SupportedContainer.new
+          end
+        end
+      end
+    end
+  end
+end