@prisma/cli 3.0.0-dev.95.1 → 3.0.0-dev.97.1

package/dist/adapters/token-storage.js

@@ -11,6 +11,7 @@ const EMPTY_AUTH_CONTEXT = {
 	activeWorkspaceId: null,
 	workspaces: {}
 };
+const UNKNOWN_WORKSPACE_NAME = "Unknown workspace";
 function getAuthContextFilePath(authFilePath) {
 	const extension = path.extname(authFilePath);
 	if (!extension) return `${authFilePath}.context.json`;
@@ -140,7 +141,7 @@ var FileTokenStorage = class {
 		return credentials.map((credential) => storedCredentialToTokens(credential)).filter((tokens) => tokens !== null).map((tokens) => {
 			const cached = context.state.workspaces[tokens.workspaceId];
 			const id = typeof cached?.id === "string" && cached.id.trim().length > 0 ? cached.id.trim() : tokens.workspaceId;
-			const name = typeof cached?.name === "string" && cached.name.trim().length > 0 ? cached.name.trim() : id;
+			const name = typeof cached?.name === "string" && cached.name.trim().length > 0 ? workspaceDisplayName(cached.name.trim(), tokens.workspaceId) : UNKNOWN_WORKSPACE_NAME;
 			const lastSeenAt = typeof cached?.lastSeenAt === "string" && cached.lastSeenAt.trim().length > 0 ? cached.lastSeenAt.trim() : null;
 			return {
 				id,
@@ -151,6 +152,10 @@ var FileTokenStorage = class {
 			};
 		});
 	}
+	async listWorkspaceTokens() {
+		this.signal?.throwIfAborted();
+		return (await this.readCredentialsFromDisk()).map((credential) => storedCredentialToTokens(credential)).filter((tokens) => tokens !== null);
+	}
 	async useWorkspace(workspaceRef) {
 		return this.withRefreshLock(() => this.useWorkspaceUnlocked(workspaceRef));
 	}
@@ -406,5 +411,8 @@ function workspaceMatchesRef(workspace, ref) {
 function stripWorkspacePrefix(value) {
 	return value.startsWith("wksp_") ? value.slice(5) : value;
 }
+function workspaceDisplayName(name, credentialWorkspaceId) {
+	return name === credentialWorkspaceId ? UNKNOWN_WORKSPACE_NAME : name;
+}
 //#endregion
 export { FileTokenStorage, WorkspaceSelectionError };

package/dist/controllers/auth.js

@@ -1,4 +1,4 @@
-import { SERVICE_TOKEN_ENV_VAR } from "../lib/auth/client.js";
+import { CLIENT_ID, SERVICE_TOKEN_ENV_VAR, getApiBaseUrl } from "../lib/auth/client.js";
 import { FileTokenStorage, WorkspaceSelectionError } from "../adapters/token-storage.js";
 import { authRequiredError, usageError, workspaceAmbiguousError, workspaceNotAuthenticatedError, workspaceSwitchUnavailableError } from "../shell/errors.js";
 import { canPrompt } from "../shell/runtime.js";
@@ -6,6 +6,7 @@ import { performLogin, performLogout, readAuthState } from "../lib/auth/auth-ops
 import { createAuthUseCases } from "../use-cases/auth.js";
 import { createCliUseCaseGateways } from "../use-cases/create-cli-gateways.js";
 import { createSelectPromptPort } from "./select-prompt-port.js";
+import { createManagementApiSdk } from "@prisma/management-api-sdk";
 //#region src/controllers/auth.ts
 function isRealMode(context) {
 	return !context.runtime.fixturePath && !context.runtime.env.PRISMA_CLI_MOCK_FIXTURE_PATH;
@@ -77,7 +78,8 @@ async function requireAuthenticatedAuthState(context) {
 }
 async function listRealAuthWorkspaces(context) {
 	const rawServiceToken = context.runtime.env[SERVICE_TOKEN_ENV_VAR];
-	const localWorkspaces = await new FileTokenStorage(context.runtime.env, context.runtime.signal).listWorkspaces();
+	const storage = new FileTokenStorage(context.runtime.env, context.runtime.signal);
+	const localWorkspaces = await hydrateLocalAuthWorkspaces(context, storage, await storage.listWorkspaces());
 	if (rawServiceToken !== void 0) {
 		const authState = await readAuthState(context.runtime.env, context.runtime.signal);
 		return {
@@ -117,6 +119,7 @@ async function listRealAuthWorkspaces(context) {
 async function useRealAuthWorkspace(context, workspaceRef) {
 	if (context.runtime.env["PRISMA_SERVICE_TOKEN"] !== void 0) throw workspaceSwitchUnavailableError();
 	const storage = new FileTokenStorage(context.runtime.env, context.runtime.signal);
+	await hydrateLocalAuthWorkspaces(context, storage, await storage.listWorkspaces());
 	try {
 		const result = await storage.useWorkspace(workspaceRef);
 		return {
@@ -137,6 +140,7 @@ async function useRealAuthWorkspace(context, workspaceRef) {
 }
 async function logoutRealAuthWorkspace(context, workspaceRef) {
 	const storage = new FileTokenStorage(context.runtime.env, context.runtime.signal);
+	await hydrateLocalAuthWorkspaces(context, storage, await storage.listWorkspaces());
 	try {
 		const result = await storage.logoutWorkspace(workspaceRef);
 		return {
@@ -171,6 +175,78 @@ async function selectWorkspaceSession(context) {
 		}))
 	})).id;
 }
+async function hydrateLocalAuthWorkspaces(context, storage, workspaces) {
+	const candidates = workspaces.filter(needsWorkspaceMetadataHydration);
+	if (candidates.length === 0) return workspaces;
+	const tokensByCredentialWorkspaceId = new Map((await storage.listWorkspaceTokens()).map((tokens) => [tokens.workspaceId, tokens]));
+	let nextWorkspaces = workspaces;
+	for (const workspace of candidates) {
+		const tokens = tokensByCredentialWorkspaceId.get(workspace.credentialWorkspaceId);
+		if (!tokens) continue;
+		const resolved = await resolveOAuthWorkspaceMetadata(context, tokens);
+		if (!resolved) continue;
+		await rememberResolvedWorkspaceMetadata(context, storage, tokens, resolved);
+		nextWorkspaces = nextWorkspaces.map((candidate) => candidate.credentialWorkspaceId === workspace.credentialWorkspaceId ? {
+			...candidate,
+			id: resolved.id,
+			name: resolved.name,
+			lastSeenAt: (/* @__PURE__ */ new Date()).toISOString()
+		} : candidate);
+	}
+	return nextWorkspaces;
+}
+async function rememberResolvedWorkspaceMetadata(context, storage, tokens, resolved) {
+	try {
+		await storage.rememberWorkspace(tokens.workspaceId, resolved);
+	} catch {
+		context.runtime.signal?.throwIfAborted();
+	}
+}
+function needsWorkspaceMetadataHydration(workspace) {
+	return workspace.id === workspace.credentialWorkspaceId || workspace.name === "Unknown workspace" || workspace.name === workspace.credentialWorkspaceId;
+}
+async function resolveOAuthWorkspaceMetadata(context, tokens) {
+	const sdk = createManagementApiSdk({
+		clientId: CLIENT_ID,
+		redirectUri: "http://localhost:0/auth/callback",
+		tokenStorage: createSingleWorkspaceTokenStorage(new FileTokenStorage(context.runtime.env, context.runtime.signal, { activateOnSetTokens: false }), tokens),
+		apiBaseUrl: getApiBaseUrl(context.runtime.env)
+	});
+	try {
+		const { data } = await sdk.client.GET("/v1/workspaces/{id}", {
+			params: { path: { id: tokens.workspaceId } },
+			signal: context.runtime.signal
+		});
+		const id = stringOrNull(data?.data?.id) ?? tokens.workspaceId;
+		const name = stringOrNull(data?.data?.name) ?? id;
+		if (id === tokens.workspaceId && name === tokens.workspaceId) return null;
+		return {
+			id,
+			name
+		};
+	} catch {
+		context.runtime.signal?.throwIfAborted();
+		return null;
+	}
+}
+function createSingleWorkspaceTokenStorage(storage, initialTokens) {
+	let currentTokens = initialTokens;
+	return {
+		getTokens: async () => currentTokens,
+		setTokens: async (tokens) => {
+			currentTokens = tokens;
+			await storage.setTokens(tokens);
+		},
+		clearTokens: async () => {
+			const tokens = currentTokens;
+			currentTokens = null;
+			if (tokens) await storage.clearTokensIfCurrent(tokens);
+		}
+	};
+}
+function stringOrNull(value) {
+	return typeof value === "string" && value.trim().length > 0 ? value.trim() : null;
+}
 function toAuthWorkspace(workspace) {
 	return {
 		id: workspace.id,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@prisma/cli",
-  "version": "3.0.0-dev.95.1",
+  "version": "3.0.0-dev.97.1",
   "description": "Command-line interface for the Prisma Developer Platform.",
   "type": "module",
   "bin": {