--- a/homescreenapp/stateplugins/hsmenuworkerstateplugin/inc/hsaddtohomescreenstate.h	Mon May 03 12:24:59 2010 +0300
+++ b/homescreenapp/stateplugins/hsmenuworkerstateplugin/inc/hsaddtohomescreenstate.h	Fri May 14 15:43:04 2010 +0300
@@ -22,23 +22,15 @@
 #include <hsmenubasestate.h>
 
 #include "hsmenustates_global.h"
+#include "hsapp_defs.h"
+
 HS_STATES_TEST_CLASS(MenuStatesTest)
 
-
+class HbAction;
 class CaEntry;
+class HbMessageBox;
 
 
-/**
- * @ingroup group_hsworkerstateplugin
- * @brief Application Library State.
- *
- * Parent state for Application Library functionality (browsing applications and collections)
- *
- * @see StateMachine
- *
- * @lib ?library
- * @since S60 ?S60_version
- */
 class HsAddToHomeScreenState: public  HsMenuBaseState
 {
     Q_OBJECT
@@ -47,48 +39,47 @@
 
 public:
 
-    /**
-     * Constructor.
-     *
-     * @since S60 ?S60_version.
-     * @param parent Owner.
-     */
     HsAddToHomeScreenState(QState *parent = 0);
 
-    /**
-     * Destructor.
-     *
-     * @since S60 ?S60_version.
-     */
     virtual ~HsAddToHomeScreenState();
 
 private slots:
 
-    /**
-     * Inherited from HsMenuBaseState.
-     *
-     * @since S60 ?S60_version.
-     */
     void onEntry(QEvent *event);
 
+    void cleanUp();
+
+    void messageWidgetCorruptedFinished(HbAction* finishedAction);
+
+signals:
+
+    void exit();
+
 private:
 
-    void showMessageWidgetCorrupted(int itemId);
+    void showMessageWidgetCorrupted();
+
+    bool addWidget(HsContentService &service, const QString &uri);
 
-    void addWidget(HsContentService &service, const QString &uri,
-                    int entryId);
+    bool addShortcut(HsContentService &contentService);
 
-    void addShortcut(HsContentService &contentService, int entryId);
-
-    void addApplication(HsContentService &contentService, CaEntry &entryId);
+    bool addApplication(HsContentService &contentService, CaEntry &entryId);
 
     void logActionResult(QString operationName, int entryId,
                          bool operationSucceded);
 
+private:
 
     // keep path in memory so it wont
     QString mLibraryPath;
 
+    int mEntryId;
+
+    HbMessageBox *mCorruptedMessage;
+
+    HbAction *mConfirmAction;
+    HsMenuMode mMenuMode;
+
 };
 
 #endif //HSADDSHORTCUTSTATE_H