addCategory corrected

1 files changed, 34 insertions, 10 deletions

diff --git a/noncore/apps/zsafe/zsafe.cpp b/noncore/apps/zsafe/zsafe.cpp
index ef16b46..e86b3d2 100644
--- a/noncore/apps/zsafe/zsafe.cpp
+++ b/noncore/apps/zsafe/zsafe.cpp
@@ -2598,4 +2598,5 @@ void ZSafe::addCategory()
         QStringList::Iterator it = list.begin();
         QString categ;
+        QString firstCategory;
         dialog->CategoryField->clear(); // remove all items
         while( it != list.end() ) 
@@ -2619,4 +2620,6 @@ void ZSafe::addCategory()
               {
                  dialog->CategoryField->insertItem (categ, -1);
+                 if (firstCategory.isEmpty())
+                    firstCategory = categ;
               }
            }
@@ -2625,5 +2628,8 @@ void ZSafe::addCategory()
 
 
-        setCategoryDialogFields(dialog);
+        if (firstCategory.isEmpty())
+           setCategoryDialogFields(dialog);
+        else
+           setCategoryDialogFields(dialog, firstCategory);
 
         // CategoryDialog *dialog = new CategoryDialog(this, "Category", TRUE);
@@ -2787,16 +2793,31 @@ void ZSafe::delCategory()
 void ZSafe::setCategoryDialogFields(CategoryDialog *dialog)
 {
-   dialog->Field1->setText(getFieldLabel (selectedItem, "1", tr("Name")));
-   dialog->Field2->setText(getFieldLabel (selectedItem, "2", tr("Username")));
-   dialog->Field3->setText(getFieldLabel (selectedItem, "3", tr("Password")));
-   dialog->Field4->setText(getFieldLabel (selectedItem, "4", tr("Comment")));
-   dialog->Field5->setText(getFieldLabel (selectedItem, "5", tr("Field 4")));
-   dialog->Field6->setText(getFieldLabel (selectedItem, "6", tr("Field 5")));
+   if (!dialog)
+      return;
 
    QString icon;
-   Category *cat= categories.find (selectedItem->text(0));
-   if (cat)
+   if (selectedItem) 
    {
-      icon = cat->getIconName();
+      dialog->Field1->setText(getFieldLabel (selectedItem, "1", tr("Name")));
+      dialog->Field2->setText(getFieldLabel (selectedItem, "2", tr("Username")));
+      dialog->Field3->setText(getFieldLabel (selectedItem, "3", tr("Password")));
+      dialog->Field4->setText(getFieldLabel (selectedItem, "4", tr("Comment")));
+      dialog->Field5->setText(getFieldLabel (selectedItem, "5", tr("Field 4")));
+      dialog->Field6->setText(getFieldLabel (selectedItem, "6", tr("Field 5")));
+
+      Category *cat= categories.find (selectedItem->text(0));
+      if (cat)
+      {
+         icon = cat->getIconName();
+      }
+   }
+   else
+   {
+      dialog->Field1->setText(tr("Name"));
+      dialog->Field2->setText(tr("Username"));
+      dialog->Field3->setText(tr("Password"));
+      dialog->Field4->setText(tr("Comment"));
+      dialog->Field5->setText(tr("Field 4"));
+      dialog->Field6->setText(tr("Field 5"));
    }
 
@@ -2839,4 +2860,7 @@ void ZSafe::setCategoryDialogFields(CategoryDialog *dialog)
 void ZSafe::setCategoryDialogFields(CategoryDialog *dialog, QString category)
 {
+   if (!dialog)
+      return;
+
    dialog->Field1->setText(getFieldLabel (category, "1", tr("Name")));
    dialog->Field2->setText(getFieldLabel (category, "2", tr("Username")));