ehbrs-tools 0.5.0 → 0.11.0

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

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

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/series/rename/line_result'
+
+module Ehbrs
+  module Videos
+    module Series
+      module Rename
+        class File < ::Ehbrs::Videos::Series::Rename::LineResult
+          module BasenameParser
+            FORMATS = [{
+              format: /(\d+)(\d{2})(\d{2})/i,
+              build: ->(m) { { s: padding(m[1]), e: padding(m[2]) + '-' + padding(m[3]) } }
+            }, {
+              format: /s(\d+)e(\d+)-(\d+)/i,
+              build: ->(m) { { s: padding(m[1]), e: padding(m[2]) + '-' + padding(m[3]) } }
+            }, {
+              format: /(\d{1,2})\s*[^\d]\s*(\d{2})/i,
+              build: ->(m) { { s: padding(m[1]), e: padding(m[2]) } }
+            }, {
+              format: /(\d+)(\d{2})/i,
+              build: ->(m) { { s: padding(m[1]), e: padding(m[2]) } }
+            }, {
+              format: /(\d{2})/i,
+              build: ->(m) { { s: padding(1), e: padding(m[1]) } }
+            }].freeze
+
+            def self.padding(string)
+              string.to_s.rjust(2, '0')
+            end
+
+            def parse_uncached
+              FORMATS.each do |format|
+                m = format.fetch(:format).match(current_name)
+                return format.fetch(:build).call(m) if m
+              end
+              nil
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Ehbrs
+  module Videos
+    module Series
+      module Rename
+        class File
+          class Options
+            def initialize(docopt_runner_options)
+              @options = docopt_runner_options
+            end
+
+            %w[kernel confirm recursive extension].each do |method|
+              class_eval <<-RUBY_EVAL, __FILE__, __LINE__ + 1
+      def #{method}
+        @options.fetch('--#{method}')
+      end
+              RUBY_EVAL
+            end
+
+            def episode_increment
+              r = @options.fetch('--episode-increment')
+              r.present? ? r.to_i : 0
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/core_ext'
+
+module Ehbrs
+  module Videos
+    module Series
+      module Rename
+        class LineResult
+          enable_simple_cache
+          enable_console_speaker
+
+          def show(level)
+            out(padding_level(level) + line_out + "\n")
+          end
+
+          private
+
+          def padding_level(level)
+            ('  ' * level.to_i) + '* '
+          end
+        end
+      end
+    end
+  end
+end