From d589a14ddd954e1dc82d2a1c0bcb73321bda6538 Mon Sep 17 00:00:00 2001
From: thePR0M3TH3AN <53631862+PR0M3TH3AN@users.noreply.github.com>
Date: Wed, 2 Jul 2025 22:48:37 -0400
Subject: [PATCH] Prompt for notes when adding or modifying

---
 src/password_manager/manager.py    | 26 ++++++++++++++++++++++++--
 src/tests/test_manager_workflow.py |  2 ++
 2 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/src/password_manager/manager.py b/src/password_manager/manager.py
index 8819155..b0595e5 100644
--- a/src/password_manager/manager.py
+++ b/src/password_manager/manager.py
@@ -801,6 +801,7 @@ class PasswordManager:
 
             username = input("Enter the username (optional): ").strip()
             url = input("Enter the URL (optional): ").strip()
+            notes = input("Enter notes (optional): ").strip()
 
             length_input = input(
                 f"Enter desired password length (default {DEFAULT_PASSWORD_LENGTH}): "
@@ -822,7 +823,12 @@ class PasswordManager:
 
             # Add the entry to the index and get the assigned index
             index = self.entry_manager.add_entry(
-                website_name, length, username, url, blacklisted=False
+                website_name,
+                length,
+                username,
+                url,
+                blacklisted=False,
+                notes=notes,
             )
 
             # Mark database as dirty for background sync
@@ -881,6 +887,10 @@ class PasswordManager:
             username = entry.get("username")
             url = entry.get("url")
             blacklisted = entry.get("blacklisted")
+            notes = entry.get("notes", "")
+            notes = entry.get("notes", "")
+            notes = entry.get("notes", "")
+            notes = entry.get("notes", "")
 
             print(
                 colored(
@@ -947,6 +957,7 @@ class PasswordManager:
             username = entry.get("username")
             url = entry.get("url")
             blacklisted = entry.get("blacklisted")
+            notes = entry.get("notes", "")
 
             # Display current values
             print(
@@ -996,9 +1007,20 @@ class PasswordManager:
                 )
                 new_blacklisted = blacklisted
 
+            new_notes = (
+                input(
+                    f'Enter new notes (leave blank to keep "{notes or "N/A"}"): '
+                ).strip()
+                or notes
+            )
+
             # Update the entry
             self.entry_manager.modify_entry(
-                index, new_username, new_url, new_blacklisted
+                index,
+                new_username,
+                new_url,
+                new_blacklisted,
+                new_notes,
             )
 
             # Mark database as dirty for background sync
diff --git a/src/tests/test_manager_workflow.py b/src/tests/test_manager_workflow.py
index 009c640..6b0833c 100644
--- a/src/tests/test_manager_workflow.py
+++ b/src/tests/test_manager_workflow.py
@@ -51,11 +51,13 @@ def test_manager_workflow(monkeypatch):
                 "",  # username
                 "",  # url
                 "",  # length (default)
+                "",  # notes
                 "0",  # retrieve index
                 "0",  # modify index
                 "user",  # new username
                 "",  # new url
                 "",  # blacklist status
+                "",  # new notes
             ]
         )
         monkeypatch.setattr("builtins.input", lambda *args, **kwargs: next(inputs))