mass-client 1.0.20 → 1.0.21

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1ee4e0f04af40b7fb0a7e26fedb26ec7fa3d76602baec49add355f620940c9b0
-  data.tar.gz: 73fa2542b417954bedc8352420b131d81dbaeea82dacca0e43dffd96002728b4
+  metadata.gz: a9aa67c61c00611402f922ed7988f67bba40ccaab9db879440512fa1b2288bc4
+  data.tar.gz: 21f416174b865c81f5122d2242aea75b55acaae7224d5bc1bc7bef13e64b1f19
 SHA512:
-  metadata.gz: 458fd9aa4929a889264c7e90b69b856e8272ee9ed26106243e64dca225ece4ec847a91b0d536036c10d8f0a43525adf6922f88587e35de8963f40fc11d2b35f7
-  data.tar.gz: a20f0ce501ed952360e071a696fc39e0c060c18a5d8d89af7cea557709f0296db2700f79152e63430766e870b1ba2e66d811bc9c116635e7cccf82a340e460c6
+  metadata.gz: fdaca0893c886f465fbc617d4b301b5fca7552354bd4daf8d96731de5a4db7b9b0eb9a78500e1a842b21f6d3d6f64f8b12be963db44c0048d4c1f426ee0eed14
+  data.tar.gz: 9dc92ae9b578167df3835499c56bf44de27b94c78356f2972aacb8390f5600f5c5bcac0dbade36dc49e3105abe11653280cb9ba665dab861e7a4a4acd87501c9

data/lib/first-line/profile_rpa.rb

@@ -175,7 +175,23 @@ module Mass
             automaticDownloads.find_element(:id => status.to_s).click()
         end # def automatic_downloads
 
-
+        # Waits for the presence of a file in specified file paths within a given timeout period.
+        # This function checks each file path in the provided array and returns the path of the
+        # first file it finds to exist. It raises an exception if none of the files are found
+        # within the specified timeout.
+        #
+        # @param file_paths [Array<String>] An array of file paths to check for the file's existence.
+        # @param timeout [Integer] The maximum number of seconds to wait for the file to appear.
+        #                          Default is 30 seconds.
+        #
+        # @return [String] The path of the first file found in `file_paths`.
+        # @raise [RuntimeError] If no file is found within the specified timeout, raises an error with a message 
+        #                       listing the paths that were checked and the timeout duration.
+        #
+        # @example
+        #   wait_for_file(['/path/to/file1', '/path/to/file2'], timeout: 60)
+        #   # => "/path/to/file1" (if the file appears within 60 seconds)
+        #
         def wait_for_file(file_paths, timeout = 30)
             Timeout.timeout(timeout) do
               loop do
@@ -189,6 +205,45 @@ module Mass
             raise "Downloaded file not found in paths #{file_paths.join(',')} after #{timeout} seconds."
         end
 
+
+        # Retrieves the URL of a file stored in Dropbox with a timeout mechanism.
+        # This function continuously attempts to retrieve the file URL until it 
+        # becomes available in Dropbox, up to a specified maximum wait time.
+        #
+        # @param path [String] The path to the file in Dropbox.
+        # @param max_wait [Integer] Maximum time to wait (in seconds) for the file to 
+        #                            appear in Dropbox. Default is 30 seconds.
+        # @param interval [Integer] Time interval (in seconds) between each retry 
+        #                            attempt. Default is 2 seconds.
+        #
+        # @return [String] The URL of the file in Dropbox, with the `&dl=1` parameter 
+        #                  replaced by `&dl=0`.
+        # @raise [RuntimeError] If the file is not available within the specified 
+        #                       maximum wait time, raises an error with a timeout message.
+        #
+        # @example
+        #   wait_for_dropbox_url('/path/to/file', max_wait: 60, interval: 3)
+        #   # => "https://www.dropbox.com/s/yourfile?dl=0"
+        #
+        def wait_for_dropbox_url(path, max_wait: 30, interval: 2)
+            start_time = Time.now
+          
+            loop do
+                begin
+                    # Try to get the file URL
+                    return BlackStack::DropBox.get_file_url(path).gsub('&dl=1', '&dl=0')
+                rescue => e
+                    # Check if the timeout has been exceeded
+                    if Time.now - start_time > max_wait
+                    raise "Timeout exceeded while waiting for file to appear in Dropbox: #{e.message}"
+                    end
+            
+                    # Wait for a short interval before retrying
+                    sleep(interval)
+                end
+            end
+        end
+
         # download image from Selenium using JavaScript and upload to Dropbox 
         # return the URL of the screenshot
         # 
@@ -274,8 +329,12 @@ module Mass
             BlackStack::DropBox.dropbox_upload_file(tmp_path, path)
             # Delete the local file
             File.delete(tmp_path)
+
             # Return the URL of the file in Dropbox
-            BlackStack::DropBox.get_file_url(path).gsub('&dl=1', '&dl=0')
+            # 
+            # Add a timeout to wait the file is present in the cloud.
+            # Reference: https://github.com/MassProspecting/docs/issues/320
+            self.wait_for_dropbox_url(path).gsub('&dl=1', '&dl=0')
         end
                 
         # download image from Selenium using JavaScript and upload to Dropbox 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mass-client
 version: !ruby/object:Gem::Version
-  version: 1.0.20
+  version: 1.0.21
 platform: ruby
 authors:
 - Leandro Daniel Sardi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-10-21 00:00:00.000000000 Z
+date: 2024-10-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: timeout