prismic-ts-codegen 0.1.4 → 0.1.5

package/src/generateTypes.ts

@@ -1,7 +1,7 @@
 import { Project, ModuleDeclarationKind } from "ts-morph";
 import type { CustomTypeModel, SharedSliceModel } from "@prismicio/types";
 
-import { BLANK_LINE_IDENTIFIER, NON_EDITABLE_FILE_HEADER } from "./constants";
+import { BLANK_LINE_IDENTIFIER } from "./constants";
 import { addTypeAliasForCustomType } from "./lib/addTypeAliasForCustomType";
 import { addTypeAliasForSharedSlice } from "./lib/addTypeAliasForSharedSlice";
 import { getSourceFileText } from "./lib/getSourceFileText";
@@ -15,6 +15,7 @@ export type GenerateTypesConfig = {
 	fieldConfigs?: FieldConfigs;
 	clientIntegration?: {
 		includeCreateClientInterface?: boolean;
+		includeContentNamespace?: boolean;
 	};
 };
 
@@ -29,12 +30,11 @@ export const generateTypes = (config: GenerateTypesConfig = {}) => {
 		moduleSpecifier: "@prismicio/types",
 		namespaceImport: "prismicT",
 		isTypeOnly: true,
-		leadingTrivia: NON_EDITABLE_FILE_HEADER,
 	});
 
 	sourceFile.addStatements(BLANK_LINE_IDENTIFIER);
 
-	sourceFile.addTypeAlias({
+	const simplifyTypeAlias = sourceFile.addTypeAlias({
 		name: "Simplify",
 		typeParameters: [
 			{
@@ -77,7 +77,10 @@ export const generateTypes = (config: GenerateTypesConfig = {}) => {
 		}
 	}
 
-	if (config.clientIntegration?.includeCreateClientInterface) {
+	if (
+		config.clientIntegration?.includeCreateClientInterface ||
+		config.clientIntegration?.includeContentNamespace
+	) {
 		sourceFile.addImportDeclaration({
 			moduleSpecifier: "@prismicio/client",
 			namespaceImport: "prismic",
@@ -90,28 +93,55 @@ export const generateTypes = (config: GenerateTypesConfig = {}) => {
 			declarationKind: ModuleDeclarationKind.Module,
 		});
 
-		clientModuleDeclaration.addInterface({
-			name: "CreateClient",
-			callSignatures: [
-				{
-					parameters: [
-						{
-							name: "repositoryNameOrEndpoint",
-							type: "string",
-						},
-						{
-							name: "options",
-							type: "prismic.ClientConfig",
-							hasQuestionToken: true,
-						},
-					],
-					returnType:
-						(config.customTypeModels?.length || 0) > 0
-							? "prismic.Client<AllDocumentTypes>"
-							: "prismic.Client",
-				},
-			],
-		});
+		if (config.clientIntegration.includeCreateClientInterface) {
+			clientModuleDeclaration.addInterface({
+				name: "CreateClient",
+				callSignatures: [
+					{
+						parameters: [
+							{
+								name: "repositoryNameOrEndpoint",
+								type: "string",
+							},
+							{
+								name: "options",
+								type: "prismic.ClientConfig",
+								hasQuestionToken: true,
+							},
+						],
+						returnType:
+							(config.customTypeModels?.length || 0) > 0
+								? "prismic.Client<AllDocumentTypes>"
+								: "prismic.Client",
+					},
+				],
+			});
+		}
+
+		if (config.clientIntegration.includeContentNamespace) {
+			const contentNamespaceDeclaration = clientModuleDeclaration.addModule({
+				name: "Content",
+				declarationKind: ModuleDeclarationKind.Namespace,
+			});
+
+			const exportSymbols = sourceFile
+				.getExportSymbols()
+				.filter((exportSymbol) => {
+					// The Simplify utility type should not
+					// be exported, but it is included in
+					// `getExportSymbols()`'s result.
+					return exportSymbol.getName() !== simplifyTypeAlias.getName();
+				});
+
+			contentNamespaceDeclaration.addExportDeclaration({
+				isTypeOnly: true,
+				namedExports: exportSymbols.map((exportSymbol) => {
+					return {
+						name: exportSymbol.getName(),
+					};
+				}),
+			});
+		}
 	}
 
 	return getSourceFileText(sourceFile);