refactor(ast): rename isKindOfLiteral to isLiteral and improve type support

Author: Rel1cx

.pkgs/eslint-plugin-local/src/utils/is-initialized-from-source.ts

@@ -33,7 +33,7 @@ export function isInitializedFromSource(
     // check for: `variable = require('source')` or `variable = require('source').variable`
     const args = getRequireExpressionArguments(init);
     const arg0 = args?.[0];
-    if (arg0 == null || !AST.isKindOfLiteral(arg0, "string")) {
+    if (arg0 == null || !AST.isLiteral(arg0, "string")) {
       return false;
     }
     // check for: `require('source')` or `require('source/...')`

packages/core/src/utils/is-initialized-from-react.ts

@@ -49,7 +49,7 @@ export function isInitializedFromReact(
     // check for: `variable = require('react')` or `variable = require('react').variable`
     const args = getRequireExpressionArguments(init);
     const arg0 = args?.[0];
-    if (arg0 == null || !AST.isKindOfLiteral(arg0, "string")) {
+    if (arg0 == null || !AST.isLiteral(arg0, "string")) {
       return false;
     }
     // check for: `require('react')` or `require('react/...')`

packages/utilities/ast/src/index.ts

@@ -13,8 +13,8 @@ export * from "./get-top-level-identifier";
 export * from "./is";
 export * from "./is-array-map-call";
 export * from "./is-empty-function";
-export * from "./is-kind-of-literal";
 export * from "./is-line-break";
+export * from "./is-literal";
 export * from "./is-multi-line";
 export * from "./is-node-equal";
 export * from "./is-this-expression";

packages/utilities/ast/src/is-kind-of-literal.ts

@@ -0,0 +1,33 @@
+/* eslint-disable local/prefer-eqeq-nullish-comparison */
+import type { TSESTree } from "@typescript-eslint/types";
+import { AST_NODE_TYPES as T } from "@typescript-eslint/types";
+
+type LiteralType =
+  | "boolean"
+  | "null"
+  | "number"
+  | "regexp"
+  | "string";
+
+export function isLiteral(node: TSESTree.Node): node is TSESTree.Literal;
+export function isLiteral(node: TSESTree.Node, type: "boolean"): node is TSESTree.BooleanLiteral;
+export function isLiteral(node: TSESTree.Node, type: "null"): node is TSESTree.NullLiteral;
+export function isLiteral(node: TSESTree.Node, type: "number"): node is TSESTree.NumberLiteral;
+export function isLiteral(node: TSESTree.Node, type: "regexp"): node is TSESTree.RegExpLiteral;
+export function isLiteral(node: TSESTree.Node, type: "string"): node is TSESTree.StringLiteral;
+export function isLiteral(node: TSESTree.Node, type?: LiteralType) {
+  if (node.type !== T.Literal) return false;
+  if (type == null) return true;
+  switch (type) {
+    case "boolean":
+      return typeof node.value === "boolean";
+    case "null":
+      return node.value === null;
+    case "number":
+      return typeof node.value === "number";
+    case "regexp":
+      return "regex" in node;
+    case "string":
+      return typeof node.value === "string";
+  }
+}