1 files changed, 21 insertions, 4 deletions
diff --git a/archiveeditwidget.h b/archiveeditwidget.h
index 2f8b382..731214a 100644
--- a/archiveeditwidget.h
+++ b/archiveeditwidget.h
@@ -8,21 +8,35 @@
 #ifndef ARCHIVEEDITWIDGET_H
 #define ARCHIVEEDITWIDGET_H
 
-#include <QWidget>
+#include <QDialog>
+#include <QDirModel>
 
 class ArchiveFileWidget;
 class ListEditor;
 class CoverEditor;
 class ListModel;
 class MovieModel;
+class QPushButton;
+class CoverItem;
 
-class ArchiveEditWidget : public QWidget {
+class ArchiveEditDialog : public QDialog {
 	Q_OBJECT
 	public:
-		ArchiveEditWidget(QWidget *parent = 0);
-		~ArchiveEditWidget() {};
+		ArchiveEditDialog(QWidget *parent = 0, Qt::WindowFlags f = 0);
+		~ArchiveEditDialog() {};
+
+	public:
+		void setFiles(const QStringList &files);
+		void setMovieModel(MovieModel *model) { mMovieModel = model; };
+		void setDirModel(QDirModel *model) { mDirModel = model; };
+
+	private slots:
+		void archive();
 
 	private:
+		bool checkInput(int genre, const QList<int> &actors, const QStringList &files, const QString &title, const QString &movie, int quality);
+		bool checkMd5(const QHash<QString, QString> &md5sums, const QStringList &files);
+		const CoverItem coverItem(const QString &filePath, const QString type, const QString &md5) const;
 		ArchiveFileWidget *mFileWidget;
 		ListEditor *mGenreEditor;
 		ListEditor *mActorsEditor;
@@ -30,6 +44,9 @@ class ArchiveEditWidget : public QWidget {
 		ListModel *mGenreModel;
 		ListModel *mActorsModel;
 		MovieModel *mMovieModel;
+		QDirModel *mDirModel;
+		QPushButton *mArchive;
+		QPushButton *mClose;
 
 };