Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save vvatikiotis/ce010daecfa5e2eadf1dea8ca3921cd6 to your computer and use it in GitHub Desktop.
Save vvatikiotis/ce010daecfa5e2eadf1dea8ca3921cd6 to your computer and use it in GitHub Desktop.
const bypass = [
// function names to avoid logging
];
const collapsed = [
// function names to groupCollapsed
];
module.exports = function(babel) {
const { types: t } = babel;
const wrapFunctionBody = babel.template(`{
try{
console.group(NAME)
BODY
}
finally {
console.groupEnd();
}
}`);
const wrapFunctionBodyCollapsed = babel.template(`{
try{
console.groupCollapsed(NAME)
BODY
}
finally {
console.groupEnd();
}
}`);
return {
visitor: {
FunctionDeclaration(path) {
if (path.node.id && path.node.id.name) {
if (bypass.includes(path.node.id.name)) {
return;
}
if (collapsed.includes(path.node.id.name)) {
path.get("body").replaceWith(
wrapFunctionBodyCollapsed({
BODY: path.node.body.body,
NAME: t.stringLiteral(path.node.id.name)
})
);
return;
}
path.get("body").replaceWith(
wrapFunctionBody({
BODY: path.node.body.body,
NAME: t.stringLiteral(path.node.id.name)
})
);
}
}
}
};
};
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment