Imported MDR. Some code still not entirely functional -- I haven't been able to figure out how to detect SSL, so IMAP and POP have it turned off. PPP auto-detect is also not functional at the moment. Other than that, it seems to work beautifully. Packaging will come later.
git-svn-id: file:///srv/svn/repos/haiku/trunk/current@9016 a95241bf-73f2-0310-859d-f6bbb57e9c96
This commit is contained in:
Nathan Whitehorn
@@ -0,0 +1,372 @@
|
||||
#include <NodeMonitor.h>
|
||||
#include <Handler.h>
|
||||
#include <Directory.h>
|
||||
#include <Entry.h>
|
||||
#include <Path.h>
|
||||
#include <String.h>
|
||||
#include <File.h>
|
||||
|
||||
#include <map>
|
||||
|
||||
class _EXPORT BMailRemoteStorageProtocol;
|
||||
|
||||
#include <RemoteStorageProtocol.h>
|
||||
#include <ChainRunner.h>
|
||||
|
||||
namespace {
|
||||
|
||||
void GetSubFolders(BDirectory *of, BStringList *folders, const char *prepend = "");
|
||||
|
||||
class UpdateHandler : public BHandler {
|
||||
public:
|
||||
UpdateHandler(BMailRemoteStorageProtocol *prot, const char *dest) : _prot(prot), _dest(dest) {
|
||||
node_ref ref;
|
||||
_dest.GetNodeRef(&ref);
|
||||
dest_node = ref.node;
|
||||
}
|
||||
virtual ~UpdateHandler() {
|
||||
stop_watching(this);
|
||||
}
|
||||
void MessageReceived(BMessage *msg) {
|
||||
switch (msg->what) {
|
||||
case 'INIT': {
|
||||
if (_prot->InitCheck() < B_OK)
|
||||
return;
|
||||
|
||||
((BMailChainRunner *)(Looper()))->ReportProgress(0,0,"Synchronizing Mailboxes");
|
||||
|
||||
BStringList subdirs;
|
||||
GetSubFolders(&_dest,&subdirs);
|
||||
BStringList to_delete;
|
||||
BStringList to_add;
|
||||
subdirs.NotThere(_prot->mailboxes,&to_add);
|
||||
if (subdirs.CountItems() != 0) // --- If it's a virgin mailfolder, the user probably just configured his machineand probably *doesn't* want all his mail folders deleted :)
|
||||
subdirs.NotHere(_prot->mailboxes,&to_delete);
|
||||
for (int32 i = 0; i < to_add.CountItems(); i++) {
|
||||
if (_prot->CreateMailbox(to_add[i]) != B_OK)
|
||||
continue;
|
||||
_prot->mailboxes += to_add[i];
|
||||
_prot->SyncMailbox(to_add[i]);
|
||||
}
|
||||
|
||||
_prot->CheckForDeletedMessages(); //--- Refresh the manifest list, delete messages in locally deleted folders
|
||||
|
||||
for (int32 i = 0; i < to_delete.CountItems(); i++) {
|
||||
if (to_delete[i][0] == 0)
|
||||
continue;
|
||||
if (_prot->DeleteMailbox(to_delete[i]) == B_OK)
|
||||
_prot->mailboxes -= to_delete[i];
|
||||
}
|
||||
|
||||
entry_ref ref;
|
||||
BEntry entry;
|
||||
_dest.GetEntry(&entry);
|
||||
entry.GetRef(&ref);
|
||||
BPath path(&ref), work_path(path);
|
||||
BNode node(&ref);
|
||||
node_ref watcher;
|
||||
node.GetNodeRef(&watcher);
|
||||
watch_node(&watcher,B_WATCH_DIRECTORY,this);
|
||||
|
||||
for (int32 i = 0; i < _prot->mailboxes.CountItems(); i++) {
|
||||
|
||||
work_path = path;
|
||||
work_path.Append(_prot->mailboxes[i]);
|
||||
node.SetTo(work_path.Path());
|
||||
node.GetNodeRef(&watcher);
|
||||
nodes[watcher.node] = strdup(_prot->mailboxes[i]);
|
||||
if (_prot->mailboxes[i][0] == 0)
|
||||
continue; //--- We've covered this in the parent monitor
|
||||
watch_node(&watcher,B_WATCH_DIRECTORY,this);
|
||||
_prot->SyncMailbox(_prot->mailboxes[i]);
|
||||
}
|
||||
((BMailChainRunner *)(Looper()))->ResetProgress();
|
||||
} break;
|
||||
|
||||
case B_NODE_MONITOR: {
|
||||
int32 opcode;
|
||||
if (msg->FindInt32("opcode",&opcode) < B_OK)
|
||||
break;
|
||||
|
||||
int64 directory, node(msg->FindInt64("node"));
|
||||
dev_t device(msg->FindInt32("device"));
|
||||
|
||||
|
||||
bool is_dir;
|
||||
{
|
||||
node_ref item_ref;
|
||||
item_ref.node = node;
|
||||
item_ref.device = device;
|
||||
BDirectory dir(&item_ref);
|
||||
is_dir = (dir.InitCheck() == B_OK);
|
||||
}
|
||||
|
||||
if (opcode == B_ENTRY_MOVED) {
|
||||
ino_t from, to;
|
||||
msg->FindInt64("from directory",&from);
|
||||
msg->FindInt64("to directory",&to);
|
||||
const char *from_mb(nodes[from]), *to_mb(nodes[to]);
|
||||
if (to == dest_node)
|
||||
to_mb = "";
|
||||
if (from == dest_node)
|
||||
from_mb = "";
|
||||
if (from_mb == NULL) {
|
||||
msg->AddInt64("directory",to);
|
||||
opcode = B_ENTRY_CREATED;
|
||||
} else if (to_mb == NULL) {
|
||||
msg->AddInt64("directory",from);
|
||||
if (is_dir)
|
||||
opcode = B_ENTRY_REMOVED;
|
||||
} else {
|
||||
if (!is_dir) {
|
||||
{
|
||||
node_ref item_ref;
|
||||
item_ref.node = to;
|
||||
item_ref.device = device;
|
||||
BDirectory dir(&item_ref);
|
||||
BNode node(&dir,msg->FindString("name"));
|
||||
//--- Why in HELL can't you make a BNode from a node_ref?
|
||||
|
||||
if (node.InitCheck() != B_OK)
|
||||
break; //-- We're late, it's already gone elsewhere. Ignore for now.
|
||||
|
||||
BString id;
|
||||
node.ReadAttrString("MAIL:unique_id",&id);
|
||||
id.Truncate(id.FindLast('/'));
|
||||
if (id == to_mb)
|
||||
break; //-- Already where it belongs, no need to do anything
|
||||
}
|
||||
|
||||
snooze(uint64(5e5));
|
||||
_prot->SyncMailbox(to_mb);
|
||||
|
||||
//node.WriteAttrString("MAIL:unique_id",&id);
|
||||
|
||||
_prot->CheckForDeletedMessages();
|
||||
} else {
|
||||
BString mb;
|
||||
if (to_mb[0] == 0)
|
||||
mb = msg->FindString("name");
|
||||
else {
|
||||
mb = to_mb;
|
||||
mb << '/' << msg->FindString("name");
|
||||
}
|
||||
if (strcmp(mb.String(),nodes[node]) == 0)
|
||||
break;
|
||||
if (_prot->CreateMailbox(mb.String()) < B_OK)
|
||||
break;
|
||||
_prot->mailboxes += mb.String();
|
||||
_prot->SyncMailbox(mb.String());
|
||||
_prot->CheckForDeletedMessages();
|
||||
_prot->DeleteMailbox(nodes[node]);
|
||||
_prot->mailboxes -= nodes[node];
|
||||
free((void *)nodes[node]);
|
||||
nodes[node] = strdup(mb.String());
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
msg->FindInt64("directory",&directory);
|
||||
switch (opcode) {
|
||||
case B_ENTRY_CREATED:
|
||||
if (!is_dir) {
|
||||
const char *dir = nodes[directory];
|
||||
snooze(500000);
|
||||
|
||||
if (dir == NULL)
|
||||
dir = "";
|
||||
|
||||
{
|
||||
node_ref item_ref;
|
||||
item_ref.node = directory;
|
||||
item_ref.device = device;
|
||||
BDirectory dir(&item_ref);
|
||||
BNode node(&dir,msg->FindString("name"));
|
||||
//--- Why in HELL can't you make a BNode from a node_ref?
|
||||
|
||||
if (node.InitCheck() != B_OK)
|
||||
break; //-- We're late, it's already gone elsewhere. Ignore for now.
|
||||
}
|
||||
|
||||
_prot->SyncMailbox(nodes[directory]);
|
||||
} else {
|
||||
BString mb;
|
||||
if (directory == dest_node)
|
||||
mb = msg->FindString("name");
|
||||
else {
|
||||
mb = nodes[directory];
|
||||
mb << '/' << msg->FindString("name");
|
||||
}
|
||||
if (_prot->CreateMailbox(mb.String()) < B_OK)
|
||||
break;
|
||||
nodes[node] = strdup(mb.String());
|
||||
_prot->mailboxes += mb.String();
|
||||
_prot->SyncMailbox(mb.String());
|
||||
node_ref ref;
|
||||
ref.device = device;
|
||||
ref.node = node;
|
||||
watch_node(&ref,B_WATCH_DIRECTORY,this);
|
||||
}
|
||||
break;
|
||||
case B_ENTRY_REMOVED:
|
||||
_prot->CheckForDeletedMessages();
|
||||
if ((is_dir) && (nodes[node] != NULL)) {
|
||||
_prot->DeleteMailbox(nodes[node]);
|
||||
_prot->mailboxes -= nodes[node];
|
||||
free((void *)nodes[node]);
|
||||
nodes[node] = NULL;
|
||||
node_ref ref;
|
||||
ref.device = device;
|
||||
ref.node = node;
|
||||
watch_node(&ref,B_STOP_WATCHING,this);
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
} break;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
private:
|
||||
BMailRemoteStorageProtocol *_prot;
|
||||
BDirectory _dest;
|
||||
|
||||
map<int64, const char *> nodes;
|
||||
ino_t dest_node;
|
||||
};
|
||||
|
||||
void GetSubFolders(BDirectory *of, BStringList *folders, const char *prepend) {
|
||||
of->Rewind();
|
||||
BEntry ent;
|
||||
BString crud;
|
||||
BDirectory sub;
|
||||
char buf[255];
|
||||
while (of->GetNextEntry(&ent) == B_OK) {
|
||||
if (ent.IsDirectory()) {
|
||||
sub.SetTo(&ent);
|
||||
ent.GetName(buf);
|
||||
crud = prepend;
|
||||
crud << buf << '/';
|
||||
GetSubFolders(&sub,folders,crud.String());
|
||||
crud = prepend;
|
||||
crud << buf;
|
||||
(*folders) += crud.String();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
BMailRemoteStorageProtocol::BMailRemoteStorageProtocol(BMessage *settings, BMailChainRunner *runner) : BMailProtocol(settings,runner) {
|
||||
handler = new UpdateHandler(this,runner->Chain()->MetaData()->FindString("path"));
|
||||
runner->AddHandler(handler);
|
||||
runner->PostMessage('INIT',handler);
|
||||
}
|
||||
|
||||
BMailRemoteStorageProtocol::~BMailRemoteStorageProtocol() {
|
||||
delete handler;
|
||||
}
|
||||
}
|
||||
|
||||
//----BMailProtocol stuff
|
||||
status_t BMailRemoteStorageProtocol::GetMessage(
|
||||
const char* uid,
|
||||
BPositionIO** out_file, BMessage* out_headers,
|
||||
BPath* out_folder_location) {
|
||||
BString folder(uid), id;
|
||||
{
|
||||
BString raw(uid);
|
||||
folder.Truncate(raw.FindLast('/'));
|
||||
raw.CopyInto(id,raw.FindLast('/') + 1,raw.Length());
|
||||
}
|
||||
|
||||
*out_folder_location = folder.String();
|
||||
return GetMessage(folder.String(),id.String(),out_file,out_headers);
|
||||
}
|
||||
|
||||
status_t BMailRemoteStorageProtocol::DeleteMessage(const char* uid) {
|
||||
BString folder(uid), id;
|
||||
{
|
||||
BString raw(uid);
|
||||
int32 j = raw.FindLast('/');
|
||||
folder.Truncate(j);
|
||||
raw.CopyInto(id,j + 1,raw.Length());
|
||||
}
|
||||
|
||||
status_t err;
|
||||
if ((err = DeleteMessage(folder.String(),id.String())) < B_OK)
|
||||
return err;
|
||||
|
||||
(*unique_ids) -= uid;
|
||||
return B_OK;
|
||||
}
|
||||
|
||||
void BMailRemoteStorageProtocol::SyncMailbox(const char *mailbox) {
|
||||
BPath path(runner->Chain()->MetaData()->FindString("path"));
|
||||
path.Append(mailbox);
|
||||
|
||||
BDirectory folder(path.Path());
|
||||
|
||||
BEntry entry;
|
||||
BFile snoodle;
|
||||
BString string;
|
||||
uint32 chain;
|
||||
bool append;
|
||||
|
||||
while (folder.GetNextEntry(&entry) == B_OK) {
|
||||
if (!entry.IsFile())
|
||||
continue;
|
||||
while (snoodle.SetTo(&entry,B_READ_WRITE) == B_BUSY) snooze(100);
|
||||
append = false;
|
||||
|
||||
while (snoodle.Lock() != B_OK) snooze(100);
|
||||
snoodle.Unlock();
|
||||
|
||||
if (snoodle.ReadAttr("MAIL:chain",B_INT32_TYPE,0,&chain,sizeof(chain)) < B_OK)
|
||||
append = true;
|
||||
if (chain != runner->Chain()->ID())
|
||||
append = true;
|
||||
if (snoodle.ReadAttrString("MAIL:unique_id",&string) < B_OK)
|
||||
append = true;
|
||||
|
||||
BString folder(string), id("");
|
||||
int32 j = string.FindLast('/');
|
||||
if ((!append) && (j >= 0)) {
|
||||
folder.Truncate(j);
|
||||
string.CopyInto(id,j + 1,string.Length());
|
||||
if (folder == mailbox)
|
||||
continue;
|
||||
} else {
|
||||
append = true;
|
||||
}
|
||||
|
||||
if (append)
|
||||
AddMessage(mailbox,&snoodle,&id); //---We should check for partial messages here
|
||||
else
|
||||
CopyMessage(folder.String(),mailbox,&id);
|
||||
|
||||
string = mailbox;
|
||||
string << '/' << id;
|
||||
/*snoodle.RemoveAttr("MAIL:unique_id");
|
||||
snoodle.RemoveAttr("MAIL:chain");*/
|
||||
chain = runner->Chain()->ID();
|
||||
snoodle.WriteAttr("MAIL:chain",B_INT32_TYPE,0,&chain,sizeof(chain));
|
||||
snoodle.WriteAttrString("MAIL:unique_id",&string);
|
||||
(*manifest) += string.String();
|
||||
(*unique_ids) += string.String();
|
||||
string = runner->Chain()->Name();
|
||||
snoodle.WriteAttrString("MAIL:account",&string);
|
||||
}
|
||||
}
|
||||
|
||||
/*status_t BMailRemoteStorageProtocol::MoveMessage(const char *mailbox, const char *to_mailbox, BString *message) {
|
||||
BString new_id(*message);
|
||||
status_t err;
|
||||
if ((err = CopyMessage(mailbox,to_mailbox,&new_id)) < B_OK)
|
||||
return err;
|
||||
if ((err = DeleteMessage(mailbox,message->String())) < B_OK)
|
||||
return err;
|
||||
*message = new_id;
|
||||
return B_OK;
|
||||
}*/
|
||||
Reference in New Issue
Block a user