Add an optional language argument

avelynhc · Sep 13, 2023 · 0f155cf · 0f155cf
1 parent 67fe86a
commit 0f155cf
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 7 deletions.
diff --git a/src/helper/convertToHTML.ts b/src/helper/convertToHTML.ts
@@ -1,8 +1,8 @@
 const htmlFs = require("fs");
 
-module.exports.convertToHTML = function (title: string, cssLink: string, body: string, outputFolder: string) {
+module.exports.convertToHTML = function (title: string, cssLink: string, body: string, selectedLang: string, outputFolder: string) {
     const result:string = `<!doctype html>
-<html lang="en">
+<html lang="${selectedLang}">
 <head>
     <meta charset="utf-8">
     <title>${title}</title>

diff --git a/src/helper/readFile.ts b/src/helper/readFile.ts
@@ -3,7 +3,7 @@ import readFilePath from "path";
 const html = require("./convertToHTML");
 let body:string = "";
 
-module.exports.readFile = function (inputPath: string, cssLink: string, outputFolder: string) {
+module.exports.readFile = function (inputPath: string, cssLink: string, selectedLang: string, outputFolder: string) {
     // parse a title from the input file, which will be used to populate <title>...</title>
     const title:string = readFilePath.basename(inputPath, ".txt");
 
@@ -18,6 +18,6 @@ module.exports.readFile = function (inputPath: string, cssLink: string, outputFo
         console.error(err);
     }
 
-    html.convertToHTML(title, cssLink, body, outputFolder);
+    html.convertToHTML(title, cssLink, body, selectedLang, outputFolder);
     return title;
 };
diff --git a/src/helper/readFolder.ts b/src/helper/readFolder.ts
@@ -1,14 +1,15 @@
 import readFolderFs from "fs";
 const fileModule = require("./readFile");
 
-module.exports.readFolder = function (path: string, cssLink: string, outputFolder: string) {
+module.exports.readFolder = function (path: string, cssLink: string, selectedLang: string,  outputFolder: string) {
     readFolderFs.readdir(path, function (err: any, files: any[]) {
         if (err) return console.log(err);
         const textFiles:any[] = files.filter(file => file.split('.').pop()==='txt');
         textFiles.forEach(function (file) {
             fileModule.readFile(
                 `${path}/${file}`,
                 cssLink,
+                selectedLang,
                 outputFolder
             );
         });

diff --git a/src/index.ts b/src/index.ts
@@ -20,6 +20,13 @@ const argv = yargs
         type: "string",
         demandOption: false,
     })
+    .option("l", {
+        alias: "lang",
+        describe: "Language used in generated file",
+        default: "en-CA",
+        type: "string",
+        demandOption: false,
+    })
     .alias("h", "help")
     .alias("v", "version")
     .version(appInfo.name + " " + appInfo.version)
@@ -41,6 +48,7 @@ if (!fs.existsSync(outputFolder)) {
 }
 console.log('Output folder is successfully created!');
 
+const selectedLang:string = argv.lang;
 // check if input is an individual file or directory
 fs.stat(argv.input, (err: any, stats: { isDirectory: () => any; isFile: () => any; }) => {
     if (err) {
@@ -49,9 +57,9 @@ fs.stat(argv.input, (err: any, stats: { isDirectory: () => any; isFile: () => an
     }
 
     if (stats.isDirectory()) {
-        folder.readFolder(argv.input, cssLink, outputFolder);
+        folder.readFolder(argv.input, cssLink, selectedLang, outputFolder);
     } else if (stats.isFile() && path.extname(argv.input) === ".txt") {
-        file.readFile(argv.input, cssLink, outputFolder);
+        file.readFile(argv.input, cssLink, selectedLang, outputFolder);
     } else {
         console.error("Error: file extension should be .txt");
     }