fix(glsl): fix console handling

Author: MrTelanie

src/glsl/index.js

@@ -246,25 +246,16 @@ function arrFun(node) {
 
 function calExp(node) {
   // console.log('calExp()', node.name, node);
-  const { arguments: args, callee: { name } } = node;
+  const { arguments: args, callee: { name, object } } = node;
+  if (object && object.name === 'console') {
+    return '/* console statement */';
+  }
   if (name === 'discard') {
     return 'discard';
   }
   if (name === 'calc') {
     return `(${handleNode(args[0].body)})`;
   }
-  if (name === 'debug') {
-    return '/* debug statement */';
-  }
-  if (name === 'log') {
-    return '/* log statement */';
-  }
-  if (name === 'warn') {
-    return '/* warn statement */';
-  }
-  if (name === 'error') {
-    return '/* error statement */';
-  }
   return `${name}(${args.map(handleNode).join(', ')})`;
 }
 

test/glsl/sim.js

@@ -324,22 +324,24 @@ float action(vec2 one, vec2 two) {
   it('works fine even with debug statements', () => {
 
     let warnRes;
-    const warn = (arg) => warnRes = arg;
+    const console = {
+      warn: (arg) => warnRes = arg
+    };
 
     const shader = ({ vec4, vec2 }) => {
       let bar = vec4((x = vec2()) => {
-        warn(`warning: ${x.x.toPrecision(8)}`);
+        console.warn(`warning: ${x.x.toPrecision(8)}`);
         return vec4(x, 1.0, 1.0);
       });
       return { bar };
     };
-    const { js, glsl } = buildGLSL(shader, { js: true });
+    const { js, glsl } = buildGLSL(shader, { js: true, ast: true });
 
     const { bar } = js;
 
     const expected = `
 vec4 bar(vec2 x) {
-\t/* warn statement */
+\t/* console statement */
 \treturn vec4(x, 1.0, 1.0);
 }
 `;