1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
//
// copyright : (C) 2013 by Eran Ifrah
// file name : sftp_writer_thread.h
//
// -------------------------------------------------------------------------
// A
// _____ _ _ _ _
// / __ \ | | | | (_) |
// | / \/ ___ __| | ___| | _| |_ ___
// | | / _ \ / _ |/ _ \ | | | __/ _ )
// | \__/\ (_) | (_| | __/ |___| | || __/
// \____/\___/ \__,_|\___\_____/_|\__\___|
//
// F i l e
//
// 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
// (at your option) any later version.
//
//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
#ifndef SFTPWRITERTHREAD_H
#define SFTPWRITERTHREAD_H
#include "worker_thread.h" // Base class: WorkerThread
#include "cl_sftp.h"
#include "ssh_account_info.h"
#include "remote_file_info.h"
class SFTP;
class SFTPThreadRequet : public ThreadRequest
{
SSHAccountInfo m_account;
wxString m_remoteFile;
wxString m_localFile;
size_t m_retryCounter;
bool m_uploadSuccess;
int m_direction;
public:
enum {
kUpload,
kDownload
};
public:
SFTPThreadRequet(const SSHAccountInfo& accountInfo, const wxString &remoteFile, const wxString &localFile);
SFTPThreadRequet(const RemoteFileInfo& remoteFile);
SFTPThreadRequet(const SFTPThreadRequet& other);
SFTPThreadRequet& operator=(const SFTPThreadRequet& other);
virtual ~SFTPThreadRequet();
void SetUploadSuccess(bool uploadSuccess) {
this->m_uploadSuccess = uploadSuccess;
}
bool IsUploadSuccess() const {
return m_uploadSuccess;
}
void SetAccount(const SSHAccountInfo& account) {
this->m_account = account;
}
void SetLocalFile(const wxString& localFile) {
this->m_localFile = localFile;
}
void SetRemoteFile(const wxString& remoteFile) {
this->m_remoteFile = remoteFile;
}
void SetRetryCounter(size_t retryCounter) {
this->m_retryCounter = retryCounter;
}
size_t GetRetryCounter() const {
return m_retryCounter;
}
const SSHAccountInfo& GetAccount() const {
return m_account;
}
const wxString& GetRemoteFile() const {
return m_remoteFile;
}
const wxString& GetLocalFile() const {
return m_localFile;
}
int GetDirection() const {
return m_direction;
}
ThreadRequest* Clone() const;
};
class SFTPThreadMessage
{
int m_status;
wxString m_message;
wxString m_account;
public:
enum {
STATUS_NONE = -1,
STATUS_OK,
STATUS_ERROR,
};
public:
SFTPThreadMessage();
virtual ~SFTPThreadMessage();
void SetAccount(const wxString& account) {
this->m_account = account;
}
void SetMessage(const wxString& message) {
this->m_message = message;
}
const wxString& GetAccount() const {
return m_account;
}
const wxString& GetMessage() const {
return m_message;
}
void SetStatus(int status) {
this->m_status = status;
}
int GetStatus() const {
return m_status;
}
};
class SFTPWorkerThread : public WorkerThread
{
static SFTPWorkerThread* ms_instance;
clSFTP::Ptr_t m_sftp;
SFTP* m_plugin;
public:
static SFTPWorkerThread* Instance();
static void Release();
private:
SFTPWorkerThread();
virtual ~SFTPWorkerThread();
void DoConnect(SFTPThreadRequet *req);
void DoReportMessage(const wxString &account, const wxString &message, int status);
void DoReportStatusBarMessage(const wxString &message);
public:
virtual void ProcessRequest(ThreadRequest* request);
void SetSftpPlugin(SFTP* sftp);
};
#endif // SFTPWRITERTHREAD_H
|
