UNPKG

ember-codemod-remove-global-styles

Version:
23 lines (22 loc) 759 B
import { AST } from '@codemod-utils/ast-javascript'; export function addStylesToClass(file, data) { if (data.isTemplateTag) { return file; } const traverse = AST.traverse(data.isTypeScript); const ast = traverse(file, { visitClassDeclaration(path) { const { body } = path.node.body; const nodesToAdd = [ AST.builders.classProperty(AST.builders.identifier('styles'), AST.builders.identifier('styles')), ]; if (body.length > 0) { // @ts-expect-error: Incorrect type nodesToAdd.push(AST.builders.noop()); } body.unshift(...nodesToAdd); return false; }, }); return AST.print(ast); }