@atlaskit/eslint-plugin-design-system
Version:
The essential plugin for use with the Atlassian Design System.
29 lines (28 loc) • 1.23 kB
JavaScript
import { getSourceCode } from '@atlaskit/eslint-utils/context-compat';
import { Import } from '../../ast-nodes/import';
import { Root } from '../../ast-nodes/root';
const TOKEN_IMPORT_MODULE = '@atlaskit/tokens';
/**
* Upserts `token` from `@atlaskit/tokens`, handling:
* 1. Already imported → no-op
* 2. `@atlaskit/tokens` exists but missing `token` → add specifier
* 3. No import → insert new `import { token } from '@atlaskit/tokens'`
*/
export function upsertTokenImport(context, fixer) {
const root = getSourceCode(context).ast.body;
const tokenImports = root.filter(node => node.type === 'ImportDeclaration' && node.source.value === TOKEN_IMPORT_MODULE);
if (tokenImports.length > 0) {
const decl = tokenImports[0];
if (Import.containsNamedSpecifier(decl, 'token')) {
return undefined;
}
const specifiers = decl.specifiers.filter(s => s.type === 'ImportSpecifier').map(s => s.local.name);
specifiers.push('token');
specifiers.sort();
return fixer.replaceText(decl, `import { ${specifiers.join(', ')} } from '${TOKEN_IMPORT_MODULE}';`);
}
return Root.upsertNamedImportDeclaration({
module: TOKEN_IMPORT_MODULE,
specifiers: ['token']
}, context, fixer);
}