From 0e4308f9b77914cdeec1a788a14212b391406e35 Mon Sep 17 00:00:00 2001
From: Nikola Petrov <nikolape7@gmail.com>
Date: Sun, 27 Aug 2023 23:15:32 +0200
Subject: [PATCH] add length to generate

---
 Password_manager/source/arg_func.cpp | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/Password_manager/source/arg_func.cpp b/Password_manager/source/arg_func.cpp
index 64e3d61..0da0a1d 100644
--- a/Password_manager/source/arg_func.cpp
+++ b/Password_manager/source/arg_func.cpp
@@ -1,7 +1,9 @@
+#include <string>
+#include <iostream>
+#include <sstream>
 
 #include "arg_func.h"
 #include "func.h"
-#include <string>
 #include "win.h"
 #include "func.h"
 #include "buffer.h"
@@ -15,8 +17,8 @@ void print_args()
 	printf_s(" -h         \t print this message\n");
 	printf_s(" <label>    \t get password of this label can use GLOB\n");
 	printf_s(" -g <label> \t generate password of this label (or update if exists)\n");
-	printf_s(" -d <label> \t delete password of this label\n");
 	printf_s(" -i <label> \t input password for this label (or update if exists)\n");
+	printf_s(" -d <label> \t delete password of this label\n");
 	printf_s(" -s <label> \t show password for this label\n");
 	printf_s(" -l         \t list all labels\n");
 	printf_s(" -p         \t print all passwords\n");
@@ -68,10 +70,23 @@ std::optional<LoginInfoPointer> arg_get(Buffer& decrypted_buffer, const char* la
 
 std::optional<LoginInfoPointer> arg_generate(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto)
 {
+	int default_length = 15;
+	printf_s("Input password length [%d]:", default_length);
+	std::string input;
+	int length = default_length;
+	
+	std::getline(std::cin, input);
+
+	if (!input.empty())
+	{
+		std::istringstream iss(input);
+		if (!(iss >> length)) length = default_length;
+	}
+
 	printf_s("Generating password for %s\n", label);
 
 	std::string password;
-	generate_password(password, 15);
+	generate_password(password, length);
 
 	int pass = find_logininfo_in_buffer(decrypted_buffer, label);
 	if (pass >= 0)	delete_logininfo_from_buffer(decrypted_buffer, pass);