ehbrs-tools 0.9.0 → 0.13.1

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.9.0'
+    VERSION = '0.13.1'
   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

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

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require_relative 'line_result'
+
+module Ehbrs
+  module Videos
+    module Series
+      module Rename
+        class LineResultGroup < ::Ehbrs::Videos::Series::Rename::LineResult
+          attr_reader :name, :children
+
+          def initialize(name, files)
+            @name = name
+            @children = build_children(files)
+          end
+
+          def show(level)
+            super(level)
+            children.each do |child|
+              child.show(level + 1)
+            end
+          end
+
+          private
+
+          def build_children(files)
+            r = {}
+            files.each do |file|
+              key = file.send(child_key)
+              r[key] ||= []
+              r[key] << file
+            end
+            r.sort.map { |v| child_class.new(v[0], v[1]) }
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'ehbrs/videos/series/rename/directory_group'
+require 'ehbrs/videos/series/rename/line_result_group'
+
+module Ehbrs
+  module Videos
+    module Series
+      module Rename
+        class ResultsBuilder < LineResultGroup
+          def initialize(files)
+            super '', files
+          end
+
+          def line_out
+            'Groups: '.cyan + children.count.to_s
+          end
+
+          protected
+
+          def child_key
+            :dirname
+          end
+
+          def child_class
+            ::Ehbrs::Videos::Series::Rename::DirectoryGroup
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require_relative 'line_result'
+
+module Ehbrs
+  module Videos
+    module Series
+      module Rename
+        class SeasonGroup < ::Ehbrs::Videos::Series::Rename::LineResult
+          attr_reader :season, :files
+
+          def initialize(season, files)
+            @season = season
+            @files = files.sort_by { |f| [f.episode] }
+          end
+
+          def line_out
+            'Season: '.cyan + "#{season} (#{first_episode} - #{last_episode})"
+          end
+
+          def show(level)
+            super
+            files.each { |file| file.show(level + 1) }
+          end
+
+          private
+
+          def first_episode
+            files.first.episode
+          end
+
+          def last_episode
+            files.last.episode
+          end
+        end
+      end
+    end
+  end
+end