mass-client 1.0.17 → 1.0.18

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 15fc20e0e46b85a66c45cbabbf3862fd1db6bff5fc254ec8b94414a0c5f086da
-  data.tar.gz: 6cd0e9274b472a4e64a07d3f9f0d846b5499b2bbb3384e3e45a322d3a1210202
+  metadata.gz: 226f48f9298fc4e264e311c3544c19935e985a9d5bab26b11740ceba9770fec7
+  data.tar.gz: 0a5943bb27e3a9b58e212ab4929dec83462712fb9614edb2019458065d53b22c
 SHA512:
-  metadata.gz: 663c292587112ca68d8a29643d7348d4510cf7028d1564d7f24c9259c518e1e25de0dfffc8bfef889f90f2f8e7a9d4625e80e4d95e8344ce81db83387c84cb66
-  data.tar.gz: 176098ed2b937fc66af21da5e0d2d585a1960c209ab0eaa2be6579aa2a05fe866f532b36011c2bcb01043e51af202e1026235e870116f519191ecc04cd954e9b
+  metadata.gz: a91705cfe04215ec98a8d2418dc87fad48fb724fc0e60a6a80d52de28f2b6631a3e88d5047673fa583d0eba9630c2fe5e917ee0f382d748f1182f79f8020d85c
+  data.tar.gz: cf042ef9442239c66aecf000392b315d804c662d0f240e79125ef90109471f28da072da178ff0069cc9ee1602f5d3d2b0c6e97b63f1eca3be4e3b644d2205abe

data/lib/first-line/profile_rpa.rb

@@ -175,96 +175,109 @@ module Mass
             automaticDownloads.find_element(:id => status.to_s).click()
         end # def automatic_downloads
 
+
+        def wait_for_file(file_paths, timeout = 30)
+            Timeout.timeout(timeout) do
+              loop do
+                file_paths.each do |path|
+                  return path if File.exist?(path)
+                end
+                sleep(1)
+              end
+            end
+        rescue Timeout::Error
+            raise "Downloaded file not found in paths #{file_paths.join(',')} after #{timeout} seconds."
+        end
+
         # download image from Selenium using JavaScript and upload to Dropbox 
         # return the URL of the screenshot
         # 
         # Parameters:
         # - url: Internet address of the image to download from the website and upload to dropbox.
         # - dropbox_folder: Dropbox folder name to store the image.
-        #
-        def download_image_0(url, dropbox_folder=nil)
+        #                
+        def download_image_0(url, dropbox_folder = nil)
             raise "Either dropbox_folder parameter or self.desc['id_account'] are required." if dropbox_folder.nil? && self.desc['id_account'].nil?
             dropbox_folder = self.desc['id_account'] if dropbox_folder.nil?
-            # parameters
+            
+            # Parameters
             id = SecureRandom.uuid
-            filename = "#{id}.png"
-            tmp_path = ["#{Mass.download_path}/#{filename}"] if Mass.download_path.is_a?(String)
-            tmp_paths = Mass.download_path.map { |s| "#{s}/#{filename}" } if Mass.download_path.is_a?(Array)
-
-            # excute JavaScript function downloadImage with the parameter src
-#binding.pry
-            # this function is to download images from the same browser, using the proxy of the browser.
-            # never download images from the server, because sites can track the IP.
-=begin
-            js0 = "
-                async function downloadImage(imageSrc, filename) {
-                    const image = await fetch(imageSrc)
-                    const imageBlog = await image.blob()
-                    const imageURL = URL.createObjectURL(imageBlog)
-                  
-                    const link = document.createElement('a')
-                    link.href = imageURL
-                    link.download = filename
-                    document.body.appendChild(link)
-                    link.click()
-                    document.body.removeChild(link)
-                }
-                downloadImage('#{url}', '#{filename}')
-            "
-=end
-            # Promise Handling: By returning a Promise in the JavaScript code, Selenium will wait for the asynchronous operation to complete.
-            # Reference: https://github.com/MassProspecting/docs/issues/253
+            
+            # JavaScript to get base64 image data
             js0 = "
-            function downloadImage(imageSrc, filename) {
+                function getImageBase64(imageSrc) {
                 return new Promise(async (resolve, reject) => {
                     try {
-                        const image = await fetch(imageSrc);
-                        const imageBlob = await image.blob();
-                        const imageURL = URL.createObjectURL(imageBlob);
-        
-                        const link = document.createElement('a');
-                        link.href = imageURL;
-                        link.download = filename;
-                        document.body.appendChild(link);
-                        link.click();
-                        document.body.removeChild(link);
-        
-                        resolve('Download initiated');
-                    } catch (error) {
+                    const response = await fetch(imageSrc);
+                    const blob = await response.blob();
+                    const reader = new FileReader();
+                    reader.onloadend = function() {
+                        resolve(reader.result);
+                    };
+                    reader.onerror = function(error) {
                         reject(error);
+                    };
+                    reader.readAsDataURL(blob);
+                    } catch (error) {
+                    reject(error);
                     }
                 });
-            }
-            return downloadImage('#{url}', '#{filename}');
+                }
+                return getImageBase64('#{url}');
             "
-#binding.pry
-            driver.execute_script(js0)
-sleep(10)
-            # code
-            year = Time.now.year.to_s.rjust(4,'0')
-            month = Time.now.month.to_s.rjust(2,'0')
+            
+            # Execute JavaScript and get base64 image data
+            base64_image = driver.execute_script(js0)
+            raise "Failed to retrieve image data from URL: #{url}" if base64_image.nil?
+            
+            # Extract MIME type and base64 data
+            mime_type_match = base64_image.match(/^data:image\/([a-zA-Z0-9.+-]+);base64,/)
+            if mime_type_match
+                mime_subtype = mime_type_match[1] # e.g., 'png', 'jpeg', 'gif'
+                # Map common MIME subtypes to file extensions
+                extension = case mime_subtype
+                            when 'jpeg' then 'jpg'
+                            else mime_subtype
+                            end
+                # Remove the data URL prefix
+                image_data = base64_image.sub(/^data:image\/[a-zA-Z0-9.+-]+;base64,/, '')
+            else
+                raise "Unsupported or invalid image data format."
+            end
+            
+            # Update filename and paths
+            filename = "#{id}.#{extension}"
+            tmp_paths = if Mass.download_path.is_a?(String)
+                            ["#{Mass.download_path}/#{filename}"]
+                        elsif Mass.download_path.is_a?(Array)
+                            Mass.download_path.map { |s| "#{s}/#{filename}" }
+                        else
+                            raise "Invalid Mass.download_path configuration."
+                        end
+            
+            # Save the image to the first available path
+            tmp_path = tmp_paths.find { |path| File.writable?(File.dirname(path)) }
+            raise "No writable path found in #{tmp_paths.join(', ')}." if tmp_path.nil?
+            
+            File.open(tmp_path, 'wb') do |file|
+                file.write(Base64.decode64(image_data))
+            end
+            
+            # Proceed with Dropbox operations
+            year = Time.now.year.to_s.rjust(4, '0')
+            month = Time.now.month.to_s.rjust(2, '0')
             folder = "/massprospecting.rpa/#{dropbox_folder}.#{year}.#{month}"
             path = "#{folder}/#{filename}"
             BlackStack::DropBox.dropbox_create_folder(folder)
-
-            # find the tmp_path that exists
-            tmp_path = nil
-            tmp_paths.each { |s|
-                if File.exist?(s)
-                    tmp_path = s
-                    break
-                end
-            }
-            raise "Downloaded file #{filename} not found in paths #{tmp_paths.join(',')}." if tmp_path.nil?
-
-            # upload the file to dropbox
+            
+            # Upload the file to Dropbox
             BlackStack::DropBox.dropbox_upload_file(tmp_path, path)
-            # delete the file
+            # Delete the local file
             File.delete(tmp_path)
-            # return the URL of the file in dropbox
+            # Return the URL of the file in Dropbox
             BlackStack::DropBox.get_file_url(path).gsub('&dl=1', '&dl=0')
-        end # def download_image_0
-
+        end
+                
         # download image from Selenium using JavaScript and upload to Dropbox 
         # return the URL of the screenshot
         # 

data/lib/mass-client.rb

@@ -5,6 +5,8 @@ require 'blackstack-core'
 require 'simple_cloud_logging'
 require 'simple_command_line_parser'
 require 'colorize'
+require 'timeout'
+require 'base64'
 
 # mass client configuration
 module Mass

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mass-client
 version: !ruby/object:Gem::Version
-  version: 1.0.17
+  version: 1.0.18
 platform: ruby
 authors:
 - Leandro Daniel Sardi
@@ -11,23 +11,45 @@ cert_chain: []
 date: 2024-09-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: uri
+  name: timeout
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.11.2
-    - - ">="
+        version: 0.4.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.11.2
+        version: 0.4.1
+- !ruby/object:Gem::Dependency
+  name: base64
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.1
+- !ruby/object:Gem::Dependency
+  name: uri
+  requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.11.2
-    - - ">="
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
       - !ruby/object:Gem::Version
         version: 0.11.2
 - !ruby/object:Gem::Dependency
@@ -37,9 +59,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.2.0
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 0.2.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -47,9 +66,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.2.0
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 0.2.0
 - !ruby/object:Gem::Dependency
   name: json
   requirement: !ruby/object:Gem::Requirement
@@ -57,9 +73,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 2.6.3
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 2.6.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -67,9 +80,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 2.6.3
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 2.6.3
 - !ruby/object:Gem::Dependency
   name: blackstack-core
   requirement: !ruby/object:Gem::Requirement
@@ -77,9 +87,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.2.20
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 1.2.20
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -87,9 +94,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.2.20
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 1.2.20
 - !ruby/object:Gem::Dependency
   name: colorize
   requirement: !ruby/object:Gem::Requirement
@@ -97,9 +101,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.8.1
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 0.8.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -107,9 +108,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.8.1
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 0.8.1
 - !ruby/object:Gem::Dependency
   name: simple_cloud_logging
   requirement: !ruby/object:Gem::Requirement
@@ -117,9 +115,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.2.6
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 1.2.6
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -127,9 +122,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.2.6
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 1.2.6
 - !ruby/object:Gem::Dependency
   name: simple_command_line_parser
   requirement: !ruby/object:Gem::Requirement
@@ -137,9 +129,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.1.2
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 1.1.2
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -147,9 +136,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.1.2
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 1.1.2
 description: Ruby library for MassProspecting API.
 email: leandro@massprospecting.com
 executables: []