fix(pgsql-types): add enum imports to generated types

Author: pyramation

packages/pgsql-types/scripts/generate-types.ts

@@ -36,10 +36,26 @@ function isPrimitiveType(type: string): boolean {
   return type in primitiveTypeMap;
 }
 
+function isEnumType(type: string): boolean {
+  return !isPrimitiveType(type) && !schemaMap.has(type) && type !== 'Node';
+}
+
 function getTsType(type: string): string {
   return primitiveTypeMap[type] || type;
 }
 
+function collectEnumTypes(): Set<string> {
+  const enumTypes = new Set<string>();
+  for (const nodeSpec of runtimeSchema) {
+    for (const field of nodeSpec.fields) {
+      if (isEnumType(field.type)) {
+        enumTypes.add(field.type);
+      }
+    }
+  }
+  return enumTypes;
+}
+
 function generateWrappedUnion(tags: string[]): string {
   if (tags.length === 0) {
     return 'Node';
@@ -118,7 +134,13 @@ function generateTypes(metadata: AllFieldMetadata): string {
   lines.push(' */');
   lines.push('');
 
+  const enumTypes = collectEnumTypes();
+  const sortedEnums = [...enumTypes].sort();
+  
   lines.push("import type { Node } from '@pgsql/types';");
+  if (sortedEnums.length > 0) {
+    lines.push(`import { ${sortedEnums.join(', ')} } from '@pgsql/enums';`);
+  }
   lines.push("export type { Node } from '@pgsql/types';");
   lines.push("export * from '@pgsql/enums';");
   lines.push('');

packages/pgsql-types/src/types.ts

@@ -7,6 +7,7 @@
  */
 
 import type { Node } from '@pgsql/types';
+import { A_Expr_Kind, AggSplit, AlterPublicationAction, AlterSubscriptionType, AlterTSConfigType, AlterTableType, BoolExprType, BoolTestType, CTEMaterialize, CmdType, CoercionContext, CoercionForm, ConstrType, DefElemAction, DiscardMode, DropBehavior, FetchDirection, FunctionParameterMode, GrantTargetType, GroupingSetKind, ImportForeignSchemaType, JoinType, JsonBehaviorType, JsonConstructorType, JsonEncoding, JsonExprOp, JsonFormatType, JsonQuotes, JsonTableColumnType, JsonValueType, JsonWrapper, KeywordKind, LimitOption, LockClauseStrength, LockWaitPolicy, MergeMatchKind, MinMaxOp, NullTestType, ObjectType, OnCommitAction, OnConflictAction, OverridingKind, ParamKind, PartitionRangeDatumKind, PartitionStrategy, PublicationObjSpecType, QuerySource, RTEKind, ReindexObjectType, RoleSpecType, RoleStmtType, RowCompareType, SQLValueFunctionOp, SetOperation, SortByDir, SortByNulls, SubLinkType, TableFuncType, Token, TransactionStmtKind, VariableSetKind, ViewCheckOption, WCOKind, XmlExprOp, XmlOptionType } from '@pgsql/enums';
 export type { Node } from '@pgsql/types';
 export * from '@pgsql/enums';