feat: support non-js languages

dotcarmen · dotcarmen · commit cec9313f6b6f · 2025-06-14T17:02:47.000+02:00
diff --git a/eslint-plugin-prettier.js b/eslint-plugin-prettier.js
@@ -12,6 +12,13 @@
  */
 
 /**
+ * @typedef {{
+ *   line: number;
+ *   column: number;
+ *   offset: number;
+ * }} Location
+ *
+ *
  * @typedef {PrettierOptions & {
  *   onDiskFilepath: string;
  *   parserMeta?: ESLint.ObjectMetaProperties['meta'];
@@ -25,6 +32,9 @@
  *   options: Options,
  *   fileInfoOptions: FileInfoOptions,
  * ) => string} PrettierFormat
+ *
+ *
+ * @typedef {Parameters<Exclude<ESLint.Plugin['rules'], undefined>[string]['create']>[0]} RuleContext
  */
 
 'use strict';
@@ -57,10 +67,31 @@ let prettierFormat;
 //  Rule Definition
 // ------------------------------------------------------------------------------
 
+/**
+ * Converts a byte offset to a Location.
+ *
+ * See also `getLocFromIndex` in `@eslint/js`.
+ *
+ * @param {number[]} lineIndexes
+ * @param {number} offset
+ * @returns {Location}
+ */
+function getLocFromOffset(lineIndexes, offset) {
+  const withSentinel = [-1, 0, ...lineIndexes];
+
+  let line = 0;
+  while (line + 1 < withSentinel.length && withSentinel[line + 1] < offset) {
+    line++;
+  }
+
+  const column = offset - withSentinel[line];
+  return { line, column, offset };
+}
+
 /**
  * Reports a difference.
  *
- * @param {Rule.RuleContext} context - The ESLint rule context.
+ * @param {RuleContext} context - The ESLint rule context.
  * @param {Difference} difference - The difference object.
  * @returns {void}
  */
@@ -71,8 +102,14 @@ function reportDifference(context, difference) {
   // `context.getSourceCode()` was deprecated in ESLint v8.40.0 and replaced
   // with the `sourceCode` property.
   // TODO: Only use property when our eslint peerDependency is >=8.40.0.
+  const sourceCode = context.sourceCode ?? context.getSourceCode();
+  if (!('text' in sourceCode)) {
+    throw new Error('prettier only supports textual source code');
+  }
+
+  const lineIndexes = [...sourceCode.text.matchAll(/\n/g)].map(match => match.index);
   const [start, end] = range.map(index =>
-    (context.sourceCode ?? context.getSourceCode()).getLocFromIndex(index),
+    getLocFromOffset(lineIndexes, index)
   );
 
   context.report({
@@ -90,7 +127,7 @@ function reportDifference(context, difference) {
 //  Module Definition
 // ------------------------------------------------------------------------------
 
-/** @type {ESLint.Plugin} */
+/** @satisfies {ESLint.Plugin} */
 const eslintPluginPrettier = {
   meta: { name, version },
   configs: {
@@ -168,7 +205,7 @@ const eslintPluginPrettier = {
         const source = sourceCode.text;
 
         return {
-          Program(node) {
+          [sourceCode.ast.type](node) {
             if (!prettierFormat) {
               // Prettier is expensive to load, so only load it if needed.
               prettierFormat = /** @type {PrettierFormat} */ (
@@ -251,7 +288,7 @@ const eslintPluginPrettier = {
 
               for (const difference of differences) {
                 reportDifference(
-                  /** @type {Rule.RuleContext} */ (context),
+                  /** @type {Rule.RuleContext} */(context),
                   difference,
                 );
               }
