server-ts: data_dir.js -> data_dir.ts

src/services/data_dir.ts

@@ -0,0 +1,78 @@
+"use strict";
+
+/*
+ * This file resolves trilium data path in this order of priority:
+ * - if TRILIUM_DATA_DIR environment variable exists, then its value is used as the path
+ * - if "trilium-data" dir exists directly in the home dir, then it is used
+ * - based on OS convention, if the "app data directory" exists, we'll use or create "trilium-data" directory there
+ * - as a fallback if the previous step fails, we'll use home dir
+ */
+
+import os = require('os');
+import fs = require('fs');
+import path = require('path');
+
+function getAppDataDir() {
+    let appDataDir = os.homedir(); // fallback if OS is not recognized
+
+    if (os.platform() === 'win32' && process.env.APPDATA) {
+        appDataDir = process.env.APPDATA;
+    }
+    else if (os.platform() === 'linux') {
+        appDataDir = `${os.homedir()}/.local/share`;
+    }
+    else if (os.platform() === 'darwin') {
+        appDataDir = `${os.homedir()}/Library/Application Support`;
+    }
+
+    if (!fs.existsSync(appDataDir)) {
+        // expected app data path doesn't exist, let's use fallback
+        appDataDir = os.homedir();
+    }
+
+    return appDataDir;
+}
+
+const DIR_NAME = 'trilium-data';
+
+function getTriliumDataDir() {
+    if (process.env.TRILIUM_DATA_DIR) {
+        if (!fs.existsSync(process.env.TRILIUM_DATA_DIR)) {
+            fs.mkdirSync(process.env.TRILIUM_DATA_DIR, 0o700);
+        }
+
+        return process.env.TRILIUM_DATA_DIR;
+    }
+
+    const homePath = os.homedir() + path.sep + DIR_NAME;
+
+    if (fs.existsSync(homePath)) {
+        return homePath;
+    }
+
+    const appDataPath = getAppDataDir() + path.sep + DIR_NAME;
+
+    if (!fs.existsSync(appDataPath)) {
+        fs.mkdirSync(appDataPath, 0o700);
+    }
+
+    return appDataPath;
+}
+
+const TRILIUM_DATA_DIR = getTriliumDataDir();
+const DIR_SEP = TRILIUM_DATA_DIR + path.sep;
+
+const DOCUMENT_PATH = process.env.TRILIUM_DOCUMENT_PATH || `${DIR_SEP}document.db`;
+const BACKUP_DIR = process.env.TRILIUM_BACKUP_DIR || `${DIR_SEP}backup`;
+const LOG_DIR = process.env.TRILIUM_LOG_DIR || `${DIR_SEP}log`;
+const ANONYMIZED_DB_DIR = process.env.TRILIUM_ANONYMIZED_DB_DIR || `${DIR_SEP}anonymized-db`;
+const CONFIG_INI_PATH = process.env.TRILIUM_CONFIG_INI_PATH || `${DIR_SEP}config.ini`;
+
+module.exports = {
+    TRILIUM_DATA_DIR,
+    DOCUMENT_PATH,
+    BACKUP_DIR,
+    LOG_DIR,
+    ANONYMIZED_DB_DIR,
+    CONFIG_INI_PATH
+};