From a2b7c0680b9752b13093aaab198b9fb2ae693369 Mon Sep 17 00:00:00 2001
From: Ian Webster <ianw_github@ianww.com>
Date: Tue, 17 Aug 2021 16:43:27 -0700
Subject: [PATCH] Add support for version param. Closes #15

---
 README.md          |  4 ++++
 index.js           | 18 +++++++++++++++---
 test/index.test.js |  3 ++-
 3 files changed, 21 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index bd287ca..4461774 100644
--- a/README.md
+++ b/README.md
@@ -76,6 +76,10 @@ Sets the background color of the chart.  Any valid HTML color works.  Defaults t
 
 Sets the device pixel ratio of the chart.  This will multiply the number of pixels by the value.  This is usually used for retina displays.  Defaults to 1.0.
 
+### setVersion(version: string)
+
+Sets the Chart.js version to use (e.g. `2.9.4` or `3.4.0`).  Valid options are shown in the [documentation](https://quickchart.io/documentation/#parameters).
+
 ## Getting outputs
 
 There are two ways to get a URL for your chart object.
diff --git a/index.js b/index.js
index 4f68297..3e39f6f 100644
--- a/index.js
+++ b/index.js
@@ -25,6 +25,7 @@ class QuickChart {
     this.devicePixelRatio = 1.0;
     this.backgroundColor = '#ffffff';
     this.format = 'png';
+    this.version = '2.9.4';
   }
 
   setConfig(chartConfig) {
@@ -57,6 +58,11 @@ class QuickChart {
     return this;
   }
 
+  setVersion(version) {
+    this.version = version;
+    return this;
+  }
+
   isValid() {
     if (!this.chart) {
       return false;
@@ -75,17 +81,20 @@ class QuickChart {
     if (this.devicePixelRatio !== 1.0) {
       ret.searchParams.append('devicePixelRatio', this.devicePixelRatio);
     }
-    if (this.backgroundColor !== 1.0) {
+    if (this.backgroundColor) {
       ret.searchParams.append('bkg', this.backgroundColor);
     }
-    if (this.format !== 1.0) {
+    if (this.format) {
       ret.searchParams.append('f', this.format);
     }
+    if (this.version) {
+      ret.searchParams.append('v', this.version);
+    }
     return ret.href;
   }
 
   getPostData() {
-    const { width, height, chart, format, backgroundColor, devicePixelRatio } = this;
+    const { width, height, chart, format, version, backgroundColor, devicePixelRatio } = this;
     const postData = {
       width,
       height,
@@ -94,6 +103,9 @@ class QuickChart {
     if (format) {
       postData.format = format;
     }
+    if (version) {
+      postData.version = version;
+    }
     if (backgroundColor) {
       postData.backgroundColor = backgroundColor;
     }
diff --git a/test/index.test.js b/test/index.test.js
index a6849c7..f2a3eba 100644
--- a/test/index.test.js
+++ b/test/index.test.js
@@ -74,12 +74,13 @@ test('basic chart, other params', () => {
     data: { labels: ['Hello world', 'Foo bar'], datasets: [{ label: 'Foo', data: [1, 2] }] },
   });
 
-  qc.setBackgroundColor('#000000').setDevicePixelRatio(2.0).setFormat('svg');
+  qc.setBackgroundColor('#000000').setDevicePixelRatio(2.0).setFormat('svg').setVersion('3');
 
   expect(qc.getUrl()).toContain('Hello+world');
   expect(qc.getUrl()).toContain('devicePixelRatio=2');
   expect(qc.getUrl()).toContain('f=svg');
   expect(qc.getUrl()).toContain('bkg=%23000000');
+  expect(qc.getUrl()).toContain('v=3');
 });
 
 test('js chart', () => {