fix(effect-schema): preserve services for $type overrides

Author: danieljvdm

drizzle-orm/src/effect-schema/column.types.ts

@@ -31,8 +31,10 @@ type IsPgArrayColumn<TColumn extends Column<any>, TType extends ColumnTypeData>
 	: GetArrayDepth<TColumn['_']['data']> extends 0 ? false
 	: true;
 
+type OverrideSchemaType<TSchema extends SchemaTop, TData> = Omit<TSchema, 'Type' | '~rebuild.out'> & Schema<TData>;
+
 type ApplyTypeOverride<TSchema extends SchemaTop, TData> = Equal<TSchema['Type'], TData> extends true ? TSchema
-	: Schema<TData>;
+	: OverrideSchemaType<TSchema, TData>;
 
 export type GetEffectSchemaType<
 	TColumn extends Column<any>,

drizzle-orm/tests/effect-schema.test.ts

@@ -19,6 +19,8 @@ test('createSelectSchema preserves $type overrides', () => {
 	expect(s.make(result.ast).toString()).toStrictEqual(s.make(expected.ast).toString());
 
 	type _ = Expect<Equal<s.Schema.Type<typeof result>['id'], MyBrandedId>>;
+	type _Services = Expect<Equal<typeof result['DecodingServices'], never>>;
+	type _FieldServices = Expect<Equal<typeof result.fields.id['DecodingServices'], never>>;
 });
 
 test('createSelectSchema preserves literal string $type overrides', () => {
@@ -35,4 +37,7 @@ test('createSelectSchema preserves literal string $type overrides', () => {
 	expect(s.make(result.ast).toString()).toStrictEqual(s.make(expected.ast).toString());
 
 	type _ = Expect<Equal<s.Schema.Type<typeof result>['id'], Foo>>;
+	type _Field = Expect<Equal<s.Schema.Type<typeof result.fields.id>, Foo>>;
+	type _Services = Expect<Equal<typeof result['DecodingServices'], never>>;
+	type _FieldServices = Expect<Equal<typeof result.fields.id['DecodingServices'], never>>;
 });