- Modified shared.ts to maintain a Map of clients per accountId - Each account gets its own Matrix client instance - Backwards compatible with single-account usage - resolveMatrixAuth now accepts accountId parameter - stopSharedClient can stop specific or all accounts Files changed: - src/matrix/client/shared.ts (main changes) - src/matrix/client/config.ts (accountId support) - src/matrix/accounts.ts (list enabled accounts) - src/matrix/monitor/index.ts (pass accountId) - src/types.ts (accounts config type)
103 lines
2.9 KiB
TypeScript
103 lines
2.9 KiB
TypeScript
import fs from "node:fs";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
|
|
import { getMatrixRuntime } from "../runtime.js";
|
|
|
|
export type MatrixStoredCredentials = {
|
|
homeserver: string;
|
|
userId: string;
|
|
accessToken: string;
|
|
deviceId?: string;
|
|
createdAt: string;
|
|
lastUsedAt?: string;
|
|
};
|
|
|
|
const CREDENTIALS_FILENAME = "credentials.json";
|
|
|
|
export function resolveMatrixCredentialsDir(
|
|
env: NodeJS.ProcessEnv = process.env,
|
|
stateDir?: string,
|
|
): string {
|
|
const resolvedStateDir =
|
|
stateDir ?? getMatrixRuntime().state.resolveStateDir(env, os.homedir);
|
|
return path.join(resolvedStateDir, "credentials", "matrix");
|
|
}
|
|
|
|
export function resolveMatrixCredentialsPath(env: NodeJS.ProcessEnv = process.env): string {
|
|
const dir = resolveMatrixCredentialsDir(env);
|
|
return path.join(dir, CREDENTIALS_FILENAME);
|
|
}
|
|
|
|
export function loadMatrixCredentials(
|
|
env: NodeJS.ProcessEnv = process.env,
|
|
): MatrixStoredCredentials | null {
|
|
const credPath = resolveMatrixCredentialsPath(env);
|
|
try {
|
|
if (!fs.existsSync(credPath)) return null;
|
|
const raw = fs.readFileSync(credPath, "utf-8");
|
|
const parsed = JSON.parse(raw) as Partial<MatrixStoredCredentials>;
|
|
if (
|
|
typeof parsed.homeserver !== "string" ||
|
|
typeof parsed.userId !== "string" ||
|
|
typeof parsed.accessToken !== "string"
|
|
) {
|
|
return null;
|
|
}
|
|
return parsed as MatrixStoredCredentials;
|
|
} catch {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
export function saveMatrixCredentials(
|
|
credentials: Omit<MatrixStoredCredentials, "createdAt" | "lastUsedAt">,
|
|
env: NodeJS.ProcessEnv = process.env,
|
|
): void {
|
|
const dir = resolveMatrixCredentialsDir(env);
|
|
fs.mkdirSync(dir, { recursive: true });
|
|
|
|
const credPath = resolveMatrixCredentialsPath(env);
|
|
|
|
const existing = loadMatrixCredentials(env);
|
|
const now = new Date().toISOString();
|
|
|
|
const toSave: MatrixStoredCredentials = {
|
|
...credentials,
|
|
createdAt: existing?.createdAt ?? now,
|
|
lastUsedAt: now,
|
|
};
|
|
|
|
fs.writeFileSync(credPath, JSON.stringify(toSave, null, 2), "utf-8");
|
|
}
|
|
|
|
export function touchMatrixCredentials(env: NodeJS.ProcessEnv = process.env): void {
|
|
const existing = loadMatrixCredentials(env);
|
|
if (!existing) return;
|
|
|
|
existing.lastUsedAt = new Date().toISOString();
|
|
const credPath = resolveMatrixCredentialsPath(env);
|
|
fs.writeFileSync(credPath, JSON.stringify(existing, null, 2), "utf-8");
|
|
}
|
|
|
|
export function clearMatrixCredentials(env: NodeJS.ProcessEnv = process.env): void {
|
|
const credPath = resolveMatrixCredentialsPath(env);
|
|
try {
|
|
if (fs.existsSync(credPath)) {
|
|
fs.unlinkSync(credPath);
|
|
}
|
|
} catch {
|
|
// ignore
|
|
}
|
|
}
|
|
|
|
export function credentialsMatchConfig(
|
|
stored: MatrixStoredCredentials,
|
|
config: { homeserver: string; userId: string },
|
|
): boolean {
|
|
// If userId is empty (token-based auth), only match homeserver
|
|
if (!config.userId) {
|
|
return stored.homeserver === config.homeserver;
|
|
}
|
|
return stored.homeserver === config.homeserver && stored.userId === config.userId;
|
|
}
|