100 lines
2.3 KiB
JavaScript
100 lines
2.3 KiB
JavaScript
import {getStaticValue} from '@eslint-community/eslint-utils';
|
|
import {switchNewExpressionToCallExpression} from './fix/index.js';
|
|
import isNumber from './utils/is-number.js';
|
|
import {isNewExpression} from './ast/index.js';
|
|
|
|
const ERROR = 'error';
|
|
const ERROR_UNKNOWN = 'error-unknown';
|
|
const SUGGESTION = 'suggestion';
|
|
const messages = {
|
|
[ERROR]: '`new Buffer()` is deprecated, use `Buffer.{{method}}()` instead.',
|
|
[ERROR_UNKNOWN]: '`new Buffer()` is deprecated, use `Buffer.alloc()` or `Buffer.from()` instead.',
|
|
[SUGGESTION]: 'Switch to `Buffer.{{replacement}}()`.',
|
|
};
|
|
|
|
const inferMethod = (bufferArguments, scope) => {
|
|
if (bufferArguments.length !== 1) {
|
|
return 'from';
|
|
}
|
|
|
|
const [firstArgument] = bufferArguments;
|
|
if (firstArgument.type === 'SpreadElement') {
|
|
return;
|
|
}
|
|
|
|
if (firstArgument.type === 'ArrayExpression' || firstArgument.type === 'TemplateLiteral') {
|
|
return 'from';
|
|
}
|
|
|
|
if (isNumber(firstArgument, scope)) {
|
|
return 'alloc';
|
|
}
|
|
|
|
const staticResult = getStaticValue(firstArgument, scope);
|
|
if (staticResult) {
|
|
const {value} = staticResult;
|
|
if (
|
|
typeof value === 'string'
|
|
|| Array.isArray(value)
|
|
) {
|
|
return 'from';
|
|
}
|
|
}
|
|
};
|
|
|
|
function fix(node, context, method) {
|
|
return function * (fixer) {
|
|
yield fixer.insertTextAfter(node.callee, `.${method}`);
|
|
yield switchNewExpressionToCallExpression(node, context, fixer);
|
|
};
|
|
}
|
|
|
|
/** @param {import('eslint').Rule.RuleContext} context */
|
|
const create = context => {
|
|
const {sourceCode} = context;
|
|
return {
|
|
NewExpression(node) {
|
|
if (!isNewExpression(node, {name: 'Buffer'})) {
|
|
return;
|
|
}
|
|
|
|
const method = inferMethod(node.arguments, sourceCode.getScope(node));
|
|
|
|
if (method) {
|
|
return {
|
|
node,
|
|
messageId: ERROR,
|
|
data: {method},
|
|
fix: fix(node, context, method),
|
|
};
|
|
}
|
|
|
|
return {
|
|
node,
|
|
messageId: ERROR_UNKNOWN,
|
|
suggest: ['from', 'alloc'].map(replacement => ({
|
|
messageId: SUGGESTION,
|
|
data: {replacement},
|
|
fix: fix(node, context, replacement),
|
|
})),
|
|
};
|
|
},
|
|
};
|
|
};
|
|
|
|
/** @type {import('eslint').Rule.RuleModule} */
|
|
const config = {
|
|
create,
|
|
meta: {
|
|
type: 'problem',
|
|
docs: {
|
|
description: 'Enforce the use of `Buffer.from()` and `Buffer.alloc()` instead of the deprecated `new Buffer()`.',
|
|
recommended: 'unopinionated',
|
|
},
|
|
fixable: 'code',
|
|
hasSuggestions: true,
|
|
messages,
|
|
},
|
|
};
|
|
|
|
export default config;
|