ehbrs-tools 0.7.0 → 0.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 72afec8feac114779c526cf0b74bfddf4bb73f3c7b72c68b96d695814773cfad
-  data.tar.gz: 456156ff658bcd0c6ccab236d1c0785706e739fe0a98ffafd7ed1c2879216bd8
+  metadata.gz: 29e80eafdb451c5637bfcafbb89ccb3cb13c157d97f8e6f0098af5ff6f512837
+  data.tar.gz: 1580aa74f7e23b3af4a55f32499044d54a2e0e41d3f97e4dd96565f3faf10d31
 SHA512:
-  metadata.gz: 549a83c6869358985e585fa9641d93b4b4260eef0da9c00ef48dd98f79c2ade46d8a1bc14dd71742bc9921001dbb066fcb93e9709cb0b82cb81e4cfb24a902ea
-  data.tar.gz: 7292c6e2fa9133ce8b9b82f8e0fce3e6183ca95932023a38172d57db1a8684bc24db517c2eaf95a42b900bb0fc22fa643469af78cb5e860bc73aee20d2e81fe9
+  metadata.gz: 9ad21b21deec42c02fcc510bf1fae9a6c397d96f86922db35e82dbeec98336b9d6a1790f34ca92d199c1916074a2cec29be5233a8154e2f13ead0ff8722a1bad
+  data.tar.gz: 7b8638979d00c8546c2b89386704bd09c2950561fb38267624357e4e640d45b6aa8685ba09d80ecf017acf8b8cdc174ac4923ee4a529b69687f875cf40b27014

data/lib/ehbrs/executables.rb

@@ -15,8 +15,10 @@ module Ehbrs
       private
 
       {
+        '-?' => %w[rar],
+        '-h' => %w[unzip],
         '-version' => %w[ffmpeg ffprobe],
-        '--version' => %w[flips wit]
+        '--version' => %w[flips tar wit]
       }.each do |validate_arg, commands|
         commands.each do |command|
           define_method("#{command}_uncached") do
@@ -24,6 +26,10 @@ module Ehbrs
           end
         end
       end
+
+      def sevenzip_uncached
+        env.executable('7z', '--help')
+      end
     end
   end
 end

data/lib/ehbrs/fs/compressed_package.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+require 'avm/files/info'
+require 'eac_ruby_utils/core_ext'
+
+module Ehbrs
+  module Fs
+    class CompressedPackage < ::Avm::Files::Info
+      MIME_TYPES = {
+        'application/zip' => :zip,
+        'application/x-7z-compressed' => :sevenzip,
+        'application/x-rar' => :rar,
+        'application/x-tar' => :tar
+      }.freeze
+
+      def extract_to(target)
+        target = target.to_pathname
+        target.mkpath
+        sub_extract_to(target)
+      end
+
+      private
+
+      def sub_extract_to(target)
+        MIME_TYPES[content_type.mime_type].if_present do |v|
+          return send("#{v}_extract_command", target).execute!
+        end
+        raise "Unknown how to extract \"#{path}\" (#{content_type})"
+      end
+
+      def sevenzip_extract_command(target_dir)
+        ::Ehbrs::Executables.sevenzip.command('x', path, '-o', target_dir)
+      end
+
+      def tar_extract_command(target_dir)
+        ::Ehbrs::Executables.tar.command('-xf', path, '-C', target_dir)
+      end
+
+      def rar_extract_command(target_dir)
+        ::Ehbrs::Executables.rar.command('x', path.expand_path).chdir(target_dir)
+      end
+
+      def zip_extract_command(target_dir)
+        ::Ehbrs::Executables.unzip.command(path, '-d', target_dir)
+      end
+    end
+  end
+end

data/lib/ehbrs/fs.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/core_ext'
+
+module Ehbrs
+  module Fs
+    require_sub __FILE__
+  end
+end

data/lib/ehbrs/runner/videos/extract.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+require 'eac_cli/default_runner'
+require 'eac_ruby_utils/core_ext'
+require 'eac_ruby_utils/console/docopt_runner'
+require 'ehbrs/videos/extract/package'
+
+module Ehbrs
+  class Runner < ::EacRubyUtils::Console::DocoptRunner
+    class Videos < ::EacRubyUtils::Console::DocoptRunner
+      class Extract < ::EacRubyUtils::Console::DocoptRunner
+        include ::EacCli::DefaultRunner
+        require_sub __FILE__
+
+        DEFAULT_QUALITIES = %w[1080 720 web webrip hdtv].freeze
+
+        runner_definition do
+          desc 'Extrai arquivos de seriados.'
+          arg_opt '-d', '--dir', 'Extraí para diretório.'
+          bool_opt '-D', '--delete', 'Remove o pacote após o processamento.'
+          arg_opt '-q', '--qualities', 'Lista de qualidades.'
+          pos_arg 'packages', repeat: true
+        end
+
+        def run
+          start_banner
+          packages.each do |package|
+            infov 'Package', package
+            package.run(options.fetch('--delete'))
+          end
+        end
+
+        private
+
+        def packages_uncached
+          options.fetch('<packages>').map do |p|
+            ::Ehbrs::Videos::Extract::Package.new(p, target_dir, qualities)
+          end
+        end
+
+        def qualities_uncached
+          (options.fetch('--qualities').to_s.split(',') + DEFAULT_QUALITIES).uniq
+        end
+
+        def start_banner
+          infov 'Packages', packages.count
+          infov 'Qualities', qualities
+          infov 'Target directory', target_dir
+        end
+
+        def target_dir_uncached
+          options.fetch('--dir').if_present(&:to_pathname) || default_target_dir
+        end
+
+        def default_target_dir
+          r = options.fetch('<packages>').first.to_pathname.basename('.*')
+          return r unless r.exist?
+
+          r = r.basename_sub { |b| "#{b}_extract" }
+          index = 0
+          loop do
+            return r unless r.exist?
+
+            index += 1
+            r = r.basename_sub { |b| "#{b}_#{index}" }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/tools/version.rb

@@ -2,6 +2,6 @@
 
 module Ehbrs
   module Tools
-    VERSION = '0.7.0'
+    VERSION = '0.9.0'
   end
 end

data/lib/ehbrs/vg/wii/game_file.rb

@@ -12,12 +12,16 @@ module Ehbrs
       class GameFile < ::Pathname
         enable_simple_cache
 
+        DISC_NUMBER_PATTERN = /disc.?(\d)/i.freeze
+
         FORMAT = ::EacRubyUtils::CustomFormat.new(
           b: :basename,
           d: :dirname,
+          D: :disc_number,
           e: :extname,
           i: :id6,
           n: :disc_name,
+          N: :nintendont_basename,
           t: :database_title,
           T: :disc_type
         )
@@ -30,6 +34,10 @@ module Ehbrs
           properties.fetch('Disc name')
         end
 
+        def disc_number
+          DISC_NUMBER_PATTERN.if_match(basename.to_s, false) { |m| m[1].to_i }.if_present(1)
+        end
+
         def disc_type
           properties.fetch('File & disc type/type')
         end
@@ -38,6 +46,11 @@ module Ehbrs
           FORMAT.format(string).with(self)
         end
 
+        def nintendont_basename
+          n = disc_number
+          n == 1 ? 'game' : "disc#{n}"
+        end
+
         def id6
           properties.fetch('Disc & part IDs/disc')
         end

data/lib/ehbrs/videos/extract/package.rb

@@ -0,0 +1,75 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/core_ext'
+require 'ehbrs/fs/compressed_package'
+require 'ehbrs/videos/extract/package_file'
+
+module Ehbrs
+  module Videos
+    module Extract
+      class Package
+        enable_simple_cache
+
+        common_constructor :path, :target_dir, :qualities do
+          self.path = path.to_pathname
+          self.target_dir = target_dir.to_pathname
+        end
+
+        delegate :to_s, to: :path
+
+        def run(delete)
+          selected_files.each(&:copy_to_selected_dir)
+          files.each(&:move_to_quality_dir)
+          extract_dir.rmdir
+          path.unlink if delete
+        end
+
+        private
+
+        def files_uncached
+          ::Pathname.glob("#{extract_dir}/**/*").map do |file|
+            ::Ehbrs::Videos::Extract::PackageFile.new(self, file)
+          end
+        end
+
+        def extract_dir_uncached
+          r = target_dir / path.basename
+          raise "Extract directory \"#{r}\" is a file" if r.file?
+
+          r.rmtree if r.directory?
+          ::Ehbrs::Fs::CompressedPackage.new(path).extract_to(r)
+          r
+        end
+
+        def files_qualities_uncached
+          qualities_with_default.select { |q| grouped_files.keys.include?(q) }
+        end
+
+        def grouped_files_uncached
+          r = {}
+          files.each do |file|
+            r[file.quality] ||= []
+            r[file.quality] << file
+          end
+          r
+        end
+
+        def qualities_with_default
+          qualities + [::Ehbrs::Videos::Extract::PackageFile::DEFAULT_QUALITY]
+        end
+
+        def selected_dir_uncached
+          target_dir / 'selected'
+        end
+
+        def selected_files
+          files.select { |f| f.quality == selected_quality }
+        end
+
+        def selected_quality_uncached
+          files_qualities.first
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/extract/package_file.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/core_ext'
+require 'ehbrs/fs/compressed_package'
+
+module Ehbrs
+  module Videos
+    module Extract
+      class PackageFile
+        DEFAULT_QUALITY = '__default__'
+
+        enable_simple_cache
+        common_constructor :package, :path do
+          self.path = path.to_pathname
+        end
+
+        def copy_to_selected_dir
+          ::FileUtils.cp(path.to_path, selected_dir.to_path)
+        end
+
+        def match_quality?(quality)
+          path.basename_sub { |b| b.to_s.downcase }.basename
+              .fnmatch?("*#{quality.downcase}*".gsub(/\A\*+/, '*').gsub(/\*+\z/, '*'))
+        end
+
+        def move_to_quality_dir
+          ::FileUtils.mv(path.to_path, quality_dir.to_path)
+        end
+
+        private
+
+        def quality_uncached
+          package.qualities.find { |q| match_quality?(q) } || DEFAULT_QUALITY
+        end
+
+        def quality_dir
+          r = package.target_dir / quality
+          r.mkpath
+          r
+        end
+
+        def selected_dir
+          r = nil
+          r = package.target_dir / 'source' if /\.torrent/ =~ path.to_path
+          r = package.target_dir / 'subtitle' if /\.srt/ =~ path.to_path
+          raise "Destination unknown: #{path}" unless r
+
+          r.mkpath
+          r
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/extract.rb

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

data/vendor/eac_ruby_utils/lib/eac_ruby_utils/envs/executable.rb

@@ -31,9 +31,9 @@ module EacRubyUtils
         raise ProgramNotFoundError, message if message
       end
 
-      def command
+      def command(*command_args)
         validate!
-        env.command(name)
+        env.command(name, *command_args)
       end
 
       private

data/vendor/eac_ruby_utils/lib/eac_ruby_utils/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module EacRubyUtils
-  VERSION = '0.31.0'
+  VERSION = '0.32.0'
 end

metadata

@@ -1,15 +1,35 @@
 --- !ruby/object:Gem::Specification
 name: ehbrs-tools
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.9.0
 platform: ruby
 authors:
 - Esquilo Azul Company
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-05-02 00:00:00.000000000 Z
+date: 2020-05-08 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: avm-tools
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.44'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.44.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.44'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.44.2
 - !ruby/object:Gem::Dependency
   name: eac_ruby_utils
   requirement: !ruby/object:Gem::Requirement
@@ -77,6 +97,8 @@ files:
 - exe/ehbrs
 - lib/ehbrs.rb
 - lib/ehbrs/executables.rb
+- lib/ehbrs/fs.rb
+- lib/ehbrs/fs/compressed_package.rb
 - lib/ehbrs/observers.rb
 - lib/ehbrs/observers/base.rb
 - lib/ehbrs/observers/with_persistence.rb
@@ -87,6 +109,7 @@ files:
 - lib/ehbrs/runner/vg/ips.rb
 - lib/ehbrs/runner/vg/wii.rb
 - lib/ehbrs/runner/videos.rb
+- lib/ehbrs/runner/videos/extract.rb
 - lib/ehbrs/runner/videos/unsupported.rb
 - lib/ehbrs/self.rb
 - lib/ehbrs/self/observers/used_space.rb
@@ -105,6 +128,9 @@ files:
 - lib/ehbrs/vg/wii/wit/path.rb
 - lib/ehbrs/videos.rb
 - lib/ehbrs/videos/convert_job.rb
+- lib/ehbrs/videos/extract.rb
+- lib/ehbrs/videos/extract/package.rb
+- lib/ehbrs/videos/extract/package_file.rb
 - lib/ehbrs/videos/file.rb
 - lib/ehbrs/videos/profiles/base.rb
 - lib/ehbrs/videos/profiles/same_quality.rb