100 lines
3.0 KiB
JavaScript
100 lines
3.0 KiB
JavaScript
/**
|
|
* @fileoverview Require all forwardRef components include a ref parameter
|
|
*/
|
|
|
|
'use strict';
|
|
|
|
const isParenthesized = require('../util/ast').isParenthesized;
|
|
const docsUrl = require('../util/docsUrl');
|
|
const report = require('../util/report');
|
|
const getMessageData = require('../util/message');
|
|
|
|
// ------------------------------------------------------------------------------
|
|
// Rule Definition
|
|
// ------------------------------------------------------------------------------
|
|
|
|
/**
|
|
* @param {ASTNode} node
|
|
* @returns {boolean} If the node represents the identifier `forwardRef`.
|
|
*/
|
|
function isForwardRefIdentifier(node) {
|
|
return node.type === 'Identifier' && node.name === 'forwardRef';
|
|
}
|
|
|
|
/**
|
|
* @param {ASTNode} node
|
|
* @returns {boolean} If the node represents a function call `forwardRef()` or `React.forwardRef()`.
|
|
*/
|
|
function isForwardRefCall(node) {
|
|
return (
|
|
node.type === 'CallExpression'
|
|
&& (
|
|
isForwardRefIdentifier(node.callee)
|
|
|| (node.callee.type === 'MemberExpression' && isForwardRefIdentifier(node.callee.property))
|
|
)
|
|
);
|
|
}
|
|
|
|
const messages = {
|
|
missingRefParameter: 'forwardRef is used with this component but no ref parameter is set',
|
|
addRefParameter: 'Add a ref parameter',
|
|
removeForwardRef: 'Remove forwardRef wrapper',
|
|
};
|
|
|
|
module.exports = {
|
|
meta: {
|
|
docs: {
|
|
description: 'Require all forwardRef components include a ref parameter',
|
|
category: 'Possible Errors',
|
|
recommended: false,
|
|
url: docsUrl('forward-ref-uses-ref'),
|
|
},
|
|
messages,
|
|
schema: [],
|
|
type: 'suggestion',
|
|
hasSuggestions: true,
|
|
},
|
|
|
|
create(context) {
|
|
const sourceCode = context.getSourceCode();
|
|
|
|
return {
|
|
'FunctionExpression, ArrowFunctionExpression'(node) {
|
|
if (!isForwardRefCall(node.parent)) {
|
|
return;
|
|
}
|
|
|
|
if (node.params.length === 1) {
|
|
report(context, messages.missingRefParameter, 'missingRefParameter', {
|
|
node,
|
|
suggest: [
|
|
Object.assign(
|
|
getMessageData('addRefParameter', messages.addRefParameter),
|
|
{
|
|
fix(fixer) {
|
|
const param = node.params[0];
|
|
// If using shorthand arrow function syntax, add parentheses around the new parameter pair
|
|
const shouldAddParentheses = node.type === 'ArrowFunctionExpression' && !isParenthesized(context, param);
|
|
return [].concat(
|
|
shouldAddParentheses ? fixer.insertTextBefore(param, '(') : [],
|
|
fixer.insertTextAfter(param, `, ref${shouldAddParentheses ? ')' : ''}`)
|
|
);
|
|
},
|
|
}
|
|
),
|
|
Object.assign(
|
|
getMessageData('removeForwardRef', messages.removeForwardRef),
|
|
{
|
|
fix(fixer) {
|
|
return fixer.replaceText(node.parent, sourceCode.getText(node));
|
|
},
|
|
}
|
|
),
|
|
],
|
|
});
|
|
}
|
|
},
|
|
};
|
|
},
|
|
};
|