feat: add create table parser (#4)

* feat: add constraint parser

* feat: add default value parsing

* feat: complete parse column def

* feat: support table constraints

* feat: add eslint check

* feat: fixing all the lint problem

* fixing while condition

.eslintrc.json

@@ -1,6 +1,16 @@
 {
-  "extends": "next/core-web-vitals",
+  "extends": ["next/core-web-vitals", "eslint:recommended", "plugin:@typescript-eslint/recommended"],
   "rules": {
     "@next/next/no-img-element": "off"
-  }
+  },
+  "parser": "@typescript-eslint/parser",
+  "plugins": ["@typescript-eslint"],
+  "overrides": [
+    {
+      "files": ["*.test.ts"],
+      "plugins": ["jest"],
+      "extends": ["plugin:jest/recommended"],
+      "rules": { "jest/prefer-expect-assertions": "off" }
+    }
+  ]
 }

package.json

@@ -13,6 +13,8 @@
   },
   "dependencies": {
     "@codemirror/lang-sql": "^6.5.5",
+    "@lezer/common": "^1.2.1",
+    "@lezer/lr": "^1.4.0",
     "@libsql/hrana-client": "^0.5.5",
     "@radix-ui/react-context-menu": "^2.1.5",
     "@radix-ui/react-dialog": "^1.0.5",
@@ -35,6 +37,7 @@
     "clsx": "^2.1.0",
     "cmdk": "^0.2.0",
     "dbgate-query-splitter": "^4.9.3",
+    "eslint-plugin-jest": "^27.6.3",
     "lucide-react": "^0.309.0",
     "next": "14.0.4",
     "react": "^18",
@@ -50,6 +53,8 @@
     "@types/node": "^20",
     "@types/react": "^18",
     "@types/react-dom": "^18",
+    "@typescript-eslint/eslint-plugin": "^6.21.0",
+    "@typescript-eslint/parser": "^6.21.0",
     "autoprefixer": "^10.0.1",
     "eslint": "^8",
     "eslint-config-next": "14.0.4",

src/app/(components)/Cells/GenericCell.tsx

@@ -106,7 +106,7 @@ export default function GenericCell({
       onMouseDown={onFocus}
       onDoubleClick={onDoubleClick}
     >
-      {(value as any).toString()}
+      {value.toString()}
     </div>
   );
 }

src/app/(components)/Cells/createEditableCell.tsx

@@ -1,5 +1,5 @@
 import { DatabaseValue } from "@/drivers/DatabaseDriver";
-import { useState, useEffect, FunctionComponent } from "react";
+import { useState, useEffect } from "react";
 import GenericCell from "./GenericCell";
 import styles from "./styles.module.css";
 

src/app/(components)/ContentMenuHandler.tsx

@@ -17,7 +17,6 @@ import {
   ContextMenuSubContent,
 } from "@/components/ui/context-menu";
 import { useEffect, useRef, useState } from "react";
-import { LucideIcon } from "lucide-react";
 
 function ContextMenuList({ menu }: { menu: OpenContextMenuList }) {
   return menu.map((item, menuIndex) => {

src/app/(components)/MainScreen.tsx

@@ -1,6 +1,6 @@
 "use client";
 import DatabaseDriver from "@/drivers/DatabaseDriver";
-import { useMemo, useEffect, useState, useLayoutEffect } from "react";
+import { useMemo, useEffect, useLayoutEffect } from "react";
 import DatabaseGui from "./DatabaseGui";
 import { DatabaseDriverProvider } from "@/context/DatabaseDriverProvider";
 import { TooltipProvider } from "@/components/ui/tooltip";

src/app/(components)/OptimizeTable/index.tsx

@@ -81,7 +81,6 @@ function handleTableCellMouseDown({
   y,
   x,
   ctrl,
-  shift,
 }: {
   y: number;
   x: number;

src/app/(components)/QueryProgressLog.tsx

@@ -17,7 +17,7 @@ export default function QueryProgressLog({
 }: {
   progress: MultipleQueryProgress;
 }) {
-  const [currentTime, setCurrentTime] = useState(() => Date.now());
+  const [, setCurrentTime] = useState(() => Date.now());
 
   useEffect(() => {
     if (!progress.error) {

src/app/(components)/SchemaView.tsx

@@ -8,7 +8,6 @@ import {
   OpenContextMenuList,
   openContextMenuFromEvent,
 } from "@/messages/openContextMenu";
-import OpacityLoading from "./OpacityLoading";
 import { useSchema } from "@/screens/DatabaseScreen/SchemaProvider";
 
 interface SchemaViewItemProps {

src/app/(components)/TopNavigation.tsx

@@ -1,7 +1,5 @@
 "use client";
 import * as React from "react";
-import Link from "next/link";
-
 import { cn } from "@/lib/utils";
 import {
   NavigationMenu,

src/drivers/DatabaseDriver.tsx

@@ -10,14 +10,57 @@ export interface DatabaseSchemaItem {
 export interface DatabaseTableColumn {
   name: string;
   type: string;
-  nullable: boolean;
-  pk: boolean;
+  pk?: boolean;
+  constraint?: DatabaseTableColumnConstraint;
+}
+
+export type DatabaseColumnConflict =
+  | "ROLLBACK"
+  | "ABORT"
+  | "FAIL"
+  | "IGNORE"
+  | "REPLACE";
+
+export interface DatabaseForeignKeyCaluse {
+  foreignTableName: string;
+  foreignColumns: string[];
+  columns?: string[];
+}
+
+export interface DatabaseTableColumnConstraint {
+  name?: string;
+
+  primaryKey?: boolean;
+  primaryColumns?: string[];
+  primaryKeyOrder?: "ASC" | "DESC";
+  primaryKeyConflict?: DatabaseColumnConflict;
+  autoIncrement?: boolean;
+
+  notNull?: boolean;
+  notNullConflict?: DatabaseColumnConflict;
+
+  unique?: boolean;
+  uniqueConflict?: DatabaseColumnConflict;
+
+  checkExpression?: string;
+
+  defaultValue?: unknown;
+  defaultExpression?: string;
+
+  collate?: string;
+
+  generatedExpression?: string;
+  generatedType?: "STORED" | "VIRTUAL";
+
+  foreignKey?: DatabaseForeignKeyCaluse;
 }
 
 export interface DatabaseTableSchema {
   columns: DatabaseTableColumn[];
   pk: string[];
   autoIncrement: boolean;
+  tableName?: string;
+  constraints?: DatabaseTableColumnConstraint[];
 }
 
 export default class DatabaseDriver {
@@ -87,7 +130,6 @@ export default class DatabaseDriver {
     const columns: DatabaseTableColumn[] = result.rows.map((row) => ({
       name: row.name?.toString() ?? "",
       type: row.type?.toString() ?? "",
-      nullable: !row.notnull,
       pk: !!row.pk,
     }));
 
@@ -105,7 +147,9 @@ export default class DatabaseDriver {
       if (seqRow && Number(seqRow[0] ?? 0) > 0) {
         hasAutoIncrement = true;
       }
-    } catch {}
+    } catch (e) {
+      console.error(e);
+    }
 
     return {
       columns,

src/lib/export-helper.ts

@@ -12,7 +12,7 @@ export function exportRowsToSqlInsert(
   headers: string[],
   records: unknown[][]
 ): string {
-  let result: string[] = [];
+  const result: string[] = [];
 
   const headersPart = headers.map(escapeIdentity).join(", ");
 
@@ -35,7 +35,7 @@ function cellToExcelValue(value: unknown) {
 }
 
 export function exportRowsToExcel(records: unknown[][]) {
-  let result: string[] = [];
+  const result: string[] = [];
 
   for (const record of records) {
     const line = record.map(cellToExcelValue).join("\t");

src/lib/internal-pubsub.ts

@@ -1,4 +1,5 @@
 export default class InternalPubSub {
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
   protected listeners: Record<string, ((obj: any) => void)[]> = {};
 
   addListener<T = unknown>(channel: string, callback: (obj: T) => void) {

src/lib/multiple-query.ts

@@ -56,7 +56,7 @@ export async function multipleQuery(
       }
     } catch (e) {
       log.end = Date.now();
-      log.error = (e as any).toString();
+      log.error = (e as Error).toString();
 
       if (onProgress) {
         onProgress({

src/lib/sql-execute-helper.ts

@@ -49,6 +49,6 @@ export async function executePlans(
     return { success: true, plans };
   } catch (e) {
     await driver.query("ROLLBACK");
-    return { success: false, error: (e as any).toString(), plans: [] };
+    return { success: false, error: (e as Error).toString(), plans: [] };
   }
 }

src/lib/sql-helper.test.ts

@@ -9,6 +9,7 @@ import {
   generateDeleteStatement,
   generateInsertStatement,
   generateUpdateStatement,
+  unescapeIdentity,
 } from "./sql-helper";
 
 describe("Escape SQL", () => {
@@ -31,6 +32,11 @@ describe("Escape SQL", () => {
     expect(escapeSqlValue(buffer)).toBe(`x'5D41402ABC4B2A76B9719D911017C592'`);
     expect(escapeSqlBinary(buffer)).toBe(`x'5D41402ABC4B2A76B9719D911017C592'`);
   });
+
+  it("unescape identity", () => {
+    expect(unescapeIdentity(`"users"`)).toBe("users");
+    expect(unescapeIdentity(`"us""ers"`)).toBe(`us"ers`);
+  });
 });
 
 describe("Generate SQL Statement", () => {
@@ -120,7 +126,9 @@ describe("Mapping sqlite column type to our table type", () => {
       expect(convertSqliteType(type)).toBe(TableColumnDataType.TEXT));
   }
 
-  expect(convertSqliteType("BLOB")).toBe(TableColumnDataType.BLOB);
+  it("BLOB column type", () => {
+    expect(convertSqliteType("BLOB")).toBe(TableColumnDataType.BLOB);
+  });
 
   const realType = ["REAL", "DOUBLE", "DOUBLE PRECISION", "FLOAT"];
   for (const type of realType) {

src/lib/sql-helper.ts

@@ -5,6 +5,13 @@ export function escapeIdentity(str: string) {
   return `"${str.replace(/"/g, `""`)}"`;
 }
 
+export function unescapeIdentity(str: string) {
+  let r = str.replace(/^["`]/g, "");
+  r = r.replace(/["`]$/g, "");
+  r = r.replace(/""/g, `"`);
+  return r;
+}
+
 export function escapeSqlString(str: string) {
   return `'${str.replace(/'/g, `''`)}'`;
 }
@@ -68,8 +75,8 @@ export function generateInsertStatement(
   tableName: string,
   value: Record<string, unknown>
 ): string {
-  let fieldPart: string[] = [];
-  let valuePart: unknown[] = [];
+  const fieldPart: string[] = [];
+  const valuePart: unknown[] = [];
 
   for (const entry of Object.entries(value)) {
     fieldPart.push(entry[0]);
@@ -84,7 +91,7 @@ export function generateDeleteStatement(
   tableName: string,
   where: Record<string, unknown>
 ) {
-  let wherePart: string = Object.entries(where)
+  const wherePart: string = Object.entries(where)
     .map(
       ([columnName, value]) =>
         `${escapeIdentity(columnName)} = ${escapeSqlValue(value)}`
@@ -105,7 +112,7 @@ export function generateUpdateStatement(
     })
     .join(", ");
 
-  let wherePart: string = Object.entries(where)
+  const wherePart: string = Object.entries(where)
     .map(
       ([columnName, value]) =>
         `${escapeIdentity(columnName)} = ${escapeSqlValue(value)}`