miteru 2.2.0 → 2.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 62e5976eed670c3d2f2b0309e7723434c1e5c04ad337d4db519a58362618d566
-  data.tar.gz: fe54d9316505b380ac50ba9ab8cf520f2fd5b0e76cf88f0e520d7fb0c18a3b56
+  metadata.gz: 3767b21d7f569a43215e2fdc0f83f9fc4f29fe2a4bb11fc3548a5d1657812774
+  data.tar.gz: 7877fb357548c619210ec63f3bd26c61c3fda68dc6d21910aa22fc0d7144ebf1
 SHA512:
-  metadata.gz: 122f011baecb0a38639610ffaf68645c48f454ef7aa6ba79d5e812f71129f303a4386308cf639e233ff43473907f9567f90625b3651e027d18d6324e694ba00e
-  data.tar.gz: 0f21e5a5b6882fb9705d9988647de16aad44693ddfbbd9983a01f16c029923629322c40165a28dcb33e4a49482bcddbc1d272825beae7b0017400c560e16a57d
+  metadata.gz: '02480c9a5604e5c7959ae4adb9e3a657d4b59f5c668be51b7d1fe120bd46e5548ecc9a26404d88d001729bf19f4f3d1b1545be80820c0717c0e027b7995030cd'
+  data.tar.gz: f2c1adc686393f3979ccf77e87c1bc5dba9cad9023903620ab61bbf6e6a2a4406dbbdd3ffb9ca73f5297c2135436dfd995f9572085fd9fde02fb5b72eac6032b

data/lib/miteru/cache.rb

@@ -19,8 +19,8 @@ module Miteru
     # @param [Integer. nil] ex
     #
     def set(key, value, ex:)
-      value = redis.set("#{prefix}:#{key}", value, ex:)
-      Miteru.logger.info("Cache:#{key} is set.") if verbose?
+      value = redis.set("#{cache_prefix}:#{key}", value, ex:)
+      logger.info("Cache:#{key} is set.") if verbose?
       value
     end
 
@@ -28,21 +28,13 @@ module Miteru
     # @param [String] key
     #
     def cached?(key)
-      value = redis.exists?("#{prefix}:#{key}")
-      Miteru.logger.info("Cache:#{key} found.") if verbose?
+      value = redis.exists?("#{cache_prefix}:#{key}")
+      logger.info("Cache:#{key} found.") if verbose?
       value
     end
 
     private
 
-    def verbose?
-      Miteru.config.verbose
-    end
-
-    def prefix
-      Miteru.config.cache_prefix
-    end
-
     #
     # @return [Redis]
     #

data/lib/miteru/concerns/url_truncatable.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Miteru
+  module Concerns
+    module UrlTruncatable
+      extend ActiveSupport::Concern
+
+      def decoded_url
+        @decoded_url ||= URI.decode_www_form_component(url)
+      end
+
+      #
+      # @return [String]
+      #
+      def truncated_url
+        @truncated_url ||= decoded_url.truncate(64)
+      end
+
+      def defanged_truncated_url
+        @defanged_truncated_url ||= truncated_url.to_s.gsub(".", "[.]")
+      end
+    end
+  end
+end

data/lib/miteru/crawler.rb

@@ -11,23 +11,21 @@ module Miteru
       Try[OpenSSL::SSL::SSLError, ::HTTP::Error, Addressable::URI::InvalidURIError] do
         info = "Website:#{website.info}."
         info = info.colorize(:red) if website.kits?
-        Miteru.logger.info(info)
+        logger.info(info)
 
         website.kits.each do |kit|
           downloader = Downloader.new(kit)
           result = downloader.result
-
           unless result.success?
-            Miteru.logger.warn("Kit:#{kit.truncated_url} failed to download - #{result.failure}.")
+            logger.warn("Kit:#{kit.truncated_url} failed to download - #{result.failure}.")
             next
           end
-
           destination = result.value!
-          Miteru.logger.info("Kit:#{kit.truncated_url} downloaded as #{destination}.")
+          logger.info("Kit:#{kit.truncated_url} downloaded as #{destination}.")
           # Remove downloaded file if auto_download is not allowed
           FileUtils.rm(destination, force: true) unless auto_download?
-          # Notify the website
-          notify website
+          # Notify the kit
+          notify(kit)
         end
 
         # Cache the website
@@ -37,39 +35,22 @@ module Miteru
 
     private
 
-    def cache?
-      Miteru.cache?
-    end
-
-    def cache
-      Miteru.cache
-    end
-
-    def cache_ex
-      Miteru.config.cache_ex
-    end
-
-    def auto_download?
-      Miteru.config.auto_download
-    end
-
     #
-    # @param [Miteru::Website] website
+    # @param [Miteru::Kit] kit
     #
-    def notify(website)
+    def notify(kit)
       notifiers.each do |notifier|
-        result = notifier.result(website)
+        result = notifier.result(kit)
         if result.success?
-          Miteru.logger.info("Notifier:#{notifier.name} succeeded.")
+          logger.info("Notifier:#{notifier.name} succeeded.")
         else
-          Miteru.logger.warn("Notifier:#{notifier.name} failed - #{result.failure}.")
+          logger.warn("Notifier:#{notifier.name} failed - #{result.failure}.")
         end
       end
     end
 
-    #
-    # @return [Array<Miteru::Notifiers::Base>]
-    #
+    private
+
     def notifiers
       @notifiers ||= Miteru.notifiers.map(&:new)
     end

data/lib/miteru/downloader.rb

@@ -43,7 +43,7 @@ module Miteru
 
       # Record a kit in DB
       Record.create_by_kit_and_hash(kit, sha256: sha256(destination))
-      Miteru.logger.info "Download #{kit.url} as #{destination}"
+      logger.info "Download #{kit.url} as #{destination}"
 
       destination
     end

data/lib/miteru/kit.rb

@@ -2,6 +2,8 @@
 
 module Miteru
   class Kit < Service
+    include Concerns::UrlTruncatable
+
     # @return [String]
     attr_reader :url
 
@@ -85,17 +87,6 @@ module Miteru
       @hostname ||= URI(url).hostname
     end
 
-    def decoded_url
-      @decoded_url ||= URI.decode_www_form_component(url)
-    end
-
-    #
-    # @return [String]
-    #
-    def truncated_url
-      url.truncate(64)
-    end
-
     private
 
     def filename_to_download

data/lib/miteru/notifiers/base.rb

@@ -3,7 +3,7 @@
 module Miteru
   module Notifiers
     class Base < Service
-      def call(website)
+      def call(kit)
         raise NotImplementedError
       end
 

data/lib/miteru/notifiers/slack.rb

@@ -77,13 +77,13 @@ module Miteru
       #
       # Notifiy to Slack
       #
-      # @param [Miteru::Website] website
+      # @param [Miteru::Kit] kit
       #
-      def call(website)
+      def call(kit)
         return unless callable?
 
-        attachment = SlackAttachment.new(website.url)
-        notifier.post(text: website.info, attachments: attachment.to_a) if website.kits?
+        attachment = SlackAttachment.new(kit.url)
+        notifier.post(text: kit.defanged_truncated_url, attachments: attachment.to_a)
       end
 
       def callable?

data/lib/miteru/notifiers/urlscan.rb

@@ -4,12 +4,12 @@ module Miteru
   module Notifiers
     class UrlScan < Base
       #
-      # @param [Miteru::Website] website
+      # @param [Miteru::Kit] kit
       #
-      def call(website)
+      def call(kit)
         return unless callable?
 
-        website.kits.each { |kit| submit(kit.url, source: website.source) }
+        submit(kit.decoded_url, source: kit.source)
       end
 
       def callable?

data/lib/miteru/orchestrator.rb

@@ -3,24 +3,31 @@
 module Miteru
   class Orchestrator < Service
     def call
-      Miteru.logger.info("#{non_cached_websites.length} websites loaded in total.") if verbose?
+      logger.info("#{non_cached_websites.length} websites loaded in total.") if verbose?
 
-      if Miteru.sidekiq?
-        non_cached_websites.each do |website|
-          Jobs::CrawleJob.perform_async(website.url, website.source)
-          Miteru.logger.info("Website:#{website.truncated_url} crawler job queued.") if verbose?
-        end
+      if sidekiq?
+        sidekiq_call
       else
-        Miteru.logger.info("Use #{threads} thread(s).") if verbose?
-        Parallel.each(non_cached_websites, in_threads: threads) do |website|
-          Miteru.logger.info("Website:#{website.truncated_url} crawling started.") if verbose?
+        parallel_call
+      end
+    end
 
-          result = Crawler.result(website)
-          if result.success?
-            Miteru.logger.info("Crawler:#{website.truncated_url} succeeded.")
-          else
-            Miteru.logger.info("Crawler:#{website.truncated_url} failed - #{result.failure}.")
-          end
+    def sidekiq_call
+      non_cached_websites.each do |website|
+        Jobs::CrawleJob.perform_async(website.url, website.source)
+        logger.info("Website:#{website.truncated_url} crawler job queued.") if verbose?
+      end
+    end
+
+    def parallel_call
+      logger.info("Use #{threads} thread(s).") if verbose?
+      Parallel.each(non_cached_websites, in_threads: threads) do |website|
+        logger.info("Website:#{website.truncated_url} crawling started.") if verbose?
+        result = Crawler.result(website)
+        if result.success?
+          logger.info("Crawler:#{website.truncated_url} succeeded.")
+        else
+          logger.info("Crawler:#{website.truncated_url} failed - #{result.failure}.")
         end
       end
     end
@@ -34,44 +41,33 @@ module Miteru
           result = feed.result
           if result.success?
             websites = result.value!
-            Miteru.logger.info("Feed:#{feed.source} has #{websites.length} websites.") if verbose?
+            logger.info("Feed:#{feed.source} has #{websites.length} websites.") if verbose?
             out << websites
           else
-            Miteru.logger.warn("Feed:#{feed.source} failed - #{result.failure}")
+            logger.warn("Feed:#{feed.source} failed - #{result.failure}")
           end
         end
       end.flatten.uniq(&:url)
     end
 
+    #
+    # @return [Array<Miteru::Website>]
+    #
     def non_cached_websites
-      return websites unless cache?
-
-      websites.reject { |website| cache.cached?(website.url) }
+      @non_cached_websites ||= [].tap do |out|
+        out << if cache?
+          websites.reject { |website| cache.cached?(website.url) }
+        else
+          websites
+        end
+      end.flatten.uniq(&:url)
     end
 
     #
     # @return [Array<Miteru::Feeds::Base>]
     #
     def feeds
-      Miteru.feeds.map(&:new)
-    end
-
-    private
-
-    def cache?
-      Miteru.cache?
-    end
-
-    def cache
-      Miteru.cache
-    end
-
-    def threads
-      Miteru.config.threads
-    end
-
-    def verbose?
-      Miteru.config.verbose
+      @feeds ||= Miteru.feeds.map(&:new)
     end
   end
 end

data/lib/miteru/service.rb

@@ -24,5 +24,49 @@ module Miteru
         new.result(...)
       end
     end
+
+    private
+
+    def auto_download?
+      Miteru.config.auto_download
+    end
+
+    #
+    # @return [SemanticLogger]
+    #
+    def logger
+      Miteru.logger
+    end
+
+    def cache?
+      Miteru.cache?
+    end
+
+    def sidekiq?
+      Miteru.sidekiq?
+    end
+
+    #
+    # @return [Miteru::Cache]
+    #
+    def cache
+      Miteru.cache
+    end
+
+    def threads
+      Miteru.config.threads
+    end
+
+    def verbose?
+      Miteru.config.verbose
+    end
+
+    def cache_prefix
+      Miteru.config.cache_prefix
+    end
+
+    def cache_ex
+      Miteru.config.cache_ex
+    end
   end
 end

data/lib/miteru/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Miteru
-  VERSION = "2.2.0"
+  VERSION = "2.2.1"
 end

data/lib/miteru/website.rb

@@ -4,6 +4,8 @@ require "oga"
 
 module Miteru
   class Website < Service
+    include Concerns::UrlTruncatable
+
     # @return [String]
     attr_reader :url
 
@@ -41,17 +43,6 @@ module Miteru
       (href_links + possible_file_links).compact.uniq
     end
 
-    #
-    # @return [String]
-    #
-    def truncated_url
-      url.truncate(64)
-    end
-
-    def defanged_truncated_url
-      truncated_url.to_s.gsub(".", "[.]")
-    end
-
     def info
       "#{defanged_truncated_url} has #{kits.length} kit(s) (Source: #{source})"
     end

data/lib/miteru.rb

@@ -31,6 +31,7 @@ require "miteru/errors"
 # Concerns
 require "miteru/concerns/database_connectable"
 require "miteru/concerns/error_unwrappable"
+require "miteru/concerns/url_truncatable"
 
 # Core classes
 require "miteru/service"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: miteru
 version: !ruby/object:Gem::Version
-  version: 2.2.0
+  version: 2.2.1
 platform: ruby
 authors:
 - Manabu Niseki
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-02-08 00:00:00.000000000 Z
+date: 2024-02-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -590,6 +590,7 @@ files:
 - lib/miteru/commands/web.rb
 - lib/miteru/concerns/database_connectable.rb
 - lib/miteru/concerns/error_unwrappable.rb
+- lib/miteru/concerns/url_truncatable.rb
 - lib/miteru/config.rb
 - lib/miteru/crawler.rb
 - lib/miteru/database.rb