1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
/*
* Copyright (C) 2021-2022 Kristóf Marussy <kristof@marussy.com>
*
* This file is part of Sophie.
*
* Sophie is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, version 3.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*
* SPDX-License-Identifier: AGPL-3.0-only
*/
import { watch } from 'fs';
import { readFile, stat, writeFile } from 'fs/promises';
import JSON5 from 'json5';
import { throttle } from 'lodash';
import { IDisposer } from 'mobx-state-tree';
import { join } from 'path';
import { CONFIG_DEBOUNCE_TIME, ConfigPersistence, ReadConfigResult } from '../ConfigPersistence';
import type { ConfigSnapshotOut } from '../../stores/Config';
export class ConfigPersistenceImpl implements ConfigPersistence {
readonly #userDataDir: string;
readonly #configFileName: string;
readonly #configFilePath: string;
constructor(
userDataDir: string,
configFileName: string,
) {
this.#userDataDir = userDataDir;
this.#configFileName = configFileName;
this.#configFilePath = join(this.#userDataDir, this.#configFileName);
}
async readConfig(): Promise<ReadConfigResult> {
let configStr;
try {
configStr = await readFile(this.#configFilePath, 'utf8');
} catch (err) {
if ((err as NodeJS.ErrnoException).code === 'ENOENT') {
return { found: false };
}
throw err;
}
return {
found: true,
data: JSON5.parse(configStr),
};
}
async writeConfig(configSnapshot: ConfigSnapshotOut): Promise<Date> {
const configJson = JSON5.stringify(configSnapshot, {
space: 2,
});
await writeFile(this.#configFilePath, configJson, 'utf8');
const { mtime } = await stat(this.#configFilePath);
return mtime;
}
watchConfig(callback: (mtime: Date) => Promise<void>): IDisposer {
const configChanged = throttle(async () => {
let mtime: Date;
try {
const stats = await stat(this.#configFilePath);
mtime = stats.mtime;
} catch (err) {
if ((err as NodeJS.ErrnoException).code === 'ENOENT') {
return;
}
throw err;
}
return callback(mtime);
}, CONFIG_DEBOUNCE_TIME);
const watcher = watch(this.#userDataDir, {
persistent: false,
});
watcher.on('change', (eventType, filename) => {
if (eventType === 'change'
&& (filename === this.#configFileName || filename === null)) {
configChanged()?.catch((err) => {
console.log('Unhandled error while listening for config changes', err);
});
}
});
return () => watcher.close();
}
}
|