108 lines
2.2 KiB
C++
108 lines
2.2 KiB
C++
/*
|
|
* Copyright 2010, Andrea Anzani. All rights reserved.
|
|
* Distributed under the terms of the MIT License.
|
|
*/
|
|
#ifndef _ACTION_DOWNLOAD_H
|
|
#define _ACTION_DOWNLOAD_H
|
|
|
|
#include <curl/curl.h>
|
|
|
|
#include <Entry.h>
|
|
#include <Looper.h>
|
|
#include <File.h>
|
|
#include <SupportDefs.h>
|
|
#include <DownloadManager.h>
|
|
|
|
#include "Action.h"
|
|
|
|
class ActionDownload : public Action
|
|
{
|
|
public:
|
|
enum Status {
|
|
ERROR_OPENFILE, // 0
|
|
ERROR_CURL_INIT, // 1
|
|
ERROR_PERFORMING, // 2
|
|
OK_CONNECTING, // 3
|
|
OK_PROGRESS, // 4
|
|
OK_DOWNLOADED // 5
|
|
};
|
|
|
|
ActionDownload(BString URL, entry_ref dest,
|
|
bool allowResume, BLooper* target = NULL, uint32 msg = 0);
|
|
virtual ~ActionDownload();
|
|
|
|
status_t Perform(BMessage* errors);
|
|
BString GetDescription();
|
|
BString GetLocalPath();
|
|
|
|
void SetLooper(BLooper* loop, uint32 msg) {
|
|
fTarget = loop;
|
|
fMsg = msg;
|
|
};
|
|
BLooper* Looper() {
|
|
return fTarget;
|
|
}
|
|
uint32 MessageWhat() {
|
|
return fMsg;
|
|
}
|
|
|
|
// For compatibility:
|
|
void SetExtraData(BString extra) {
|
|
extraInfo.AddString("extra", extra);
|
|
}
|
|
|
|
void SetKey(BString key, BString data) {
|
|
extraInfo.AddString(key.String(), data);
|
|
}
|
|
|
|
void SetRef(BString key, entry_ref* ref) {
|
|
extraInfo.AddRef(key.String(), ref);
|
|
}
|
|
|
|
status_t GetRef(BString key, entry_ref* ref) {
|
|
return extraInfo.FindRef(key.String(), ref);
|
|
}
|
|
|
|
BString GetKey(BString key) {
|
|
BString data;
|
|
extraInfo.FindString(key.String(), &data);
|
|
return data;
|
|
}
|
|
|
|
|
|
void SetShouldStop(bool stop) {
|
|
fShouldStop = stop;
|
|
}
|
|
bool ShouldStop() {
|
|
return fShouldStop;
|
|
}
|
|
|
|
void SetDownloadManager(DownloadManager* downloadManager);
|
|
|
|
private:
|
|
status_t openFile(BMessage*);
|
|
|
|
static void fillMessage(ActionDownload*, BMessage*);
|
|
static void sendProgressX(ActionDownload*, double max, double current);
|
|
|
|
//curl callbacks:
|
|
static size_t save_file ( void* ptr, size_t size, size_t nmemb, void* stream);
|
|
static int callback (void* clientp, double dltotal, double dlnow, double , double);
|
|
|
|
CURL* curl;
|
|
BString fUrl;
|
|
entry_ref fDest;
|
|
BLooper* fTarget;
|
|
uint32 fMsg;
|
|
BMessage extraInfo;
|
|
bool fShouldStop;
|
|
BFile* file;
|
|
double resuming;
|
|
bool fAllowResume;
|
|
DownloadManager* fDownloadManager;
|
|
|
|
BString fFileType;
|
|
};
|
|
|
|
#endif // _ACTION_DOWNLOAD_H
|