ElectronPlatform: Implement the EventIndexManager for Seshat.

pull/11125/head
Damir Jelić 2019-11-13 12:15:26 +01:00
parent a6839afc1f
commit c3c5756c7a
2 changed files with 131 additions and 55 deletions

View File

@ -155,17 +155,6 @@ autoUpdater.on('update-downloaded', (ev, releaseNotes, releaseName, releaseDate,
ipcMain.on('ipcCall', async function(ev, payload) { ipcMain.on('ipcCall', async function(ev, payload) {
if (!mainWindow) return; if (!mainWindow) return;
const sendError = (id, e) => {
const error = {
message: e.message
}
mainWindow.webContents.send('ipcReply', {
id:id,
error: error
});
}
const args = payload.args || []; const args = payload.args || [];
let ret; let ret;
@ -218,12 +207,50 @@ ipcMain.on('ipcCall', async function(ev, payload) {
ret = vectorConfig; ret = vectorConfig;
break; break;
default:
mainWindow.webContents.send('ipcReply', {
id: payload.id,
error: "Unknown IPC Call: " + payload.name,
});
return;
}
mainWindow.webContents.send('ipcReply', {
id: payload.id,
reply: ret,
});
});
ipcMain.on('seshat', async function(ev, payload) {
if (!mainWindow) return;
const sendError = (id, e) => {
const error = {
message: e.message
}
mainWindow.webContents.send('seshatReply', {
id:id,
error: error
});
}
const args = payload.args || [];
let ret;
switch (payload.name) {
case 'supportsEventIndexing':
if (Seshat === null) ret = false;
else ret = true;
break;
case 'initEventIndex': case 'initEventIndex':
if (args[0] && eventIndex === null) { if (args[0] && eventIndex === null) {
let p = path.normalize(path.join(eventStorePath, args[0])); let p = path.normalize(path.join(eventStorePath, args[0]));
try { try {
await makeDir(p); await makeDir(p);
eventIndex = new Seshat(p); eventIndex = new Seshat(p);
// eventIndex = new Seshat(p, {passphrase: "DEFAULT_PASSPHRASE"});
console.log("Initialized event store"); console.log("Initialized event store");
} catch (e) { } catch (e) {
sendError(payload.id, e); sendError(payload.id, e);
@ -317,14 +344,14 @@ ipcMain.on('ipcCall', async function(ev, payload) {
break; break;
default: default:
mainWindow.webContents.send('ipcReply', { mainWindow.webContents.send('seshatReply', {
id: payload.id, id: payload.id,
error: "Unknown IPC Call: " + payload.name, error: "Unknown IPC Call: " + payload.name,
}); });
return; return;
} }
mainWindow.webContents.send('ipcReply', { mainWindow.webContents.send('seshatReply', {
id: payload.id, id: payload.id,
reply: ret, reply: ret,
}); });

View File

@ -20,6 +20,7 @@ limitations under the License.
*/ */
import VectorBasePlatform, {updateCheckStatusEnum} from './VectorBasePlatform'; import VectorBasePlatform, {updateCheckStatusEnum} from './VectorBasePlatform';
import BaseEventIndexManager from 'matrix-react-sdk/lib/BaseEventIndexManager';
import dis from 'matrix-react-sdk/lib/dispatcher'; import dis from 'matrix-react-sdk/lib/dispatcher';
import { _t } from 'matrix-react-sdk/lib/languageHandler'; import { _t } from 'matrix-react-sdk/lib/languageHandler';
import Promise from 'bluebird'; import Promise from 'bluebird';
@ -66,12 +67,100 @@ function getUpdateCheckStatus(status) {
} }
} }
class SeshatIndexerManager extends BaseEventIndexManager {
constructor() {
super();
this._pendingIpcCalls = {};
this._nextIpcCallId = 0;
ipcRenderer.on('seshatReply', this._onIpcReply.bind(this));
}
async _ipcCall(name: string, ...args: []): Promise<{}> {
// TODO this should be moved into the preload.js file.
const ipcCallId = ++this._nextIpcCallId;
return new Promise((resolve, reject) => {
this._pendingIpcCalls[ipcCallId] = {resolve, reject};
window.ipcRenderer.send('seshat', {id: ipcCallId, name, args});
});
}
_onIpcReply(ev: {}, payload: {}) {
if (payload.id === undefined) {
console.warn("Ignoring IPC reply with no ID");
return;
}
if (this._pendingIpcCalls[payload.id] === undefined) {
console.warn("Unknown IPC payload ID: " + payload.id);
return;
}
const callbacks = this._pendingIpcCalls[payload.id];
delete this._pendingIpcCalls[payload.id];
if (payload.error) {
callbacks.reject(payload.error);
} else {
callbacks.resolve(payload.reply);
}
}
async supportsEventIndexing(): Promise<boolean> {
return this._ipcCall('supportsEventIndexing');
}
async initEventIndex(userId: string): Promise<> {
return this._ipcCall('initEventIndex', userId);
}
async addEventToIndex(ev: MatrixEvent, profile: MatrixProfile): Promise<> {
return this._ipcCall('addEventToIndex', ev, profile);
}
async isEventIndexEmpty(): Promise<boolean> {
return this._ipcCall('isEventIndexEmpty');
}
async commitLiveEvents(): Promise<> {
return this._ipcCall('commitLiveEvents');
}
async searchEventIndex(searchConfig: SearchConfig): Promise<SearchResult> {
return this._ipcCall('searchEventIndex', searchConfig);
}
async addHistoricEvents(
events: [HistoricEvent],
checkpoint: CrawlerCheckpoint | null = null,
oldCheckpoint: CrawlerCheckpoint | null = null,
): Promise<> {
return this._ipcCall('addHistoricEvents', events, checkpoint, oldCheckpoint);
}
async addCrawlerCheckpoint(checkpoint: CrawlerCheckpoint): Promise<> {
return this._ipcCall('addCrawlerCheckpoint', checkpoint);
}
async removeCrawlerCheckpoint(checkpoint: CrawlerCheckpoint): Promise<> {
return this._ipcCall('removeCrawlerCheckpoint', checkpoint);
}
async loadCheckpoints(): Promise<[CrawlerCheckpoint]> {
return this._ipcCall('loadCheckpoints');
}
async deleteEventIndex(): Promise<> {
return this._ipcCall('deleteEventIndex');
}
}
export default class ElectronPlatform extends VectorBasePlatform { export default class ElectronPlatform extends VectorBasePlatform {
constructor() { constructor() {
super(); super();
this._pendingIpcCalls = {}; this._pendingIpcCalls = {};
this._nextIpcCallId = 0; this._nextIpcCallId = 0;
this.eventIndexManager = new SeshatIndexerManager();
dis.register(_onAction); dis.register(_onAction);
/* /*
@ -294,47 +383,7 @@ export default class ElectronPlatform extends VectorBasePlatform {
} }
} }
async initEventIndex(userId: string): void { getEventIndexingManager(): BaseEventIndexManager | null {
return this._ipcCall('initEventIndex', userId); return this.eventIndexManager;
}
supportsEventIndexing(): boolean {
return true;
}
async addEventToIndex(ev: {}, profile: {}): void {
return this._ipcCall('addEventToIndex', ev, profile);
}
async isEventIndexEmpty(): Promise<boolean> {
return this._ipcCall('isEventIndexEmpty');
}
async commitLiveEvents(): Promise<{}> {
return this._ipcCall('commitLiveEvents');
}
async searchEventIndex(term: string): Promise<{}> {
return this._ipcCall('searchEventIndex', term);
}
async addHistoricEvents(events: string, checkpoint: {} = null, oldCheckpoint: {} = null): Promise<{}> {
return this._ipcCall('addHistoricEvents', events, checkpoint, oldCheckpoint);
}
async addCrawlerCheckpoint(checkpoint: {}): Promise<{}> {
return this._ipcCall('addCrawlerCheckpoint', checkpoint);
}
async removeCrawlerCheckpoint(checkpoint: {}): Promise<{}> {
return this._ipcCall('removeCrawlerCheckpoint', checkpoint);
}
async loadCheckpoints(checkpoint: {}): Promise<[{}]> {
return this._ipcCall('loadCheckpoints');
}
async deleteEventIndex(): Promise<> {
return this._ipcCall('deleteEventIndex');
} }
} }