fix: next build (#1472)

Author: eemmiillyy

cli/src/commands/schema/edit.test.ts

@@ -1,5 +1,7 @@
 import { beforeEach, expect, test, describe } from 'vitest';
-import {
+import { PgRollMigration } from '@xata.io/pgroll';
+import EditSchema, {
+  editsToMigrations,
   AddColumnPayload,
   AddTablePayload,
   ColumnAdditions,
@@ -8,9 +10,7 @@ import {
   DeleteColumnPayload,
   DeleteTablePayload,
   EditTablePayload
-} from './types';
-import { PgRollMigration } from '@xata.io/pgroll';
-import EditSchema, { editsToMigrations } from './edit';
+} from './edit';
 
 const column: AddColumnPayload['column'] = {
   name: 'col1',

cli/src/commands/schema/edit.ts

@@ -15,19 +15,6 @@ import {
 } from '@xata.io/pgroll';
 import chalk from 'chalk';
 import enquirer from 'enquirer';
-import {
-  AddColumnPayload,
-  AddTablePayload,
-  BranchSchemaFormatted,
-  ColumnAdditions,
-  ColumnEdits,
-  DeleteColumnPayload,
-  DeleteTablePayload,
-  EditColumnPayload,
-  EditTablePayload,
-  SelectChoice,
-  ValidationState
-} from './types.js';
 import {
   exhaustiveCheck,
   generateLinkReference,
@@ -42,6 +29,104 @@ import {
   xataColumnTypeToZeroValue
 } from '../../migrations/pgroll.js';
 
+export type BranchSchemaFormatted =
+  | {
+      schema: {
+        tables: {
+          name: string;
+          uniqueConstraints: Schemas.BranchSchema['tables'][number]['uniqueConstraints'];
+          checkConstraints: Schemas.BranchSchema['tables'][number]['checkConstraints'];
+          foreignKeys: Schemas.BranchSchema['tables'][number]['foreignKeys'];
+          columns: {
+            name: string;
+            type: string;
+            unique: boolean;
+            notNull: boolean;
+            defaultValue: any;
+            comment: string;
+          }[];
+        }[];
+      };
+    }
+  | undefined;
+
+export type ColumnData = {
+  name: string;
+  type: string;
+  unique: boolean;
+  nullable: boolean;
+  defaultValue?: string;
+  vector?: {
+    dimension: number;
+  };
+  originalName: string;
+  tableName: string;
+  link?: {
+    table: string;
+  };
+  file?: {
+    defaultPublicAccess: boolean;
+  };
+  'file[]'?: {
+    defaultPublicAccess: boolean;
+  };
+};
+
+export type AddTablePayload = {
+  type: 'add-table';
+  table: {
+    name: string;
+  };
+};
+
+export type EditTablePayload = {
+  type: 'edit-table';
+  table: {
+    name: string;
+    newName: string;
+  };
+};
+
+export type DeleteTablePayload = {
+  name: string;
+};
+
+export type AddColumnPayload = {
+  type: 'add-column';
+  tableName: string;
+  column: ColumnData;
+};
+
+export type EditColumnPayload = {
+  type: 'edit-column';
+  column: ColumnData;
+};
+
+export type DeleteColumnPayload = { [tableName: string]: string[] };
+
+export type FormatPayload = {
+  type: 'space' | 'migrate' | 'schema';
+};
+
+export type SelectChoice = {
+  name: FormatPayload | AddTablePayload | EditTablePayload | AddColumnPayload | EditColumnPayload;
+  message: string;
+  role?: string;
+  choices?: SelectChoice[];
+  disabled?: boolean;
+  hint?: string;
+};
+
+export type ValidationState = {
+  values: { name: string };
+  items: { name: string; input: string }[];
+  fields: { name: string; initial: string }[];
+};
+
+export type ColumnAdditions = { [tableName: string]: { [columnName: string]: AddColumnPayload['column'] } };
+
+export type ColumnEdits = { [tableName: string]: { [columnName: string]: AddColumnPayload['column'] } };
+
 const { Select, Snippet, Confirm } = enquirer as any;
 
 const xataTypes = [

cli/src/commands/schema/types.ts

@@ -4,7 +4,7 @@ import { OpRawSQL, OpRenameConstraint, PgRollOperation } from '@xata.io/pgroll';
 import path from 'path';
 import z from 'zod';
 import { XataClient } from '../base.js';
-import { BranchSchemaFormatted } from '../commands/schema/types.js';
+import { BranchSchemaFormatted } from '../commands/schema/edit.js';
 import { safeJSONParse, safeReadFile } from '../utils/files.js';
 import { migrationsDir, readMigrationsDir } from './files.js';
 import { MigrationFilePgroll, migrationFilePgroll } from './schema.js';