Writing generic functions with interdependent / recursive schemas

[rhengles]

I'm having a problem related to #558. I'm trying to use recursive types with z.lazy but with generic functions of my own.
Basically I want to create a Zod parser to a GraphQL schema, and validate all the relations that the GraphQL schema can return. It's objects all the way down...

I successfully created a generic function that returns the accurate type for the object without subobjects. But whatever I type, including forcibly casting the result, something like this:

export function getJoinedEntityParser<
	OwnShape extends z.ZodRawShape,
	OwnParser extends z.ZodObject<OwnShape>,
	RefsShape extends z.ZodRawShape,
	RefsParser extends z.ZodObject<RefsShape>,
>(
	ownParser: OwnParser,
	refsParser: RefsParser,
) {
	type JoinedRawShape = z.objectUtil.MergeShapes<OwnShape, RefsShape> // I also tried z.extendShape<>
	type JoinedRawParser = z.ZodType<JoinedRawShape>
	const parser = z.lazy(() => ownParser.merge(refsParser)) as unknown as JoinedRawParser
	return parser
}

When I call this, with the following:

export const entityOneJoinedParser = getJoinedEntityParser(
	entityOneOnlyParser,
	z.object({
		entityTwo: entityTwoJoinedParser,
	}),
)

// then I try to extract the type
export type EntityOneJoined = z.output<typeof entityOneJoinedParser>

Then VSCode only says this when I hover the type (using z.objectUtil.MergeShapes):

type EntityOneJoined = {} & z.ZodRawShape

And this is what it says when I try z.extendShape instead:

type EntityOneJoined = Omit<z.ZodRawShape, string | number> & z.ZodRawShape

I want to get it to say:

type EntityOneJoined = {
	entityOneProp1: <type>,
	// ... entityOnePropN: <type>,
	entityTwo: {
		entityTwoProp1: <type>,
		// ... entityTwoPropN: <type>,
	},
}

Is this possible?

[rhengles]

Another attempt:

export function getJoinedEntityParser<
	OwnShape extends z.ZodRawShape,
	RefsShape extends z.ZodRawShape,
	OwnParser extends z.ZodObject<OwnShape>,
	RefsParser extends z.ZodObject<RefsShape>,
>(
	ownParser: OwnParser,
	refsParser: RefsParser,
) {
	// The result is the same if I use z.objectUtil.MergeShapes below
	type JoinedShape = z.extendShape<OwnShape, RefsParser['_shape']> // I'd use simply 'RefsShape', but TS complains
	type JoinedParser = z.ZodType<
		z.baseObjectOutputType<JoinedShape>,
		z.ZodObjectDef<JoinedShape, OwnParser['_unknownKeys'], OwnParser['_catchall']>,
		z.baseObjectInputType<JoinedShape>
	>
	const parser = z.lazy(() => ownParser.merge(refsParser)) as unknown as JoinedParser
	return parser
}

Then this is what I get when I try to use it:

export const entityOneJoinedParser = getJoinedEntityParser(
	entityOneOnlyParser,
	z.object({
		entityTwo: entityTwoJoinedParser,
	}),
)

// then I try to extract the type
export type EntityOneJoined = z.output<typeof entityOneJoinedParser>

// this is what VSCode shows when I hover the type
type EntityOneJoined = {
	[x: string]: any;
	[x: number]: any;
}