@mathieuc/tradingview 2.0.5 → 3.0.0-pre

package/examples/SimpleChart.js

@@ -0,0 +1,63 @@
+const TradingView = require('../main');
+
+/*
+  This example creates a BTCEUR daily chart
+*/
+
+const client = new TradingView.Client(); // Creates a websocket client
+
+const chart = new client.Session.Chart(); // Init a Chart session
+
+chart.setMarket('BINANCE:BTCEUR', { // Set the market
+  timeframe: 'D',
+});
+
+chart.onError((...err) => { // Listen for errors (can avoid crash)
+  console.error('Chart error:', ...err);
+  // Do something...
+});
+
+chart.onSymbolLoaded(() => { // When the symbol is successfully loaded
+  console.log(`Market "${chart.infos.description}" loaded !`);
+});
+
+chart.onUpdate(() => { // When price changes
+  if (!chart.periods[0]) return;
+  console.log(`[${chart.infos.description}]: ${chart.periods[0].close} ${chart.infos.currency_id}`);
+  // Do something...
+});
+
+// Wait 5 seconds and set the market to BINANCE:ETHEUR
+setTimeout(() => {
+  console.log('\nSetting market to BINANCE:ETHEUR...');
+  chart.setMarket('BINANCE:ETHEUR', {
+    timeframe: 'D',
+  });
+}, 5000);
+
+// Wait 10 seconds and set the timeframe to 15 minutes
+setTimeout(() => {
+  console.log('\nSetting timeframe to 15 minutes...');
+  chart.setSeries('15');
+}, 10000);
+
+// Wait 15 seconds and set the chart type to "Heikin Ashi"
+setTimeout(() => {
+  console.log('\nSetting the chart type to "Heikin Ashi"s...');
+  chart.setMarket('BINANCE:ETHEUR', {
+    timeframe: 'D',
+    type: 'HeikinAshi',
+  });
+}, 15000);
+
+// Wait 20 seconds and close the chart
+setTimeout(() => {
+  console.log('\nClosing the chart...');
+  chart.delete();
+}, 20000);
+
+// Wait 25 seconds and close the client
+setTimeout(() => {
+  console.log('\nClosing the client...');
+  client.end();
+}, 25000);

package/examples/_RenkoIndicator.js

@@ -0,0 +1,41 @@
+const TradingView = require('../main');
+
+if (!process.argv[2]) throw Error('Please specify your \'sessionid\' cookie');
+
+const client = new TradingView.Client({
+  token: process.argv[2],
+});
+
+const chart = new client.Session.Chart();
+chart.setMarket('NASDAQ:AAPL', {
+  type: 'Renko',
+  timeframe: '1',
+  inputs: {
+    source: 'close',
+    sources: 'Close',
+    boxSize: 3,
+    style: 'Traditional',
+    atrLength: 14,
+    wicks: true,
+  },
+});
+
+TradingView.searchIndicator('CM_Ultimate_MA_MTF_V2').then(async (indicator) => {
+  console.log(
+    '*** Indicator information\n',
+    indicator[0].id,
+  );
+  const pineIndicator = await indicator[0].get();
+  const study = new chart.Study(pineIndicator);
+  study.onUpdate(() => {
+    console.log(
+      '\n*** Chart time and close:\n',
+      chart.periods[0].time,
+      chart.periods[0].close,
+    );
+    console.log(
+      '\n*** Indicator values:\n',
+      study.periods[0].MultiTimeframe_Moving_Avg,
+    );
+  });
+});

package/main.js

@@ -1,337 +1,9 @@
-const WebSocket = require('ws');
-const {
-  search, getScreener, getTA, getIndicator, getUserToken,
-} = require('./miscRequests');
-
-let onPacket = () => null;
-
-function parse(str) {
-  const packets = str.replace(/~h~/g, '').split(/~m~[0-9]{1,}~m~/g).map((p) => {
-    if (!p) return false;
-    try {
-      return JSON.parse(p);
-    } catch (error) {
-      console.warn('Cant parse', p);
-      return false;
-    }
-  }).filter((p) => p);
-
-  packets.forEach((packet) => {
-    if (packet.m === 'protocol_error') {
-      return onPacket({
-        type: 'error',
-        syntax: packet.p[0],
-      });
-    }
-
-    if (packet.m && packet.p) {
-      return onPacket({
-        type: packet.m,
-        session: packet.p[0],
-        data: packet.p[1],
-      });
-    }
-
-    if (typeof packet === 'number') return onPacket({ type: 'ping', ping: packet });
-
-    return onPacket({ type: 'info', ...packet });
-  });
-}
-
-function genSession() {
-  let r = '';
-  const c = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
-  for (let i = 0; i < 12; i += 1) r += c.charAt(Math.floor(Math.random() * c.length));
-  return `qs_${r}`;
-}
-
-module.exports = (autoInit = true) => {
-  const callbacks = {
-    connected: [],
-    disconnected: [],
-    logged: [],
-    subscribed: [],
-    ping: [],
-    price: [],
-    data: [],
-
-    error: [],
-    event: [],
-  };
-
-  const chartEventNames = [
-    'du', 'timescale_update',
-    'series_loading', 'series_completed', 'series_error',
-    'symbol_resolved', 'symbol_error',
-    'study_loading', 'study_error',
-  ];
-  const chartCBs = {};
-
-  function handleEvent(ev, ...data) {
-    callbacks[ev].forEach((e) => e(...data));
-    callbacks.event.forEach((e) => e(ev, ...data));
-  }
-
-  function handleError(...msgs) {
-    if (callbacks.error.length === 0) console.error(...msgs);
-    else handleEvent('error', ...msgs);
-  }
-
-  const ws = new WebSocket('wss://widgetdata.tradingview.com/socket.io/websocket', {
-    origin: 'https://s.tradingview.com',
-  });
-
-  let logged = false;
-  /** ID of the quote session */
-  let sessionId = '';
-
-  /** List of subscribed symbols */
-  let subscribed = [];
-
-  /**
-   * Send a custom packet
-   * @param {string} t Packet type
-   * @param {string[]} p Packet data
-   * @example
-   * // Subscribe manualy to BTCEUR
-   * send('quote_add_symbols', [sessionId, 'BTCEUR']);
-  */
-  function send(t, p = []) {
-    if (!sessionId) return;
-
-    const msg = JSON.stringify({ m: t, p });
-    ws.send(`~m~${msg.length}~m~${msg}`);
-  }
-
-  ws.on('open', () => {
-    sessionId = genSession();
-    handleEvent('connected');
-  });
-
-  ws.on('close', () => {
-    logged = false;
-    sessionId = '';
-    handleEvent('disconnected');
-  });
-
-  ws.on('message', parse);
-
-  onPacket = (packet) => {
-    if (packet.type === 'ping') {
-      const pingStr = `~h~${packet.ping}`;
-      ws.send(`~m~${pingStr.length}~m~${pingStr}`);
-      handleEvent('ping', packet.ping);
-      return;
-    }
-
-    if (packet.type === 'quote_completed' && packet.data) {
-      handleEvent('subscribed', packet.data);
-      return;
-    }
-
-    if (packet.type === 'qsd' && packet.data.n && packet.data.v.lp) {
-      handleEvent('price', {
-        symbol: packet.data.n,
-        price: packet.data.v.lp,
-      });
-
-      return;
-    }
-
-    if (chartEventNames.includes(packet.type) && chartCBs[packet.session]) {
-      chartCBs[packet.session](packet);
-      return;
-    }
-
-    if (!logged && packet.type === 'info') {
-      if (autoInit) {
-        send('set_auth_token', ['unauthorized_user_token']);
-        send('quote_create_session', [sessionId]);
-        send('quote_set_fields', [sessionId, 'lp']);
-
-        subscribed.forEach((symbol) => send('quote_add_symbols', [sessionId, symbol]));
-      }
-
-      handleEvent('logged', packet);
-      return;
-    }
-
-    if (packet.type === 'error') {
-      handleError(`Market API critical error: ${packet.syntax}`);
-      ws.close();
-      return;
-    }
-
-    handleEvent('data', packet);
-  };
-
-  return {
-    /** Event listener
-     * @param { 'connected' | 'disconnected' | 'logged'
-     * | 'subscribed' | 'price' | 'data' | 'error' | 'ping' } event Event
-     * @param {(...data: object) => null} cb Callback
-     */
-    on(event, cb) {
-      if (!callbacks[event]) {
-        console.log('Wrong event:', event);
-        console.log('Available events:', Object.keys(callbacks));
-        return;
-      }
-
-      callbacks[event].push(cb);
-    },
-
-    /**
-     * Close the websocket connection
-     * @param {string} symbol Market symbol (Example: BTCEUR or COINBASE:BTCEUR)
-     */
-    end() { ws.close(); },
-
-    search,
-    getScreener,
-    getTA,
-    subscribed,
-
-    /**
-     * Unsubscribe to a market
-     * @param {string} symbol Market symbol (Example: BTCEUR or COINBASE:BTCEUR)
-     */
-    subscribe(symbol) {
-      if (subscribed.includes(symbol)) return;
-      send('quote_add_symbols', [sessionId, symbol]);
-      subscribed.push(symbol);
-    },
-
-    /**
-     * Unsubscribe from a market
-     * @param {string} symbol Market symbol (Example: BTCEUR or COINBASE:BTCEUR)
-     */
-    unsubscribe(symbol) {
-      if (!subscribed.includes(symbol)) return;
-      send('quote_remove_symbols', [sessionId, symbol]);
-      subscribed = subscribed.filter((s) => s !== symbol);
-    },
-
-    /**
-     * @typedef {Object} IndicatorInfos Indicator infos
-     * @property {string} id ID of the indicator (Like: XXX;XXXXXXXXXXXXXXXXXXXXX)
-     * @property {string} version Wanted version of the indicator
-     * @property {(string | number | boolean | null)[]} [settings] Indicator settings value
-     *
-     * @typedef {Object} ChartInfos
-     * @property {string} [session] User 'sessionid' cookie
-     * @property {string} symbol Market symbol (Example: BTCEUR or COINBASE:BTCEUR)
-     * @property { '1' | '3' | '5' | '15' | '30' | '45'
-     *  | '60' | '120' | '180' | '240'
-     *  | '1D' | '1W' | '1M'
-     * } [period] Period
-     * @property {number} [range] Number of loaded periods
-     * @property {string} [timezone] Timezone in 'Europe/Paris' format
-     * @property {IndicatorInfos[]} [indicators] List of indicators
-     */
-
-    /**
-     * @typedef {Object} Period List of prices / indicator values
-     * @property {number} $time
-     * @property {{
-     *  time: number, open: number, close: number,
-     *  max: number, min: number, change: number,
-     * }} $prices
-     */
-
-    /**
-     * Init a chart instance
-     * @param {ChartInfos} chart
-     * @param {{(prices: Period[]): null}} onUpdate List of periods starting from the lastest
-     */
-    async initChart(chart, onUpdate) {
-      const chartSession = genSession();
-      const periods = [];
-      const indicators = await Promise.all(
-        (chart.indicators || []).map((i) => getIndicator(i.id, i.version, i.settings)),
-      );
-
-      function updatePeriods(packet) {
-        const newData = packet.data;
-
-        Object.keys(newData).forEach((type) => {
-          (newData[type].s || newData[type].st || []).forEach((p) => {
-            if (!periods[p.i]) periods[p.i] = {};
-
-            if (newData[type].s) {
-              [periods[p.i].$time] = p.v;
-
-              periods[p.i][type] = {
-                open: p.v[1],
-                close: p.v[4],
-                max: p.v[2],
-                min: p.v[3],
-                change: Math.round(p.v[5] * 100) / 100,
-              };
-            }
-
-            if (newData[type].st) {
-              const period = {};
-              const indicator = indicators[parseInt(type, 10)];
-
-              p.v.forEach((val, i) => {
-                if (i === 0) return;
-                if (indicator.plots[`plot_${i - 1}`]) period[indicator.plots[`plot_${i - 1}`]] = val;
-                else period[`_plot_${i - 1}`] = val;
-              });
-              periods[p.i][chart.indicators[parseInt(type, 10)].name || `st${type}`] = period;
-            }
-          });
-        });
-      }
-
-      chartCBs[chartSession] = (packet) => {
-        if (['timescale_update', 'du'].includes(packet.type)) {
-          updatePeriods(packet);
-          onUpdate([...periods].reverse());
-          return;
-        }
-
-        if (packet.type.endsWith('_error')) {
-          handleError(`Error on '${chart.symbol}' (${chartSession}) chart: "${packet.type}"`);
-        }
-      };
-
-      if (chart.session) send('set_auth_token', [await getUserToken(chart.session)]);
-      send('chart_create_session', [chartSession, '']);
-      if (chart.timezone) send('switch_timezone', [chartSession, chart.timezone]);
-      send('resolve_symbol', [chartSession, 'sds_sym_1', `={"symbol":"${chart.symbol || 'BTCEUR'}","adjustment":"splits"}`]);
-      send('create_series', [chartSession, '$prices', 's1', 'sds_sym_1', (chart.period || '240'), (chart.range || 100), '']);
-
-      indicators.forEach(async (indicator, i) => {
-        const pineInfos = {
-          pineId: indicator.pineId,
-          pineVersion: indicator.pineVersion,
-          text: indicator.script,
-        };
-
-        Object.keys(indicator.inputs).forEach((inputID, inp) => {
-          const input = indicator.inputs[inputID];
-          if (input.type === 'bool' && typeof input.value !== 'boolean') handleError(`Input '${input.name}' (${inp}) must be a boolean !`);
-          if (input.type === 'integer' && typeof input.value !== 'number') handleError(`Input '${input.name}' (${inp}) must be a number !`);
-          if (input.type === 'float' && typeof input.value !== 'number') handleError(`Input '${input.name}' (${inp}) must be a number !`);
-          if (input.type === 'text' && typeof input.value !== 'string') handleError(`Input '${input.name}' (${inp}) must be a string !`);
-          if (input.options && !input.options.includes(input.value)) {
-            handleError(`Input '${input.name}' (${inp}) must be one of these values:`, input.options);
-          }
-
-          pineInfos[inputID] = {
-            v: input.value,
-            f: input.isFake,
-            t: input.type,
-          };
-        });
-
-        send('create_study', [chartSession, `${i}`, 'st1', '$prices', 'Script@tv-scripting-101!', pineInfos]);
-      });
-    },
-
-    send,
-    sessionId,
-  };
-};
+const miscRequests = require('./src/miscRequests');
+const Client = require('./src/client');
+const BuiltInIndicator = require('./src/classes/BuiltInIndicator');
+const PineIndicator = require('./src/classes/PineIndicator');
+
+module.exports = { ...miscRequests };
+module.exports.Client = Client;
+module.exports.BuiltInIndicator = BuiltInIndicator;
+module.exports.PineIndicator = PineIndicator;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mathieuc/tradingview",
-  "version": "2.0.5",
+  "version": "3.0.0-pre",
   "description": "Tradingview instant stocks API, indicator alerts, trading bot, and more !",
   "main": "main.js",
   "scripts": {
@@ -11,6 +11,10 @@
     "test-strat-alerts": "node ./tests/alerts/strategyAlert.js",
     "test": "npm run test-prices"
   },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/Mathieu2301/TradingView-API.git"
+  },
   "keywords": [
     "tradingwiew",
     "market",
@@ -24,11 +28,15 @@
   "author": "Mathieu Colmon",
   "license": "ISC",
   "dependencies": {
+    "jszip": "^3.7.1",
     "ws": "^7.4.3"
   },
   "devDependencies": {
+    "@babel/eslint-parser": "^7.15.7",
+    "@mathieuc/console": "^1.0.1",
     "eslint": "^7.25.0",
     "eslint-config-airbnb-base": "^14.2.1",
-    "eslint-plugin-import": "^2.22.1"
+    "eslint-plugin-import": "^2.22.1",
+    "jsdoc-to-markdown": "^7.1.0"
   }
 }

package/src/chart/graphicParser.js

@@ -0,0 +1,110 @@
+/**
+ * @typedef {Object} GraphicLabel
+ * @prop {number} id Drawing ID
+ */
+
+/**
+ * @typedef {Object} GraphicLine
+ * @prop {number} id Drawing ID
+ */
+
+/**
+ * @typedef {Object} GraphicBox
+ * @prop {number} id Drawing ID
+ */
+
+/**
+ * @typedef {Object} GraphicTable
+ * @prop {number} id Drawing ID
+ */
+
+/**
+ * @typedef {Object} GraphicHorizline
+ * @prop {number} id Drawing ID
+ * @prop {number} level Y position of the line
+ * @prop {number} startIndex Start index of the line (`chart.periods[line.startIndex]`)
+ * @prop {number} endIndex End index of the line (`chart.periods[line.endIndex]`)
+ * @prop {boolean} extendRight Is the line extended to the right
+ * @prop {boolean} extendLeft Is the line extended to the left
+ */
+
+/**
+ * @typedef {Object} GraphicPoint
+ * @prop {number} index X position of the point
+ * @prop {number} level Y position of the point
+ */
+
+/**
+ * @typedef {Object} GraphicPolygon
+ * @prop {number} id Drawing ID
+ * @prop {GraphicPoint[]} points List of polygon points
+ */
+
+/**
+ * @typedef {Object} GraphicHorizHist
+ * @prop {number} id Drawing ID
+ * @prop {number} priceLow Low Y position
+ * @prop {number} priceHigh High Y position
+ * @prop {number} firstBarTime First X position
+ * @prop {number} lastBarTime Last X position
+ * @prop {number[]} rate List of values
+ */
+
+/**
+ * @typedef {Object} GraphicData List of drawings indexed by type
+ * @prop {GraphicLabel[]} labels List of labels drawings
+ * @prop {GraphicLine[]} lines List of lines drawings
+ * @prop {GraphicBox[]} boxes List of boxes drawings
+ * @prop {GraphicTable[]} tables List of tables drawings
+ * @prop {GraphicPolygon[]} polygons List of polygons drawings
+ * @prop {GraphicHorizHist[]} horizHists List of horizontal histograms drawings
+ * @prop {GraphicHorizline[]} horizLines List of horizontal lines drawings
+ */
+
+/**
+ * @param {Object} rawGraphic Raw graphic data
+ * @param {Object} indexes Drawings xPos indexes
+ * @returns {GraphicData}
+ */
+module.exports = function graphicParse(rawGraphic = {}, indexes = []) {
+  // console.log('indexes', indexes);
+  return {
+    labels: Object.values(rawGraphic.dwglabels ?? {}).map((l) => ({
+      ...l,
+    })),
+
+    lines: Object.values(rawGraphic.dwglines ?? {}).map((l) => ({
+      ...l,
+    })),
+
+    boxes: Object.values(rawGraphic.dwgboxes ?? {}).map((b) => ({
+      ...b,
+    })),
+
+    tables: Object.values(rawGraphic.dwgtables ?? {}).map((t) => ({
+      ...t,
+    })),
+
+    horizLines: Object.values(rawGraphic.horizlines ?? {}).map((h) => ({
+      ...h,
+      startIndex: indexes[h.startIndex],
+      endIndex: indexes[h.endIndex],
+    })),
+
+    polygons: Object.values(rawGraphic.polygons ?? {}).map((p) => ({
+      ...p,
+      points: p.points.map((pt) => ({
+        ...pt,
+        index: indexes[pt.index],
+      })),
+    })),
+
+    horizHists: Object.values(rawGraphic.hhists ?? {}).map((h) => ({
+      ...h,
+      firstBarTime: indexes[h.firstBarTime],
+      lastBarTime: indexes[h.lastBarTime],
+    })),
+
+    raw: () => rawGraphic,
+  };
+};