allow for SWDM version change

thegreenwebfoundation · May 27, 2024 · 15e33a5 · 15e33a5
1 parent 91b73b1
commit 15e33a5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/src/co2.js b/src/co2.js
@@ -61,6 +61,7 @@
 
 import OneByte from "./1byte.js";
 import SustainableWebDesignV3 from "./sustainable-web-design-v3.js";
+import SustainableWebDesignV4 from "./sustainable-web-design-v4.js";
 
 import {
   GLOBAL_GRID_INTENSITY,
@@ -77,6 +78,9 @@ class CO2 {
       this.model = new OneByte();
     } else if (options?.model === "swd") {
       this.model = new SustainableWebDesignV3();
+      if (options?.version === 4) {
+        this.model = new SustainableWebDesignV4();
+      }
     } else if (options?.model) {
       throw new Error(
         `"${options.model}" is not a valid model. Please use "1byte" for the OneByte model, and "swd" for the Sustainable Web Design model.\nSee https://developers.thegreenwebfoundation.org/co2js/models/ to learn more about the models available in CO2.js.`

diff --git a/src/sustainable-web-design-v3.js b/src/sustainable-web-design-v3.js
@@ -37,6 +37,7 @@ class SustainableWebDesign {
   constructor(options) {
     this.allowRatings = true;
     this.options = options;
+    this.version = 3;
   }
 
   /**

diff --git a/src/sustainable-web-design-v4.js b/src/sustainable-web-design-v4.js
@@ -24,6 +24,7 @@ const {
 class SustainableWebDesign {
   constructor(options) {
     this.options = options;
+    this.version = 4;
   }
 
   /**