Tools/Notes
1
0
Fork 0
mirror of https://github.com/TriliumNext/Notes.git synced 2025-08-10 10:22:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed problem with using existing databases.

Browse Source
This commit is contained in:
chesspro13 2024-09-14 10:04:39 -07:00
parent 2e21a45fc1
commit d92820280d
No known key found for this signature in database
GPG Key ID: 5FEAE94D298066E5
2 changed files with 20 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/services/encryption/open_id_encryption.ts
View File

@ -4,26 +4,11 @@ import utils from "../utils.js";
import dataEncryptionService from "./data_encryption.js";
import sql from "../sql.js";
import sqlInit from "../sql_init.js";
import OpenIDError from "../../errors/open_id_error.js";
function saveUser(subjectIdentifier: string, name: string, email: string) {
if (isUserSaved()) return false;
// Allows setup with existing instances of trilium
sql.execute(`
CREATE TABLE IF NOT EXISTS "user_data"
(
tmpID INT,
username TEXT,
email TEXT,
userIDEcnryptedDataKey TEXT,
userIDVerificationHash TEXT,
salt TEXT,
derivedKey TEXT,
isSetup TEXT DEFAULT "false",
UNIQUE (tmpID),
PRIMARY KEY (tmpID)
);`);
const verificationSalt = utils.randomSecureToken(32);
const derivedKeySalt = utils.randomSecureToken(32);
@ -32,8 +17,7 @@ function saveUser(subjectIdentifier: string, name: string, email: string) {
verificationSalt
);
if (verificationHash === undefined) {
console.log("Verification hash undefined!");
return undefined;
throw new OpenIDError("Verification hash undefined!")
}
const userIDEncryptedDataKey = setDataKey(
@ -75,13 +59,11 @@ function isUserSaved() {
function verifyOpenIDSubjectIdentifier(subjectIdentifier: string) {
if (!sqlInit.isDbInitialized()) {
console.log("Database not initialized!");
return undefined;
throw new OpenIDError("Database not initialized!");
}
if (!isUserSaved()) {
console.log("DATABASE NOT SETUP");
return undefined;
if (isUserSaved()) {
return false;
}
const salt = sql.getValue("SELECT salt FROM user_data;");
@ -115,7 +97,6 @@ function setDataKey(
plainTextDataKey: string | Buffer,
salt: string
) {
console.log("Subject Identifier: " + subjectIdentifier);
const subjectIdentifierDerivedKey =
myScryptService.getSubjectIdentifierDerivedKey(subjectIdentifier, salt);
@ -132,7 +113,6 @@ function setDataKey(
}
function getDataKey(subjectIdentifier: string) {
console.log("Subject Identifier: " + subjectIdentifier);
const subjectIdentifierDerivedKey =
myScryptService.getSubjectIdentifierDerivedKey(subjectIdentifier);

15
src/services/sql_init.ts
View File

@ -48,6 +48,21 @@ async function initDbConnection() {
sql.execute('CREATE TEMP TABLE "param_list" (`paramId` TEXT NOT NULL PRIMARY KEY)');
sql.execute(`
CREATE TABLE IF NOT EXISTS "user_data"
(
tmpID INT,
username TEXT,
email TEXT,
userIDEcnryptedDataKey TEXT,
userIDVerificationHash TEXT,
salt TEXT,
derivedKey TEXT,
isSetup TEXT DEFAULT "false",
UNIQUE (tmpID),
PRIMARY KEY (tmpID)
);`)
dbReady.resolve();
}