@nocobase/cli 2.1.0-beta.2 → 2.1.0-beta.21

package/dist/lib/auth-store.js

@@ -0,0 +1,357 @@
+/**
+ * This file is part of the NocoBase (R) project.
+ * Copyright (c) 2020-2024 NocoBase Co., Ltd.
+ * Authors: NocoBase Team.
+ *
+ * This project is dual-licensed under AGPL-3.0 and NocoBase Commercial License.
+ * For more information, please refer to: https://www.nocobase.com/agreement.
+ */
+import { promises as fs } from 'node:fs';
+import path from 'node:path';
+import { resolveCliHomeDir, resolveConfiguredEnvPath, resolveDefaultConfigScope, resolveEnvRelativePath, } from './cli-home.js';
+function normalizeStoredEnvKind(value) {
+    const kind = String(value ?? '').trim();
+    if (kind === 'remote') {
+        return 'http';
+    }
+    if (kind === 'local' || kind === 'http' || kind === 'docker' || kind === 'ssh') {
+        return kind;
+    }
+    return undefined;
+}
+function normalizeOptionalString(value) {
+    const normalized = String(value ?? '').trim();
+    return normalized || undefined;
+}
+export function readEnvApiBaseUrl(config) {
+    if (!config) {
+        return undefined;
+    }
+    return (normalizeOptionalString(config.apiBaseUrl)
+        ?? normalizeOptionalString(config.baseUrl)
+        ?? normalizeOptionalString(config.apibaseUrl));
+}
+export function resolveEnvKind(config) {
+    if (!config) {
+        return undefined;
+    }
+    const explicitKind = normalizeStoredEnvKind(config.kind);
+    if (explicitKind) {
+        return explicitKind;
+    }
+    const source = String(config.source ?? '').trim();
+    if (source === 'docker') {
+        return 'docker';
+    }
+    if (source === 'npm' || source === 'git' || source === 'local') {
+        return 'local';
+    }
+    if (String(config.appRootPath ?? '').trim()) {
+        return 'local';
+    }
+    if (readEnvApiBaseUrl(config) || config.auth) {
+        return 'http';
+    }
+    return undefined;
+}
+function normalizeEnvConfigEntry(entry) {
+    if (!entry) {
+        return entry;
+    }
+    const { kind: _kind, apiBaseUrl: _apiBaseUrl, baseUrl: _baseUrl, apibaseUrl: _legacyApiBaseUrl, ...rest } = entry;
+    const normalizedKind = resolveEnvKind(entry);
+    const apiBaseUrl = readEnvApiBaseUrl(entry);
+    return {
+        ...rest,
+        ...(normalizedKind ? { kind: normalizedKind } : {}),
+        ...(apiBaseUrl !== undefined ? { apiBaseUrl } : {}),
+    };
+}
+function normalizeAuthConfig(config) {
+    return {
+        name: config.name || config.dockerResourcePrefix,
+        currentEnv: config.currentEnv || 'default',
+        envs: Object.fromEntries(Object.entries(config.envs || {}).map(([envName, entry]) => [envName, normalizeEnvConfigEntry(entry) ?? {}])),
+    };
+}
+function getConfigFile(options = {}) {
+    return path.join(resolveCliHomeDir(options.scope), 'config.json');
+}
+function createDefaultConfig() {
+    return {
+        currentEnv: 'default',
+        envs: {},
+    };
+}
+function hasConfiguredEnvs(config) {
+    return Object.keys(config.envs).length > 0;
+}
+function shouldFallbackToLegacyProjectScope(options = {}) {
+    const requestedScope = options.scope ?? resolveDefaultConfigScope();
+    return requestedScope === 'global';
+}
+async function loadExactAuthConfig(options = {}) {
+    try {
+        const content = await fs.readFile(getConfigFile(options), 'utf8');
+        const parsed = JSON.parse(content);
+        return normalizeAuthConfig(parsed);
+    }
+    catch (_error) {
+        return createDefaultConfig();
+    }
+}
+async function resolveEnvStorageScope(envName, options = {}) {
+    const requestedScope = options.scope ?? resolveDefaultConfigScope();
+    if (requestedScope !== 'global') {
+        return { ...options, scope: requestedScope };
+    }
+    const globalConfig = await loadExactAuthConfig({ scope: 'global' });
+    if (globalConfig.envs[envName]) {
+        return { ...options, scope: 'global' };
+    }
+    const projectConfig = await loadExactAuthConfig({ scope: 'project' });
+    if (projectConfig.envs[envName]) {
+        return { ...options, scope: 'project' };
+    }
+    return { ...options, scope: 'global' };
+}
+export async function loadAuthConfig(options = {}) {
+    const config = await loadExactAuthConfig(options);
+    if (!shouldFallbackToLegacyProjectScope(options) || hasConfiguredEnvs(config)) {
+        return config;
+    }
+    const legacyProjectConfig = await loadExactAuthConfig({ scope: 'project' });
+    return hasConfiguredEnvs(legacyProjectConfig) ? legacyProjectConfig : config;
+}
+export async function saveAuthConfig(config, options = {}) {
+    const filePath = getConfigFile(options);
+    await fs.mkdir(path.dirname(filePath), { recursive: true });
+    await fs.writeFile(filePath, JSON.stringify(normalizeAuthConfig(config), null, 2));
+}
+export async function listEnvs(options = {}) {
+    const config = await loadAuthConfig(options);
+    return {
+        currentEnv: config.currentEnv || 'default',
+        envs: config.envs,
+    };
+}
+export async function getCurrentEnvName(options = {}) {
+    const config = await loadAuthConfig(options);
+    return config.currentEnv || 'default';
+}
+export async function setCurrentEnv(envName, options = {}) {
+    const writeOptions = await resolveEnvStorageScope(envName, options);
+    const config = await loadExactAuthConfig(writeOptions);
+    if (!config.envs[envName]) {
+        throw new Error(`Env "${envName}" is not configured`);
+    }
+    config.currentEnv = envName;
+    await saveAuthConfig(config, writeOptions);
+}
+export async function ensureWorkspaceName(defaultName, options = {}) {
+    const config = await loadExactAuthConfig(options);
+    const existing = config.name?.trim();
+    if (existing) {
+        return existing;
+    }
+    const next = defaultName.trim();
+    config.name = next;
+    await saveAuthConfig(config, options);
+    return next;
+}
+export class Env {
+    config;
+    constructor(config = {}) {
+        this.config = config;
+    }
+    get name() {
+        return this.config.name;
+    }
+    get baseUrl() {
+        return readEnvApiBaseUrl(this.config);
+    }
+    get apiBaseUrl() {
+        return readEnvApiBaseUrl(this.config);
+    }
+    get auth() {
+        return this.config.auth;
+    }
+    get runtime() {
+        return this.config.runtime;
+    }
+    get kind() {
+        return resolveEnvKind(this.config);
+    }
+    get appRootPath() {
+        if (this.kind === 'ssh') {
+            const configuredPath = String(this.config.appRootPath ?? '').trim();
+            if (configuredPath) {
+                return configuredPath;
+            }
+        }
+        return resolveConfiguredEnvPath(this.config.appRootPath) ?? resolveEnvRelativePath('.');
+    }
+    get storagePath() {
+        if (this.kind === 'ssh') {
+            const configuredPath = String(this.config.storagePath ?? '').trim();
+            if (configuredPath) {
+                return configuredPath;
+            }
+        }
+        return resolveConfiguredEnvPath(this.config.storagePath) ?? resolveEnvRelativePath('.');
+    }
+    get appPort() {
+        return this.config.appPort;
+    }
+    get envVars() {
+        const out = {
+            STORAGE_PATH: this.storagePath,
+        };
+        const put = (key, value) => {
+            if (value === undefined || value === null) {
+                return;
+            }
+            out[key] = String(value);
+        };
+        put('APP_PORT', this.appPort);
+        put('APP_KEY', this.config.appKey);
+        put('TZ', this.config.timezone);
+        put('DB_DIALECT', this.config.dbDialect);
+        put('DB_HOST', this.config.dbHost);
+        put('DB_PORT', this.config.dbPort);
+        put('DB_DATABASE', this.config.dbDatabase);
+        put('DB_USER', this.config.dbUser);
+        put('DB_PASSWORD', this.config.dbPassword);
+        return out;
+    }
+}
+export async function getEnv(envName, options = {}) {
+    const { config: snapshot, ...loadOptions } = options;
+    const config = snapshot ?? (await loadAuthConfig(loadOptions));
+    const resolved = envName?.trim() || config.currentEnv || 'default';
+    const envConfig = config.envs[resolved];
+    if (!envConfig) {
+        if (!shouldFallbackToLegacyProjectScope(loadOptions)) {
+            return undefined;
+        }
+        const legacyProjectConfig = await loadExactAuthConfig({ scope: 'project' });
+        const legacyResolved = envName?.trim() || legacyProjectConfig.currentEnv || 'default';
+        const legacyEnvConfig = legacyProjectConfig.envs[legacyResolved];
+        if (!legacyEnvConfig) {
+            return undefined;
+        }
+        return new Env({ ...(normalizeEnvConfigEntry(legacyEnvConfig) ?? {}), name: legacyResolved });
+    }
+    return new Env({ ...(normalizeEnvConfigEntry(envConfig) ?? {}), name: resolved });
+}
+function areAuthConfigsEquivalent(left, right) {
+    if (!left && !right) {
+        return true;
+    }
+    if (!left || !right || left.type !== right.type) {
+        return false;
+    }
+    if (left.type === 'token' && right.type === 'token') {
+        return left.accessToken === right.accessToken;
+    }
+    if (left.type === 'oauth' && right.type === 'oauth') {
+        return (left.accessToken === right.accessToken &&
+            left.refreshToken === right.refreshToken &&
+            left.expiresAt === right.expiresAt &&
+            left.scope === right.scope &&
+            left.issuer === right.issuer &&
+            left.clientId === right.clientId &&
+            left.resource === right.resource);
+    }
+    return false;
+}
+async function writeEnv(envName, updater, options = {}) {
+    const writeOptions = await resolveEnvStorageScope(envName, options);
+    const config = await loadExactAuthConfig(writeOptions);
+    const previous = config.envs[envName];
+    config.envs[envName] = updater(previous);
+    config.currentEnv = envName;
+    await saveAuthConfig(config, writeOptions);
+}
+export async function upsertEnv(envName, config, options = {}) {
+    await writeEnv(envName, (previous) => {
+        const { apiBaseUrl: _apiBaseUrl, baseUrl: _baseUrl, apibaseUrl: _legacyApiBaseUrl, accessToken, ...rest } = config;
+        const nextApiBaseUrl = readEnvApiBaseUrl(config);
+        const previousApiBaseUrl = readEnvApiBaseUrl(previous);
+        const baseUrlChanged = previousApiBaseUrl !== nextApiBaseUrl;
+        const nextAuth = accessToken
+            ? {
+                type: 'token',
+                accessToken,
+            }
+            : baseUrlChanged || previous?.auth?.type === 'token'
+                ? undefined
+                : previous?.auth;
+        const authChanged = !areAuthConfigsEquivalent(previous?.auth, nextAuth);
+        return {
+            ...previous,
+            apiBaseUrl: nextApiBaseUrl,
+            auth: nextAuth,
+            ...rest,
+            runtime: baseUrlChanged || authChanged ? undefined : previous?.runtime,
+        };
+    }, options);
+}
+export async function updateEnvConnection(envName, updates, options = {}) {
+    await writeEnv(envName, (previous) => {
+        const nextApiBaseUrl = readEnvApiBaseUrl(updates) ?? readEnvApiBaseUrl(previous);
+        const previousApiBaseUrl = readEnvApiBaseUrl(previous);
+        const baseUrlChanged = previousApiBaseUrl !== nextApiBaseUrl;
+        const nextAuth = updates.accessToken
+            ? {
+                type: 'token',
+                accessToken: updates.accessToken,
+            }
+            : baseUrlChanged || previous?.auth?.type === 'token'
+                ? undefined
+                : previous?.auth;
+        const authChanged = !areAuthConfigsEquivalent(previous?.auth, nextAuth);
+        return {
+            ...previous,
+            ...(nextApiBaseUrl !== undefined ? { apiBaseUrl: nextApiBaseUrl } : {}),
+            auth: nextAuth,
+            runtime: baseUrlChanged || authChanged ? undefined : previous?.runtime,
+        };
+    }, options);
+}
+export async function setEnvOauthSession(envName, auth, options = {}) {
+    await writeEnv(envName, (previous) => ({
+        ...previous,
+        auth,
+        runtime: options.preserveRuntime ? previous?.runtime : undefined,
+    }), options);
+}
+export async function setEnvRuntime(envName, runtime, options = {}) {
+    const writeOptions = await resolveEnvStorageScope(envName, options);
+    const config = await loadExactAuthConfig(writeOptions);
+    const current = config.envs[envName] ?? {};
+    config.envs[envName] = {
+        ...current,
+        runtime,
+    };
+    config.currentEnv = envName;
+    await saveAuthConfig(config, writeOptions);
+}
+export async function removeEnv(envName, options = {}) {
+    const writeOptions = await resolveEnvStorageScope(envName, options);
+    const config = await loadExactAuthConfig(writeOptions);
+    if (!config.envs[envName]) {
+        throw new Error(`Env "${envName}" is not configured`);
+    }
+    delete config.envs[envName];
+    if (config.currentEnv === envName) {
+        const nextEnv = Object.keys(config.envs).sort()[0];
+        config.currentEnv = nextEnv ?? 'default';
+    }
+    await saveAuthConfig(config, writeOptions);
+    return {
+        removed: envName,
+        currentEnv: config.currentEnv || 'default',
+        hasEnvs: Object.keys(config.envs).length > 0,
+    };
+}