Added filename to download path

Author: rristov60

puppeteer/page/download.html

@@ -10,6 +10,8 @@
       delay: { value: 0 },
       downloadPath: { value: "", required: false },
       downloadPathtype: { value: "str" },
+      fileName: { value: "", required: false },
+      fileNametype: { value: "str" },
       name: { value: "" },
     },
     inputs: 1,
@@ -36,6 +38,11 @@
         types: ["str", "msg", "flow", "global"],
         typeField: "#node-input-downloadPathtype",
       });
+      $("#node-input-fileName").typedInput({
+        type: "str",
+        types: ["str", "msg", "flow", "global"],
+        typeField: "#node-input-fileNametype",
+      });
     },
   });
 </script>
@@ -69,26 +76,38 @@
     <input id="node-input-downloadPath"></input>
     <input type="hidden" id="node-input-downloadPathtype"></input>
   </div>
+  <div class="form-row">
+    <label for="node-input-fileName"><i class="fa fa-file-o"></i> File name</label>
+    <input id="node-input-fileName"></input>
+    <input type="hidden" id="node-input-fileNametype"></input>
+  </div>
   <div class="form-row">
       <label for="node-input-name"><i class="fa fa-tag"></i> Name</label>
       <input type="text" id="node-input-name" style="width:70%;" placeholder="Name" />
   </div>
+  <div class="form-tips">
+    <span data-i18n="[html]tip1">This node downloads a file by clicking on a specified selector. Essetially that can be done with the click node as well, but here you can specify download location and also if this node is followed by rename node with specified download path and filename, it can change the filename directly, but in order to get this to work, in the rename node the old file path needs to be set to <code>msg.old</code> and the new file path needs to be set to <code>msg.new</code>. The output of this node is a <b>msg</b> which contains <code>msg.puppeteer</code>, <code>msg.new & msg.old</code> (<b><u style="text-decoration-color: #fe6d73">only when download path and file path are specified</u></b>) and <code>msg.error</code> when an error ocurred.
+  </div>
 </script>
 
 <script type="text/x-red" data-help-name="puppeteer-page-download">
-      <p>Click on Chrome</p>
+      <p>Dowload a file</p>
       <h3>Inputs</h3>
       <dl class="message-properties">
           <dt>Selector<span class="property-type">string</span></dt>
           <dd>A <a href="#interface-selector" title="Selector">selector</a> to search for element to click. If there are multiple elements satisfying the selector, the first will be clicked.</dd>
           <dt>Button<span class="property-type">string</span></dt>
           <dd>Slows down Puppeteer operations by the specified amount of milliseconds. Useful so that you can see what is going on.&lt;"left"|"right"|"middle"&gt; Defaults to <code>left</code>.</dd>
           <dt>Click Count<span class="property-type">number</span></dt>
-          <dd>Whether to run browser in <a href="https://developers.google.com/web/updates/2017/04/headless-chrome" rel="nofollow">headless mode</a>. Defaults to <code>false</code>, it will show Chrome when <code>devtools</code> option is <code>true</code>.defaults to 1. See <a href="https://developer.mozilla.org/en-US/docs/Web/API/UIEvent/detail" title="UIEvent.detail" rel="nofollow">UIEvent.detail</a>.</dd>
           <dt>Delay<span class="property-type">number</span></dt>
-          <dd>Specify custom debugging port. Pass <code>0</code> to discover a random port. Defaults to <code>0</code>.Time to wait between <code>mousedown</code> and <code>mouseup</code> in milliseconds. Defaults to 0.</dd>
+          <dt>Download path<span class="property-type">string</span></dt>
+          <dd>Specify custom download path. Leave the field blank for default download path. Defaults to user's default browser download path.</dd>
+          <dt>File name<span class="property-type">string</span></dt>
+          <dd>Specify custom file name. Leave the field blank for default file name. Please note that in order for this to work this node needs to be followed by the rename node. Defaults to default file name.</dd>
       </dl>
       <h3>Details</h3>
           <p>This method fetches an element with <code>selector</code>, scrolls it into view if needed, and then uses <code>page.mouse</code> to click in the center of the element.
-  If there's no element matching <code>selector</code>, the method throws an error.</p>
+          If there's no element matching <code>selector</code>, the method throws an error. The main difference between this and click node is that with this node you can specify
+          download path and file name. But, in order the custom filename to work, this node must have download path and file name specified and be followed by a rename node where
+          the old file path should be set to <code>msg.old</code> and the new file path should be set to <code>msg.new</code></p>
 </script>

puppeteer/page/download.js

@@ -39,6 +39,22 @@ module.exports = function (RED) {
           );
         }
 
+        // Parsing the fileName from string input or from msg object
+        let fileName =
+          nodeConfig.fileNametype == "msg"
+            ? eval(nodeConfig.fileNametype + "." + nodeConfig.fileName)
+            : nodeConfig.fileName;
+        // If the type of fileName is set to flow or global, it needs to be parsed differently
+        if (
+          nodeConfig.fileNametype == "flow" ||
+          nodeConfig.fileNametype == "global"
+        ) {
+          // Parsing the fileName
+          fileName = this.context()[nodeConfig.fileNametype].get(
+            nodeConfig.fileName
+          );
+        }
+
         // If download path is defined
         if (downloadPath && downloadPath != "") {
           // Enable requests interception
@@ -53,7 +69,37 @@ module.exports = function (RED) {
               interceptedRequest.continue(); // Continue the requets as usual
             }
           });
+
+          if (fileName && fileName != "") {
+            msg.puppeteer.page.on("response", async (response) => {
+              // node.warn(response);
+              const url = response.url();
+              const headers = response.headers();
+              const regex = /filename\*?=[^'']*''([^']*)/;
+
+              if (headers["content-disposition"]) {
+                const match = regex.exec(headers["content-disposition"]);
+                node.warn(headers["content-disposition"]);
+                if (match && match[1]) {
+                  const fs = require("fs");
+                  const downloadedFileName = match[1];
+                  const extensionRegex = /\.[^.]*$/;
+                  const extension = extensionRegex.exec(downloadedFileName);
+                  if(extension) {
+                    msg.old = `${downloadPath}/${downloadedFileName}`;
+                    msg.new = `${downloadPath}/${fileName}${extension[0]}`;
+                    node.send(msg);
+                  }
+                  // extension ? msg.old
+                  //   : node.send({ payload: null });
+                  // const fileExtension = extension ? fs.renameSync(`${downloadPath}/${downloadedFileName}`, `${downloadPath}/${fileName}.${extension[0]}`) : node.warn(`File has no extension! saved as ${downloadedFileName} in ${downloadPath}!`);
+                  // fs.renameSync(`${downloadPath}/${downloadedFileName}`, `${downloadPath}/${fileName}.${extension}`);
+                }
+              }
+            });
+          }
         }
+
         // Waiting for the specified selector
         node.status({
           fill: "blue",
@@ -81,9 +127,9 @@ module.exports = function (RED) {
           shape: "dot",
           text: `Clicked ${selector}`,
         });
-        // Sending the msg
-        send(msg);
-
+        if (!fileName || fileName == "")
+          // Sending the msg
+          send(msg);
       } catch (e) {
         // If an error occured
         node.error(e);
@@ -95,10 +141,13 @@ module.exports = function (RED) {
       }
 
       // Clear status of the node
-      setTimeout(() => {
-        done();
-        node.status({});
-      }, (msg.error) ? 10000 : 3000);
+      setTimeout(
+        () => {
+          done();
+          node.status({});
+        },
+        msg.error ? 10000 : 3000
+      );
     });
     this.on("close", function () {
       node.status({});