1 files changed, 9 insertions, 6 deletions
diff --git a/itemselectionwidget.h b/itemselectionwidget.h
index 0b8b48b..f78b9d1 100644
--- a/itemselectionwidget.h
+++ b/itemselectionwidget.h
@@ -2,6 +2,7 @@
 #define DIRSELECTIONWIDGET_H
 
 #include <QWidget>
+#include <QIcon>
 
 class QStandardItemModel;
 class QListView;
@@ -11,19 +12,21 @@ class ItemSelectionWidget : public QWidget
 {
     Q_OBJECT
     public:
-        explicit ItemSelectionWidget(QWidget *parent = nullptr);
-        const QStringList directories() const;
-        void setDirectories(const QStringList &dirs);
+        enum Mode { Item, Directories };
+        explicit ItemSelectionWidget(int mode, QString labelText, QWidget *parent = nullptr);
+        const QStringList items() const;
+        void setItems(const QStringList &items);
 
     public slots:
         void browseDir();
-        void addDir();
-        void removeDir();
+        void addItem();
+        void removeItem();
 
     private:
         QStandardItemModel *mModel;
         QListView *mView;
-        QLineEdit *mDir;
+        QLineEdit *mItem;
+        QIcon mIcon;
 };
 
 #endif // DIRSELECTIONWIDGET_H