pulse-downloader 0.1.22 → 0.1.27

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 00f53bf497e8c65d6211956814fa94613857b23bc0b1e3f35c24d69998645941
-  data.tar.gz: 77447e1a3b1df593d5c836074c41e6caa57fef28f8926a5f7e1859fb1b58b39d
+  metadata.gz: ed736b97e09aaec1104ac4e817b981c2ebb88fbac583abe0ebf44e721143b087
+  data.tar.gz: a81646945f24e428738e8a514e671f2f14682f5824faee1e63e07bd498fe519c
 SHA512:
-  metadata.gz: 627db0d3d117a199943ba7ade0c27af871f4934e6879bb788fcade5946b95bfa0378d237fb5eee0f68685cf211b042a3b35986441466b0d6811875fe7103304f
-  data.tar.gz: 1f7c99a75487b6c615cb94bd8def2ed1f5316036339318244329648be8800dc5ab867a8022e956e1a46ac0e0cb58f4502030453c7efadd7131d7edbb69a5a538
+  metadata.gz: b684940cab23055ec977672dd8472fcb38ebf8e70e59d18f2a76270a3234f5873c7a6bee47877760c54733af2c073bbaa7a05eec33266a33df316482baa6d836
+  data.tar.gz: 9b1737bc6607585c6fb6081d9c1b3031bd53ffccc5cce72a0650b163792742dde2b7dffa6ed663775a7cd1cad50f1fb964c07bc555ceee7a2ca66637af25b41a

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    pulse-downloader (0.1.11)
+    pulse-downloader (0.1.26)
       active_attr (~> 0.15)
       httparty (~> 0.18)
       nokogiri (~> 1.11)
@@ -69,7 +69,7 @@ GEM
       ruby-progressbar
     mocha (1.11.2)
     multi_xml (0.6.0)
-    nokogiri (1.11.3-x86_64-linux)
+    nokogiri (1.11.5-x86_64-linux)
       racc (~> 1.4)
     options (2.3.2)
     progress_bar (1.3.3)

data/README.md

@@ -33,7 +33,8 @@ client = Pulse::Downloader::Client.new(
   drop_exitsing_files_in_path: false,
   save_and_dont_return: true,
   report_time: false,
-  progress_bar: true
+  progress_bar: true,
+  scrape_images: true
 )
 
 client.call!

data/lib/pulse/downloader/client.rb

@@ -13,6 +13,7 @@ module Pulse
         :save_path,
         :read_from_save_path,
         :verify_ssl,
+        :headers,
         :drop_exitsing_files_in_path,
         :save_and_dont_return,
         :report_time,
@@ -36,6 +37,7 @@ module Pulse
         save_path: '',
         read_from_save_path: false,
         verify_ssl: true,
+        headers: nil,
         drop_exitsing_files_in_path: false,
         save_and_dont_return: true,
         report_time: false,
@@ -48,6 +50,7 @@ module Pulse
         @save_path = save_path
         @read_from_save_path = read_from_save_path
         @verify_ssl = verify_ssl
+        @headers = headers
         @drop_exitsing_files_in_path = drop_exitsing_files_in_path
         @save_and_dont_return = save_and_dont_return
         @report_time = report_time

data/lib/pulse/downloader/file_downloader.rb

@@ -8,7 +8,11 @@ module Pulse
 
         @start_time = get_micro_second_time
 
-        file_data = HTTParty.get(escape(compute_file_link(file_path)), verify: verify_ssl)
+        file_data = HTTParty.get(
+          escape(compute_file_link(file_path)),
+          verify: verify_ssl,
+          headers: headers
+        )
 
         @end_time = get_micro_second_time
 

data/lib/pulse/downloader/version.rb

@@ -1,5 +1,5 @@
 module Pulse
   module Downloader
-    VERSION = "0.1.22"
+    VERSION = "0.1.27"
   end
 end

data/lib/pulse/downloader/web_page_parser.rb

@@ -1,7 +1,7 @@
 module Pulse
   module Downloader
     module WebPageParser
-      def fetch_file_paths
+      def fetch_file_paths(custom_path_root=nil)
         @start_time = get_micro_second_time
 
         response = HTTParty.get(url, verify: verify_ssl)
@@ -12,29 +12,47 @@ module Pulse
           print_time
         end
 
-        extract_file_urls(response)
+        if file_type.is_a?(Array)
+          file_type.flat_map do |type|
+            extract_file_urls(response, custom_path_root, type)
+          end
+        else
+          extract_file_urls(response, custom_path_root, file_type)
+        end
       end
 
       private
 
-      def extract_file_urls(response)
+      def extract_file_urls(response, custom_path_root, type)
         return [] if response.body.nil? || response.body.empty?
-        (
-          extract_download_links(response) + extract_embedded_images(response)
+        remove_base64(
+          extract_all_urls(response, custom_path_root, type) +
+            extract_download_links(response, custom_path_root, type) +
+            extract_embedded_images(response, custom_path_root, type)
         ).uniq
       end
 
-      def extract_download_links(response)
+      def extract_all_urls(response, custom_path_root, type)
+        parse_html(response.body)
+          .to_s
+          .split(/\s+/)
+          .find_all { |u| u =~ /^https?:/ }
+          .compact
+          .select { |link| (link.include? type || link.include?(custom_path_root)) }
+          .map { |link| add_base_url(link) }
+      end
+
+      def extract_download_links(response, custom_path_root, type)
         parse_html(response.body)
           .css('a')
           .to_a
           .map { |link| link['href'] }
           .compact
-          .select { |link| link.include? file_type }
+          .select { |link| (link.include? type || link.include?(custom_path_root)) }
           .map { |link| add_base_url(link) }
       end
 
-      def extract_embedded_images(response)
+      def extract_embedded_images(response, custom_path_root, type)
         return [] unless scrape_images
 
         parse_html(response.body)
@@ -42,10 +60,16 @@ module Pulse
           .to_a
           .map { |e| e["src"] }
           .compact
-          .select { |link| link.include? file_type }
+          .select { |link| (link.include? type || link.include?(custom_path_root)) }
           .map { |link| add_base_url(link) }
       end
 
+      def remove_base64(urls)
+        urls.reject do |url|
+          url.include?(':image/') || url.include?('base64')
+        end
+      end
+
       def parse_html(raw_html)
         Nokogiri::HTML(raw_html)
       end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pulse-downloader
 version: !ruby/object:Gem::Version
-  version: 0.1.22
+  version: 0.1.27
 platform: ruby
 authors:
 - trex22
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2021-05-10 00:00:00.000000000 Z
+date: 2021-05-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty