Forgot the actual download class

1 files changed, 108 insertions, 0 deletions

diff --git a/src/net/download.h b/src/net/download.h
new file mode 100644
index 00000000..1e1f806f
--- /dev/null
+++ b/src/net/download.h
@@ -0,0 +1,108 @@
+/*
+ *  The Mana World
+ *  Copyright (C) 2009  The Mana World Development Team
+ *
+ *  This file is part of The Mana World.
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#include <SDL_types.h>
+#include <stdio.h>
+#include <string>
+
+#ifndef NET_DOWNLOAD_H
+#define NET_DOWNLOAD_H
+
+enum DownloadStatus
+{
+    DOWNLOAD_STATUS_CANCELLED = -3,
+    DOWNLOAD_STATUS_THREAD_ERROR = -2,
+    DOWNLOAD_STATUS_ERROR = -1,
+    DOWNLOAD_STATUS_STARTING = 0,
+    DOWNLOAD_STATUS_IDLE,
+    DOWNLOAD_STATUS_COMPLETE
+};
+
+typedef int (*DownloadUpdate)(void *ptr, DownloadStatus status,
+                              size_t total, size_t remaining);
+
+// Matches what CURL expects
+typedef size_t (*WriteFunction)( void *ptr, size_t size, size_t nmemb,
+                                 void *stream);
+
+struct SDL_Thread;
+typedef void CURL;
+struct curl_slist;
+
+namespace Net {
+class Download
+{
+    public:
+        Download(void *ptr, const std::string &url, DownloadUpdate updateFunction);
+
+        ~Download();
+
+        void addHeader(const std::string &header);
+
+        /**
+         * Convience method for adding no-cache headers.
+         */
+        void noCache();
+
+        void setFile(const std::string &filename, Sint64 adler32 = -1);
+
+        void setWriteFunction(WriteFunction write);
+
+        /**
+         * Starts the download thread.
+         * @returns true  if thread was created
+         *          false if the thread could not be made or download wasn't
+         *                properly setup
+         */
+        bool start();
+
+        /**
+         * Cancels the download. Returns immediately, the cancelled status will
+         * be noted in the next avialable update call.
+         */
+        void cancel();
+
+        char *getError();
+
+    private:
+        static int downloadThread(void *ptr);
+        static int downloadProgress(void *clientp, double dltotal, double dlnow,
+                                        double ultotal, double ulnow);
+        void *mPtr;
+        std::string mUrl;
+        struct {
+            unsigned cancel : 1;
+            unsigned memoryWrite: 1;
+            unsigned checkAdler: 1;
+        } mOptions;
+        std::string mFileName;
+        WriteFunction mWriteFunction;
+        unsigned long mAdler;
+        DownloadUpdate mUpdateFunction;
+        SDL_Thread *mThread;
+        CURL *mCurl;
+        curl_slist *mHeaders;
+        char *mError;
+};
+
+} // namespace Net
+
+#endif // NET_DOWNLOAD_H