no-unused-class-name.ts

import { createRule } from "../utils"
import type {
  SourceLocation,
  SvelteAttribute,
  SvelteDirective,
  SvelteShorthandAttribute,
  SvelteSpecialDirective,
  SvelteSpreadAttribute,
  SvelteStyleDirective,
} from "svelte-eslint-parser/lib/ast"
import type { AnyNode } from "postcss"
import {
  default as selectorParser,
  type Node as SelectorNode,
} from "postcss-selector-parser"

export default createRule("no-unused-class-name", {
  meta: {
    docs: {
      description:
        "disallow the use of a class in the template without a corresponding style",
      category: "Best Practices",
      recommended: false,
    },
    schema: [],
    messages: {},
    type: "suggestion",
  },
  create(context) {
    const classesUsedInTemplate: Record<string, SourceLocation> = {}

    return {
      SvelteElement(node) {
        if (node.kind !== "html") {
          return
        }
        const classes = node.startTag.attributes.flatMap(findClassesInAttribute)
        for (const className of classes) {
          classesUsedInTemplate[className] = node.startTag.loc
        }
      },
      "Program:exit"() {
        const styleContext = context.parserServices.getStyleContext()
        if (["parse-error", "unknown-lang"].includes(styleContext.status)) {
          return
        }
        const classesUsedInStyle =
          styleContext.sourceAst != null
            ? findClassesInPostCSSNode(styleContext.sourceAst)
            : []
        for (const className in classesUsedInTemplate) {
          if (!classesUsedInStyle.includes(className)) {
            context.report({
              loc: classesUsedInTemplate[className],
              message: `Unused class "${className}".`,
            })
          }
        }
      },
    }
  },
})

/**
 * Extract all class names used in a HTML element attribute.
 */
function findClassesInAttribute(
  attribute:
    | SvelteAttribute
    | SvelteShorthandAttribute
    | SvelteSpreadAttribute
    | SvelteDirective
    | SvelteStyleDirective
    | SvelteSpecialDirective,
): string[] {
  if (attribute.type === "SvelteAttribute" && attribute.key.name === "class") {
    return attribute.value.flatMap((value) =>
      value.type === "SvelteLiteral" ? value.value.trim().split(/\s+/u) : [],
    )
  }
  if (attribute.type === "SvelteDirective" && attribute.kind === "Class") {
    return [attribute.key.name.name]
  }
  return []
}

/**
 * Extract all class names used in a PostCSS node.
 */
function findClassesInPostCSSNode(node: AnyNode): string[] {
  if (node.type === "rule") {
    let classes = node.nodes.flatMap(findClassesInPostCSSNode)
    const processor = selectorParser()
    classes = classes.concat(
      findClassesInSelector(processor.astSync(node.selector)),
    )
    return classes
  }
  if (node.type === "root" || node.type === "atrule") {
    return node.nodes.flatMap(findClassesInPostCSSNode)
  }
  return []
}

/**
 * Extract all class names used in a PostCSS selector.
 */
function findClassesInSelector(node: SelectorNode): string[] {
  if (node.type === "class") {
    return [node.value]
  }
  if (
    node.type === "pseudo" ||
    node.type === "root" ||
    node.type === "selector"
  ) {
    return node.nodes.flatMap(findClassesInSelector)
  }
  return []
}