101 lines
2.9 KiB
JavaScript
101 lines
2.9 KiB
JavaScript
'use strict';
|
|
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
const os = require('os');
|
|
|
|
// 生产环境用 /etc/clawd/，开发环境用 ~/.clawd/
|
|
const CONFIG_DIR = process.env.CLAWD_CONFIG_DIR
|
|
|| (process.getuid && process.getuid() === 0
|
|
? '/etc/clawd'
|
|
: path.join(os.homedir(), '.clawd'));
|
|
|
|
const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
|
|
|
|
const DEFAULTS = {
|
|
server: 'wss://claw.cutos.ai/ws',
|
|
claw_id: null,
|
|
token: null,
|
|
heartbeat_interval: 30, // 秒
|
|
/** 云端已激活：用于启动/重连时立即点亮 alarm（pwr），不等首包 connected */
|
|
activated: false,
|
|
ssh_secret_key: null,
|
|
share_key: null,
|
|
headscale_server: 'https://hs.claw.cutos.ai',
|
|
};
|
|
|
|
function _generateSshSecretKey() {
|
|
const bytes = require('crypto').randomBytes(16);
|
|
return 'sk-' + bytes.toString('hex');
|
|
}
|
|
|
|
/**
|
|
* 生成可读性好的 Samba 共享密码，格式：xxxxxx-xxxxxx-xxxxXX（三段各6字符，连字符分隔）。
|
|
* 最后一段包含至少1个大写字母和1个数字，其余为小写字母。
|
|
* 示例：juqhuf-hykgyh-mykGi3
|
|
*/
|
|
function _generateShareKey() {
|
|
const crypto = require('crypto');
|
|
const lower = 'abcdefghijklmnopqrstuvwxyz';
|
|
const upper = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ';
|
|
const digits = '0123456789';
|
|
|
|
function rndChars(chars, n) {
|
|
const bytes = crypto.randomBytes(n);
|
|
return Array.from(bytes).map(b => chars[b % chars.length]).join('');
|
|
}
|
|
|
|
const g1 = rndChars(lower, 6);
|
|
const g2 = rndChars(lower, 6);
|
|
// 第三段：4个小写 + 1个大写 + 1个数字，随机打乱顺序
|
|
const g3raw = (rndChars(lower, 4) + rndChars(upper, 1) + rndChars(digits, 1)).split('');
|
|
for (let i = g3raw.length - 1; i > 0; i--) {
|
|
const j = crypto.randomBytes(1)[0] % (i + 1);
|
|
[g3raw[i], g3raw[j]] = [g3raw[j], g3raw[i]];
|
|
}
|
|
return `${g1}-${g2}-${g3raw.join('')}`;
|
|
}
|
|
|
|
function load() {
|
|
let cfg;
|
|
try {
|
|
if (fs.existsSync(CONFIG_FILE)) {
|
|
const raw = fs.readFileSync(CONFIG_FILE, 'utf8');
|
|
cfg = Object.assign({}, DEFAULTS, JSON.parse(raw));
|
|
}
|
|
} catch (e) {
|
|
const log = require('./logger');
|
|
log.error('config', '读取配置失败，使用默认值:', e.message);
|
|
}
|
|
if (!cfg) cfg = Object.assign({}, DEFAULTS);
|
|
|
|
let dirty = false;
|
|
if (!cfg.ssh_secret_key) {
|
|
cfg.ssh_secret_key = _generateSshSecretKey();
|
|
dirty = true;
|
|
}
|
|
if (!cfg.share_key) {
|
|
cfg.share_key = _generateShareKey();
|
|
dirty = true;
|
|
}
|
|
if (dirty) save(cfg);
|
|
|
|
return cfg;
|
|
}
|
|
|
|
function save(data) {
|
|
try {
|
|
fs.mkdirSync(CONFIG_DIR, { recursive: true });
|
|
fs.writeFileSync(CONFIG_FILE, JSON.stringify(data, null, 2), 'utf8');
|
|
} catch (e) {
|
|
const log = require('./logger');
|
|
log.error('config', '写入配置失败:', e.message);
|
|
}
|
|
}
|
|
|
|
function getConfigPath() {
|
|
return CONFIG_FILE;
|
|
}
|
|
|
|
module.exports = { load, save, getConfigPath };
|