Implements genAuthCookies

Author: Mathieu2301

src/classes/PinePermManager.js

@@ -1,4 +1,5 @@
 const axios = require('axios');
+const { genAuthCookies } = require('../utils');
 
 /**
  * @typedef {Object} AuthorizationUser
@@ -50,7 +51,7 @@ class PinePermManager {
           headers: {
             origin: 'https://www.tradingview.com',
             'Content-Type': 'application/x-www-form-urlencoded',
-            cookie: `sessionid=${this.sessionId};sessionid_sign=${this.signature};`,
+            cookie: genAuthCookies(this.sessionId, this.signature),
           },
         },
       );
@@ -84,7 +85,7 @@ class PinePermManager {
           headers: {
             origin: 'https://www.tradingview.com',
             'Content-Type': 'application/x-www-form-urlencoded',
-            cookie: `sessionid=${this.sessionId};sessionid_sign=${this.signature};`,
+            cookie: genAuthCookies(this.sessionId, this.signature),
           },
         },
       );
@@ -118,7 +119,7 @@ class PinePermManager {
           headers: {
             origin: 'https://www.tradingview.com',
             'Content-Type': 'application/x-www-form-urlencoded',
-            cookie: `sessionid=${this.sessionId};sessionid_sign=${this.signature};`,
+            cookie: genAuthCookies(this.sessionId, this.signature),
           },
         },
       );
@@ -143,7 +144,7 @@ class PinePermManager {
           headers: {
             origin: 'https://www.tradingview.com',
             'Content-Type': 'application/x-www-form-urlencoded',
-            cookie: `sessionid=${this.sessionId};sessionid_sign=${this.signature};`,
+            cookie: genAuthCookies(this.sessionId, this.signature),
           },
         },
       );

src/miscRequests.js

@@ -2,6 +2,7 @@ const os = require('os');
 const axios = require('axios');
 
 const PineIndicator = require('./classes/PineIndicator');
+const { genAuthCookies } = require('./utils');
 
 const validateStatus = (status) => status < 500;
 
@@ -279,7 +280,7 @@ module.exports = {
       `https://pine-facade.tradingview.com/pine-facade/translate/${indicID}/${version}`,
       {
         headers: {
-          cookie: `${session ? `sessionid=${session};` : ''}${signature ? `sessionid_sign=${signature};` : ''}`,
+          cookie: genAuthCookies(session, signature),
         },
         validateStatus,
       },
@@ -429,7 +430,7 @@ module.exports = {
   async getUser(session, signature = '', location = 'https://www.tradingview.com/') {
     const { data } = await axios.get(location, {
       headers: {
-        cookie: `sessionid=${session}${signature ? `;sessionid_sign=${signature};` : ''}`,
+        cookie: genAuthCookies(session, signature),
       },
       validateStatus,
     });
@@ -471,7 +472,7 @@ module.exports = {
       'https://pine-facade.tradingview.com/pine-facade/list',
       {
         headers: {
-          cookie: `sessionid=${session}${signature ? `;sessionid_sign=${signature};` : ''}`,
+          cookie: genAuthCookies(session, signature),
         },
         params: {
           filter: 'saved',
@@ -529,9 +530,7 @@ module.exports = {
       'https://www.tradingview.com/chart-token',
       {
         headers: {
-          cookie: session
-            ? `sessionid=${session}${signature ? `;sessionid_sign=${signature};` : ''}`
-            : '',
+          cookie: genAuthCookies(session, signature),
         },
         params: {
           image_url: layout,

src/utils.js

@@ -11,4 +11,10 @@ module.exports = {
     for (let i = 0; i < 12; i += 1) r += c.charAt(Math.floor(Math.random() * c.length));
     return `${type}_${r}`;
   },
+
+  genAuthCookies(sessionId = '', signature = '') {
+    if (!sessionId) return '';
+    if (!signature) return `sessionid=${sessionId}`;
+    return `sessionid=${sessionId};sessionid_sign=${signature}`;
+  },
 };