From ebbf75386499670209b4facf58022a556357c087 Mon Sep 17 00:00:00 2001
From: Kononnable <kononnable@gmail.com>
Date: Mon, 12 Aug 2019 21:35:37 +0200
Subject: [PATCH] option to add visibility to constructor #169

---
 src/entity.mst |  2 +-
 src/index.ts   | 21 ++++++++++++++++-----
 2 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/src/entity.mst b/src/entity.mst
index 634a2bb..97de093 100644
--- a/src/entity.mst
+++ b/src/entity.mst
@@ -37,7 +37,7 @@ import {BaseEntity,Column,Entity,Index,JoinColumn,JoinTable,ManyToMany,ManyToOne
     {{printPropertyVisibility}}{{toPropertyName ../tsName}}Id: {{#if isOneToOne}}{{toLazy ../tsType}}{{else}}{{toLazy (concat  ../tsType "[]")}}{{/if}};{{/if}}{{/relations}}
     {{/Columns}}
     {{#if GenerateConstructor}}
-    constructor(init?: Partial<{{toEntityName tsEntityName}}>) {
+    {{printPropertyVisibility}}constructor(init?: Partial<{{toEntityName tsEntityName}}>) {
 		Object.assign(this, init);
 	}
     {{/if}}
diff --git a/src/index.ts b/src/index.ts
index 6bd2858..b090f8d 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -378,6 +378,18 @@ async function GetUtilParametersByInquirer() {
                 type: "checkbox"
             }
         ])) as any).selected;
+
+        generationOptions.propertyVisibility = ((await inquirer.prompt([
+            {
+                choices: ["public", "protected", "private", "none"],
+                message:
+                    "Defines which visibility should have the generated property",
+                name: "propertyVisibility",
+                default: "none",
+                type: "list"
+            }
+        ])) as any).propertyVisibility;
+
         generationOptions.noConfigs = !customizations.includes("config");
         generationOptions.lazy = customizations.includes("lazy");
         generationOptions.activeRecord = customizations.includes(
@@ -450,11 +462,10 @@ async function GetUtilParametersByInquirer() {
         }
     ])) as any;
     if (saveConfig) {
-        await fs.writeJson(
-            path.resolve(process.cwd(), ".tomg-config"),
-            [connectionOptions, generationOptions],
-            { spaces: "\t" }
-        );
+        await fs.writeJson(path.resolve(process.cwd(), ".tomg-config"), [
+            connectionOptions,
+            generationOptions
+        ]);
         console.log(`[${new Date().toLocaleTimeString()}] Config file saved.`);
         console.warn(
             `\x1b[33m[${new Date().toLocaleTimeString()}] WARNING: Password was saved as plain text.\x1b[0m`