ehbrs-tools 0.7.0 → 0.13.0

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/runner/videos/series.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require 'eac_cli/default_runner'
+require 'eac_ruby_utils/core_ext'
+require 'eac_ruby_utils/console/docopt_runner'
+
+module Ehbrs
+  class Runner < ::EacRubyUtils::Console::DocoptRunner
+    class Videos < ::EacRubyUtils::Console::DocoptRunner
+      class Series < ::EacRubyUtils::Console::DocoptRunner
+        include ::EacCli::DefaultRunner
+        require_sub __FILE__
+
+        runner_definition do
+          desc 'Ferramentas para seriados.'
+          subcommands
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/runner/videos/series/rename.rb

@@ -0,0 +1,75 @@
+# 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/series/rename/file'
+require 'ehbrs/videos/series/rename/file/options'
+require 'ehbrs/videos/series/rename/results_builder'
+
+module Ehbrs
+  class Runner < ::EacRubyUtils::Console::DocoptRunner
+    class Videos < ::EacRubyUtils::Console::DocoptRunner
+      class Series < ::EacRubyUtils::Console::DocoptRunner
+        class Rename < ::EacRubyUtils::Console::DocoptRunner
+          include ::EacCli::DefaultRunner
+          include ::EacRubyUtils::Fs::Traversable
+
+          runner_definition do
+            desc 'Renomeia arquivos de séries.'
+            bool_opt '-r', '--recursive', 'Recursivo.'
+            bool_opt '-c', '--confirm', 'Confirmação a renomeação.'
+            arg_opt '-k', '--kernel', 'Determina o kernel para os nomes dos arquivos.'
+            arg_opt '-E', '--extension', 'Substitui a extensão dos arquivos.'
+            arg_opt '-e', '--episode-increment', 'Aumenta o número do episódio.'
+            pos_arg 'paths', repeat: true, optional: true
+          end
+
+          def run
+            banner
+            show_results
+            rename_files
+          end
+
+          def banner
+            infov 'Paths', paths.count
+            infov 'Files found', files.count
+          end
+
+          def files_uncached
+            @files = []
+            paths.each { |path| traverser_check_path(path) }
+            @files
+          end
+
+          def paths
+            options.fetch('<paths>').if_present(['.'])
+          end
+
+          def traverser_recursive
+            options.fetch('--recursive')
+          end
+
+          def traverser_check_file(path)
+            @files << ::Ehbrs::Videos::Series::Rename::File.new(path, series_file_options)
+          end
+
+          def series_file_options_uncached
+            ::Ehbrs::Videos::Series::Rename::File::Options.new(options)
+          end
+
+          def show_results
+            ::Ehbrs::Videos::Series::Rename::ResultsBuilder.new(files).show(0)
+          end
+
+          def rename_files
+            return unless series_file_options.confirm
+
+            infom 'Renaming files...'
+            files.each(&:rename)
+          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.13.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.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/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/series.rb

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

data/lib/ehbrs/videos/series/rename.rb

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

data/lib/ehbrs/videos/series/rename/directory_group.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/series/rename/line_result_group'
+require 'ehbrs/videos/series/rename/season_group'
+
+module Ehbrs
+  module Videos
+    module Series
+      module Rename
+        class DirectoryGroup < LineResultGroup
+          def line_out
+            name.light_yellow
+          end
+
+          protected
+
+          def child_key
+            :season
+          end
+
+          def child_class
+            ::Ehbrs::Videos::Series::Rename::SeasonGroup
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ehbrs/videos/series/rename/file.rb

@@ -0,0 +1,114 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/fs/extname'
+require 'ostruct'
+require_relative 'line_result'
+require_relative 'file/basename_parser'
+
+module Ehbrs
+  module Videos
+    module Series
+      module Rename
+        class File < ::Ehbrs::Videos::Series::Rename::LineResult
+          include ::Ehbrs::Videos::Series::Rename::File::BasenameParser
+
+          attr_reader :file, :options
+
+          def initialize(file, options)
+            @file = file
+            @options = options
+            @options = ::OpenStruct.new(options) if options.is_a?(::Hash)
+          end
+
+          def rename
+            target = ::File.expand_path(new_name, dirname)
+            return if ::File.exist?(target)
+
+            infov 'Renaming', @file
+            FileUtils.mv(@file, target)
+          end
+
+          def season
+            return '??' if parse.blank?
+
+            parse.fetch(:s)
+          end
+
+          def episode
+            return '??' if parse.blank?
+
+            parse.fetch(:e)
+          end
+
+          def dirname
+            d = @file
+            while d != '/'
+              d = ::File.dirname(d)
+              return ::File.expand_path(d) + '/' unless ::File.basename(d).starts_with?('_')
+            end
+            raise "series_dirname not found for #{@file}"
+          end
+
+          def line_out
+            "#{episode.green}: " + if new_name == current_name
+                                     current_name.light_black
+                                   else
+                                     "#{new_name} <= #{current_name}"
+                                   end
+          end
+
+          def new_name
+            return "#{kernel}-s#{parse[:s]}e#{new_name_episodes}#{extension}" if parse.present?
+
+            current_name
+          end
+
+          private
+
+          def current_name
+            ::File.expand_path(@file).gsub(/\A#{::Regexp.quote(dirname)}/, '')
+          end
+
+          def rename?
+            return false unless new_name
+
+            current_name != new_name
+          end
+
+          def kernel
+            options.kernel || kernel_from_directory_name ||
+              raise("Kernel undefined (File: #{@file})")
+          end
+
+          def kernel_from_directory_name
+            dir = ::File.basename(dirname).sub(/\([0-9]+\)/, '')
+            dir.split(/\W+/).select { |w| /\A[a-z0-9]/.match(w) }.map { |p| p[0].downcase }.join('')
+          end
+
+          def extension
+            return options.extension if options.extension
+
+            extension_by_directory || ::EacRubyUtils::Fs.extname2(current_name)
+          end
+
+          def extension_by_directory
+            r = ::File.dirname(current_name)
+            return nil if r == '.'
+
+            r.gsub(/\A_/, '.')
+          end
+
+          def source_episodes
+            parse[:e].split('-').map(&:to_i)
+          end
+
+          def new_name_episodes
+            source_episodes.map do |i|
+              (i + options.episode_increment).to_s.rjust(2, '0')
+            end.join('-')
+          end
+        end
+      end
+    end
+  end
+end