ehbrs-tools 0.9.0 → 0.10.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 29e80eafdb451c5637bfcafbb89ccb3cb13c157d97f8e6f0098af5ff6f512837
-  data.tar.gz: 1580aa74f7e23b3af4a55f32499044d54a2e0e41d3f97e4dd96565f3faf10d31
+  metadata.gz: ffc4f5477aac7e6c40bc373d77e239599afaf91fe96f8f1b4c2248a4da8d7fa4
+  data.tar.gz: 7950f28d7c9222ce50c645cc4ee36c8be3e45bcaf89cc03cf8fcf1c6355cf88b
 SHA512:
-  metadata.gz: 9ad21b21deec42c02fcc510bf1fae9a6c397d96f86922db35e82dbeec98336b9d6a1790f34ca92d199c1916074a2cec29be5233a8154e2f13ead0ff8722a1bad
-  data.tar.gz: 7b8638979d00c8546c2b89386704bd09c2950561fb38267624357e4e640d45b6aa8685ba09d80ecf017acf8b8cdc174ac4923ee4a529b69687f875cf40b27014
+  metadata.gz: 3179aedf31490fc080ff805c6418c38477a7a7575d18415a8081beba23162f7e42efbb50c553deb871df640d1fc1d2f997a4f53caaf39cdb7a453ded3a6f65fd
+  data.tar.gz: 0532f3b45ad4014d11dab09e5947c95da727a0422806ad29f9125011d7b0722fc94e3e89b1f8317aeed76cab96a5a857e08fb68c61785830e858ba925cd75b38

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/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/tools/version.rb

@@ -2,6 +2,6 @@
 
 module Ehbrs
   module Tools
-    VERSION = '0.9.0'
+    VERSION = '0.10.0'
   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/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/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/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

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.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/vendor/eac_ruby_utils/eac_ruby_utils.gemspec

@@ -20,5 +20,5 @@ Gem::Specification.new do |s|
   s.add_dependency 'colorize', '~> 0.8.1'
   s.add_dependency 'docopt', '~> 0.6.1'
   s.add_dependency 'net-ssh', '~> 4.2'
-  s.add_development_dependency 'eac_ruby_gem_support', '~> 0.1'
+  s.add_development_dependency 'eac_ruby_gem_support', '~> 0.1', '>= 0.1.1'
 end

data/vendor/eac_ruby_utils/lib/eac_ruby_utils/fs/extname.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module EacRubyUtils
+  module Fs
+    class << self
+      # A [File.extname] which find multiple extensions (Ex.: .tar.gz).
+      def extname(path, limit = -1)
+        recursive_extension(::File.basename(path), limit)
+      end
+
+      # Shortcut to +extname(2)+.
+      def extname2(path)
+        extname(path, 2)
+      end
+
+      private
+
+      def recursive_extension(basename, limit)
+        return '' if limit.zero?
+
+        m = /\A(.+)(\.[a-z][a-z0-9]*)\z/i.match(basename)
+        if m
+          "#{recursive_extension(m[1], limit - 1)}#{m[2]}"
+        else
+          ''
+        end
+      end
+    end
+  end
+end

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

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

data/vendor/eac_ruby_utils/spec/lib/eac_ruby_utils/fs/extname_spec.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/fs/extname'
+
+RSpec.describe ::EacRubyUtils::Fs do
+  describe '#extname' do
+    {
+      'After.Life.S01E01.WEBRip.x264-ION10.mp4' => '.mp4',
+      's01e01.en.srt' => '.en.srt',
+      's01e01.srt' => '.srt',
+      '/path/to/file.tar.gz' => '.tar.gz'
+    }.each do |source, expected|
+      context "when source is \"#{source}\"" do
+        it { expect(described_class.extname(source)).to eq(expected) }
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ehbrs-tools
 version: !ruby/object:Gem::Version
-  version: 0.9.0
+  version: 0.10.0
 platform: ruby
 authors:
 - Esquilo Azul Company
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-05-08 00:00:00.000000000 Z
+date: 2020-05-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: avm-tools
@@ -110,6 +110,8 @@ files:
 - lib/ehbrs/runner/vg/wii.rb
 - lib/ehbrs/runner/videos.rb
 - lib/ehbrs/runner/videos/extract.rb
+- lib/ehbrs/runner/videos/series.rb
+- lib/ehbrs/runner/videos/series/rename.rb
 - lib/ehbrs/runner/videos/unsupported.rb
 - lib/ehbrs/self.rb
 - lib/ehbrs/self/observers/used_space.rb
@@ -134,6 +136,16 @@ files:
 - lib/ehbrs/videos/file.rb
 - lib/ehbrs/videos/profiles/base.rb
 - lib/ehbrs/videos/profiles/same_quality.rb
+- lib/ehbrs/videos/series.rb
+- lib/ehbrs/videos/series/rename.rb
+- lib/ehbrs/videos/series/rename/directory_group.rb
+- lib/ehbrs/videos/series/rename/file.rb
+- lib/ehbrs/videos/series/rename/file/basename_parser.rb
+- lib/ehbrs/videos/series/rename/file/options.rb
+- lib/ehbrs/videos/series/rename/line_result.rb
+- lib/ehbrs/videos/series/rename/line_result_group.rb
+- lib/ehbrs/videos/series/rename/results_builder.rb
+- lib/ehbrs/videos/series/rename/season_group.rb
 - lib/ehbrs/videos/track.rb
 - lib/ehbrs/videos/unsupported/check_result.rb
 - lib/ehbrs/videos/unsupported/check_set.rb
@@ -206,6 +218,7 @@ files:
 - vendor/eac_ruby_utils/lib/eac_ruby_utils/envs/ssh_env.rb
 - vendor/eac_ruby_utils/lib/eac_ruby_utils/filesystem_cache.rb
 - vendor/eac_ruby_utils/lib/eac_ruby_utils/fs.rb
+- vendor/eac_ruby_utils/lib/eac_ruby_utils/fs/extname.rb
 - vendor/eac_ruby_utils/lib/eac_ruby_utils/fs/temp.rb
 - vendor/eac_ruby_utils/lib/eac_ruby_utils/fs/temp/directory.rb
 - vendor/eac_ruby_utils/lib/eac_ruby_utils/fs/temp/file.rb
@@ -275,6 +288,7 @@ files:
 - vendor/eac_ruby_utils/spec/lib/eac_ruby_utils/envs/executable_spec.rb
 - vendor/eac_ruby_utils/spec/lib/eac_ruby_utils/envs/ssh_env_spec.rb
 - vendor/eac_ruby_utils/spec/lib/eac_ruby_utils/filesystem_cache_spec.rb
+- vendor/eac_ruby_utils/spec/lib/eac_ruby_utils/fs/extname_spec.rb
 - vendor/eac_ruby_utils/spec/lib/eac_ruby_utils/fs/temp/temp_spec.rb
 - vendor/eac_ruby_utils/spec/lib/eac_ruby_utils/fs/temp_spec.rb
 - vendor/eac_ruby_utils/spec/lib/eac_ruby_utils/listable_spec.rb