/**
 * @fileoverview Rule to replace assignment expressions with logical operator assignment
 * @author Daniel Martens
 */
"use strict";

//------------------------------------------------------------------------------
// Requirements
//------------------------------------------------------------------------------
const astUtils = require("./utils/ast-utils.js");

//------------------------------------------------------------------------------
// Helpers
//------------------------------------------------------------------------------

const baseTypes = new Set(["Identifier", "Super", "ThisExpression"]);

/**
 * Returns true iff either "undefined" or a void expression (eg. "void 0")
 * @param {ASTNode} expression Expression to check
 * @param {import('eslint-scope').Scope} scope Scope of the expression
 * @returns {boolean} True iff "undefined" or "void ..."
 */
function isUndefined(expression, scope) {
	if (expression.type === "Identifier" && expression.name === "undefined") {
		return astUtils.isReferenceToGlobalVariable(scope, expression);
	}

	return (
		expression.type === "UnaryExpression" &&
		expression.operator === "void" &&
		expression.argument.type === "Literal" &&
		expression.argument.value === 0
	);
}

/**
 * Returns true iff the reference is either an identifier or member expression
 * @param {ASTNode} expression Expression to check
 * @returns {boolean} True for identifiers and member expressions
 */
function isReference(expression) {
	return (
		(expression.type === "Identifier" && expression.name !== "undefined") ||
		expression.type === "MemberExpression"
	);
}

/**
 * Returns true iff the expression checks for nullish with loose equals.
 * Examples: value == null, value == void 0
 * @param {ASTNode} expression Test condition
 * @param {import('eslint-scope').Scope} scope Scope of the expression
 * @returns {boolean} True iff implicit nullish comparison
 */
function isImplicitNullishComparison(expression, scope) {
	if (
		expression.type !== "BinaryExpression" ||
		expression.operator !== "=="
	) {
		return false;
	}

	const reference = isReference(expression.left) ? "left" : "right";
	const nullish = reference === "left" ? "right" : "left";

	return (
		isReference(expression[reference]) &&
		(astUtils.isNullLiteral(expression[nullish]) ||
			isUndefined(expression[nullish], scope))
	);
}

/**
 * Condition with two equal comparisons.
 * @param {ASTNode} expression Condition
 * @returns {boolean} True iff matches ? === ? || ? === ?
 */
function isDoubleComparison(expression) {
	return (
		expression.type === "LogicalExpression" &&
		expression.operator === "||" &&
		expression.left.type === "BinaryExpression" &&
		expression.left.operator === "===" &&
		expression.right.type === "BinaryExpression" &&
		expression.right.operator === "==="
	);
}

/**
 * Returns true iff the expression checks for undefined and null.
 * Example: value === null || value === undefined
 * @param {ASTNode} expression Test condition
 * @param {import('eslint-scope').Scope} scope Scope of the expression
 * @returns {boolean} True iff explicit nullish comparison
 */
function isExplicitNullishComparison(expression, scope) {
	if (!isDoubleComparison(expression)) {
		return false;
	}
	const leftReference = isReference(expression.left.left) ? "left" : "right";
	const leftNullish = leftReference === "left" ? "right" : "left";
	const rightReference = isReference(expression.right.left)
		? "left"
		: "right";
	const rightNullish = rightReference === "left" ? "right" : "left";

	return (
		astUtils.isSameReference(
			expression.left[leftReference],
			expression.right[rightReference],
		) &&
		((astUtils.isNullLiteral(expression.left[leftNullish]) &&
			isUndefined(expression.right[rightNullish], scope)) ||
			(isUndefined(expression.left[leftNullish], scope) &&
				astUtils.isNullLiteral(expression.right[rightNullish])))
	);
}

/**
 * Returns true for Boolean(arg) calls
 * @param {ASTNode} expression Test condition
 * @param {import('eslint-scope').Scope} scope Scope of the expression
 * @returns {boolean} Whether the expression is a boolean cast
 */
function isBooleanCast(expression, scope) {
	return (
		expression.type === "CallExpression" &&
		expression.callee.name === "Boolean" &&
		expression.arguments.length === 1 &&
		astUtils.isReferenceToGlobalVariable(scope, expression.callee)
	);
}

/**
 * Returns true for:
 * truthiness checks:  value, Boolean(value), !!value
 * falsiness checks:   !value, !Boolean(value)
 * nullish checks:     value == null, value === undefined || value === null
 * @param {ASTNode} expression Test condition
 * @param {import('eslint-scope').Scope} scope Scope of the expression
 * @returns {?{ reference: ASTNode, operator: '??'|'||'|'&&'}} Null if not a known existence
 */
function getExistence(expression, scope) {
	const isNegated =
		expression.type === "UnaryExpression" && expression.operator === "!";
	const base = isNegated ? expression.argument : expression;

	switch (true) {
		case isReference(base):
			return { reference: base, operator: isNegated ? "||" : "&&" };
		case base.type === "UnaryExpression" &&
			base.operator === "!" &&
			isReference(base.argument):
			return { reference: base.argument, operator: "&&" };
		case isBooleanCast(base, scope) && isReference(base.arguments[0]):
			return {
				reference: base.arguments[0],
				operator: isNegated ? "||" : "&&",
			};
		case isImplicitNullishComparison(expression, scope):
			return {
				reference: isReference(expression.left)
					? expression.left
					: expression.right,
				operator: "??",
			};
		case isExplicitNullishComparison(expression, scope):
			return {
				reference: isReference(expression.left.left)
					? expression.left.left
					: expression.left.right,
				operator: "??",
			};
		default:
			return null;
	}
}

/**
 * Returns true iff the node is inside a with block
 * @param {ASTNode} node Node to check
 * @returns {boolean} True iff passed node is inside a with block
 */
function isInsideWithBlock(node) {
	if (node.type === "Program") {
		return false;
	}

	return node.parent.type === "WithStatement" && node.parent.body === node
		? true
		: isInsideWithBlock(node.parent);
}

/**
 * Gets the leftmost operand of a consecutive logical expression.
 * @param {SourceCode} sourceCode The ESLint source code object
 * @param {LogicalExpression} node LogicalExpression
 * @returns {Expression} Leftmost operand
 */
function getLeftmostOperand(sourceCode, node) {
	let left = node.left;

	while (
		left.type === "LogicalExpression" &&
		left.operator === node.operator
	) {
		if (astUtils.isParenthesised(sourceCode, left)) {
			/*
			 * It should have associativity,
			 * but ignore it if use parentheses to make the evaluation order clear.
			 */
			return left;
		}
		left = left.left;
	}
	return left;
}

//------------------------------------------------------------------------------
// Rule Definition
//------------------------------------------------------------------------------
/** @type {import('../types').Rule.RuleModule} */
module.exports = {
	meta: {
		type: "suggestion",

		docs: {
			description:
				"Require or disallow logical assignment operator shorthand",
			recommended: false,
			frozen: true,
			url: "https://eslint.org/docs/latest/rules/logical-assignment-operators",
		},

		schema: {
			type: "array",
			oneOf: [
				{
					items: [
						{ const: "always" },
						{
							type: "object",
							properties: {
								enforceForIfStatements: {
									type: "boolean",
								},
							},
							additionalProperties: false,
						},
					],
					minItems: 0, // 0 for allowing passing no options
					maxItems: 2,
				},
				{
					items: [{ const: "never" }],
					minItems: 1,
					maxItems: 1,
				},
			],
		},

		defaultOptions: ["always"],

		fixable: "code",
		hasSuggestions: true,
		messages: {
			assignment:
				"Assignment (=) can be replaced with operator assignment ({{operator}}).",
			useLogicalOperator:
				"Convert this assignment to use the operator {{ operator }}.",
			logical:
				"Logical expression can be replaced with an assignment ({{ operator }}).",
			convertLogical:
				"Replace this logical expression with an assignment with the operator {{ operator }}.",
			if: "'if' statement can be replaced with a logical operator assignment with operator {{ operator }}.",
			convertIf:
				"Replace this 'if' statement with a logical assignment with operator {{ operator }}.",
			unexpected:
				"Unexpected logical operator assignment ({{operator}}) shorthand.",
			separate:
				"Separate the logical assignment into an assignment with a logical operator.",
		},
	},

	create(context) {
		const mode = context.options[0];
		const checkIf =
			mode === "always" &&
			context.options.length > 1 &&
			context.options[1].enforceForIfStatements;
		const sourceCode = context.sourceCode;
		const isStrict = sourceCode.getScope(sourceCode.ast).isStrict;

		/**
		 * Returns false if the access could be a getter
		 * @param {ASTNode} node Assignment expression
		 * @returns {boolean} True iff the fix is safe
		 */
		function cannotBeGetter(node) {
			return (
				node.type === "Identifier" &&
				(isStrict || !isInsideWithBlock(node))
			);
		}

		/**
		 * Check whether only a single property is accessed
		 * @param {ASTNode} node reference
		 * @returns {boolean} True iff a single property is accessed
		 */
		function accessesSingleProperty(node) {
			if (!isStrict && isInsideWithBlock(node)) {
				return node.type === "Identifier";
			}

			return (
				node.type === "MemberExpression" &&
				baseTypes.has(node.object.type) &&
				(!node.computed ||
					(node.property.type !== "MemberExpression" &&
						node.property.type !== "ChainExpression"))
			);
		}

		/**
		 * Adds a fixer or suggestion whether on the fix is safe.
		 * @param {{ messageId: string, node: ASTNode }} descriptor Report descriptor without fix or suggest
		 * @param {{ messageId: string, fix: Function }} suggestion Adds the fix or the whole suggestion as only element in "suggest" to suggestion
		 * @param {boolean} shouldBeFixed Fix iff the condition is true
		 * @returns {Object} Descriptor with either an added fix or suggestion
		 */
		function createConditionalFixer(descriptor, suggestion, shouldBeFixed) {
			if (shouldBeFixed) {
				return {
					...descriptor,
					fix: suggestion.fix,
				};
			}

			return {
				...descriptor,
				suggest: [suggestion],
			};
		}

		/**
		 * Returns the operator token for assignments and binary expressions
		 * @param {ASTNode} node AssignmentExpression or BinaryExpression
		 * @returns {import('eslint').AST.Token} Operator token between the left and right expression
		 */
		function getOperatorToken(node) {
			return sourceCode.getFirstTokenBetween(
				node.left,
				node.right,
				token => token.value === node.operator,
			);
		}

		if (mode === "never") {
			return {
				// foo ||= bar
				AssignmentExpression(assignment) {
					if (
						!astUtils.isLogicalAssignmentOperator(
							assignment.operator,
						)
					) {
						return;
					}

					const descriptor = {
						messageId: "unexpected",
						node: assignment,
						data: { operator: assignment.operator },
					};
					const suggestion = {
						messageId: "separate",
						*fix(ruleFixer) {
							if (
								sourceCode.getCommentsInside(assignment)
									.length > 0
							) {
								return;
							}

							const operatorToken = getOperatorToken(assignment);

							// -> foo = bar
							yield ruleFixer.replaceText(operatorToken, "=");

							const assignmentText = sourceCode.getText(
								assignment.left,
							);
							const operator = assignment.operator.slice(0, -1);

							// -> foo = foo || bar
							yield ruleFixer.insertTextAfter(
								operatorToken,
								` ${assignmentText} ${operator}`,
							);

							const precedence =
								astUtils.getPrecedence(assignment.right) <=
								astUtils.getPrecedence({
									type: "LogicalExpression",
									operator,
								});

							// ?? and || / && cannot be mixed but have same precedence
							const mixed =
								assignment.operator === "??=" &&
								astUtils.isLogicalExpression(assignment.right);

							if (
								!astUtils.isParenthesised(
									sourceCode,
									assignment.right,
								) &&
								(precedence || mixed)
							) {
								// -> foo = foo || (bar)
								yield ruleFixer.insertTextBefore(
									assignment.right,
									"(",
								);
								yield ruleFixer.insertTextAfter(
									assignment.right,
									")",
								);
							}
						},
					};

					context.report(
						createConditionalFixer(
							descriptor,
							suggestion,
							cannotBeGetter(assignment.left),
						),
					);
				},
			};
		}

		return {
			// foo = foo || bar
			"AssignmentExpression[operator='='][right.type='LogicalExpression']"(
				assignment,
			) {
				const leftOperand = getLeftmostOperand(
					sourceCode,
					assignment.right,
				);

				if (!astUtils.isSameReference(assignment.left, leftOperand)) {
					return;
				}

				const descriptor = {
					messageId: "assignment",
					node: assignment,
					data: { operator: `${assignment.right.operator}=` },
				};
				const suggestion = {
					messageId: "useLogicalOperator",
					data: { operator: `${assignment.right.operator}=` },
					*fix(ruleFixer) {
						if (
							sourceCode.getCommentsInside(assignment).length > 0
						) {
							return;
						}

						// No need for parenthesis around the assignment based on precedence as the precedence stays the same even with changed operator
						const assignmentOperatorToken =
							getOperatorToken(assignment);

						// -> foo ||= foo || bar
						yield ruleFixer.insertTextBefore(
							assignmentOperatorToken,
							assignment.right.operator,
						);

						// -> foo ||= bar
						const logicalOperatorToken = getOperatorToken(
							leftOperand.parent,
						);
						const firstRightOperandToken =
							sourceCode.getTokenAfter(logicalOperatorToken);

						yield ruleFixer.removeRange([
							leftOperand.parent.range[0],
							firstRightOperandToken.range[0],
						]);
					},
				};

				context.report(
					createConditionalFixer(
						descriptor,
						suggestion,
						cannotBeGetter(assignment.left),
					),
				);
			},

			// foo || (foo = bar)
			'LogicalExpression[right.type="AssignmentExpression"][right.operator="="]'(
				logical,
			) {
				// Right side has to be parenthesized, otherwise would be parsed as (foo || foo) = bar which is illegal
				if (
					isReference(logical.left) &&
					astUtils.isSameReference(logical.left, logical.right.left)
				) {
					const descriptor = {
						messageId: "logical",
						node: logical,
						data: { operator: `${logical.operator}=` },
					};
					const suggestion = {
						messageId: "convertLogical",
						data: { operator: `${logical.operator}=` },
						*fix(ruleFixer) {
							if (
								sourceCode.getCommentsInside(logical).length > 0
							) {
								return;
							}

							const parentPrecedence = astUtils.getPrecedence(
								logical.parent,
							);
							const requiresOuterParenthesis =
								logical.parent.type !== "ExpressionStatement" &&
								(parentPrecedence === -1 ||
									astUtils.getPrecedence({
										type: "AssignmentExpression",
									}) < parentPrecedence);

							if (
								!astUtils.isParenthesised(
									sourceCode,
									logical,
								) &&
								requiresOuterParenthesis
							) {
								yield ruleFixer.insertTextBefore(logical, "(");
								yield ruleFixer.insertTextAfter(logical, ")");
							}

							// Also removes all opening parenthesis
							yield ruleFixer.removeRange([
								logical.range[0],
								logical.right.range[0],
							]); // -> foo = bar)

							// Also removes all ending parenthesis
							yield ruleFixer.removeRange([
								logical.right.range[1],
								logical.range[1],
							]); // -> foo = bar

							const operatorToken = getOperatorToken(
								logical.right,
							);

							yield ruleFixer.insertTextBefore(
								operatorToken,
								logical.operator,
							); // -> foo ||= bar
						},
					};
					const fix =
						cannotBeGetter(logical.left) ||
						accessesSingleProperty(logical.left);

					context.report(
						createConditionalFixer(descriptor, suggestion, fix),
					);
				}
			},

			// if (foo) foo = bar
			"IfStatement[alternate=null]"(ifNode) {
				if (!checkIf) {
					return;
				}

				const hasBody = ifNode.consequent.type === "BlockStatement";

				if (hasBody && ifNode.consequent.body.length !== 1) {
					return;
				}

				const body = hasBody
					? ifNode.consequent.body[0]
					: ifNode.consequent;
				const scope = sourceCode.getScope(ifNode);
				const existence = getExistence(ifNode.test, scope);

				if (
					body.type === "ExpressionStatement" &&
					body.expression.type === "AssignmentExpression" &&
					body.expression.operator === "=" &&
					existence !== null &&
					astUtils.isSameReference(
						existence.reference,
						body.expression.left,
					)
				) {
					const descriptor = {
						messageId: "if",
						node: ifNode,
						data: { operator: `${existence.operator}=` },
					};
					const suggestion = {
						messageId: "convertIf",
						data: { operator: `${existence.operator}=` },
						*fix(ruleFixer) {
							if (
								sourceCode.getCommentsInside(ifNode).length > 0
							) {
								return;
							}

							const firstBodyToken =
								sourceCode.getFirstToken(body);
							const prevToken = sourceCode.getTokenBefore(ifNode);

							if (
								prevToken !== null &&
								prevToken.value !== ";" &&
								prevToken.value !== "{" &&
								firstBodyToken.type !== "Identifier" &&
								firstBodyToken.type !== "Keyword"
							) {
								// Do not fix if the fixed statement could be part of the previous statement (eg. fn() if (a == null) (a) = b --> fn()(a) ??= b)
								return;
							}

							const operatorToken = getOperatorToken(
								body.expression,
							);

							yield ruleFixer.insertTextBefore(
								operatorToken,
								existence.operator,
							); // -> if (foo) foo ||= bar

							yield ruleFixer.removeRange([
								ifNode.range[0],
								body.range[0],
							]); // -> foo ||= bar

							yield ruleFixer.removeRange([
								body.range[1],
								ifNode.range[1],
							]); // -> foo ||= bar, only present if "if" had a body

							const nextToken = sourceCode.getTokenAfter(
								body.expression,
							);

							if (
								hasBody &&
								nextToken !== null &&
								nextToken.value !== ";"
							) {
								yield ruleFixer.insertTextAfter(ifNode, ";");
							}
						},
					};
					const shouldBeFixed =
						cannotBeGetter(existence.reference) ||
						(ifNode.test.type !== "LogicalExpression" &&
							accessesSingleProperty(existence.reference));

					context.report(
						createConditionalFixer(
							descriptor,
							suggestion,
							shouldBeFixed,
						),
					);
				}
			},
		};
	},
};
