fix: preprocess types (#45)

diplodoc-platform · Apr 15, 2024 · 6b21901 · 6b21901
1 parent c75a06a
commit 6b21901
Showing 1 changed file with 52 additions and 13 deletions.
diff --git a/src/preprocess.ts b/src/preprocess.ts
@@ -1,5 +1,16 @@
-import {Block, ConstructorBlock, Lang} from '@gravity-ui/page-constructor';
-import {config, contentTransformer} from '@gravity-ui/page-constructor/server';
+import {
+    Block,
+    ConstructorBlock,
+    NavigationData as ConstructorNavigaitonData,
+    PageContent as ConstructorPageContentBase,
+} from '@gravity-ui/page-constructor';
+import {
+    BlocksConfig,
+    Parser,
+    TransformerRaw,
+    config,
+    contentTransformer,
+} from '@gravity-ui/page-constructor/server';
 
 export interface MetaData {
     title: string;
@@ -10,22 +21,37 @@ export interface PageContentBase {
     meta?: MetaData;
 }
 
+export enum Lang {
+    RU = 'ru',
+    EN = 'en',
+}
+
 export type PageContent<T> = T & PageContentBase;
 export type ConstructorPageContent = PageContent<ConstructorPageContentBase>;
 export type NavigationData = PageContent<ConstructorNavigaitonData>;
 export type ConfigData = ConstructorPageContent | NavigationData;
 
 export interface PreloadParams {
-    locale: string;
+    lang: Lang;
     pageName: string;
     pageReferer?: string;
 }
 
+interface BlockConfig {
+    transformer: TransformerRaw;
+    fields?: string[];
+    parser?: Parser;
+}
+
 export function isPageConfig(config: ConfigData): config is ConstructorPageContent {
     return 'blocks' in config;
 }
 
-export function preprocess(content: ConfigData, params: PreloadParams, customYfmTransformer) {
+export function preprocess(
+    content: ConfigData,
+    params: PreloadParams,
+    customYfmTransformer: TransformerRaw,
+) {
     const {lang} = params;
 
     if (isPageConfig(content) && content.blocks) {
@@ -38,19 +64,32 @@ export function preprocess(content: ConfigData, params: PreloadParams, customYfm
     return content;
 }
 
-function replaceTransformer(config: BlocksConfig, newTransformer: Function): BlocksConfig {
-    return Object.keys(config).reduce((newConfig, key) => {
-        const subBlockType = key as SubBlockType;
-        newConfig[subBlockType] = config[subBlockType].map((block) => {
-            return block.transformer.name === 'yfmTransformer'
-                ? {...block, transformer: newTransformer}
-                : block;
-        });
+function replaceTransformer(config: BlocksConfig, newTransformer: TransformerRaw): BlocksConfig {
+    return Object.keys(config).reduce((newConfig, subBlockType) => {
+        const subBlock = config[subBlockType];
+
+        if (Array.isArray(subBlock)) {
+            newConfig[subBlockType] = subBlock.map((block: {transformer: {name: string}}) => {
+                return block.transformer.name === 'yfmTransformer'
+                    ? {...block, transformer: newTransformer}
+                    : block;
+            }) as BlockConfig[];
+        } else {
+            newConfig[subBlockType] =
+                subBlock.transformer.name === 'yfmTransformer'
+                    ? {...subBlock, transformer: newTransformer}
+                    : subBlock;
+        }
+
         return newConfig;
     }, {} as BlocksConfig);
 }
 
-function transformBlocks(blocks: ConstructorBlock[], lang: Lang, customYfmTransformer) {
+function transformBlocks(
+    blocks: ConstructorBlock[],
+    lang: Lang,
+    customYfmTransformer: TransformerRaw,
+) {
     const customConfig = replaceTransformer(config, customYfmTransformer);
 
     return contentTransformer({