fix(oklab): make sure color.set and color.get still work with oklab

gka · Feb 3, 2022 · 93af49d · 93af49d
1 parent 0d0c494
commit 93af49d
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 15 deletions.
diff --git a/src/ops/get.js b/src/ops/get.js
@@ -1,13 +1,13 @@
 const Color = require('../Color');
 
-Color.prototype.get = function(mc) {
-    const [mode,channel] = mc.split('.');
+Color.prototype.get = function (mc) {
+    const [mode, channel] = mc.split('.');
     const src = this[mode]();
     if (channel) {
-        const i = mode.indexOf(channel);
+        const i = mode.indexOf(channel) - (mode.substr(0, 2) === 'ok' ? 2 : 0);
         if (i > -1) return src[i];
         throw new Error(`unknown channel ${channel} in mode ${mode}`);
     } else {
         return src;
     }
-}
+};
diff --git a/src/ops/set.js b/src/ops/set.js
@@ -1,19 +1,28 @@
 const Color = require('../Color');
-const {type} = require('../utils');
+const { type } = require('../utils');
 
-Color.prototype.set = function(mc, value, mutate=false) {
-    const [mode,channel] = mc.split('.');
+Color.prototype.set = function (mc, value, mutate = false) {
+    const [mode, channel] = mc.split('.');
     const src = this[mode]();
     if (channel) {
-        const i = mode.indexOf(channel);
+        const i = mode.indexOf(channel) - (mode.substr(0, 2) === 'ok' ? 2 : 0);
         if (i > -1) {
             if (type(value) == 'string') {
-                switch(value.charAt(0)) {
-                    case '+': src[i] += +value; break;
-                    case '-': src[i] += +value; break;
-                    case '*': src[i] *= +(value.substr(1)); break;
-                    case '/': src[i] /= +(value.substr(1)); break;
-                    default: src[i] = +value;
+                switch (value.charAt(0)) {
+                    case '+':
+                        src[i] += +value;
+                        break;
+                    case '-':
+                        src[i] += +value;
+                        break;
+                    case '*':
+                        src[i] *= +value.substr(1);
+                        break;
+                    case '/':
+                        src[i] /= +value.substr(1);
+                        break;
+                    default:
+                        src[i] = +value;
                 }
             } else if (type(value) === 'number') {
                 src[i] = value;
@@ -31,4 +40,4 @@ Color.prototype.set = function(mc, value, mutate=false) {
     } else {
         return src;
     }
-}
+};