mirror of
https://github.com/nodemailer/wildduck.git
synced 2024-12-31 04:33:09 +08:00
1120 lines
33 KiB
JavaScript
1120 lines
33 KiB
JavaScript
|
'use strict';
|
||
|
|
|
||
|
|
const log = require('npmlog');
|
||
|
|
const uuidV1 = require('uuid/v1');
|
||
|
|
const config = require('config');
|
||
|
|
const IMAPServerModule = require('./imap-core');
|
||
|
|
const IMAPServer = IMAPServerModule.IMAPServer;
|
||
|
|
const mongodb = require('mongodb');
|
||
|
|
const MongoClient = mongodb.MongoClient;
|
||
|
|
const ImapNotifier = require('./imap-notifier');
|
||
|
|
const imapHandler = IMAPServerModule.imapHandler;
|
||
|
|
const bcrypt = require('bcryptjs');
|
||
|
|
//const fs = require('fs');
|
||
|
|
const Indexer = require('./imap-core/lib/indexer/indexer');
|
||
|
|
|
||
|
|
// Setup server
|
||
|
|
let server = new IMAPServer({
|
||
|
|
secure: true,
|
||
|
|
//key: config.imap.key ? fs.readFileSync(config.imap.key) : false,
|
||
|
|
//cert: config.imap.cert ? fs.readFileSync(config.imap.cert) : false,
|
||
|
|
id: {
|
||
|
|
name: 'test'
|
||
|
|
},
|
||
|
|
logger: {
|
||
|
|
info: log.silly.bind(log, 'IMAP'),
|
||
|
|
debug: log.silly.bind(log, 'IMAP'),
|
||
|
|
error: log.error.bind(log, 'IMAP')
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
let database;
|
||
|
|
|
||
|
|
|
||
|
|
server.onAuth = function (login, session, callback) {
|
||
|
|
let username = (login.username || '').toString().replace(/\./g, '').trim();
|
||
|
|
|
||
|
|
database.collection('users').findOne({
|
||
|
|
username
|
||
|
|
}, (err, user) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!user) {
|
||
|
|
return callback();
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!bcrypt.compareSync(login.password, user.password)) {
|
||
|
|
return callback();
|
||
|
|
}
|
||
|
|
|
||
|
|
let ensureInitial = next => {
|
||
|
|
let collection = database.collection('mailboxes');
|
||
|
|
collection.findOne({
|
||
|
|
username,
|
||
|
|
path: 'INBOX'
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (mailbox) {
|
||
|
|
return next();
|
||
|
|
}
|
||
|
|
|
||
|
|
let uidValidity = Math.floor(Date.now() / 1000);
|
||
|
|
|
||
|
|
collection.insertMany([{
|
||
|
|
username,
|
||
|
|
path: 'INBOX',
|
||
|
|
uidValidity,
|
||
|
|
uidNext: 1,
|
||
|
|
modifyIndex: 0,
|
||
|
|
subscribed: true
|
||
|
|
}, {
|
||
|
|
username,
|
||
|
|
path: 'Sent Mail',
|
||
|
|
specialUse: '\\Sent',
|
||
|
|
uidValidity,
|
||
|
|
uidNext: 1,
|
||
|
|
modifyIndex: 0,
|
||
|
|
subscribed: true
|
||
|
|
}, {
|
||
|
|
username,
|
||
|
|
path: 'Trash',
|
||
|
|
specialUse: '\\Trash',
|
||
|
|
uidValidity,
|
||
|
|
uidNext: 1,
|
||
|
|
modifyIndex: 0,
|
||
|
|
subscribed: true
|
||
|
|
}, {
|
||
|
|
username: login.username,
|
||
|
|
path: 'Junk',
|
||
|
|
specialUse: '\\Junk',
|
||
|
|
uidValidity,
|
||
|
|
uidNext: 1,
|
||
|
|
modifyIndex: 0,
|
||
|
|
subscribed: true
|
||
|
|
}], {
|
||
|
|
w: 1,
|
||
|
|
ordered: false
|
||
|
|
}, err => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
return next();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
ensureInitial(() => callback(null, {
|
||
|
|
user: {
|
||
|
|
username
|
||
|
|
}
|
||
|
|
}));
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// LIST "" "*"
|
||
|
|
// Returns all folders, query is informational
|
||
|
|
// folders is either an Array or a Map
|
||
|
|
server.onList = function (query, session, callback) {
|
||
|
|
this.logger.debug('[%s] LIST for "%s"', session.id, query);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').find({
|
||
|
|
username
|
||
|
|
}).toArray(callback);
|
||
|
|
};
|
||
|
|
|
||
|
|
// LSUB "" "*"
|
||
|
|
// Returns all subscribed folders, query is informational
|
||
|
|
// folders is either an Array or a Map
|
||
|
|
server.onLsub = function (query, session, callback) {
|
||
|
|
this.logger.debug('[%s] LSUB for "%s"', session.id, query);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').find({
|
||
|
|
username,
|
||
|
|
subscribed: true
|
||
|
|
}).toArray(callback);
|
||
|
|
};
|
||
|
|
|
||
|
|
// SUBSCRIBE "path/to/mailbox"
|
||
|
|
server.onSubscribe = function (path, session, callback) {
|
||
|
|
this.logger.debug('[%s] SUBSCRIBE to "%s"', session.id, path);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOneAndUpdate({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, {
|
||
|
|
$set: {
|
||
|
|
subscribed: true
|
||
|
|
}
|
||
|
|
}, {}, (err, item) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!item || !item.value) {
|
||
|
|
// was not able to acquire a lock
|
||
|
|
return callback(null, 'NONEXISTENT');
|
||
|
|
}
|
||
|
|
|
||
|
|
callback(null, true);
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// UNSUBSCRIBE "path/to/mailbox"
|
||
|
|
server.onUnsubscribe = function (path, session, callback) {
|
||
|
|
this.logger.debug('[%s] UNSUBSCRIBE from "%s"', session.id, path);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOneAndUpdate({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, {
|
||
|
|
$set: {
|
||
|
|
subscribed: false
|
||
|
|
}
|
||
|
|
}, {}, (err, item) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!item || !item.value) {
|
||
|
|
// was not able to acquire a lock
|
||
|
|
return callback(null, 'NONEXISTENT');
|
||
|
|
}
|
||
|
|
|
||
|
|
callback(null, true);
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// CREATE "path/to/mailbox"
|
||
|
|
server.onCreate = function (path, session, callback) {
|
||
|
|
this.logger.debug('[%s] CREATE "%s"', session.id, path);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (mailbox) {
|
||
|
|
return callback(null, 'ALREADYEXISTS');
|
||
|
|
}
|
||
|
|
|
||
|
|
mailbox = {
|
||
|
|
username,
|
||
|
|
path,
|
||
|
|
uidValidity: Math.floor(Date.now() / 1000),
|
||
|
|
uidNext: 1,
|
||
|
|
modifyIndex: 0,
|
||
|
|
subscribed: true
|
||
|
|
};
|
||
|
|
|
||
|
|
database.collection('mailboxes').insertOne(mailbox, err => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
return callback(null, true);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// RENAME "path/to/mailbox" "new/path"
|
||
|
|
// NB! RENAME affects child and hierarchy mailboxes as well, this example does not do this
|
||
|
|
server.onRename = function (path, newname, session, callback) {
|
||
|
|
this.logger.debug('[%s] RENAME "%s" to "%s"', session.id, path, newname);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path: newname
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (mailbox) {
|
||
|
|
return callback(null, 'ALREADYEXISTS');
|
||
|
|
}
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOneAndUpdate({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, {
|
||
|
|
$set: {
|
||
|
|
path: newname
|
||
|
|
}
|
||
|
|
}, {}, (err, item) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!item || !item.value) {
|
||
|
|
// was not able to acquire a lock
|
||
|
|
return callback(null, 'NONEXISTENT');
|
||
|
|
}
|
||
|
|
|
||
|
|
callback(null, true);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// DELETE "path/to/mailbox"
|
||
|
|
server.onDelete = function (path, session, callback) {
|
||
|
|
this.logger.debug('[%s] DELETE "%s"', session.id, path);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!mailbox) {
|
||
|
|
return callback(null, 'NONEXISTENT');
|
||
|
|
}
|
||
|
|
if (mailbox.specialUse) {
|
||
|
|
return callback(null, 'CANNOT');
|
||
|
|
}
|
||
|
|
|
||
|
|
database.collection('mailboxes').deleteOne({
|
||
|
|
_id: mailbox._id
|
||
|
|
}, err => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
database.collection('journal').deleteMany({
|
||
|
|
mailbox: mailbox._id
|
||
|
|
}, err => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
database.collection('messages').deleteMany({
|
||
|
|
mailbox: mailbox._id
|
||
|
|
}, err => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
callback(null, true);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// SELECT/EXAMINE
|
||
|
|
server.onOpen = function (path, session, callback) {
|
||
|
|
this.logger.debug('[%s] Opening "%s"', session.id, path);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!mailbox) {
|
||
|
|
return callback(null, 'NONEXISTENT');
|
||
|
|
}
|
||
|
|
|
||
|
|
database.collection('messages').find({
|
||
|
|
mailbox: mailbox._id
|
||
|
|
}).project({
|
||
|
|
uid: true
|
||
|
|
}).sort([
|
||
|
|
['uid', 1]
|
||
|
|
]).toArray((err, messages) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
mailbox.uidList = messages.map(message => message.uid);
|
||
|
|
callback(null, mailbox);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// STATUS (X Y X)
|
||
|
|
server.onStatus = function (path, session, callback) {
|
||
|
|
this.logger.debug('[%s] Requested status for "%s"', session.id, path);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!mailbox) {
|
||
|
|
return callback(null, 'NONEXISTENT');
|
||
|
|
}
|
||
|
|
|
||
|
|
database.collection('messages').find({
|
||
|
|
mailbox: mailbox._id
|
||
|
|
}).count((err, total) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
database.collection('messages').find({
|
||
|
|
mailbox: mailbox._id,
|
||
|
|
unseen: true
|
||
|
|
}).count((err, unseen) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
return callback(null, {
|
||
|
|
messages: total,
|
||
|
|
uidNext: mailbox.uidNext,
|
||
|
|
uidValidity: mailbox.uidValidity,
|
||
|
|
unseen
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// APPEND mailbox (flags) date message
|
||
|
|
server.onAppend = function (path, flags, date, raw, session, callback) {
|
||
|
|
this.logger.debug('[%s] Appending message to "%s"', session.id, path);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
this.addToMailbox(username, path, {
|
||
|
|
source: 'IMAP',
|
||
|
|
user: username,
|
||
|
|
time: Date.now()
|
||
|
|
}, date, flags, raw, (err, status, data) => {
|
||
|
|
if (err) {
|
||
|
|
if (err.imapResponse) {
|
||
|
|
return callback(null, err.imapResponse);
|
||
|
|
}
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
callback(null, status, data);
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// STORE / UID STORE, updates flags for selected UIDs
|
||
|
|
server.onUpdate = function (path, update, session, callback) {
|
||
|
|
this.logger.debug('[%s] Updating messages in "%s"', session.id, path);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!mailbox) {
|
||
|
|
return callback(null, 'NONEXISTENT');
|
||
|
|
}
|
||
|
|
|
||
|
|
let cursor = database.collection('messages').find({
|
||
|
|
mailbox: mailbox._id,
|
||
|
|
uid: {
|
||
|
|
$in: update.messages
|
||
|
|
}
|
||
|
|
}).project({
|
||
|
|
_id: true,
|
||
|
|
uid: true,
|
||
|
|
flags: true
|
||
|
|
}).sort([
|
||
|
|
['uid', 1]
|
||
|
|
]);
|
||
|
|
|
||
|
|
let processNext = () => {
|
||
|
|
cursor.next((err, message) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!message) {
|
||
|
|
return cursor.close(() => {
|
||
|
|
this.notifier.fire(username, path);
|
||
|
|
return callback(null, true);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
let updated = false;
|
||
|
|
switch (update.action) {
|
||
|
|
case 'set':
|
||
|
|
// check if update set matches current or is different
|
||
|
|
if (message.flags.length !== update.value.length || update.value.filter(flag => message.flags.indexOf(flag) < 0).length) {
|
||
|
|
updated = true;
|
||
|
|
}
|
||
|
|
// set flags
|
||
|
|
message.flags = [].concat(update.value);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 'add':
|
||
|
|
message.flags = message.flags.concat(update.value.filter(flag => {
|
||
|
|
if (message.flags.indexOf(flag) < 0) {
|
||
|
|
updated = true;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}));
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 'remove':
|
||
|
|
message.flags = message.flags.filter(flag => {
|
||
|
|
if (update.value.indexOf(flag) < 0) {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
updated = true;
|
||
|
|
return false;
|
||
|
|
});
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!update.silent) {
|
||
|
|
session.writeStream.write(session.formatResponse('FETCH', message.uid, {
|
||
|
|
uid: update.isUid ? message.uid : false,
|
||
|
|
flags: message.flags
|
||
|
|
}));
|
||
|
|
}
|
||
|
|
|
||
|
|
if (updated) {
|
||
|
|
database.collection('messages').findOneAndUpdate({
|
||
|
|
_id: message._id
|
||
|
|
}, {
|
||
|
|
$set: {
|
||
|
|
flags: message.flags,
|
||
|
|
unseen: !message.flags.includes('\\Seen')
|
||
|
|
}
|
||
|
|
}, {}, err => {
|
||
|
|
if (err) {
|
||
|
|
return cursor.close(() => callback(err));
|
||
|
|
}
|
||
|
|
this.notifier.addEntries(username, path, {
|
||
|
|
command: 'FETCH',
|
||
|
|
ignore: session.id,
|
||
|
|
uid: message.uid,
|
||
|
|
flags: message.flags,
|
||
|
|
message: message._id
|
||
|
|
}, processNext);
|
||
|
|
});
|
||
|
|
} else {
|
||
|
|
processNext();
|
||
|
|
}
|
||
|
|
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
processNext();
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// EXPUNGE deletes all messages in selected mailbox marked with \Delete
|
||
|
|
// EXPUNGE deletes all messages in selected mailbox marked with \Delete
|
||
|
|
server.onExpunge = function (path, update, session, callback) {
|
||
|
|
this.logger.debug('[%s] Deleting messages from "%s"', session.id, path);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!mailbox) {
|
||
|
|
return callback(null, 'NONEXISTENT');
|
||
|
|
}
|
||
|
|
|
||
|
|
let cursor = database.collection('messages').find({
|
||
|
|
mailbox: mailbox._id,
|
||
|
|
flags: '\\Deleted'
|
||
|
|
}).project({
|
||
|
|
_id: true,
|
||
|
|
uid: true
|
||
|
|
}).sort([
|
||
|
|
['uid', 1]
|
||
|
|
]);
|
||
|
|
|
||
|
|
let processNext = () => {
|
||
|
|
cursor.next((err, message) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!message) {
|
||
|
|
return cursor.close(() => {
|
||
|
|
this.notifier.fire(username, path);
|
||
|
|
return callback(null, true);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!update.silent) {
|
||
|
|
session.writeStream.write(session.formatResponse('EXPUNGE', message.uid));
|
||
|
|
}
|
||
|
|
|
||
|
|
database.collection('messages').deleteOne({
|
||
|
|
_id: message._id
|
||
|
|
}, err => {
|
||
|
|
if (err) {
|
||
|
|
return cursor.close(() => callback(err));
|
||
|
|
}
|
||
|
|
this.notifier.addEntries(username, path, {
|
||
|
|
command: 'EXPUNGE',
|
||
|
|
ignore: session.id,
|
||
|
|
uid: message.uid,
|
||
|
|
message: message._id
|
||
|
|
}, processNext);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
processNext();
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// COPY / UID COPY sequence mailbox
|
||
|
|
server.onCopy = function (path, update, session, callback) {
|
||
|
|
this.logger.debug('[%s] Copying messages from "%s" to "%s"', session.id, path, update.destination);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!mailbox) {
|
||
|
|
return callback(null, 'NONEXISTENT');
|
||
|
|
}
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path: update.destination
|
||
|
|
}, (err, target) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!target) {
|
||
|
|
return callback(null, 'TRYCREATE');
|
||
|
|
}
|
||
|
|
|
||
|
|
let cursor = database.collection('messages').find({
|
||
|
|
mailbox: mailbox._id,
|
||
|
|
uid: {
|
||
|
|
$in: update.messages
|
||
|
|
}
|
||
|
|
}); // no projection as we need to copy the entire message
|
||
|
|
|
||
|
|
let sourceUid = [];
|
||
|
|
let destinationUid = [];
|
||
|
|
let processNext = () => {
|
||
|
|
cursor.next((err, message) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!message) {
|
||
|
|
return cursor.close(() => {
|
||
|
|
this.notifier.fire(username, target.path);
|
||
|
|
return callback(null, true, {
|
||
|
|
uidValidity: target.uidValidity,
|
||
|
|
sourceUid,
|
||
|
|
destinationUid
|
||
|
|
});
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
sourceUid.unshift(message.uid);
|
||
|
|
database.collection('mailboxes').findOneAndUpdate({
|
||
|
|
_id: target._id
|
||
|
|
}, {
|
||
|
|
$inc: {
|
||
|
|
uidNext: 1
|
||
|
|
}
|
||
|
|
}, {
|
||
|
|
uidNext: true
|
||
|
|
}, (err, item) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!item || !item.value) {
|
||
|
|
// was not able to acquire a lock
|
||
|
|
return callback(null, 'TRYCREATE');
|
||
|
|
}
|
||
|
|
|
||
|
|
let uidNext = item.value.uidNext;
|
||
|
|
destinationUid.unshift(uidNext);
|
||
|
|
|
||
|
|
message._id = null;
|
||
|
|
message.mailbox = target._id;
|
||
|
|
message.uid = uidNext;
|
||
|
|
|
||
|
|
if (!message.meta) {
|
||
|
|
message.meta = {};
|
||
|
|
}
|
||
|
|
message.meta.source = 'IMAPCOPY';
|
||
|
|
|
||
|
|
database.collection('messages').insertOne(message, err => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
this.notifier.addEntries(username, target.path, {
|
||
|
|
command: 'EXISTS',
|
||
|
|
uid: message.uid,
|
||
|
|
message: message._id
|
||
|
|
}, processNext);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
processNext();
|
||
|
|
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
// sends results to socket
|
||
|
|
server.onFetch = function (path, options, session, callback) {
|
||
|
|
this.logger.debug('[%s] Requested FETCH for "%s"', session.id, path);
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!mailbox) {
|
||
|
|
return callback(null, 'NONEXISTENT');
|
||
|
|
}
|
||
|
|
|
||
|
|
let projection = {
|
||
|
|
uid: true,
|
||
|
|
modseq: true,
|
||
|
|
internaldate: true,
|
||
|
|
flags: true,
|
||
|
|
envelope: true,
|
||
|
|
bodystructure: true
|
||
|
|
};
|
||
|
|
|
||
|
|
if (!options.metadataOnly) {
|
||
|
|
//projection.raw = true;
|
||
|
|
projection.mimeTree = true;
|
||
|
|
}
|
||
|
|
|
||
|
|
let query = {
|
||
|
|
mailbox: mailbox._id,
|
||
|
|
uid: {
|
||
|
|
$in: options.messages
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
if (options.changedSince) {
|
||
|
|
query.modseq = {
|
||
|
|
$gt: options.changedSince
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
let cursor = database.collection('messages').find(query).project(projection).sort([
|
||
|
|
['uid', 1]
|
||
|
|
]);
|
||
|
|
|
||
|
|
let processNext = () => {
|
||
|
|
cursor.next((err, message) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!message) {
|
||
|
|
return cursor.close(() => {
|
||
|
|
this.notifier.fire(username, path);
|
||
|
|
return callback(null, true);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
let stream = imapHandler.compileStream(session.formatResponse('FETCH', message.uid, {
|
||
|
|
query: options.query,
|
||
|
|
values: session.getQueryResponse(options.query, message, {
|
||
|
|
logger: this.logger,
|
||
|
|
fetchOptions: {}
|
||
|
|
})
|
||
|
|
}));
|
||
|
|
|
||
|
|
stream.on('error', err => {
|
||
|
|
session.socket.write('INTERNAL ERROR\n');
|
||
|
|
session.socket.destroy(); // ended up in erroneus state, kill the connection to abort
|
||
|
|
return cursor.close(() => callback(err));
|
||
|
|
});
|
||
|
|
|
||
|
|
// send formatted response to socket
|
||
|
|
session.writeStream.write(stream, () => {
|
||
|
|
|
||
|
|
if (!options.markAsSeen || message.flags.includes('\\Seen')) {
|
||
|
|
return processNext();
|
||
|
|
}
|
||
|
|
|
||
|
|
message.flags.unshift('\\Seen');
|
||
|
|
|
||
|
|
database.collection('messages').findOneAndUpdate({
|
||
|
|
_id: message._id
|
||
|
|
}, {
|
||
|
|
$set: {
|
||
|
|
flags: message.flags,
|
||
|
|
unseen: false
|
||
|
|
}
|
||
|
|
}, {}, err => {
|
||
|
|
if (err) {
|
||
|
|
return cursor.close(() => callback(err));
|
||
|
|
}
|
||
|
|
this.notifier.addEntries(username, path, {
|
||
|
|
command: 'FETCH',
|
||
|
|
ignore: session.id,
|
||
|
|
uid: message.uid,
|
||
|
|
flags: message.flags,
|
||
|
|
message: message._id
|
||
|
|
}, processNext);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
processNext();
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Returns an array of matching UID values
|
||
|
|
*
|
||
|
|
* IMAP search can be quite complex, so we optimize here for most common queries to be handled
|
||
|
|
* by MongoDB and then do the final filtering on the client side. This allows
|
||
|
|
*/
|
||
|
|
server.onSearch = function (path, options, session, callback) {
|
||
|
|
|
||
|
|
let username = session.user.username;
|
||
|
|
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!mailbox) {
|
||
|
|
return callback(null, 'NONEXISTENT');
|
||
|
|
}
|
||
|
|
|
||
|
|
// prepare query
|
||
|
|
|
||
|
|
let query = {
|
||
|
|
mailbox: mailbox._id
|
||
|
|
};
|
||
|
|
|
||
|
|
let projection = {
|
||
|
|
uid: true,
|
||
|
|
internaldate: true,
|
||
|
|
flags: true,
|
||
|
|
modseq: true
|
||
|
|
};
|
||
|
|
|
||
|
|
if (options.terms.includes('body') || options.terms.includes('text') || options.terms.includes('header')) {
|
||
|
|
projection.raw = true;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!options.terms.includes('all')) {
|
||
|
|
options.query.forEach(term => {
|
||
|
|
switch (term.key) {
|
||
|
|
case 'modseq':
|
||
|
|
query.modseq = {
|
||
|
|
$gte: term.value
|
||
|
|
};
|
||
|
|
break;
|
||
|
|
case 'uid':
|
||
|
|
if (Array.isArray(term.value)) {
|
||
|
|
if (!term.value.length) {
|
||
|
|
// trying to find a message that does not exist
|
||
|
|
return callback(null, {
|
||
|
|
uidList: [],
|
||
|
|
highestModseq: 0
|
||
|
|
});
|
||
|
|
}
|
||
|
|
query.uid = {
|
||
|
|
$in: term.value
|
||
|
|
};
|
||
|
|
} else {
|
||
|
|
query.uid = term.value;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case 'flag':
|
||
|
|
{
|
||
|
|
let entry = term.exists ? term.value : {
|
||
|
|
$ne: term.value
|
||
|
|
};
|
||
|
|
|
||
|
|
if (!query.$and) {
|
||
|
|
query.$and = [];
|
||
|
|
}
|
||
|
|
query.$and.push({
|
||
|
|
flags: entry
|
||
|
|
});
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case 'not':
|
||
|
|
[].concat(term.value || []).forEach(term => {
|
||
|
|
switch (term.key) {
|
||
|
|
case 'flag':
|
||
|
|
{
|
||
|
|
let entry = !term.exists ? term.value : {
|
||
|
|
$ne: term.value
|
||
|
|
};
|
||
|
|
|
||
|
|
if (!query.$and) {
|
||
|
|
query.$and = [];
|
||
|
|
}
|
||
|
|
query.$and.push({
|
||
|
|
flags: entry
|
||
|
|
});
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
});
|
||
|
|
break;
|
||
|
|
case 'internaldate':
|
||
|
|
{
|
||
|
|
let op = false;
|
||
|
|
let value = new Date(term.value + ' GMT');
|
||
|
|
switch (term.operator) {
|
||
|
|
case '<':
|
||
|
|
op = '$lt';
|
||
|
|
break;
|
||
|
|
case '<=':
|
||
|
|
op = '$lte';
|
||
|
|
break;
|
||
|
|
case '>':
|
||
|
|
op = '$gt';
|
||
|
|
break;
|
||
|
|
case '>=':
|
||
|
|
op = '$gte';
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
let entry = !op ? [{
|
||
|
|
$gte: value
|
||
|
|
}, {
|
||
|
|
$lt: new Date(value.getTime() + 24 * 3600 * 1000)
|
||
|
|
}] : {
|
||
|
|
[op]: value
|
||
|
|
};
|
||
|
|
|
||
|
|
if (!query.$and) {
|
||
|
|
query.$and = [];
|
||
|
|
}
|
||
|
|
query.$and.push({
|
||
|
|
internaldate: entry
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
let cursor = database.collection('messages').find(query).
|
||
|
|
project(projection).
|
||
|
|
sort([
|
||
|
|
['uid', 1]
|
||
|
|
]);
|
||
|
|
|
||
|
|
let highestModseq = 0;
|
||
|
|
let uidList = [];
|
||
|
|
|
||
|
|
let processNext = () => {
|
||
|
|
cursor.next((err, message) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
if (!message) {
|
||
|
|
return cursor.close(() => callback(null, {
|
||
|
|
uidList,
|
||
|
|
highestModseq
|
||
|
|
}));
|
||
|
|
}
|
||
|
|
|
||
|
|
if (message.raw) {
|
||
|
|
message.raw = message.raw.toString();
|
||
|
|
}
|
||
|
|
|
||
|
|
let match = session.matchSearchQuery(message, options.query);
|
||
|
|
if (match && highestModseq < message.modseq) {
|
||
|
|
highestModseq = message.modseq;
|
||
|
|
}
|
||
|
|
if (match) {
|
||
|
|
uidList.push(message.uid);
|
||
|
|
}
|
||
|
|
|
||
|
|
processNext();
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
processNext();
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
server.addToMailbox = (username, path, meta, date, flags, raw, callback) => {
|
||
|
|
server.logger.debug('[%s] Appending message to "%s" for "%s"', 'API', path, username);
|
||
|
|
|
||
|
|
let mimeTree = server.indexer.parseMimeTree(raw);
|
||
|
|
let envelope = server.indexer.getEnvelope(mimeTree);
|
||
|
|
let bodystructure = server.indexer.getBodyStructure(mimeTree);
|
||
|
|
let messageId = envelope[9] || uuidV1() + '@wildduck.email';
|
||
|
|
|
||
|
|
// check if mailbox exists
|
||
|
|
database.collection('mailboxes').findOne({
|
||
|
|
username,
|
||
|
|
path
|
||
|
|
}, (err, mailbox) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!mailbox) {
|
||
|
|
let err = new Error('Mailbox is missing');
|
||
|
|
err.imapResponse = 'TRYCREATE';
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
// check if message with same Message-ID exists
|
||
|
|
database.collection('messages').findOne({
|
||
|
|
mailbox: mailbox._id,
|
||
|
|
messageId
|
||
|
|
}, (err, message) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (message) {
|
||
|
|
// message already exists, skip
|
||
|
|
return callback(null, true, {
|
||
|
|
uidValidity: mailbox.uidValidity,
|
||
|
|
uid: message.uid
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
// acquire new UID
|
||
|
|
database.collection('mailboxes').findOneAndUpdate({
|
||
|
|
_id: mailbox._id
|
||
|
|
}, {
|
||
|
|
$inc: {
|
||
|
|
uidNext: 1
|
||
|
|
}
|
||
|
|
}, {}, (err, item) => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!item || !item.value) {
|
||
|
|
// was not able to acquire a lock
|
||
|
|
let err = new Error('Mailbox is missing');
|
||
|
|
err.imapResponse = 'TRYCREATE';
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
|
||
|
|
let mailbox = item.value;
|
||
|
|
|
||
|
|
let message = {
|
||
|
|
mailbox: mailbox._id,
|
||
|
|
uid: mailbox.uidNext,
|
||
|
|
internaldate: date && new Date(date) || new Date(),
|
||
|
|
raw,
|
||
|
|
flags,
|
||
|
|
unseen: !flags.includes('\\Seen'),
|
||
|
|
meta,
|
||
|
|
modseq: 0,
|
||
|
|
mimeTree,
|
||
|
|
envelope,
|
||
|
|
bodystructure,
|
||
|
|
messageId
|
||
|
|
};
|
||
|
|
|
||
|
|
database.collection('messages').insertOne(message, err => {
|
||
|
|
if (err) {
|
||
|
|
return callback(err);
|
||
|
|
}
|
||
|
|
server.notifier.addEntries(username, path, {
|
||
|
|
command: 'EXISTS',
|
||
|
|
uid: message.uid,
|
||
|
|
message: message._id
|
||
|
|
}, () => {
|
||
|
|
|
||
|
|
let uidValidity = mailbox.uidValidity;
|
||
|
|
let uid = message.uid;
|
||
|
|
|
||
|
|
server.notifier.fire(username, path);
|
||
|
|
|
||
|
|
return callback(null, true, {
|
||
|
|
uidValidity,
|
||
|
|
uid
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
module.exports = done => {
|
||
|
|
MongoClient.connect(config.mongo, (err, db) => {
|
||
|
|
if (err) {
|
||
|
|
server.logger.error('Queue', 'Could not initialize MongoDB: %s', err.message);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
database = db;
|
||
|
|
|
||
|
|
server.indexer = new Indexer();
|
||
|
|
|
||
|
|
// setup notification system for updates
|
||
|
|
server.notifier = new ImapNotifier({
|
||
|
|
database
|
||
|
|
});
|
||
|
|
|
||
|
|
let started = false;
|
||
|
|
|
||
|
|
server.on('error', err => {
|
||
|
|
if (!started) {
|
||
|
|
started = true;
|
||
|
|
return done(err);
|
||
|
|
}
|
||
|
|
log.error('IMAP', err);
|
||
|
|
});
|
||
|
|
|
||
|
|
// start listening
|
||
|
|
server.listen(config.imap.port, config.imap.host, () => {
|
||
|
|
if (started) {
|
||
|
|
return server.close();
|
||
|
|
}
|
||
|
|
started = true;
|
||
|
|
done(null, server);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|