1 files changed, 34 insertions, 23 deletions
diff --git a/Swiften/FileTransfer/FileTransfer.h b/Swiften/FileTransfer/FileTransfer.h
index 336c51c..c01aadb 100644
--- a/Swiften/FileTransfer/FileTransfer.h
+++ b/Swiften/FileTransfer/FileTransfer.h
@@ -5,4 +5,10 @@
  */
 
+/*
+ * Copyright (c) 2013 Remko Tronçon
+ * Licensed under the GNU General Public License.
+ * See the COPYING file for more information.
+ */
+
 #pragma once
 
@@ -15,45 +21,50 @@
 
 namespace Swift {
-
 	class FileTransfer {
 		public:
 			struct State {
-		enum FTState {
-			Canceled,
-			Failed,
-			Finished,
-			Negotiating,
-			Transferring,
+				enum Type {
+					Initial,
 					WaitingForStart,
+					Negotiating,
 					WaitingForAccept,
+					Transferring,
+					Canceled,
+					Failed,
+					Finished
 				};
 
-		FTState state;
-		std::string message;
+				State(Type type, const std::string& message = "") : type(type), message(message) {}
 
-		State(FTState state) : state(state), message("") {}
-		State(FTState state, std::string message) : state(state), message(message) {}
+				Type type;
+				std::string message;
 			};
-
-public:
 			typedef boost::shared_ptr<FileTransfer> ref;
 
 		public:
-	boost::uintmax_t fileSizeInBytes;
-	std::string filename;
-	std::string algo;
-	std::string hash;
+			FileTransfer();
+			virtual ~FileTransfer();
 
-public:
 			virtual void cancel() = 0;
 
+			const std::string& getFileName() const {
+				return filename;
+			}
+
+			boost::uintmax_t getFileSizeInBytes() const {
+				return fileSizeInBytes;
+			}
+
 		public:
-	boost::signal<void (int /* proccessedBytes */)> onProcessedBytes;
-	boost::signal<void (State)> onStateChange;
+			boost::signal<void (size_t /* proccessedBytes */)> onProcessedBytes;
+			boost::signal<void (const State&)> onStateChanged;
 			boost::signal<void (boost::optional<FileTransferError>)> onFinished;
 
-public:
-	virtual ~FileTransfer() {}
-};
+		protected:
+			void setFileInfo(const std::string& name, boost::uintmax_t size);
 
+		private:
+			boost::uintmax_t fileSizeInBytes;
+			std::string filename;
+	};
 }