5 files changed, 68 insertions, 3 deletions

diff --git a/kabc/addressee.cpp b/kabc/addressee.cpp
index eec0f1f..3ce733d 100644
--- a/kabc/addressee.cpp
+++ b/kabc/addressee.cpp
@@ -1993,12 +1993,16 @@ Resource *Addressee::resource() const
 
 //US
 QString Addressee::resourceLabel()
 {
   return i18n("Resource");
 }
+QString Addressee::categoryLabel()
+{
+  return i18n("Category");
+}
 
 void Addressee::setChanged( bool value )
 {
     detach();
     mData->changed = value;
 }
diff --git a/kabc/addressee.h b/kabc/addressee.h
index 2c81c1a..08d2f56 100644
--- a/kabc/addressee.h
+++ b/kabc/addressee.h
@@ -820,13 +820,13 @@ class Addressee
 
     /**
       Return resourcelabel.
      */
     //US
     static QString resourceLabel();
-    
+    static QString categoryLabel();
     /**
             Mark addressee as changed.
            */
     void setChanged( bool value );
 
           /**
diff --git a/kabc/field.cpp b/kabc/field.cpp
index d95cd19..89d0b77 100644
--- a/kabc/field.cpp
+++ b/kabc/field.cpp
@@ -54,12 +54,13 @@ class Field::FieldImpl
       GivenName,
       AdditionalName,
       Prefix,
       Suffix,
       NickName,
       Birthday,
+      Category,
       HomeAddressStreet,
       HomeAddressLocality,
       HomeAddressRegion,
       HomeAddressPostalCode,
       HomeAddressCountry,
       HomeAddressLabel,
@@ -192,12 +193,14 @@ QString Field::label()
     case FieldImpl::Note:
       return Addressee::noteLabel();
     case FieldImpl::Url:
       return Addressee::urlLabel();
     case FieldImpl::Resource:
       return Addressee::resourceLabel();
+    case FieldImpl::Category:
+      return Addressee::categoryLabel();
     case FieldImpl::Sip:
       return Addressee::sipLabel();
     case FieldImpl::CustomField:
       return mImpl->label();
     default:
       return i18n("Unknown Field");
@@ -274,12 +277,14 @@ QString Field::value( const KABC::Addressee &a )
         return QString::null;
     case FieldImpl::Url:
       return a.url().prettyURL();
 //US    
     case FieldImpl::Resource:
       return a.resource()->resourceName();
+    case FieldImpl::Category:
+        return a.categories().join(",");
     case FieldImpl::HomePhone:
       return a.phoneNumber( PhoneNumber::Home ).number();
     case FieldImpl::BusinessPhone:
       return a.phoneNumber( PhoneNumber::Work ).number();
     case FieldImpl::MobilePhone:
       return a.phoneNumber( PhoneNumber::Cell ).number();
@@ -392,12 +397,13 @@ Field::List Field::allFields()
     createField( FieldImpl::GivenName, Frequent );
     createField( FieldImpl::AdditionalName );
     createField( FieldImpl::Prefix );
     createField( FieldImpl::Suffix );
     createField( FieldImpl::NickName, Personal );
     createField( FieldImpl::Birthday, Personal );
+    createField( FieldImpl::Category );
     createField( FieldImpl::HomeAddressStreet, Address|Personal );
     createField( FieldImpl::HomeAddressLocality, Address|Personal );
     createField( FieldImpl::HomeAddressRegion, Address|Personal );
     createField( FieldImpl::HomeAddressPostalCode, Address|Personal );
     createField( FieldImpl::HomeAddressCountry, Address|Personal );
     createField( FieldImpl::HomeAddressLabel, Address|Personal );
diff --git a/kaddressbook/kabcore.cpp b/kaddressbook/kabcore.cpp
index e6bdde9..2c2f1a0 100644
--- a/kaddressbook/kabcore.cpp
+++ b/kaddressbook/kabcore.cpp
@@ -1873,21 +1873,29 @@ void KABCore::initActions()
 
   mActionWhoAmI = new KAction( i18n( "Set Who Am I" ), "personal", 0, this,
                                SLOT( setWhoAmI() ), actionCollection(),
                                "set_personal" );
 
 
-
-
   mActionCategories = new KAction( i18n( "Set Categories" ), 0, this,
                                    SLOT( setCategories() ), actionCollection(),
                                    "edit_set_categories" );
 
   mActionRemoveVoice = new KAction( i18n( "Remove \"voice\"..." ), 0, this,
                                     SLOT( removeVoice() ), actionCollection(),
                                    "remove_voice" );
+
+  mActionSetCat= new KAction( i18n( "Set categories..." ), 0, this,
+                                    SLOT( setCat() ), actionCollection(),
+                                   "remove_voice" );
+
+
+  mActionAddCat= new KAction( i18n( "Add categories..." ), 0, this,
+                                    SLOT( addCat() ), actionCollection(),
+                                   "remove_voice" );
+
   mActionImportOL = new KAction( i18n( "Import from Outlook..." ), 0, this,
                                     SLOT( importFromOL() ), actionCollection(),
                                    "import_OL" );
 #ifdef KAB_EMBEDDED
  mActionLicence = new KAction( i18n( "Licence" ), 0,
                  this, SLOT( showLicence() ), actionCollection(),
@@ -2011,12 +2019,14 @@ void KABCore::addActionsManually()
   mActionPaste->plug( editMenu );
   mActionDelete->plug( editMenu );
   editMenu->insertSeparator();
   mActionSelectAll->plug( editMenu );
 
   mActionRemoveVoice->plug( changeMenu );
+  mActionSetCat->plug( changeMenu );
+  mActionAddCat->plug( changeMenu );
   // settings menu
 //US special menuentry to configure the addressbook resources. On KDE
 // you do that through the control center !!!
   mActionConfigResources->plug( settingsMenu );
   settingsMenu->insertSeparator();
 
@@ -2093,12 +2103,52 @@ void KABCore::addActionsManually()
 
 }
 void KABCore::showLicence()
 {
     KApplication::showLicence();
 }
+void KABCore::setCat()
+{
+    setCategories( true );
+}
+void KABCore::addCat()
+{
+    setCategories( false );
+}
+void KABCore::setCategories( bool removeOld )
+{
+    KPIM::CategorySelectDialog* csd = new KPIM::CategorySelectDialog( KABPrefs::instance(), 0 );
+    if (! csd->exec()) { 
+        message( i18n("Setting categories cancelled") );
+        delete csd;
+        return;
+    }
+    message( i18n("Setting categories ... please wait!") );
+    QStringList catList = csd->selectedCategories();
+    delete csd;
+    catList.sort();
+    QStringList newList;
+    KABC::Addressee::List list = mViewManager->selectedAddressees();
+    KABC::Addressee::List::Iterator it;
+    for ( it = list.begin(); it != list.end(); ++it ) {
+        if ( removeOld ) {
+            (*it).setCategories( catList );
+        } else {
+            newList = (*it).categories();
+            int i;
+            for( i = 0; i<   catList.count(); ++i ) {
+                if (  !newList.contains (catList[i]))
+                    newList.append( catList[i] );
+            }
+            newList.sort();
+            (*it).setCategories( newList );
+        }
+        contactModified((*it) );
+    }
+    message( i18n("Setting categories completed!") );
+}
 void KABCore::removeVoice()
 {
     if ( KMessageBox::questionYesNo( this, i18n("After importing, phone numbers\nmay have two or more types.\n(E.g. work+voice)\nThese numbers are shown as \"other\".\nClick Yes to remove the voice type\nfrom numbers with more than one type.\n\nRemove voice type?") ) == KMessageBox::No )
         return;
     KABC::Addressee::List list = mViewManager->selectedAddressees();
     KABC::Addressee::List::Iterator it;
diff --git a/kaddressbook/kabcore.h b/kaddressbook/kabcore.h
index c01d598..1bdae5f 100644
--- a/kaddressbook/kabcore.h
+++ b/kaddressbook/kabcore.h
@@ -367,12 +367,15 @@ class KABCore : public QWidget, public KSyncInterface
     void extensionModified( const KABC::Addressee::List &list );
     void extensionChanged( int id );
     void clipboardDataChanged();
     void updateActionMenu();
     void configureKeyBindings();
     void removeVoice();
+    void setCat();
+    void addCat();
+    void setCategories( bool removeOld );
 #ifdef KAB_EMBEDDED
     void configureResources();
 #endif //KAB_EMBEDDED
 
     void slotEditorDestroyed( const QString &uid );
     void configurationChanged();
@@ -467,12 +470,14 @@ class KABCore : public QWidget, public KSyncInterface
     QPopupMenu *changeMenu;
 //US    QAction *mActionSave;
     QPopupMenu *ImportMenu;
     QPopupMenu *ExportMenu;
     //LR additional methods
     KAction *mActionRemoveVoice;
+    KAction *mActionSetCat;
+    KAction *mActionAddCat;
     KAction * mActionImportOL;
 
 #ifndef KAB_EMBEDDED
     KAddressBookService *mAddressBookService;
 #endif //KAB_EMBEDDED