ehbrs-tools 0.3.0

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

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/file'
+require 'ehbrs/videos/unsupported/file/fix'
+require 'ehbrs/videos/unsupported/check_support'
+require 'ehbrs/videos/unsupported/track'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      class File < ::Ehbrs::Videos::File
+        include ::Ehbrs::Videos::Unsupported::CheckSupport
+        include ::Ehbrs::Videos::Unsupported::File::Fix
+
+        attr_reader :options
+
+        def initialize(file, options)
+          super file
+          @options = options
+        end
+
+        def banner
+          infov 'File', path
+          pad_speaker do
+            aggressions_banner('Self')
+            tracks.each(&:banner)
+          end
+        end
+
+        def all_passed?
+          passed? && tracks.all?(&:passed?)
+        end
+
+        def all_fixes
+          fixes + tracks.flat_map(&:fixes)
+        end
+
+        def check_set_key
+          :file_check_set
+        end
+
+        private
+
+        def tracks_uncached
+          super.map { |t| ::Ehbrs::Videos::Unsupported::Track.new(self, t) }
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/file/fix.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/convert_job'
+require 'ehbrs/videos/file'
+require 'ehbrs/videos/unsupported/fix_profile'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      class File < ::Ehbrs::Videos::File
+        module Fix
+          def check_fix
+            return unless options.fetch(:fix)
+
+            if fix_blocks.any?
+              infom '  * Cannot fix:'
+              fix_blocks.each do |fb|
+                infom "    * #{fb.check.check_name}"
+              end
+            else
+              infom '  * Fixing...'
+              infom "  * Fixed in: \"#{fix}\""
+            end
+          end
+
+          private
+
+          def all_fix_blocks_uncached
+            fix_blocks + tracks.flat_map(&:fix_blocks)
+          end
+
+          def fix_profile_uncached
+            ::Ehbrs::Videos::Unsupported::FixProfile.new(self)
+          end
+
+          def fix
+            job = ::Ehbrs::Videos::ConvertJob.new(path, fix_profile)
+            job.run
+            job.target
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/profiles/same_quality'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      class FixProfile
+        include ::Ehbrs::Videos::Profiles::SameQuality
+
+        common_constructor :video
+        set_callback :swap, :after do
+          video.all_fixes.each do |fix|
+            next unless fix.respond_to?(:after_swap)
+
+            fix.after_swap(self)
+          end
+        end
+
+        def name
+          "fix_for_#{::File.basename(video.file)}"
+        end
+
+        def ffmpeg_args
+          r = fix_args
+          r += container_fix_args unless r.include?('-f')
+          r + super
+        end
+
+        private
+
+        def fix_args
+          ['-c', 'copy'] + video.ffmpeg_fix_args +
+            video.tracks.flat_map(&:ffmpeg_fix_args)
+        end
+
+        def track_fix_args(track)
+          ["#{track_codec_option_by_type(track.type)}:#{track.number}",
+           track.passed? ? 'copy' : track_codec_fix_by_type(track.type)]
+        end
+
+        def track_codec_option_by_type(track_type)
+          TRACK_TYPE_OPTIONS.fetch(track_type)
+        end
+
+        def track_codec_fix_by_type(track_type)
+          TRACK_TYPE_FIX_CODECS.fetch(track_type)
+        end
+
+        def container_fix_args
+          ['-f', ::Ehbrs::Videos::Unsupported::Fixes::SupportedContainer::FIX_FORMAT]
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/fixes/supported_codec.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module Fixes
+        class SupportedCodec
+          TRACK_TYPE_OPTIONS = {
+            'Audio' => '-acodec',
+            'Video' => '-vcodec',
+            'Subtitle' => '-scodec'
+          }.freeze
+
+          TRACK_TYPE_FIX_CODECS = {
+            'Audio' => 'aac',
+            'Video' => 'libx264',
+            'Subtitle' => 'ass'
+          }.freeze
+
+          def ffmpeg_args(track)
+            ["#{track_codec_option_by_type(track.type)}:#{track.number}",
+             track_codec_fix_by_type(track.type)]
+          end
+
+          def track_codec_option_by_type(track_type)
+            TRACK_TYPE_OPTIONS.fetch(track_type)
+          end
+
+          def track_codec_fix_by_type(track_type)
+            TRACK_TYPE_FIX_CODECS.fetch(track_type)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/fixes/supported_container.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module Fixes
+        class SupportedContainer
+          FIX_FORMAT = 'matroska'
+
+          def ffmpeg_args(_video)
+            ['-f', FIX_FORMAT]
+          end
+        end
+      end
+    end
+  end
+end

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

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

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

@@ -0,0 +1,102 @@
+# frozen_string_literal: true
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module Profiles
+        class Base
+          enable_simple_cache
+          include ::Singleton
+
+          CONSTANT_PREFIXES = %w[video audio subtitle other].freeze
+
+          def added_checks
+            @added_checks ||= []
+          end
+
+          def add_check(name, *args)
+            check_path = "ehbrs/videos/unsupported/checks/#{name}"
+            added_checks << check_path.camelize.constantize.new(*args)
+          end
+
+          def file_checks
+            added_checks
+          end
+
+          def track_checks
+            [unlisted_codec_check] + unsupported_codec_checks +
+              supported_codecs.flat_map { |codec| codec_extra_checks(codec) }
+          end
+
+          def codec_extra_checks(codec)
+            codec_unlisted_extra_check(codec).if_present([]) { |v| [v] } +
+              codec_unsupported_extras(codec).map do |extra|
+                Ehbrs::Videos::Unsupported::Checks::CodecExtraUnsupported.new(codec, extra)
+              end
+          end
+
+          def codec_unlisted_extra_check(codec)
+            extras = codec_unsupported_extras(codec) + codec_supported_extras(codec)
+            return nil unless extras.any?
+
+            Ehbrs::Videos::Unsupported::Checks::CodecExtraUnlisted.new(
+              codec, extras.sort.uniq
+            )
+          end
+
+          def name
+            self.class.name.demodulize.underscore
+          end
+
+          def to_s
+            name
+          end
+
+          def unsupported_codec_checks
+            unsupported_codecs.map do |codec|
+              Ehbrs::Videos::Unsupported::Checks::CodecUnsupported.new(codec)
+            end
+          end
+
+          def unsupported_codecs_uncached
+            codecs_by_constant('unsupported')
+          end
+
+          def unlisted_codec_check
+            ::Ehbrs::Videos::Unsupported::Checks::CodecUnlisted.new(
+              (supported_codecs + unsupported_codecs).sort.uniq
+            )
+          end
+
+          def supported_codecs_uncached
+            codecs_by_constant('supported')
+          end
+
+          def codecs_by_constant(middle)
+            CONSTANT_PREFIXES.inject([]) { |a, e| a + codecs_by_prefix(e, middle) }
+          end
+
+          def codecs_by_prefix(prefix, middle)
+            self.class.const_get("#{prefix}_#{middle}_codecs".upcase)
+          rescue NameError
+            []
+          end
+
+          def codec_extras(codec, suffix)
+            self.class.const_get("#{codec}_extra_#{suffix}".upcase)
+          rescue NameError
+            []
+          end
+
+          def codec_unsupported_extras(codec)
+            codec_extras(codec, 'unsupported')
+          end
+
+          def codec_supported_extras(codec)
+            codec_extras(codec, 'supported')
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/profiles/philco.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/unsupported/profiles/base'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module Profiles
+        class Philco < ::Ehbrs::Videos::Unsupported::Profiles::Base
+          AUDIO_SUPPORTED_CODECS = %w[aac ac3 eac3 mp3].freeze
+          AUDIO_UNSUPPORTED_CODECS = %w[dts].freeze
+
+          VIDEO_SUPPORTED_CODECS = %w[h264 mpeg4].freeze
+          VIDEO_UNSUPPORTED_CODECS = %w[hevc msmpeg4v3].freeze
+
+          SUBTITLE_SUPPORTED_CODECS = %w[ass dvd subrip].freeze
+          SUBTITLE_UNSUPPORTED_CODECS = %w[mov].freeze
+
+          OTHER_SUPPORTED_CODECS = %w[ttf].freeze
+
+          MPEG4_EXTRA_SUPPORTED = %w[xvid].freeze
+          MPEG4_EXTRA_UNSUPPORTED = %w[dx50].freeze
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/unsupported/profiles/samsung.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/unsupported/checks/invalid_extension'
+require 'ehbrs/videos/unsupported/profiles/base'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      module Profiles
+        class Samsung < ::Ehbrs::Videos::Unsupported::Profiles::Base
+          AUDIO_SUPPORTED_CODECS = %w[aac ac3 dts eac3 mp3 vorbis].freeze
+          AUDIO_UNSUPPORTED_CODECS = %w[].freeze
+
+          VIDEO_SUPPORTED_CODECS = %w[h264 mpeg4 hevc].freeze
+          VIDEO_UNSUPPORTED_CODECS = %w[].freeze
+
+          SUBTITLE_SUPPORTED_CODECS = %w[ass dvd subrip].freeze
+          SUBTITLE_UNSUPPORTED_CODECS = %w[mov].freeze
+
+          OTHER_SUPPORTED_CODECS = %w[png ttf].freeze
+
+          MPEG4_EXTRA_SUPPORTED = %w[].freeze
+          MPEG4_EXTRA_UNSUPPORTED = %w[dx50 xvid].freeze
+
+          def initialize
+            add_check('invalid_extension', '.m4v')
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/filesystem_traverser'
+require 'eac_ruby_utils/core_ext'
+require 'ehbrs/videos/unsupported/file'
+
+module Ehbrs
+  module Videos
+    module Unsupported
+      class Search
+        enable_console_speaker
+
+        VALID_EXTENSIONS = %w[.avi .mp4 .mkv .m4v].freeze
+
+        def initialize(root, file_options)
+          @root = root
+          @file_options = file_options
+          @files = 0
+          @videos = 0
+          @unsupported = 0
+          run
+        end
+
+        private
+
+        def run
+          start_banner
+          traverser = ::EacRubyUtils::FilesystemTraverser.new
+          traverser.recursive = true
+          traverser.check_file = method(:check_file)
+          traverser.check_path(@root)
+          end_banner
+        end
+
+        def recursive?
+          true
+        end
+
+        def check_file(file)
+          @files += 1
+          check_video(file) if video_file?(file)
+        end
+
+        def check_video(file)
+          @videos += 1
+          v = ::Ehbrs::Videos::Unsupported::File.new(file, @file_options)
+          return if v.all_passed?
+
+          @unsupported += 1
+          v.banner
+          v.check_fix
+        end
+
+        def start_banner
+          infom "Searching in: \"#{@root}\""
+        end
+
+        def end_banner
+          infom "Unsupported/Videos/Files: #{@unsupported}/#{@videos}/#{@files}"
+        end
+
+        def video_file?(path)
+          valid_extensions.include?(::File.extname(path))
+        end
+
+        def valid_extensions
+          self.class::VALID_EXTENSIONS
+        end
+      end
+    end
+  end
+end