Convert to TypeScript

package.json

@@ -16,8 +16,8 @@
     "name": "Kristian Dupont",
     "url": "http://www.kristiandupont.com"
   },
-  "main": "build/engine.js",
-  "types": "build/engine.d.ts",
+  "main": "build/index.js",
+  "types": "build/index.d.ts",
   "bin": {
     "schemalint": "./bin/schemalint"
   },
@@ -45,6 +45,7 @@
   },
   "devDependencies": {
     "@kristiandupont/dev-deps": "^2.20.0",
+    "@types/pg": "^8.11.0",
     "@types/ramda": "0.29.10",
     "np": "9.2.0",
     "testcontainers": "10.7.1"

src/Config.ts

@@ -0,0 +1,76 @@
+import { ClientConfig } from "pg";
+
+/**
+ * The configuration for schemalint.
+ */
+type Config = {
+  /**
+   * The connection configuration for the database.
+   * @see https://node-postgres.com/apis/client
+   */
+  connection: ClientConfig;
+  /**
+   * The plugins to be used.
+   */
+  plugins?: string[];
+  /**
+   * The rules to be used.
+   */
+  rules: Record<string, RuleConfig>;
+  /**
+   * The schemas to be linted.
+   */
+  schemas: SchemaConfig[];
+  /**
+   * The configuration for ignoring problems.
+   */
+  ignores?: IgnoreConfig[];
+};
+
+/**
+ * A schema to be linted.
+ */
+export type SchemaConfig = {
+  /**
+   * The name of the schema to be linted.
+   */
+  name: string;
+  /**
+   * The rules to be used spefically for this schema. These rules will be merged with the global rules.
+   */
+  rules?: Record<string, RuleConfig>;
+};
+
+/**
+ * A rule configuration. The first element is the severity of the rule, and the rest of the elements are the options for the rule.
+ */
+export type RuleConfig = [Severity, ...unknown[]];
+
+/**
+ * The severity of a rule. `off` means the rule is disabled, `error` means the rule is enabled.
+ */
+export type Severity = "off" | "error";
+
+/**
+ * A configuration for ignoring problems.
+ */
+export type IgnoreConfig = {
+  /**
+   * The rule name to ignore. `rule` or `rulePattern` must be provided.
+   */
+  rule?: string;
+  /**
+   * A pattern to match against the rule name. `rule` or `rulePattern` must be provided.
+   */
+  rulePattern?: string;
+  /**
+   * The identifier to ignore. `identifier` or `identifierPattern` must be provided.
+   */
+  identifier?: string;
+  /**
+   * A pattern to match against the identifier. `identifier` or `identifierPattern` must be provided.
+   */
+  identifierPattern?: string;
+};
+
+export default Config;

src/Rule.ts

@@ -1,11 +1,13 @@
 import { Schema } from "extract-pg-schema";
 
-export type Reporter = (p: {
+export type Issue = {
   rule: string;
   identifier: string;
   message: string;
   suggestedMigration?: string;
-}) => void;
+};
+
+export type Reporter = (p: Issue) => void;
 
 type Rule = {
   name: string;

src/cli.js → src/cli.ts

@@ -3,9 +3,8 @@ import chalk from "chalk";
 import optionator from "optionator";
 import path from "path";
 
+import { version } from "../package.json";
 import { processDatabase } from "./engine";
-// @ts-ignore
-const { version } = require("../package.json");
 
 async function main() {
   const o = optionator({
@@ -34,11 +33,11 @@ async function main() {
     ],
   });
 
-  let options;
+  let options: any;
 
   try {
     options = o.parseArgv(process.argv);
-  } catch (error) {
+  } catch (error: any) {
     console.error(error.message);
     process.exit(1);
   }
@@ -60,6 +59,7 @@ async function main() {
   );
 
   try {
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
     const config = require(configFile);
     const exitCode = await processDatabase(config);
     process.exit(exitCode);

src/engine.js → src/engine.ts

@@ -3,19 +3,23 @@ import { extractSchemas } from "extract-pg-schema";
 import path from "path";
 import { indexBy, keys, prop, values } from "ramda";
 
+import Config from "./Config";
+import Rule, { Issue, Reporter } from "./Rule";
 import * as builtinRules from "./rules";
 
-function consoleReporter({ rule, identifier, message }) {
+type IgnoreMatcher = (rule: string, identifier: string) => boolean;
+
+function consoleReporter({ rule, identifier, message }: Issue) {
   console.error(
     `${chalk.yellow(identifier)}: error ${chalk.red(rule)} : ${message}`,
   );
 }
 
 let anyIssues = false;
-const suggestedMigrations = [];
+const suggestedMigrations: string[] = [];
 
 const createReportFunction =
-  (reporter, ignoreMatchers) =>
+  (reporter: Reporter, ignoreMatchers: IgnoreMatcher[]): Reporter =>
   ({ rule, identifier, message, suggestedMigration }) => {
     if (ignoreMatchers.some((im) => im(rule, identifier))) {
       // This one is ignored.
@@ -36,8 +40,11 @@ export async function processDatabase({
   rules,
   schemas,
   ignores = [],
-}) {
-  const pluginRules = plugins.map((p) => require(path.join(process.cwd(), p)));
+}: Config): Promise<number> {
+  const pluginRules = plugins.map(
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    (p) => require(path.join(process.cwd(), p)) as Record<string, Rule>,
+  );
   const allRules = [builtinRules, ...pluginRules].reduce(
     (acc, elem) => ({ ...acc, ...elem }),
     {},
@@ -50,35 +57,38 @@ export async function processDatabase({
     }`,
   );
 
-  const ignoreMatchers = ignores.map((i) => (rule, identifier) => {
-    let ruleMatch;
-    if (i.rule) {
-      ruleMatch = rule === i.rule;
-    } else if (i.rulePattern) {
-      ruleMatch = new RegExp(i.rulePattern).test(rule);
-    } else {
-      throw new Error(
-        `Ignore object is missing a rule or rulePattern property: ${JSON.stringify(
-          i,
-        )}`,
-      );
-    }
-
-    let identifierMatch;
-    if (i.identifier) {
-      identifierMatch = identifier === i.identifier;
-    } else if (i.identifierPattern) {
-      identifierMatch = new RegExp(i.identifierPattern).test(identifier);
-    } else {
-      throw new Error(
-        `Ignore object is missing an identifier or identifierPattern property: ${JSON.stringify(
-          i,
-        )}`,
-      );
-    }
-
-    return ruleMatch && identifierMatch;
-  });
+  const ignoreMatchers = ignores.map(
+    (i): IgnoreMatcher =>
+      (rule, identifier) => {
+        let ruleMatch: boolean;
+        if (i.rule) {
+          ruleMatch = rule === i.rule;
+        } else if (i.rulePattern) {
+          ruleMatch = new RegExp(i.rulePattern).test(rule);
+        } else {
+          throw new Error(
+            `Ignore object is missing a rule or rulePattern property: ${JSON.stringify(
+              i,
+            )}`,
+          );
+        }
+
+        let identifierMatch: boolean;
+        if (i.identifier) {
+          identifierMatch = identifier === i.identifier;
+        } else if (i.identifierPattern) {
+          identifierMatch = new RegExp(i.identifierPattern).test(identifier);
+        } else {
+          throw new Error(
+            `Ignore object is missing an identifier or identifierPattern property: ${JSON.stringify(
+              i,
+            )}`,
+          );
+        }
+
+        return ruleMatch && identifierMatch;
+      },
+  );
 
   const report = createReportFunction(consoleReporter, ignoreMatchers);
   const extractedSchemas = await extractSchemas(connection, {

src/index.ts

@@ -0,0 +1,3 @@
+export type { default as Config } from "./Config";
+export * from "./engine";
+export type { default as Rule } from "./Rule";