const { IMAPConnection, PubsubConnector, DatabaseConnector, } = require('nylas-core'); const FetchCategoryList = require('./imap/fetch-category-list') const FetchMessagesInCategory = require('./imap/fetch-messages-in-category') // // account.syncPolicy = { // afterSync: 'idle', // limit: { // after: Date.now() - 7 * 24 * 60 * 60 * 1000, // count: 10000, // }, // interval: 60 * 1000, // folderSyncOptions: { // deepFolderScan: 5 * 60 * 1000, // }, // expiration: Date.now() + 60 * 60 * 1000, // } class SyncWorker { constructor(account, db) { this._db = db; this._conn = null; this._account = account; this._lastSyncTime = null; this._syncTimer = null; this._expirationTimer = null; this._destroyed = false; this.syncNow(); this._listener = PubsubConnector.observableForAccountChanges(account.id).subscribe(() => { this.onAccountChanged(); }); } cleanup() { this._destroyed = true; this._listener.dispose(); this._conn.end(); } onAccountChanged() { console.log("SyncWorker: Detected change to account. Reloading and syncing now.") DatabaseConnector.forShared().then(({Account}) => { Account.find({where: {id: this._account.id}}).then((account) => { this._account = account; this.syncNow(); }) }); } onSyncDidComplete() { const {afterSync} = this._account.syncPolicy; if (afterSync === 'idle') { this.getInboxCategory().then((inboxCategory) => { this._conn.openBox(inboxCategory.name, true).then(() => { console.log("SyncWorker: - Idling on inbox category"); }); }); } else if (afterSync === 'close') { console.log("SyncWorker: - Closing connection"); this._conn.end(); this._conn = null; } else { throw new Error(`onSyncDidComplete: Unknown afterSync behavior: ${afterSync}`) } } onConnectionIdleUpdate() { this.syncNow(); } getInboxCategory() { return this._db.Category.find({where: {role: 'inbox'}}) } ensureConnection() { if (this._conn) { return this._conn.connect(); } return new Promise((resolve) => { const settings = this._account.connectionSettings; const credentials = this._account.decryptedCredentials(); if (!settings || !settings.imap_host) { throw new Error("ensureConnection: There are no IMAP connection settings for this account.") } if (!credentials || !credentials.imap_username) { throw new Error("ensureConnection: There are no IMAP connection credentials for this account.") } const conn = new IMAPConnection(this._db, Object.assign({}, settings, credentials)); conn.on('mail', () => { this.onConnectionIdleUpdate(); }) conn.on('update', () => { this.onConnectionIdleUpdate(); }) conn.on('queue-empty', () => { }); this._conn = conn; resolve(this._conn.connect()); }); } fetchCategoryList() { // todo: syncback operations belong here! return this._conn.runOperation(new FetchCategoryList()) } fetchMessagesInCategory() { const {Category} = this._db; const {folderSyncOptions} = this._account.syncPolicy; return Category.findAll().then((categories) => { const priority = ['inbox', 'drafts', 'sent'].reverse(); const categoriesToSync = categories.sort((a, b) => (priority.indexOf(a.role) - priority.indexOf(b.role)) * -1 ) // const filtered = sorted.filter(cat => // ['[Gmail]/All Mail', '[Gmail]/Trash', '[Gmail]/Spam'].includes(cat.name) // ) return Promise.all(categoriesToSync.map((cat) => this._conn.runOperation(new FetchMessagesInCategory(cat, folderSyncOptions)) )) }); } syncNow() { clearTimeout(this._syncTimer); this.ensureConnection() .then(this.fetchCategoryList.bind(this)) .then(this.syncbackMessageActions.bind(this)) .then(this.fetchMessagesInCategory.bind(this)) .then(() => { this._lastSyncTime = Date.now() }) .catch(console.error) .finally(() => { this.onSyncDidComplete(); this.scheduleNextSync(); }); } scheduleNextSync() { const {interval} = this._account.syncPolicy; if (interval) { const target = this._lastSyncTime + interval; console.log(`SyncWorker: Next sync scheduled for ${new Date(target).toLocaleString()}`); this._syncTimer = setTimeout(() => { this.syncNow(); }, target - Date.now()); } } } module.exports = SyncWorker;