eslint-plugin-unicorn/rules/fix/remove-argument.js

Various awesome ESLint rules

'use strict';
const {isCommaToken} = require('eslint-utils');
const {getParentheses} = require('../utils/parentheses.js');

function removeArgument(fixer, node, sourceCode) {
	const callExpression = node.parent;
	const index = callExpression.arguments.indexOf(node);
	const parentheses = getParentheses(node, sourceCode);
	const firstToken = parentheses[0] || node;
	const lastToken = parentheses[parentheses.length - 1] || node;

	let [start] = firstToken.range;
	let [, end] = lastToken.range;

	if (index !== 0) {
		start = sourceCode.getTokenBefore(firstToken).range[0];
	}

	// If the removed argument is the only argument, the trailing comma must be removed too
	/* istanbul ignore next: Not reachable for now */
	if (callExpression.arguments.length === 1) {
		const tokenAfter = sourceCode.getTokenBefore(lastToken);
		if (isCommaToken(tokenAfter)) {
			end = tokenAfter[1];
		}
	}

	return fixer.replaceTextRange([start, end], '');
}

module.exports = removeArgument;