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

More than 300 powerful ESLint rules

import {isSemicolonToken} from '@eslint-community/eslint-utils';

const isWhitespaceOnly = text => /^\s*$/.test(text);

// Removes a statement node along with its surrounding whitespace, while preserving comments.
export default function removeStatement(statement, context, fixer, preserveSemiColon = false) {
	const {sourceCode} = context;
	const {lines} = sourceCode;
	let endToken = statement;

	if (preserveSemiColon) {
		const [penultimateToken, lastToken] = sourceCode.getLastTokens(statement, 2);

		if (isSemicolonToken(lastToken)) {
			endToken = penultimateToken;
		}
	}

	const startLocation = sourceCode.getLoc(statement).start;
	const endLocation = sourceCode.getLoc(endToken).end;

	const textBefore = lines[startLocation.line - 1].slice(0, startLocation.column);
	const textAfter = lines[endLocation.line - 1].slice(endLocation.column);

	let [start] = sourceCode.getRange(statement);
	let [, end] = sourceCode.getRange(endToken);

	if (isWhitespaceOnly(textBefore) && isWhitespaceOnly(textAfter)) {
		end += textAfter.length;

		// A line terminator can be a two-character `\r\n`, so it has to be measured instead of assumed
		const {text} = sourceCode;

		// Absorb the own line, including its indentation
		start = sourceCode.getIndexFromLoc({line: startLocation.line, column: 0});

		if (start === 0) {
			// On the first line there is no preceding line terminator to absorb, so take the trailing one instead, unless it is the end of the file.
			end += text.startsWith('\r\n', end) ? 2 : Math.min(1, text.length - end);
		} else {
			// And the line terminator before it, a line always starts right after one
			start -= text.startsWith('\r\n', start - 2) ? 2 : 1;
		}
	}

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