Add custom field support

src/password_manager/entry_management.py

@@ -112,6 +112,7 @@ class EntryManager:
         url: Optional[str] = None,
         blacklisted: bool = False,
         notes: str = "",
+        custom_fields: List[Dict[str, Any]] | None = None,
     ) -> int:
         """
         Adds a new entry to the encrypted JSON index file.
@@ -138,6 +139,7 @@ class EntryManager:
                 "type": EntryType.PASSWORD.value,
                 "kind": EntryType.PASSWORD.value,
                 "notes": notes,
+                "custom_fields": custom_fields or [],
             }
 
             logger.debug(f"Added entry at index {index}: {data['entries'][str(index)]}")
@@ -414,6 +416,8 @@ class EntryManager:
             entry = data.get("entries", {}).get(str(index))
 
             if entry:
+                if entry.get("type", entry.get("kind")) == EntryType.PASSWORD.value:
+                    entry.setdefault("custom_fields", [])
                 logger.debug(f"Retrieved entry at index {index}: {entry}")
                 return entry
             else:
@@ -441,6 +445,7 @@ class EntryManager:
         label: Optional[str] = None,
         period: Optional[int] = None,
         digits: Optional[int] = None,
+        custom_fields: List[Dict[str, Any]] | None = None,
     ) -> None:
         """
         Modifies an existing entry based on the provided index and new values.
@@ -500,6 +505,12 @@ class EntryManager:
                 entry["notes"] = notes
                 logger.debug(f"Updated notes for index {index}.")
 
+            if custom_fields is not None:
+                entry["custom_fields"] = custom_fields
+                logger.debug(
+                    f"Updated custom fields for index {index}: {custom_fields}"
+                )
+
             data["entries"][str(index)] = entry
             logger.debug(f"Modified entry at index {index}: {entry}")
 
@@ -629,11 +640,18 @@ class EntryManager:
                 username = entry.get("username", "")
                 url = entry.get("url", "")
                 notes = entry.get("notes", "")
+                custom_fields = entry.get("custom_fields", [])
+                custom_match = any(
+                    query_lower in str(cf.get("label", "")).lower()
+                    or query_lower in str(cf.get("value", "")).lower()
+                    for cf in custom_fields
+                )
                 if (
                     query_lower in website.lower()
                     or query_lower in username.lower()
                     or query_lower in url.lower()
                     or query_lower in notes.lower()
+                    or custom_match
                 ):
                     results.append(
                         (

src/password_manager/manager.py

@@ -867,6 +867,18 @@ class PasswordManager:
             url = input("Enter the URL (optional): ").strip()
             notes = input("Enter notes (optional): ").strip()
 
+            custom_fields: list[dict[str, object]] = []
+            while True:
+                add_field = input("Add custom field? (y/N): ").strip().lower()
+                if add_field != "y":
+                    break
+                label = input("  Field label: ").strip()
+                value = input("  Field value: ").strip()
+                hidden = input("  Hidden field? (y/N): ").strip().lower() == "y"
+                custom_fields.append(
+                    {"label": label, "value": value, "is_hidden": hidden}
+                )
+
             length_input = input(
                 f"Enter desired password length (default {DEFAULT_PASSWORD_LENGTH}): "
             ).strip()
@@ -893,6 +905,7 @@ class PasswordManager:
                 url,
                 blacklisted=False,
                 notes=notes,
+                custom_fields=custom_fields,
             )
 
             # Mark database as dirty for background sync
@@ -1429,6 +1442,20 @@ class PasswordManager:
                     or notes
                 )
 
+                edit_fields = input("Edit custom fields? (y/N): ").strip().lower()
+                custom_fields = None
+                if edit_fields == "y":
+                    custom_fields = []
+                    while True:
+                        label = input("  Field label (leave blank to finish): ").strip()
+                        if not label:
+                            break
+                        value = input("  Field value: ").strip()
+                        hidden = input("  Hidden field? (y/N): ").strip().lower() == "y"
+                        custom_fields.append(
+                            {"label": label, "value": value, "is_hidden": hidden}
+                        )
+
                 self.entry_manager.modify_entry(
                     index,
                     blacklisted=new_blacklisted,
@@ -1436,6 +1463,7 @@ class PasswordManager:
                     label=new_label,
                     period=new_period,
                     digits=new_digits,
+                    custom_fields=custom_fields,
                 )
             else:
                 website_name = entry.get("website")
@@ -1500,12 +1528,27 @@ class PasswordManager:
                     or notes
                 )
 
+                edit_fields = input("Edit custom fields? (y/N): ").strip().lower()
+                custom_fields = None
+                if edit_fields == "y":
+                    custom_fields = []
+                    while True:
+                        label = input("  Field label (leave blank to finish): ").strip()
+                        if not label:
+                            break
+                        value = input("  Field value: ").strip()
+                        hidden = input("  Hidden field? (y/N): ").strip().lower() == "y"
+                        custom_fields.append(
+                            {"label": label, "value": value, "is_hidden": hidden}
+                        )
+
                 self.entry_manager.modify_entry(
                     index,
                     new_username,
                     new_url,
                     new_blacklisted,
                     new_notes,
+                    custom_fields=custom_fields,
                 )
 
             # Mark database as dirty for background sync