528 lines
19 KiB
JavaScript
528 lines
19 KiB
JavaScript
/*
|
|
Copyright 2017 OpenMarket Ltd
|
|
Copyright 2018 New Vector Ltd
|
|
Copyright 2019 The Matrix.org Foundation C.I.C.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
// This module contains all the code needed to log the console, persist it to
|
|
// disk and submit bug reports. Rationale is as follows:
|
|
// - Monkey-patching the console is preferable to having a log library because
|
|
// we can catch logs by other libraries more easily, without having to all
|
|
// depend on the same log framework / pass the logger around.
|
|
// - We use IndexedDB to persists logs because it has generous disk space
|
|
// limits compared to local storage. IndexedDB does not work in incognito
|
|
// mode, in which case this module will not be able to write logs to disk.
|
|
// However, the logs will still be stored in-memory, so can still be
|
|
// submitted in a bug report should the user wish to: we can also store more
|
|
// logs in-memory than in local storage, which does work in incognito mode.
|
|
// We also need to handle the case where there are 2+ tabs. Each JS runtime
|
|
// generates a random string which serves as the "ID" for that tab/session.
|
|
// These IDs are stored along with the log lines.
|
|
// - Bug reports are sent as a POST over HTTPS: it purposefully does not use
|
|
// Matrix as bug reports may be made when Matrix is not responsive (which may
|
|
// be the cause of the bug). We send the most recent N MB of UTF-8 log data,
|
|
// starting with the most recent, which we know because the "ID"s are
|
|
// actually timestamps. We then purge the remaining logs. We also do this
|
|
// purge on startup to prevent logs from accumulating.
|
|
|
|
// the frequency with which we flush to indexeddb
|
|
const FLUSH_RATE_MS = 30 * 1000;
|
|
|
|
// the length of log data we keep in indexeddb (and include in the reports)
|
|
const MAX_LOG_SIZE = 1024 * 1024 * 5; // 5 MB
|
|
|
|
// A class which monkey-patches the global console and stores log lines.
|
|
class ConsoleLogger {
|
|
constructor() {
|
|
this.logs = "";
|
|
}
|
|
|
|
monkeyPatch(consoleObj) {
|
|
// Monkey-patch console logging
|
|
const consoleFunctionsToLevels = {
|
|
log: "I",
|
|
info: "I",
|
|
warn: "W",
|
|
error: "E",
|
|
};
|
|
Object.keys(consoleFunctionsToLevels).forEach((fnName) => {
|
|
const level = consoleFunctionsToLevels[fnName];
|
|
const originalFn = consoleObj[fnName].bind(consoleObj);
|
|
consoleObj[fnName] = (...args) => {
|
|
this.log(level, ...args);
|
|
originalFn(...args);
|
|
};
|
|
});
|
|
}
|
|
|
|
log(level, ...args) {
|
|
// We don't know what locale the user may be running so use ISO strings
|
|
const ts = new Date().toISOString();
|
|
|
|
// Convert objects and errors to helpful things
|
|
args = args.map((arg) => {
|
|
if (arg instanceof DOMException) {
|
|
return arg.message + ` (${arg.name} | ${arg.code}) ` + (arg.stack ? `\n${arg.stack}` : '');
|
|
} else if (arg instanceof Error) {
|
|
return arg.message + (arg.stack ? `\n${arg.stack}` : '');
|
|
} else if (typeof (arg) === 'object') {
|
|
try {
|
|
return JSON.stringify(arg);
|
|
} catch (e) {
|
|
// In development, it can be useful to log complex cyclic
|
|
// objects to the console for inspection. This is fine for
|
|
// the console, but default `stringify` can't handle that.
|
|
// We workaround this by using a special replacer function
|
|
// to only log values of the root object and avoid cycles.
|
|
return JSON.stringify(arg, (key, value) => {
|
|
if (key && typeof value === "object") {
|
|
return "<object>";
|
|
}
|
|
return value;
|
|
});
|
|
}
|
|
} else {
|
|
return arg;
|
|
}
|
|
});
|
|
|
|
// Some browsers support string formatting which we're not doing here
|
|
// so the lines are a little more ugly but easy to implement / quick to
|
|
// run.
|
|
// Example line:
|
|
// 2017-01-18T11:23:53.214Z W Failed to set badge count
|
|
let line = `${ts} ${level} ${args.join(' ')}\n`;
|
|
// Do some cleanup
|
|
line = line.replace(/token=[a-zA-Z0-9-]+/gm, 'token=xxxxx');
|
|
// Using + really is the quickest way in JS
|
|
// http://jsperf.com/concat-vs-plus-vs-join
|
|
this.logs += line;
|
|
}
|
|
|
|
/**
|
|
* Retrieve log lines to flush to disk.
|
|
* @param {boolean} keepLogs True to not delete logs after flushing.
|
|
* @return {string} \n delimited log lines to flush.
|
|
*/
|
|
flush(keepLogs) {
|
|
// The ConsoleLogger doesn't care how these end up on disk, it just
|
|
// flushes them to the caller.
|
|
if (keepLogs) {
|
|
return this.logs;
|
|
}
|
|
const logsToFlush = this.logs;
|
|
this.logs = "";
|
|
return logsToFlush;
|
|
}
|
|
}
|
|
|
|
// A class which stores log lines in an IndexedDB instance.
|
|
class IndexedDBLogStore {
|
|
constructor(indexedDB, logger) {
|
|
this.indexedDB = indexedDB;
|
|
this.logger = logger;
|
|
this.id = "instance-" + Math.random() + Date.now();
|
|
this.index = 0;
|
|
this.db = null;
|
|
|
|
// these promises are cleared as soon as fulfilled
|
|
this.flushPromise = null;
|
|
// set if flush() is called whilst one is ongoing
|
|
this.flushAgainPromise = null;
|
|
}
|
|
|
|
/**
|
|
* @return {Promise} Resolves when the store is ready.
|
|
*/
|
|
connect() {
|
|
const req = this.indexedDB.open("logs");
|
|
return new Promise((resolve, reject) => {
|
|
req.onsuccess = (event) => {
|
|
this.db = event.target.result;
|
|
// Periodically flush logs to local storage / indexeddb
|
|
setInterval(this.flush.bind(this), FLUSH_RATE_MS);
|
|
resolve();
|
|
};
|
|
|
|
req.onerror = (event) => {
|
|
const err = (
|
|
"Failed to open log database: " + event.target.error.name
|
|
);
|
|
console.error(err);
|
|
reject(new Error(err));
|
|
};
|
|
|
|
// First time: Setup the object store
|
|
req.onupgradeneeded = (event) => {
|
|
const db = event.target.result;
|
|
const logObjStore = db.createObjectStore("logs", {
|
|
keyPath: ["id", "index"],
|
|
});
|
|
// Keys in the database look like: [ "instance-148938490", 0 ]
|
|
// Later on we need to query everything based on an instance id.
|
|
// In order to do this, we need to set up indexes "id".
|
|
logObjStore.createIndex("id", "id", { unique: false });
|
|
|
|
logObjStore.add(
|
|
this._generateLogEntry(
|
|
new Date() + " ::: Log database was created.",
|
|
),
|
|
);
|
|
|
|
const lastModifiedStore = db.createObjectStore("logslastmod", {
|
|
keyPath: "id",
|
|
});
|
|
lastModifiedStore.add(this._generateLastModifiedTime());
|
|
};
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Flush logs to disk.
|
|
*
|
|
* There are guards to protect against race conditions in order to ensure
|
|
* that all previous flushes have completed before the most recent flush.
|
|
* Consider without guards:
|
|
* - A calls flush() periodically.
|
|
* - B calls flush() and wants to send logs immediately afterwards.
|
|
* - If B doesn't wait for A's flush to complete, B will be missing the
|
|
* contents of A's flush.
|
|
* To protect against this, we set 'flushPromise' when a flush is ongoing.
|
|
* Subsequent calls to flush() during this period will chain another flush,
|
|
* then keep returning that same chained flush.
|
|
*
|
|
* This guarantees that we will always eventually do a flush when flush() is
|
|
* called.
|
|
*
|
|
* @return {Promise} Resolved when the logs have been flushed.
|
|
*/
|
|
flush() {
|
|
// check if a flush() operation is ongoing
|
|
if (this.flushPromise) {
|
|
if (this.flushAgainPromise) {
|
|
// this is the 3rd+ time we've called flush() : return the same promise.
|
|
return this.flushAgainPromise;
|
|
}
|
|
// queue up a flush to occur immediately after the pending one completes.
|
|
this.flushAgainPromise = this.flushPromise.then(() => {
|
|
return this.flush();
|
|
}).then(() => {
|
|
this.flushAgainPromise = null;
|
|
});
|
|
return this.flushAgainPromise;
|
|
}
|
|
// there is no flush promise or there was but it has finished, so do
|
|
// a brand new one, destroying the chain which may have been built up.
|
|
this.flushPromise = new Promise((resolve, reject) => {
|
|
if (!this.db) {
|
|
// not connected yet or user rejected access for us to r/w to the db.
|
|
reject(new Error("No connected database"));
|
|
return;
|
|
}
|
|
const lines = this.logger.flush();
|
|
if (lines.length === 0) {
|
|
resolve();
|
|
return;
|
|
}
|
|
const txn = this.db.transaction(["logs", "logslastmod"], "readwrite");
|
|
const objStore = txn.objectStore("logs");
|
|
txn.oncomplete = (event) => {
|
|
resolve();
|
|
};
|
|
txn.onerror = (event) => {
|
|
console.error(
|
|
"Failed to flush logs : ", event,
|
|
);
|
|
reject(
|
|
new Error("Failed to write logs: " + event.target.errorCode),
|
|
);
|
|
};
|
|
objStore.add(this._generateLogEntry(lines));
|
|
const lastModStore = txn.objectStore("logslastmod");
|
|
lastModStore.put(this._generateLastModifiedTime());
|
|
}).then(() => {
|
|
this.flushPromise = null;
|
|
});
|
|
return this.flushPromise;
|
|
}
|
|
|
|
/**
|
|
* Consume the most recent logs and return them. Older logs which are not
|
|
* returned are deleted at the same time, so this can be called at startup
|
|
* to do house-keeping to keep the logs from growing too large.
|
|
*
|
|
* @return {Promise<Object[]>} Resolves to an array of objects. The array is
|
|
* sorted in time (oldest first) based on when the log file was created (the
|
|
* log ID). The objects have said log ID in an "id" field and "lines" which
|
|
* is a big string with all the new-line delimited logs.
|
|
*/
|
|
async consume() {
|
|
const db = this.db;
|
|
|
|
// Returns: a string representing the concatenated logs for this ID.
|
|
// Stops adding log fragments when the size exceeds maxSize
|
|
function fetchLogs(id, maxSize) {
|
|
const objectStore = db.transaction("logs", "readonly").objectStore("logs");
|
|
|
|
return new Promise((resolve, reject) => {
|
|
const query = objectStore.index("id").openCursor(IDBKeyRange.only(id), 'prev');
|
|
let lines = '';
|
|
query.onerror = (event) => {
|
|
reject(new Error("Query failed: " + event.target.errorCode));
|
|
};
|
|
query.onsuccess = (event) => {
|
|
const cursor = event.target.result;
|
|
if (!cursor) {
|
|
resolve(lines);
|
|
return; // end of results
|
|
}
|
|
lines = cursor.value.lines + lines;
|
|
if (lines.length >= maxSize) {
|
|
resolve(lines);
|
|
} else {
|
|
cursor.continue();
|
|
}
|
|
};
|
|
});
|
|
}
|
|
|
|
// Returns: A sorted array of log IDs. (newest first)
|
|
function fetchLogIds() {
|
|
// To gather all the log IDs, query for all records in logslastmod.
|
|
const o = db.transaction("logslastmod", "readonly").objectStore(
|
|
"logslastmod",
|
|
);
|
|
return selectQuery(o, undefined, (cursor) => {
|
|
return {
|
|
id: cursor.value.id,
|
|
ts: cursor.value.ts,
|
|
};
|
|
}).then((res) => {
|
|
// Sort IDs by timestamp (newest first)
|
|
return res.sort((a, b) => {
|
|
return b.ts - a.ts;
|
|
}).map((a) => a.id);
|
|
});
|
|
}
|
|
|
|
function deleteLogs(id) {
|
|
return new Promise((resolve, reject) => {
|
|
const txn = db.transaction(
|
|
["logs", "logslastmod"], "readwrite",
|
|
);
|
|
const o = txn.objectStore("logs");
|
|
// only load the key path, not the data which may be huge
|
|
const query = o.index("id").openKeyCursor(IDBKeyRange.only(id));
|
|
query.onsuccess = (event) => {
|
|
const cursor = event.target.result;
|
|
if (!cursor) {
|
|
return;
|
|
}
|
|
o.delete(cursor.primaryKey);
|
|
cursor.continue();
|
|
};
|
|
txn.oncomplete = () => {
|
|
resolve();
|
|
};
|
|
txn.onerror = (event) => {
|
|
reject(
|
|
new Error(
|
|
"Failed to delete logs for " +
|
|
`'${id}' : ${event.target.errorCode}`,
|
|
),
|
|
);
|
|
};
|
|
// delete last modified entries
|
|
const lastModStore = txn.objectStore("logslastmod");
|
|
lastModStore.delete(id);
|
|
});
|
|
}
|
|
|
|
const allLogIds = await fetchLogIds();
|
|
let removeLogIds = [];
|
|
const logs = [];
|
|
let size = 0;
|
|
for (let i = 0; i < allLogIds.length; i++) {
|
|
const lines = await fetchLogs(allLogIds[i], MAX_LOG_SIZE - size);
|
|
|
|
// always add the log file: fetchLogs will truncate once the maxSize we give it is
|
|
// exceeded, so we'll go over the max but only by one fragment's worth.
|
|
logs.push({
|
|
lines: lines,
|
|
id: allLogIds[i],
|
|
});
|
|
size += lines.length;
|
|
|
|
// If fetchLogs truncated we'll now be at or over the size limit,
|
|
// in which case we should stop and remove the rest of the log files.
|
|
if (size >= MAX_LOG_SIZE) {
|
|
// the remaining log IDs should be removed. If we go out of
|
|
// bounds this is just []
|
|
removeLogIds = allLogIds.slice(i + 1);
|
|
break;
|
|
}
|
|
}
|
|
if (removeLogIds.length > 0) {
|
|
console.log("Removing logs: ", removeLogIds);
|
|
// Don't await this because it's non-fatal if we can't clean up
|
|
// logs.
|
|
Promise.all(removeLogIds.map((id) => deleteLogs(id))).then(() => {
|
|
console.log(`Removed ${removeLogIds.length} old logs.`);
|
|
}, (err) => {
|
|
console.error(err);
|
|
});
|
|
}
|
|
return logs;
|
|
}
|
|
|
|
_generateLogEntry(lines) {
|
|
return {
|
|
id: this.id,
|
|
lines: lines,
|
|
index: this.index++,
|
|
};
|
|
}
|
|
|
|
_generateLastModifiedTime() {
|
|
return {
|
|
id: this.id,
|
|
ts: Date.now(),
|
|
};
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Helper method to collect results from a Cursor and promiseify it.
|
|
* @param {ObjectStore|Index} store The store to perform openCursor on.
|
|
* @param {IDBKeyRange=} keyRange Optional key range to apply on the cursor.
|
|
* @param {Function} resultMapper A function which is repeatedly called with a
|
|
* Cursor.
|
|
* Return the data you want to keep.
|
|
* @return {Promise<T[]>} Resolves to an array of whatever you returned from
|
|
* resultMapper.
|
|
*/
|
|
function selectQuery(store, keyRange, resultMapper) {
|
|
const query = store.openCursor(keyRange);
|
|
return new Promise((resolve, reject) => {
|
|
const results = [];
|
|
query.onerror = (event) => {
|
|
reject(new Error("Query failed: " + event.target.errorCode));
|
|
};
|
|
// collect results
|
|
query.onsuccess = (event) => {
|
|
const cursor = event.target.result;
|
|
if (!cursor) {
|
|
resolve(results);
|
|
return; // end of results
|
|
}
|
|
results.push(resultMapper(cursor));
|
|
cursor.continue();
|
|
};
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Configure rage shaking support for sending bug reports.
|
|
* Modifies globals.
|
|
* @param {boolean} setUpPersistence When true (default), the persistence will
|
|
* be set up immediately for the logs.
|
|
* @return {Promise} Resolves when set up.
|
|
*/
|
|
export function init(setUpPersistence = true) {
|
|
if (global.mx_rage_initPromise) {
|
|
return global.mx_rage_initPromise;
|
|
}
|
|
global.mx_rage_logger = new ConsoleLogger();
|
|
global.mx_rage_logger.monkeyPatch(window.console);
|
|
|
|
if (setUpPersistence) {
|
|
return tryInitStorage();
|
|
}
|
|
|
|
global.mx_rage_initPromise = Promise.resolve();
|
|
return global.mx_rage_initPromise;
|
|
}
|
|
|
|
/**
|
|
* Try to start up the rageshake storage for logs. If not possible (client unsupported)
|
|
* then this no-ops.
|
|
* @return {Promise} Resolves when complete.
|
|
*/
|
|
export function tryInitStorage() {
|
|
if (global.mx_rage_initStoragePromise) {
|
|
return global.mx_rage_initStoragePromise;
|
|
}
|
|
|
|
console.log("Configuring rageshake persistence...");
|
|
|
|
// just *accessing* indexedDB throws an exception in firefox with
|
|
// indexeddb disabled.
|
|
let indexedDB;
|
|
try {
|
|
indexedDB = window.indexedDB;
|
|
} catch (e) {}
|
|
|
|
if (indexedDB) {
|
|
global.mx_rage_store = new IndexedDBLogStore(indexedDB, global.mx_rage_logger);
|
|
global.mx_rage_initStoragePromise = global.mx_rage_store.connect();
|
|
return global.mx_rage_initStoragePromise;
|
|
}
|
|
global.mx_rage_initStoragePromise = Promise.resolve();
|
|
return global.mx_rage_initStoragePromise;
|
|
}
|
|
|
|
export function flush() {
|
|
if (!global.mx_rage_store) {
|
|
return;
|
|
}
|
|
global.mx_rage_store.flush();
|
|
}
|
|
|
|
/**
|
|
* Clean up old logs.
|
|
* @return {Promise} Resolves if cleaned logs.
|
|
*/
|
|
export async function cleanup() {
|
|
if (!global.mx_rage_store) {
|
|
return;
|
|
}
|
|
await global.mx_rage_store.consume();
|
|
}
|
|
|
|
/**
|
|
* Get a recent snapshot of the logs, ready for attaching to a bug report
|
|
*
|
|
* @return {Array<{lines: string, id, string}>} list of log data
|
|
*/
|
|
export async function getLogsForReport() {
|
|
if (!global.mx_rage_logger) {
|
|
throw new Error(
|
|
"No console logger, did you forget to call init()?",
|
|
);
|
|
}
|
|
// If in incognito mode, store is null, but we still want bug report
|
|
// sending to work going off the in-memory console logs.
|
|
if (global.mx_rage_store) {
|
|
// flush most recent logs
|
|
await global.mx_rage_store.flush();
|
|
return await global.mx_rage_store.consume();
|
|
} else {
|
|
return [{
|
|
lines: global.mx_rage_logger.flush(true),
|
|
id: "-",
|
|
}];
|
|
}
|
|
}
|