import { createRule } from "../utils.js";
const OVERRIDE_SCHEMA = { oneOf: [{
	type: "string",
	enum: [
		"before",
		"after",
		"both",
		"neither"
	]
}, {
	type: "object",
	properties: {
		before: { type: "boolean" },
		after: { type: "boolean" }
	},
	additionalProperties: false
}] };
var generator_star_spacing_default = createRule({
	name: "generator-star-spacing",
	meta: {
		type: "layout",
		docs: { description: "Enforce consistent spacing around `*` operators in generator functions" },
		fixable: "whitespace",
		schema: [{ oneOf: [{
			type: "string",
			enum: [
				"before",
				"after",
				"both",
				"neither"
			]
		}, {
			type: "object",
			properties: {
				before: { type: "boolean" },
				after: { type: "boolean" },
				named: OVERRIDE_SCHEMA,
				anonymous: OVERRIDE_SCHEMA,
				method: OVERRIDE_SCHEMA,
				shorthand: OVERRIDE_SCHEMA
			},
			additionalProperties: false
		}] }],
		defaultOptions: [{
			before: true,
			after: false
		}],
		messages: {
			missingBefore: "Missing space before *.",
			missingAfter: "Missing space after *.",
			unexpectedBefore: "Unexpected space before *.",
			unexpectedAfter: "Unexpected space after *."
		}
	},
	create(context, [options]) {
		const optionDefinitions = {
			before: {
				before: true,
				after: false
			},
			after: {
				before: false,
				after: true
			},
			both: {
				before: true,
				after: true
			},
			neither: {
				before: false,
				after: false
			}
		};
		function optionToDefinition(option, defaults) {
			if (!option) return defaults;
			return typeof option === "string" ? optionDefinitions[option] : Object.assign({}, defaults, option);
		}
		const modes = function() {
			const defaults = optionToDefinition(options, optionDefinitions.before);
			const { named, anonymous, method, shorthand } = options;
			return {
				named: optionToDefinition(named, defaults),
				anonymous: optionToDefinition(anonymous, defaults),
				method: optionToDefinition(method, defaults),
				shorthand: optionToDefinition(shorthand ?? method, defaults)
			};
		}();
		const sourceCode = context.sourceCode;
		function isStarToken(token) {
			return token.value === "*" && token.type === "Punctuator";
		}
		function getStarToken(node) {
			return sourceCode.getFirstToken("method" in node.parent && node.parent.method || node.parent.type === "MethodDefinition" ? node.parent : node, isStarToken);
		}
		function capitalize(str) {
			return str[0].toUpperCase() + str.slice(1);
		}
		function checkSpacing(kind, side, leftToken, rightToken) {
			if (!!(rightToken.range[0] - leftToken.range[1]) !== modes[kind][side]) {
				const after = leftToken.value === "*";
				const spaceRequired = modes[kind][side];
				const node = after ? leftToken : rightToken;
				const messageId = `${spaceRequired ? "missing" : "unexpected"}${capitalize(side)}`;
				context.report({
					node,
					messageId,
					fix(fixer) {
						if (spaceRequired) {
							if (after) return fixer.insertTextAfter(node, " ");
							return fixer.insertTextBefore(node, " ");
						}
						return fixer.removeRange([leftToken.range[1], rightToken.range[0]]);
					}
				});
			}
		}
		function checkFunction(node) {
			if (!node.generator) return;
			const starToken = getStarToken(node);
			const prevToken = sourceCode.getTokenBefore(starToken);
			const nextToken = sourceCode.getTokenAfter(starToken);
			let kind = "named";
			if (node.parent.type === "Property" && node.parent.method) kind = "shorthand";
			else if (node.parent.type === "MethodDefinition") kind = "method";
			else if (!node.id) kind = "anonymous";
			if (!((kind === "method" || kind === "shorthand") && starToken === sourceCode.getFirstToken(node.parent))) checkSpacing(kind, "before", prevToken, starToken);
			checkSpacing(kind, "after", starToken, nextToken);
		}
		return {
			FunctionDeclaration: checkFunction,
			FunctionExpression: checkFunction
		};
	}
});
export { generator_star_spacing_default };
