ehbrs-tools 0.7.0 → 0.13.0

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

data/vendor/aranha-selenium/Gemfile

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+source 'https://rubygems.org'
+
+gemspec

data/vendor/aranha-selenium/aranha-selenium.gemspec

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+$LOAD_PATH.push File.expand_path('lib', __dir__)
+
+require 'aranha/selenium/version'
+
+Gem::Specification.new do |s|
+  s.name        = 'aranha-selenium'
+  s.version     = ::Aranha::Selenium::VERSION
+  s.authors     = ['Esquilo Azul Company']
+  s.summary     = 'Selenium\' utilities for Ruby.'
+
+  s.files = Dir['{lib}/**/*', 'Gemfile']
+
+  s.add_dependency 'activesupport'
+  s.add_dependency 'selenium-webdriver', '~> 3.142', '>= 3.142.3'
+
+  s.add_development_dependency 'eac_ruby_gem_support', '~> 0.1', '>= 0.1.1'
+end

data/vendor/aranha-selenium/lib/aranha/selenium.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module Aranha
+  module Selenium
+    require 'aranha/selenium/driver_factory'
+    require 'aranha/selenium/session'
+  end
+end

data/vendor/aranha-selenium/lib/aranha/selenium/driver_factory.rb

@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+require 'active_support/core_ext/hash/indifferent_access'
+require 'aranha/selenium/driver_factory/base'
+require 'aranha/selenium/driver_factory/chrome'
+require 'aranha/selenium/driver_factory/firefox'
+
+module Aranha
+  module Selenium
+    class DriverFactory
+      class << self
+        def create_driver(options = {})
+          new(options).create_driver
+        end
+      end
+
+      attr_reader :options
+
+      def initialize(options)
+        @options = options.with_indifferent_access.freeze
+      end
+
+      def create_driver
+        driver_class.new(driver_options).build
+      end
+
+      def driver_name
+        (options[:driver] || default_driver_name).to_s
+      end
+
+      def driver_class
+        Aranha::Selenium::DriverFactory.const_get(driver_name.classify)
+      rescue NameError
+        raise "Unknown Aranha Selenium driver: \"#{driver_name}\"" \
+          " (Class \"Aranha::Selenium::DriverFactory::#{driver_name.classify}\" not found)"
+      end
+
+      def default_driver_name
+        :firefox
+      end
+
+      def driver_options
+        options.except(:driver)
+      end
+    end
+  end
+end

data/vendor/aranha-selenium/lib/aranha/selenium/driver_factory/base.rb

@@ -0,0 +1,76 @@
+# frozen_string_literal: true
+
+require 'active_support/json'
+require 'active_support/core_ext/hash/indifferent_access'
+require 'selenium-webdriver'
+require 'tmpdir'
+
+module Aranha
+  module Selenium
+    class DriverFactory
+      class Base
+        DEFAULT_DOWNLOADS_DIR = ::File.join(::Dir.tmpdir, 'aranha_downloads_dir')
+        DEFAULT_ACCEPT_INSECURE_CERTS = false
+        DEFAULT_HEADLESS = false
+        DEFAULT_USER_AGENT = nil
+
+        class << self
+          attr_writer :default_accept_insecure_certs, :default_downloads_dir, :default_headless,
+                      :default_user_agent
+
+          def default_downloads_dir
+            @default_downloads_dir || DEFAULT_DOWNLOADS_DIR
+          end
+
+          def default_accept_insecure_certs
+            @default_accept_insecure_certs || DEFAULT_ACCEPT_INSECURE_CERTS
+          end
+
+          def default_headless
+            @default_headless || DEFAULT_HEADLESS
+          end
+
+          def default_user_agent
+            @default_user_agent || DEFAULT_USER_AGENT
+          end
+        end
+
+        attr_reader :options
+
+        def initialize(options)
+          @options = options.with_indifferent_access.freeze
+        end
+
+        def build
+          raise 'Must be overrided'
+        end
+
+        def downloads_dir
+          option_value(:downloads_dir)
+        end
+
+        def accept_insecure_certs?
+          option_value(:accept_insecure_certs)
+        end
+
+        def headless?
+          option_value(:headless)
+        end
+
+        def user_agent
+          option_value(:user_agent)
+        end
+
+        private
+
+        def option_value(key)
+          if options.key?(key)
+            options.fetch(key)
+          else
+            ::Aranha::Selenium::DriverFactory::Base.send("default_#{key}")
+          end
+        end
+      end
+    end
+  end
+end

data/vendor/aranha-selenium/lib/aranha/selenium/driver_factory/chrome.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require 'aranha/selenium/driver_factory/base'
+
+module Aranha
+  module Selenium
+    class DriverFactory
+      class Chrome < ::Aranha::Selenium::DriverFactory::Base
+        def build
+          ::Selenium::WebDriver.for :chrome, options: chrome_options
+        end
+
+        private
+
+        def chrome_options
+          r = ::Selenium::WebDriver::Chrome::Options.new
+          r.add_argument('--ignore-certificate-errors') if accept_insecure_certs?
+          r.add_argument('--headless') if headless?
+          r.add_argument('--disable-popup-blocking')
+          r.add_argument('--disable-translate')
+          r.add_argument("user-agent=#{user_agent}") if user_agent.present?
+          r.add_preference(:download, prompt_for_download: false, default_directory: downloads_dir)
+          r
+        end
+      end
+    end
+  end
+end

data/vendor/aranha-selenium/lib/aranha/selenium/driver_factory/firefox.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require 'aranha/selenium/driver_factory/base'
+
+module Aranha
+  module Selenium
+    class DriverFactory
+      class Firefox < ::Aranha::Selenium::DriverFactory::Base
+        def build
+          ::Selenium::WebDriver.for(
+            :firefox,
+            options: ::Selenium::WebDriver::Firefox::Options.new(
+              profile: build_profile,
+              args: build_args
+            ),
+            desired_capabilities: build_capabilities
+          )
+        end
+
+        private
+
+        def build_args
+          r = []
+          r << '-headless' if headless?
+          r
+        end
+
+        def build_capabilities
+          if accept_insecure_certs?
+            ::Selenium::WebDriver::Remote::Capabilities.firefox(accept_insecure_certs: true)
+          else
+            ::Selenium::WebDriver::Remote::Capabilities.firefox
+          end
+        end
+
+        def build_profile
+          r = ::Selenium::WebDriver::Firefox::Profile.new
+          r['browser.download.dir'] = downloads_dir
+          r['browser.download.folderList'] = 2
+          r['browser.helperApps.neverAsk.saveToDisk'] = auto_download_mime_types.join(';')
+          r['pdfjs.disabled'] = true
+          r['general.useragent.override'] = user_agent if user_agent.present?
+          r
+        end
+
+        def auto_download_mime_types
+          ::File.read(
+            ::File.join(__dir__, 'firefox_auto_download_mime_types')
+          ).each_line.map(&:strip)
+        end
+      end
+    end
+  end
+end