"use strict";

const _ = require("lodash");
const ruleMessages = require("../../utils/ruleMessages");
const selectorAttributeOperatorSpaceChecker = require("../selectorAttributeOperatorSpaceChecker");
const validateOptions = require("../../utils/validateOptions");
const whitespaceChecker = require("../../utils/whitespaceChecker");

const ruleName = "selector-attribute-operator-space-after";

const messages = ruleMessages(ruleName, {
  expectedAfter: operator => `Expected single space after "${operator}"`,
  rejectedAfter: operator => `Unexpected whitespace after "${operator}"`
});

const rule = function(expectation, options, context) {
  return (root, result) => {
    const checker = whitespaceChecker("space", expectation, messages);
    const validOptions = validateOptions(result, ruleName, {
      actual: expectation,
      possible: ["always", "never"]
    });

    if (!validOptions) {
      return;
    }

    selectorAttributeOperatorSpaceChecker({
      root,
      result,
      locationChecker: checker.after,
      checkedRuleName: ruleName,
      checkBeforeOperator: false,
      fix: context.fix
        ? attributeNode => {
            const { operatorAfter, setOperatorAfter } = (() => {
              const rawOperator = _.get(attributeNode, "raws.operator");

              if (rawOperator) {
                const operatorAfter = rawOperator.slice(
                  attributeNode.operator.length
                );

                return {
                  operatorAfter,
                  setOperatorAfter(fixed) {
                    delete attributeNode.raws.operator;
                    _.set(attributeNode, "raws.spaces.operator.after", fixed);
                  }
                };
              }

              const rawOperatorAfter = _.get(
                attributeNode,
                "raws.spaces.operator.after"
              );

              if (rawOperatorAfter) {
                return {
                  operatorAfter: rawOperatorAfter,
                  setOperatorAfter(fixed) {
                    attributeNode.raws.spaces.operator.after = fixed;
                  }
                };
              }

              return {
                operatorAfter: _.get(
                  attributeNode,
                  "spaces.operator.after",
                  ""
                ),
                setOperatorAfter(fixed) {
                  _.set(attributeNode, "spaces.operator.after", fixed);
                }
              };
            })();

            if (expectation === "always") {
              setOperatorAfter(operatorAfter.replace(/^\s*/, " "));

              return true;
            } else if (expectation === "never") {
              setOperatorAfter(operatorAfter.replace(/^\s*/, ""));

              return true;
            }
          }
        : null
    });
  };
};

rule.ruleName = ruleName;
rule.messages = messages;
module.exports = rule;