space-infix-ops.ts

import {
  AST_TOKEN_TYPES,
  TSESTree,
} from '@typescript-eslint/experimental-utils';
import baseRule from 'eslint/lib/rules/space-infix-ops';
import * as util from '../util';

export type Options = util.InferOptionsTypeFromRule<typeof baseRule>;
export type MessageIds = util.InferMessageIdsTypeFromRule<typeof baseRule>;

const UNIONS = ['|', '&'];

export default util.createRule<Options, MessageIds>({
  name: 'space-infix-ops',
  meta: {
    type: 'layout',
    docs: {
      description:
        'This rule is aimed at ensuring there are spaces around infix operators.',
      category: 'Stylistic Issues',
      recommended: false,
      extendsBaseRule: true,
    },
    fixable: baseRule.meta.fixable,
    schema: baseRule.meta.schema,
    messages: baseRule.meta.messages ?? {
      missingSpace: "Operator '{{operator}}' must be spaced.",
    },
  },
  defaultOptions: [
    {
      int32Hint: false,
    },
  ],
  create(context) {
    const rules = baseRule.create(context);
    const sourceCode = context.getSourceCode();

    const report = (
      node: TSESTree.Node | TSESTree.Token,
      operator: TSESTree.Token,
    ): void => {
      context.report({
        node: node,
        loc: operator.loc,
        messageId: 'missingSpace',
        data: {
          operator: operator.value,
        },
        fix(fixer) {
          const previousToken = sourceCode.getTokenBefore(operator);
          const afterToken = sourceCode.getTokenAfter(operator);
          let fixString = '';

          if (operator.range[0] - previousToken!.range[1] === 0) {
            fixString = ' ';
          }

          fixString += operator.value;

          if (afterToken!.range[0] - operator.range[1] === 0) {
            fixString += ' ';
          }

          return fixer.replaceText(operator, fixString);
        },
      });
    };

    function isSpaceChar(token: TSESTree.Token): boolean {
      return token.type === AST_TOKEN_TYPES.Punctuator && token.value === '=';
    }

    function checkAndReportAssignmentSpace(
      node: TSESTree.Node,
      leftNode: TSESTree.Token,
      rightNode?: TSESTree.Token | null,
    ): void {
      if (!rightNode) {
        return;
      }

      const operator = sourceCode.getFirstTokenBetween(
        leftNode,
        rightNode,
        isSpaceChar,
      );

      const prev = sourceCode.getTokenBefore(operator!);
      const next = sourceCode.getTokenAfter(operator!);

      if (
        !sourceCode.isSpaceBetween!(prev!, operator!) ||
        !sourceCode.isSpaceBetween!(operator!, next!)
      ) {
        report(node, operator!);
      }
    }

    /**
     * Check if it has an assignment char and report if it's faulty
     * @param node The node to report
     */
    function checkForEnumAssignmentSpace(node: TSESTree.TSEnumMember): void {
      if (!node.initializer) {
        return;
      }

      const leftNode = sourceCode.getTokenByRangeStart(node.id.range[0])!;
      const rightNode = sourceCode.getTokenByRangeStart(
        node.initializer.range[0],
      )!;

      checkAndReportAssignmentSpace(node, leftNode, rightNode);
    }

    /**
     * Check if it has an assignment char and report if it's faulty
     * @param node The node to report
     */
    function checkForClassPropertyAssignmentSpace(
      node: TSESTree.ClassProperty,
    ): void {
      const leftNode = sourceCode.getTokenByRangeStart(
        node.typeAnnotation?.range[0] ?? node.range[0],
      )!;
      const rightNode = node.value
        ? sourceCode.getTokenByRangeStart(node.value.range[0])
        : undefined;

      checkAndReportAssignmentSpace(node, leftNode, rightNode);
    }

    /**
     * Check if it is missing spaces between type annotations chaining
     * @param typeAnnotation TypeAnnotations list
     */
    function checkForTypeAnnotationSpace(
      typeAnnotation: TSESTree.TSIntersectionType | TSESTree.TSUnionType,
    ): void {
      const types = typeAnnotation.types;

      types.forEach(type => {
        const operator = sourceCode.getTokenBefore(type);

        if (operator != null && UNIONS.includes(operator.value)) {
          const prev = sourceCode.getTokenBefore(operator);
          const next = sourceCode.getTokenAfter(operator);

          if (
            !sourceCode.isSpaceBetween!(prev!, operator) ||
            !sourceCode.isSpaceBetween!(operator, next!)
          ) {
            report(typeAnnotation, operator);
          }
        }
      });
    }

    /**
     * Check if it has an assignment char and report if it's faulty
     * @param node The node to report
     */
    function checkForTypeAliasAssignment(
      node: TSESTree.TSTypeAliasDeclaration,
    ): void {
      const leftNode = sourceCode.getTokenByRangeStart(node.id.range[0])!;
      const rightNode = sourceCode.getTokenByRangeStart(
        node.typeAnnotation.range[0],
      );

      checkAndReportAssignmentSpace(node, leftNode, rightNode);
    }

    return {
      ...rules,
      TSEnumMember: checkForEnumAssignmentSpace,
      ClassProperty: checkForClassPropertyAssignmentSpace,
      TSTypeAliasDeclaration: checkForTypeAliasAssignment,
      TSUnionType: checkForTypeAnnotationSpace,
      TSIntersectionType: checkForTypeAnnotationSpace,
    };
  },
});