pulse-downloader 0.1.33 → 0.1.37

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ea8f7f1cba1f575af4fb82c8d83d7108decbdc8eabf38be38693b2f94c5d4b95
-  data.tar.gz: 69deb413bdce72404ca9a76e5d5d01014214559c71a1aad421c8b3e32c400f52
+  metadata.gz: b0db887eeb1a5bdd42489b608d9ed59322af9532aa904a5c8ad6d0181aab5277
+  data.tar.gz: 1e0fd11d6462bd9c0f2c731c3e6fc1028a78577a1874d8959b1fd3b8af21123a
 SHA512:
-  metadata.gz: 1f1633832d05cf665d116cd9180c66b02d7c0a0c5dbf77db03d758fa9f0f051982761125cbdfc4914e3dd53b5ee6c91fc9b74fe16cd9c992098e7f72df78ac52
-  data.tar.gz: 4f571acd0d6019506a9cb58b5e86bcb8c79cbc856406342a1325864ee6f771b7c3bdc09e65d8bba6ee79d1d77b59f4a37b5e771e6b8b51d959350d32142bf378
+  metadata.gz: 189b1b83effd9aa0a71c4cf41dfcdbdc97d25dee9f5ee14a0a9d7e2a8de4652147a7df31575dfea6379ace00e45d21ada15a891cf1277c178e47622e0d573bd6
+  data.tar.gz: 35193c286218971b45a8a75b716c5f503ac239f7f7efb5f7cca7593827585bdb2194eae6879590bda4bb971ac4080677db560296d62fdf3c405b2635c4470411

data/README.md

@@ -29,6 +29,7 @@ client = Pulse::Downloader::Client.new(
   save_data: true,
   save_path: '',
   read_from_save_path: false,
+  traverse_folders: false,
   verify_ssl: true,
   drop_exitsing_files_in_path: false,
   save_and_dont_return: true,
@@ -38,6 +39,8 @@ client = Pulse::Downloader::Client.new(
 )
 
 client.call!
+
+client.file_paths # Will give you the list of file paths
 ```
 
 ## Development

data/lib/pulse/downloader/client.rb

@@ -12,6 +12,7 @@ module Pulse
         :save_data,
         :save_path,
         :read_from_save_path,
+        :traverse_folders,
         :verify_ssl,
         :headers,
         :drop_exitsing_files_in_path,
@@ -21,7 +22,8 @@ module Pulse
         :end_time,
         :progress_bar,
         :base_url,
-        :file_paths
+        :file_paths,
+        :folder_urls
 
       # Does not continue downloads-
       # Will only save once the file has been downloaded in memory
@@ -29,7 +31,6 @@ module Pulse
       # TODO: Validation
       # TODO: Retry
       # TODO: DNS
-      # TODO: Multiple filetypes
       # TODO: lib/pulse/downloader/file_downloader.rb:13: warning: URI.escape is obsolete
       def initialize(url:,
         file_type:,
@@ -37,6 +38,7 @@ module Pulse
         save_data: false,
         save_path: '',
         read_from_save_path: false,
+        traverse_folders: false,
         verify_ssl: true,
         headers: nil,
         drop_exitsing_files_in_path: false,
@@ -50,6 +52,7 @@ module Pulse
         @save_data = save_data
         @save_path = save_path
         @read_from_save_path = read_from_save_path
+        @traverse_folders = traverse_folders
         @verify_ssl = verify_ssl
         @headers = headers
         @drop_exitsing_files_in_path = drop_exitsing_files_in_path
@@ -58,6 +61,7 @@ module Pulse
         @progress_bar = progress_bar
 
         @base_url = get_base_url
+        @folder_urls = []
       end
 
       def call!

data/lib/pulse/downloader/version.rb

@@ -1,5 +1,5 @@
 module Pulse
   module Downloader
-    VERSION = "0.1.33"
+    VERSION = "0.1.37"
   end
 end

data/lib/pulse/downloader/web_page_parser.rb

@@ -2,9 +2,38 @@ module Pulse
   module Downloader
     module WebPageParser
       def fetch_file_paths(custom_path_root=nil)
+        if traverse_folders
+          fetch_folders(url, custom_path_root).each do |folder_url|
+            fetch_and_parse_response(folder_url, custom_path_root)
+          end
+        else
+          fetch_and_parse_response(url, custom_path_root)
+        end
+      end
+
+      private
+
+      def fetch_folders(folder_url, custom_path_root)
+        current_paths = extract_hrefs(get_response(folder_url), custom_path_root)
+        return unless current_paths.compact.size > 0
+
+        @folder_urls = folder_urls.union(current_paths).uniq.compact
+
+        current_paths.each do |path|
+          fetch_folders(path, custom_path_root)
+        end
+
+        folder_urls
+      end
+
+      def fetch_and_parse_response(folder_url, custom_path_root)
+        parse_response(get_response(folder_url), custom_path_root, file_type)
+      end
+
+      def get_response(folder_url)
         @start_time = get_micro_second_time
 
-        response = HTTParty.get(url, verify: verify_ssl, headers: headers)
+        response = HTTParty.get(folder_url, verify: verify_ssl, headers: headers)
 
         @end_time = get_micro_second_time
 
@@ -12,6 +41,10 @@ module Pulse
           print_time
         end
 
+        response
+      end
+
+      def parse_response(response, custom_path_root, file_type)
         if file_type.is_a?(Array)
           file_type.flat_map do |type|
             extract_file_urls(response, custom_path_root, type)
@@ -21,8 +54,6 @@ module Pulse
         end
       end
 
-      private
-
       def extract_file_urls(response, custom_path_root, type)
         return [] if response.body.nil? || response.body.empty?
 
@@ -33,6 +64,14 @@ module Pulse
         ).uniq
       end
 
+      def extract_hrefs(response, custom_path_root)
+        parse_html(response.body)
+          .css('a')
+          .map { |link| link['href'] }
+          .reject { |link| link == "../" }
+          .map { |link| add_base_url(link, custom_path_root) }
+      end
+
       def extract_all_urls(response, custom_path_root, type)
         parse_html(response.body)
           .to_s

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pulse-downloader
 version: !ruby/object:Gem::Version
-  version: 0.1.33
+  version: 0.1.37
 platform: ruby
 authors:
 - trex22