feat: add disableHtmlExt config feature to links and redirect

diplodoc-platform · Apr 26, 2024 · 7e65877 · 7e65877
1 parent 98ca605
commit 7e65877
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/constants.ts b/src/constants.ts
@@ -133,6 +133,8 @@ export const REGEXP_INCLUDE_FILE_PATH = /(?<=[(]).+(?=[)])/g;
 // Regexp result: authorLogin
 export const REGEXP_AUTHOR = /(?<=author:\s).+(?=\r?\n)/g;
 
+export const REGEXP_EXT_HTML = /(\.html)$/gi;
+
 export const MIN_CHUNK_SIZE = Number(process.env.MIN_CHUNK_SIZE) || 1000;
 export const WORKERS_COUNT = Number(process.env.WORKERS_COUNT) || os.cpus().length - 1;
 export const PAGE_PROCESS_CONCURRENCY = Number(process.env.PAGE_PROCESS_CONCURRENCY) || 500;

diff --git a/src/steps/processPages.ts b/src/steps/processPages.ts
@@ -10,6 +10,7 @@ import shell from 'shelljs';
 import {
     Lang,
     PAGE_PROCESS_CONCURRENCY,
+    REGEXP_EXT_HTML,
     ResourceType,
     SINGLE_PAGE_DATA_FILENAME,
     SINGLE_PAGE_FILENAME,
@@ -189,15 +190,18 @@ async function saveSinglePages() {
 }
 
 function saveRedirectPage(outputDir: string): void {
-    const {lang, langs} = ArgvService.getConfig();
+    const {lang, langs, disableHtmlExt} = ArgvService.getConfig();
 
     const redirectLang = lang || langs?.[0] || Lang.RU;
-    const redirectLangRelativePath = `./${redirectLang}/index.html`;
+    let redirectLangRelativePath = `./${redirectLang}/index.html`;
 
     const redirectPagePath = join(outputDir, 'index.html');
     const redirectLangPath = join(outputDir, redirectLangRelativePath);
 
     if (!existsSync(redirectPagePath) && existsSync(redirectLangPath)) {
+        if (disableHtmlExt) {
+            redirectLangRelativePath = redirectLangRelativePath.replace(REGEXP_EXT_HTML, '');
+        }
         const content = generateStaticRedirect(redirectLang, redirectLangRelativePath);
         writeFileSync(redirectPagePath, content);
     }