fix setup of new document, closes #966

src/public/app/services/server.js

@@ -0,0 +1,163 @@
+import utils from './utils.js';
+
+const REQUEST_LOGGING_ENABLED = false;
+
+function getHeaders(headers) {
+    // headers need to be lowercase because node.js automatically converts them to lower case
+    // so hypothetical protectedSessionId becomes protectedsessionid on the backend
+    // also avoiding using underscores instead of dashes since nginx filters them out by default
+    const allHeaders = {
+        'trilium-source-id': glob.sourceId,
+        'x-csrf-token': glob.csrfToken
+    };
+
+    for (const headerName in headers) {
+        if (headers[headerName]) {
+            allHeaders[headerName] = headers[headerName];
+        }
+    }
+
+    if (utils.isElectron()) {
+        // passing it explicitely here because of the electron HTTP bypass
+        allHeaders.cookie = document.cookie;
+    }
+
+    return allHeaders;
+}
+
+async function get(url, sourceId) {
+    return await call('GET', url, null, {'trilium-source-id': sourceId});
+}
+
+async function post(url, data, sourceId) {
+    return await call('POST', url, data, {'trilium-source-id': sourceId});
+}
+
+async function put(url, data, sourceId) {
+    return await call('PUT', url, data, {'trilium-source-id': sourceId});
+}
+
+async function remove(url, sourceId) {
+    return await call('DELETE', url, null, {'trilium-source-id': sourceId});
+}
+
+let i = 1;
+const reqResolves = {};
+
+let maxKnownSyncId = 0;
+
+async function call(method, url, data, headers = {}) {
+    let resp;
+
+    const start = Date.now();
+
+    if (utils.isElectron()) {
+        const ipc = utils.dynamicRequire('electron').ipcRenderer;
+        const requestId = i++;
+
+        resp = await new Promise((resolve, reject) => {
+            reqResolves[requestId] = resolve;
+
+            if (REQUEST_LOGGING_ENABLED) {
+                console.log(utils.now(), "Request #" + requestId + " to " + method + " " + url);
+            }
+
+            ipc.send('server-request', {
+                requestId: requestId,
+                headers: getHeaders(headers),
+                method: method,
+                url: "/" + baseApiUrl + url,
+                data: data
+            });
+        });
+    }
+    else {
+        resp = await ajax(url, method, data, headers);
+    }
+
+    const end = Date.now();
+
+    if (glob.PROFILING_LOG) {
+        console.log(`${method} ${url} took ${end - start}ms`);
+    }
+
+    const maxSyncIdStr = resp.headers['trilium-max-sync-id'];
+
+    if (maxSyncIdStr && maxSyncIdStr.trim()) {
+        maxKnownSyncId = Math.max(maxKnownSyncId, parseInt(maxSyncIdStr));
+    }
+
+    return resp.body;
+}
+
+function ajax(url, method, data, headers) {
+    return new Promise((res, rej) => {
+        const options = {
+            url: baseApiUrl + url,
+            type: method,
+            headers: getHeaders(headers),
+            timeout: 60000,
+            success: (body, textStatus, jqXhr) => {
+                const respHeaders = {};
+
+                jqXhr.getAllResponseHeaders().trim().split(/[\r\n]+/).forEach(line => {
+                    const parts = line.split(': ');
+                    const header = parts.shift();
+                    respHeaders[header] = parts.join(': ');
+                });
+
+                res({
+                    body,
+                    headers: respHeaders
+                });
+            },
+            error: async (jqXhr, textStatus, error) => {
+                const message = "Error when calling " + method + " " + url + ": " + textStatus + " - " + error;
+                const toastService = (await import("./toast.js")).default;
+                toastService.showError(message);
+                toastService.throwError(message);
+
+                rej(error);
+            }
+        };
+
+        if (data) {
+            try {
+                options.data = JSON.stringify(data);
+            } catch (e) {
+                console.log("Can't stringify data: ", data, " because of error: ", e)
+            }
+            options.contentType = "application/json";
+        }
+
+        $.ajax(options);
+    });
+}
+
+if (utils.isElectron()) {
+    const ipc = utils.dynamicRequire('electron').ipcRenderer;
+
+    ipc.on('server-response', (event, arg) => {
+        if (REQUEST_LOGGING_ENABLED) {
+            console.log(utils.now(), "Response #" + arg.requestId + ": " + arg.statusCode);
+        }
+
+        reqResolves[arg.requestId]({
+            body: arg.body,
+            headers: arg.headers
+        });
+
+        delete reqResolves[arg.requestId];
+    });
+}
+
+export default {
+    get,
+    post,
+    put,
+    remove,
+    ajax,
+    // don't remove, used from CKEditor image upload!
+    getHeaders,
+    getMaxKnownSyncId: () => maxKnownSyncId
+};