repoburg 1.0.68 → 1.1.0

package/backend/dist/packages/gemini-core/src/code_assist/oauth2.js

@@ -0,0 +1,371 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getOauthClient = getOauthClient;
+exports.getAvailablePort = getAvailablePort;
+exports.clearOauthClientCache = clearOauthClientCache;
+exports.clearCachedCredentialFile = clearCachedCredentialFile;
+exports.resetOauthClientForTesting = resetOauthClientForTesting;
+const google_auth_library_1 = require("google-auth-library");
+const http = require("node:http");
+const node_url_1 = require("node:url");
+const crypto = require("node:crypto");
+const net = require("node:net");
+const path = require("node:path");
+const node_fs_1 = require("node:fs");
+const errors_1 = require("../utils/errors");
+const userAccountManager_1 = require("../utils/userAccountManager");
+const contentGenerator_1 = require("../core/contentGenerator");
+const node_readline_1 = require("node:readline");
+const storage_1 = require("../config/storage");
+const oauth_credential_storage_1 = require("./oauth-credential-storage");
+const index_1 = require("../mcp/token-storage/index");
+const userAccountManager = new userAccountManager_1.UserAccountManager();
+const OAUTH_CLIENT_ID = '681255809395-oo8ft2oprdrnp9e3aqf6av3hmdib135j.apps.googleusercontent.com';
+const OAUTH_CLIENT_SECRET = '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';
+const OAUTH_SCOPE = [
+    'https://www.googleapis.com/auth/cloud-platform',
+    'https://www.googleapis.com/auth/userinfo.email',
+    'https://www.googleapis.com/auth/userinfo.profile',
+];
+const HTTP_REDIRECT = 301;
+const SIGN_IN_SUCCESS_URL = 'https://developers.google.com/gemini-code-assist/auth_success_gemini';
+const SIGN_IN_FAILURE_URL = 'https://developers.google.com/gemini-code-assist/auth_failure_gemini';
+const oauthClientPromises = new Map();
+function getUseEncryptedStorageFlag() {
+    return process.env[index_1.FORCE_ENCRYPTED_FILE_ENV_VAR] === 'true';
+}
+async function initOauthClient(authType, config) {
+    const client = new google_auth_library_1.OAuth2Client({
+        clientId: OAUTH_CLIENT_ID,
+        clientSecret: OAUTH_CLIENT_SECRET,
+    });
+    const useEncryptedStorage = getUseEncryptedStorageFlag();
+    if (process.env['GOOGLE_GENAI_USE_GCA'] &&
+        process.env['GOOGLE_CLOUD_ACCESS_TOKEN']) {
+        client.setCredentials({
+            access_token: process.env['GOOGLE_CLOUD_ACCESS_TOKEN'],
+        });
+        await fetchAndCacheUserInfo(client);
+        return client;
+    }
+    client.on('tokens', async (tokens) => {
+        if (useEncryptedStorage) {
+            await oauth_credential_storage_1.OAuthCredentialStorage.saveCredentials(tokens);
+        }
+        else {
+            await cacheCredentials(tokens);
+        }
+    });
+    if (await loadCachedCredentials(client)) {
+        if (!userAccountManager.getCachedGoogleAccount()) {
+            try {
+                await fetchAndCacheUserInfo(client);
+            }
+            catch (error) {
+                console.warn('Failed to fetch user info:', (0, errors_1.getErrorMessage)(error));
+            }
+        }
+        console.log('Loaded cached credentials.');
+        return client;
+    }
+    if (authType === contentGenerator_1.AuthType.CLOUD_SHELL) {
+        try {
+            console.log("Attempting to authenticate via Cloud Shell VM's ADC.");
+            const computeClient = new google_auth_library_1.Compute({});
+            await computeClient.getAccessToken();
+            console.log('Authentication successful.');
+            return computeClient;
+        }
+        catch (e) {
+            throw new Error(`Could not authenticate using Cloud Shell credentials. Please select a different authentication method or ensure you are in a properly configured environment. Error: ${(0, errors_1.getErrorMessage)(e)}`);
+        }
+    }
+    if (config.isBrowserLaunchSuppressed()) {
+        let success = false;
+        const maxRetries = 2;
+        for (let i = 0; !success && i < maxRetries; i++) {
+            success = await authWithUserCode(client);
+            if (!success) {
+                console.error('\nFailed to authenticate with user code.', i === maxRetries - 1 ? '' : 'Retrying...\n');
+            }
+        }
+        if (!success) {
+            throw new errors_1.FatalAuthenticationError('Failed to authenticate with user code.');
+        }
+    }
+    else {
+        const webLogin = await authWithWeb(client);
+        console.log(`\n\nCode Assist login required.\n` +
+            `Attempting to open authentication page in your browser.\n` +
+            `Otherwise navigate to:\n\n${webLogin.authUrl}\n\n`);
+        try {
+            const { default: open } = await Promise.resolve().then(() => require('open'));
+            const childProcess = await open(webLogin.authUrl);
+            childProcess.on('error', (error) => {
+                console.error('Failed to open browser automatically. Please try running again with NO_BROWSER=true set.');
+                console.error('Browser error details:', (0, errors_1.getErrorMessage)(error));
+            });
+        }
+        catch (err) {
+            console.error('An unexpected error occurred while trying to open the browser:', (0, errors_1.getErrorMessage)(err), '\nThis might be due to browser compatibility issues or system configuration.', '\nPlease try running again with NO_BROWSER=true set for manual authentication.');
+            throw new errors_1.FatalAuthenticationError(`Failed to open browser: ${(0, errors_1.getErrorMessage)(err)}`);
+        }
+        console.log('Waiting for authentication...');
+        const authTimeout = 5 * 60 * 1000;
+        const timeoutPromise = new Promise((_, reject) => {
+            setTimeout(() => {
+                reject(new errors_1.FatalAuthenticationError('Authentication timed out after 5 minutes. The browser tab may have gotten stuck in a loading state. ' +
+                    'Please try again or use NO_BROWSER=true for manual authentication.'));
+            }, authTimeout);
+        });
+        await Promise.race([webLogin.loginCompletePromise, timeoutPromise]);
+    }
+    return client;
+}
+async function getOauthClient(authType, config) {
+    if (!oauthClientPromises.has(authType)) {
+        oauthClientPromises.set(authType, initOauthClient(authType, config));
+    }
+    return oauthClientPromises.get(authType);
+}
+async function authWithUserCode(client) {
+    const redirectUri = 'https://codeassist.google.com/authcode';
+    const codeVerifier = await client.generateCodeVerifierAsync();
+    const state = crypto.randomBytes(32).toString('hex');
+    const authUrl = client.generateAuthUrl({
+        redirect_uri: redirectUri,
+        access_type: 'offline',
+        scope: OAUTH_SCOPE,
+        code_challenge_method: google_auth_library_1.CodeChallengeMethod.S256,
+        code_challenge: codeVerifier.codeChallenge,
+        state,
+    });
+    console.log('Please visit the following URL to authorize the application:');
+    console.log('');
+    console.log(authUrl);
+    console.log('');
+    const code = await new Promise((resolve) => {
+        const rl = node_readline_1.default.createInterface({
+            input: process.stdin,
+            output: process.stdout,
+        });
+        rl.question('Enter the authorization code: ', (code) => {
+            rl.close();
+            resolve(code.trim());
+        });
+    });
+    if (!code) {
+        console.error('Authorization code is required.');
+        return false;
+    }
+    try {
+        const { tokens } = await client.getToken({
+            code,
+            codeVerifier: codeVerifier.codeVerifier,
+            redirect_uri: redirectUri,
+        });
+        client.setCredentials(tokens);
+    }
+    catch (error) {
+        console.error('Failed to authenticate with authorization code:', (0, errors_1.getErrorMessage)(error));
+        return false;
+    }
+    return true;
+}
+async function authWithWeb(client) {
+    const port = await getAvailablePort();
+    const host = process.env['OAUTH_CALLBACK_HOST'] || 'localhost';
+    const redirectUri = `http://localhost:${port}/oauth2callback`;
+    const state = crypto.randomBytes(32).toString('hex');
+    const authUrl = client.generateAuthUrl({
+        redirect_uri: redirectUri,
+        access_type: 'offline',
+        scope: OAUTH_SCOPE,
+        state,
+    });
+    const loginCompletePromise = new Promise((resolve, reject) => {
+        const server = http.createServer(async (req, res) => {
+            try {
+                if (req.url.indexOf('/oauth2callback') === -1) {
+                    res.writeHead(HTTP_REDIRECT, { Location: SIGN_IN_FAILURE_URL });
+                    res.end();
+                    reject(new errors_1.FatalAuthenticationError('OAuth callback not received. Unexpected request: ' + req.url));
+                }
+                const qs = new node_url_1.default.URL(req.url, 'http://localhost:3000').searchParams;
+                if (qs.get('error')) {
+                    res.writeHead(HTTP_REDIRECT, { Location: SIGN_IN_FAILURE_URL });
+                    res.end();
+                    const errorCode = qs.get('error');
+                    const errorDescription = qs.get('error_description') || 'No additional details provided';
+                    reject(new errors_1.FatalAuthenticationError(`Google OAuth error: ${errorCode}. ${errorDescription}`));
+                }
+                else if (qs.get('state') !== state) {
+                    res.end('State mismatch. Possible CSRF attack');
+                    reject(new errors_1.FatalAuthenticationError('OAuth state mismatch. Possible CSRF attack or browser session issue.'));
+                }
+                else if (qs.get('code')) {
+                    try {
+                        const { tokens } = await client.getToken({
+                            code: qs.get('code'),
+                            redirect_uri: redirectUri,
+                        });
+                        client.setCredentials(tokens);
+                        try {
+                            await fetchAndCacheUserInfo(client);
+                        }
+                        catch (error) {
+                            console.warn('Failed to retrieve Google Account ID during authentication:', (0, errors_1.getErrorMessage)(error));
+                        }
+                        res.writeHead(HTTP_REDIRECT, { Location: SIGN_IN_SUCCESS_URL });
+                        res.end();
+                        resolve();
+                    }
+                    catch (error) {
+                        res.writeHead(HTTP_REDIRECT, { Location: SIGN_IN_FAILURE_URL });
+                        res.end();
+                        reject(new errors_1.FatalAuthenticationError(`Failed to exchange authorization code for tokens: ${(0, errors_1.getErrorMessage)(error)}`));
+                    }
+                }
+                else {
+                    reject(new errors_1.FatalAuthenticationError('No authorization code received from Google OAuth. Please try authenticating again.'));
+                }
+            }
+            catch (e) {
+                if (e instanceof errors_1.FatalAuthenticationError) {
+                    reject(e);
+                }
+                else {
+                    reject(new errors_1.FatalAuthenticationError(`Unexpected error during OAuth authentication: ${(0, errors_1.getErrorMessage)(e)}`));
+                }
+            }
+            finally {
+                server.close();
+            }
+        });
+        server.listen(port, host, () => {
+        });
+        server.on('error', (err) => {
+            reject(new errors_1.FatalAuthenticationError(`OAuth callback server error: ${(0, errors_1.getErrorMessage)(err)}`));
+        });
+    });
+    return {
+        authUrl,
+        loginCompletePromise,
+    };
+}
+function getAvailablePort() {
+    return new Promise((resolve, reject) => {
+        let port = 0;
+        try {
+            const portStr = process.env['OAUTH_CALLBACK_PORT'];
+            if (portStr) {
+                port = parseInt(portStr, 10);
+                if (isNaN(port) || port <= 0 || port > 65535) {
+                    return reject(new Error(`Invalid value for OAUTH_CALLBACK_PORT: "${portStr}"`));
+                }
+                return resolve(port);
+            }
+            const server = net.createServer();
+            server.listen(0, () => {
+                const address = server.address();
+                port = address.port;
+            });
+            server.on('listening', () => {
+                server.close();
+                server.unref();
+            });
+            server.on('error', (e) => reject(e));
+            server.on('close', () => resolve(port));
+        }
+        catch (e) {
+            reject(e);
+        }
+    });
+}
+async function loadCachedCredentials(client) {
+    const useEncryptedStorage = getUseEncryptedStorageFlag();
+    if (useEncryptedStorage) {
+        const credentials = await oauth_credential_storage_1.OAuthCredentialStorage.loadCredentials();
+        if (credentials) {
+            client.setCredentials(credentials);
+            return true;
+        }
+        return false;
+    }
+    const pathsToTry = [
+        storage_1.Storage.getOAuthCredsPath(),
+        process.env['GOOGLE_APPLICATION_CREDENTIALS'],
+    ].filter((p) => !!p);
+    for (const keyFile of pathsToTry) {
+        try {
+            const creds = await node_fs_1.promises.readFile(keyFile, 'utf-8');
+            client.setCredentials(JSON.parse(creds));
+            const { token } = await client.getAccessToken();
+            if (!token) {
+                continue;
+            }
+            await client.getTokenInfo(token);
+            return true;
+        }
+        catch (error) {
+            console.debug(`Failed to load credentials from ${keyFile}:`, (0, errors_1.getErrorMessage)(error));
+        }
+    }
+    return false;
+}
+async function cacheCredentials(credentials) {
+    const filePath = storage_1.Storage.getOAuthCredsPath();
+    await node_fs_1.promises.mkdir(path.dirname(filePath), { recursive: true });
+    const credString = JSON.stringify(credentials, null, 2);
+    await node_fs_1.promises.writeFile(filePath, credString, { mode: 0o600 });
+    try {
+        await node_fs_1.promises.chmod(filePath, 0o600);
+    }
+    catch {
+    }
+}
+function clearOauthClientCache() {
+    oauthClientPromises.clear();
+}
+async function clearCachedCredentialFile() {
+    try {
+        const useEncryptedStorage = getUseEncryptedStorageFlag();
+        if (useEncryptedStorage) {
+            await oauth_credential_storage_1.OAuthCredentialStorage.clearCredentials();
+        }
+        else {
+            await node_fs_1.promises.rm(storage_1.Storage.getOAuthCredsPath(), { force: true });
+        }
+        await userAccountManager.clearCachedGoogleAccount();
+        clearOauthClientCache();
+    }
+    catch (e) {
+        console.error('Failed to clear cached credentials:', e);
+    }
+}
+async function fetchAndCacheUserInfo(client) {
+    try {
+        const { token } = await client.getAccessToken();
+        if (!token) {
+            return;
+        }
+        const response = await fetch('https://www.googleapis.com/oauth2/v2/userinfo', {
+            headers: {
+                Authorization: `Bearer ${token}`,
+            },
+        });
+        if (!response.ok) {
+            console.error('Failed to fetch user info:', response.status, response.statusText);
+            return;
+        }
+        const userInfo = await response.json();
+        await userAccountManager.cacheGoogleAccount(userInfo.email);
+    }
+    catch (error) {
+        console.error('Error retrieving user info:', error);
+    }
+}
+function resetOauthClientForTesting() {
+    oauthClientPromises.clear();
+}
+//# sourceMappingURL=oauth2.js.map

package/backend/dist/packages/gemini-core/src/code_assist/oauth2.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"oauth2.js","sourceRoot":"","sources":["../../../../../packages/gemini-core/src/code_assist/oauth2.ts"],"names":[],"mappings":";;AAqNA,wCAQC;AAgLD,4CA6BC;AA0DD,sDAEC;AAED,8DAeC;AAmCD,gEAEC;AArhBD,6DAI6B;AAC7B,kCAAkC;AAClC,uCAA2B;AAC3B,sCAAsC;AACtC,gCAAgC;AAChC,kCAAkC;AAClC,qCAAyC;AAEzC,4CAA4E;AAC5E,oEAAiE;AACjE,+DAAoD;AACpD,iDAAqC;AACrC,+CAA4C;AAC5C,yEAAoE;AACpE,sDAA0E;AAE1E,MAAM,kBAAkB,GAAG,IAAI,uCAAkB,EAAE,CAAC;AAGpD,MAAM,eAAe,GACnB,0EAA0E,CAAC;AAQ7E,MAAM,mBAAmB,GAAG,qCAAqC,CAAC;AAGlE,MAAM,WAAW,GAAG;IAClB,gDAAgD;IAChD,gDAAgD;IAChD,kDAAkD;CACnD,CAAC;AAEF,MAAM,aAAa,GAAG,GAAG,CAAC;AAC1B,MAAM,mBAAmB,GACvB,sEAAsE,CAAC;AACzE,MAAM,mBAAmB,GACvB,sEAAsE,CAAC;AAYzE,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAmC,CAAC;AAEvE,SAAS,0BAA0B;IACjC,OAAO,OAAO,CAAC,GAAG,CAAC,oCAA4B,CAAC,KAAK,MAAM,CAAC;AAC9D,CAAC;AAED,KAAK,UAAU,eAAe,CAC5B,QAAkB,EAClB,MAAc;IAEd,MAAM,MAAM,GAAG,IAAI,kCAAY,CAAC;QAC9B,QAAQ,EAAE,eAAe;QACzB,YAAY,EAAE,mBAAmB;KAClC,CAAC,CAAC;IACH,MAAM,mBAAmB,GAAG,0BAA0B,EAAE,CAAC;IAEzD,IACE,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,EACxC,CAAC;QACD,MAAM,CAAC,cAAc,CAAC;YACpB,YAAY,EAAE,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC;SACvD,CAAC,CAAC;QACH,MAAM,qBAAqB,CAAC,MAAM,CAAC,CAAC;QACpC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,MAAM,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAmB,EAAE,EAAE;QAChD,IAAI,mBAAmB,EAAE,CAAC;YACxB,MAAM,iDAAsB,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACvD,CAAC;aAAM,CAAC;YACN,MAAM,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACjC,CAAC;IACH,CAAC,CAAC,CAAC;IAGH,IAAI,MAAM,qBAAqB,CAAC,MAAM,CAAC,EAAE,CAAC;QAGxC,IAAI,CAAC,kBAAkB,CAAC,sBAAsB,EAAE,EAAE,CAAC;YACjD,IAAI,CAAC;gBACH,MAAM,qBAAqB,CAAC,MAAM,CAAC,CAAC;YACtC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAEf,OAAO,CAAC,IAAI,CAAC,4BAA4B,EAAE,IAAA,wBAAe,EAAC,KAAK,CAAC,CAAC,CAAC;YACrE,CAAC;QACH,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAC1C,OAAO,MAAM,CAAC;IAChB,CAAC;IAKD,IAAI,QAAQ,KAAK,2BAAQ,CAAC,WAAW,EAAE,CAAC;QACtC,IAAI,CAAC;YACH,OAAO,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAC;YACpE,MAAM,aAAa,GAAG,IAAI,6BAAO,CAAC,EAGjC,CAAC,CAAC;YACH,MAAM,aAAa,CAAC,cAAc,EAAE,CAAC;YACrC,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;YAG1C,OAAO,aAAa,CAAC;QACvB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CACb,wKAAwK,IAAA,wBAAe,EACrL,CAAC,CACF,EAAE,CACJ,CAAC;QACJ,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,yBAAyB,EAAE,EAAE,CAAC;QACvC,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,MAAM,UAAU,GAAG,CAAC,CAAC;QACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,OAAO,IAAI,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YAChD,OAAO,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,CAAC;YACzC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CACX,0CAA0C,EAC1C,CAAC,KAAK,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,eAAe,CAC5C,CAAC;YACJ,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,iCAAwB,CAChC,wCAAwC,CACzC,CAAC;QACJ,CAAC;IACH,CAAC;SAAM,CAAC;QACN,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,CAAC;QAE3C,OAAO,CAAC,GAAG,CACT,mCAAmC;YACjC,2DAA2D;YAC3D,6BAA6B,QAAQ,CAAC,OAAO,MAAM,CACtD,CAAC;QACF,IAAI,CAAC;YACH,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,2CAAa,MAAM,EAAC,CAAC;YAI/C,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAMlD,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACjC,OAAO,CAAC,KAAK,CACX,0FAA0F,CAC3F,CAAC;gBACF,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,IAAA,wBAAe,EAAC,KAAK,CAAC,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CACX,gEAAgE,EAChE,IAAA,wBAAe,EAAC,GAAG,CAAC,EACpB,8EAA8E,EAC9E,gFAAgF,CACjF,CAAC;YACF,MAAM,IAAI,iCAAwB,CAChC,2BAA2B,IAAA,wBAAe,EAAC,GAAG,CAAC,EAAE,CAClD,CAAC;QACJ,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;QAG7C,MAAM,WAAW,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QAClC,MAAM,cAAc,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;YACtD,UAAU,CAAC,GAAG,EAAE;gBACd,MAAM,CACJ,IAAI,iCAAwB,CAC1B,sGAAsG;oBACpG,oEAAoE,CACvE,CACF,CAAC;YACJ,CAAC,EAAE,WAAW,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,oBAAoB,EAAE,cAAc,CAAC,CAAC,CAAC;IACtE,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAEM,KAAK,UAAU,cAAc,CAClC,QAAkB,EAClB,MAAc;IAEd,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QACvC,mBAAmB,CAAC,GAAG,CAAC,QAAQ,EAAE,eAAe,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;IACvE,CAAC;IACD,OAAO,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC;AAC5C,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,MAAoB;IAClD,MAAM,WAAW,GAAG,wCAAwC,CAAC;IAC7D,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,yBAAyB,EAAE,CAAC;IAC9D,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACrD,MAAM,OAAO,GAAW,MAAM,CAAC,eAAe,CAAC;QAC7C,YAAY,EAAE,WAAW;QACzB,WAAW,EAAE,SAAS;QACtB,KAAK,EAAE,WAAW;QAClB,qBAAqB,EAAE,yCAAmB,CAAC,IAAI;QAC/C,cAAc,EAAE,YAAY,CAAC,aAAa;QAC1C,KAAK;KACN,CAAC,CAAC;IACH,OAAO,CAAC,GAAG,CAAC,8DAA8D,CAAC,CAAC;IAC5E,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAChB,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACrB,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAEhB,MAAM,IAAI,GAAG,MAAM,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,EAAE;QACjD,MAAM,EAAE,GAAG,uBAAQ,CAAC,eAAe,CAAC;YAClC,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;QACH,EAAE,CAAC,QAAQ,CAAC,gCAAgC,EAAE,CAAC,IAAI,EAAE,EAAE;YACrD,EAAE,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACjD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,CAAC;QACH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;YACvC,IAAI;YACJ,YAAY,EAAE,YAAY,CAAC,YAAY;YACvC,YAAY,EAAE,WAAW;SAC1B,CAAC,CAAC;QACH,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CACX,iDAAiD,EACjD,IAAA,wBAAe,EAAC,KAAK,CAAC,CACvB,CAAC;QACF,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,MAAoB;IAC7C,MAAM,IAAI,GAAG,MAAM,gBAAgB,EAAE,CAAC;IAEtC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,IAAI,WAAW,CAAC;IAK/D,MAAM,WAAW,GAAG,oBAAoB,IAAI,iBAAiB,CAAC;IAC9D,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACrD,MAAM,OAAO,GAAG,MAAM,CAAC,eAAe,CAAC;QACrC,YAAY,EAAE,WAAW;QACzB,WAAW,EAAE,SAAS;QACtB,KAAK,EAAE,WAAW;QAClB,KAAK;KACN,CAAC,CAAC;IAEH,MAAM,oBAAoB,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACjE,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;YAClD,IAAI,CAAC;gBACH,IAAI,GAAG,CAAC,GAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;oBAC/C,GAAG,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE,CAAC,CAAC;oBAChE,GAAG,CAAC,GAAG,EAAE,CAAC;oBACV,MAAM,CACJ,IAAI,iCAAwB,CAC1B,mDAAmD,GAAG,GAAG,CAAC,GAAG,CAC9D,CACF,CAAC;gBACJ,CAAC;gBAED,MAAM,EAAE,GAAG,IAAI,kBAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAI,EAAE,uBAAuB,CAAC,CAAC,YAAY,CAAC;gBACvE,IAAI,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBACpB,GAAG,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE,CAAC,CAAC;oBAChE,GAAG,CAAC,GAAG,EAAE,CAAC;oBAEV,MAAM,SAAS,GAAG,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBAClC,MAAM,gBAAgB,GACpB,EAAE,CAAC,GAAG,CAAC,mBAAmB,CAAC,IAAI,gCAAgC,CAAC;oBAClE,MAAM,CACJ,IAAI,iCAAwB,CAC1B,uBAAuB,SAAS,KAAK,gBAAgB,EAAE,CACxD,CACF,CAAC;gBACJ,CAAC;qBAAM,IAAI,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,KAAK,EAAE,CAAC;oBACrC,GAAG,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;oBAEhD,MAAM,CACJ,IAAI,iCAAwB,CAC1B,sEAAsE,CACvE,CACF,CAAC;gBACJ,CAAC;qBAAM,IAAI,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC1B,IAAI,CAAC;wBACH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;4BACvC,IAAI,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAE;4BACrB,YAAY,EAAE,WAAW;yBAC1B,CAAC,CAAC;wBACH,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;wBAG9B,IAAI,CAAC;4BACH,MAAM,qBAAqB,CAAC,MAAM,CAAC,CAAC;wBACtC,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,OAAO,CAAC,IAAI,CACV,6DAA6D,EAC7D,IAAA,wBAAe,EAAC,KAAK,CAAC,CACvB,CAAC;wBAEJ,CAAC;wBAED,GAAG,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE,CAAC,CAAC;wBAChE,GAAG,CAAC,GAAG,EAAE,CAAC;wBACV,OAAO,EAAE,CAAC;oBACZ,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,GAAG,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE,CAAC,CAAC;wBAChE,GAAG,CAAC,GAAG,EAAE,CAAC;wBACV,MAAM,CACJ,IAAI,iCAAwB,CAC1B,qDAAqD,IAAA,wBAAe,EAAC,KAAK,CAAC,EAAE,CAC9E,CACF,CAAC;oBACJ,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,MAAM,CACJ,IAAI,iCAAwB,CAC1B,oFAAoF,CACrF,CACF,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBAEX,IAAI,CAAC,YAAY,iCAAwB,EAAE,CAAC;oBAC1C,MAAM,CAAC,CAAC,CAAC,CAAC;gBACZ,CAAC;qBAAM,CAAC;oBACN,MAAM,CACJ,IAAI,iCAAwB,CAC1B,iDAAiD,IAAA,wBAAe,EAAC,CAAC,CAAC,EAAE,CACtE,CACF,CAAC;gBACJ,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;QAE/B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACzB,MAAM,CACJ,IAAI,iCAAwB,CAC1B,gCAAgC,IAAA,wBAAe,EAAC,GAAG,CAAC,EAAE,CACvD,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,OAAO;QACP,oBAAoB;KACrB,CAAC;AACJ,CAAC;AAED,SAAgB,gBAAgB;IAC9B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;YACnD,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,GAAG,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBAC7B,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,GAAG,KAAK,EAAE,CAAC;oBAC7C,OAAO,MAAM,CACX,IAAI,KAAK,CAAC,2CAA2C,OAAO,GAAG,CAAC,CACjE,CAAC;gBACJ,CAAC;gBACD,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;YACvB,CAAC;YACD,MAAM,MAAM,GAAG,GAAG,CAAC,YAAY,EAAE,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,EAAE;gBACpB,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAsB,CAAC;gBACrD,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;YACtB,CAAC,CAAC,CAAC;YACH,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE;gBAC1B,MAAM,CAAC,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,CAAC,CAAC,CAAC;YACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,CAAC,CAAC,CAAC,CAAC;QACZ,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,qBAAqB,CAAC,MAAoB;IACvD,MAAM,mBAAmB,GAAG,0BAA0B,EAAE,CAAC;IACzD,IAAI,mBAAmB,EAAE,CAAC;QACxB,MAAM,WAAW,GAAG,MAAM,iDAAsB,CAAC,eAAe,EAAE,CAAC;QACnE,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;YACnC,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,UAAU,GAAG;QACjB,iBAAO,CAAC,iBAAiB,EAAE;QAC3B,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC;KAC9C,CAAC,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAElC,KAAK,MAAM,OAAO,IAAI,UAAU,EAAE,CAAC;QACjC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,kBAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAClD,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;YAGzC,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,cAAc,EAAE,CAAC;YAChD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,SAAS;YACX,CAAC;YAGD,MAAM,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;YAEjC,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEf,OAAO,CAAC,KAAK,CACX,mCAAmC,OAAO,GAAG,EAC7C,IAAA,wBAAe,EAAC,KAAK,CAAC,CACvB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,WAAwB;IACtD,MAAM,QAAQ,GAAG,iBAAO,CAAC,iBAAiB,EAAE,CAAC;IAC7C,MAAM,kBAAE,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAE5D,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACxD,MAAM,kBAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1D,IAAI,CAAC;QACH,MAAM,kBAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAClC,CAAC;IAAC,MAAM,CAAC;IAET,CAAC;AACH,CAAC;AAED,SAAgB,qBAAqB;IACnC,mBAAmB,CAAC,KAAK,EAAE,CAAC;AAC9B,CAAC;AAEM,KAAK,UAAU,yBAAyB;IAC7C,IAAI,CAAC;QACH,MAAM,mBAAmB,GAAG,0BAA0B,EAAE,CAAC;QACzD,IAAI,mBAAmB,EAAE,CAAC;YACxB,MAAM,iDAAsB,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,MAAM,kBAAE,CAAC,EAAE,CAAC,iBAAO,CAAC,iBAAiB,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,kBAAkB,CAAC,wBAAwB,EAAE,CAAC;QAEpD,qBAAqB,EAAE,CAAC;IAC1B,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,CAAC,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC;AAED,KAAK,UAAU,qBAAqB,CAAC,MAAoB;IACvD,IAAI,CAAC;QACH,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,cAAc,EAAE,CAAC;QAChD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,+CAA+C,EAC/C;YACE,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,KAAK,EAAE;aACjC;SACF,CACF,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,OAAO,CAAC,KAAK,CACX,4BAA4B,EAC5B,QAAQ,CAAC,MAAM,EACf,QAAQ,CAAC,UAAU,CACpB,CAAC;YACF,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACvC,MAAM,kBAAkB,CAAC,kBAAkB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC9D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;IACtD,CAAC;AACH,CAAC;AAGD,SAAgB,0BAA0B;IACxC,mBAAmB,CAAC,KAAK,EAAE,CAAC;AAC9B,CAAC"}

package/backend/dist/packages/gemini-core/src/code_assist/server.d.ts

@@ -0,0 +1,28 @@
+import type { OAuth2Client } from 'google-auth-library';
+import type { LoadCodeAssistRequest, LoadCodeAssistResponse, LongRunningOperationResponse, OnboardUserRequest } from './types';
+import type { CountTokensParameters, CountTokensResponse, EmbedContentParameters, EmbedContentResponse, GenerateContentParameters, GenerateContentResponse } from '@google/genai';
+import type { ContentGenerator } from '../core/contentGenerator';
+import { UserTierId } from './types';
+export interface HttpOptions {
+    headers?: Record<string, string>;
+}
+export declare const CODE_ASSIST_ENDPOINT = "https://cloudcode-pa.googleapis.com";
+export declare const CODE_ASSIST_API_VERSION = "v1internal";
+export declare class CodeAssistServer implements ContentGenerator {
+    readonly client: OAuth2Client;
+    readonly projectId?: string;
+    readonly httpOptions: HttpOptions;
+    readonly sessionId?: string;
+    readonly userTier?: UserTierId;
+    constructor(client: OAuth2Client, projectId?: string, httpOptions?: HttpOptions, sessionId?: string, userTier?: UserTierId);
+    generateContentStream(req: GenerateContentParameters, userPromptId: string): Promise<AsyncGenerator<GenerateContentResponse>>;
+    generateContent(req: GenerateContentParameters, userPromptId: string): Promise<GenerateContentResponse>;
+    onboardUser(req: OnboardUserRequest): Promise<LongRunningOperationResponse>;
+    loadCodeAssist(req: LoadCodeAssistRequest): Promise<LoadCodeAssistResponse>;
+    countTokens(req: CountTokensParameters): Promise<CountTokensResponse>;
+    embedContent(_req: EmbedContentParameters): Promise<EmbedContentResponse>;
+    requestPost<T>(method: string, req: object, signal?: AbortSignal): Promise<T>;
+    requestGet<T>(method: string, signal?: AbortSignal): Promise<T>;
+    requestStreamingPost<T>(method: string, req: object, signal?: AbortSignal): Promise<AsyncGenerator<T>>;
+    getMethodUrl(method: string): string;
+}

package/backend/dist/packages/gemini-core/src/code_assist/server.js

@@ -0,0 +1,135 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CodeAssistServer = exports.CODE_ASSIST_API_VERSION = exports.CODE_ASSIST_ENDPOINT = void 0;
+const readline = require("node:readline");
+const types_1 = require("./types");
+const converter_1 = require("./converter");
+exports.CODE_ASSIST_ENDPOINT = 'https://cloudcode-pa.googleapis.com';
+exports.CODE_ASSIST_API_VERSION = 'v1internal';
+class CodeAssistServer {
+    constructor(client, projectId, httpOptions = {}, sessionId, userTier) {
+        this.client = client;
+        this.projectId = projectId;
+        this.httpOptions = httpOptions;
+        this.sessionId = sessionId;
+        this.userTier = userTier;
+    }
+    async generateContentStream(req, userPromptId) {
+        const resps = await this.requestStreamingPost('streamGenerateContent', (0, converter_1.toGenerateContentRequest)(req, userPromptId, this.projectId, this.sessionId), req.config?.abortSignal);
+        return (async function* () {
+            for await (const resp of resps) {
+                yield (0, converter_1.fromGenerateContentResponse)(resp);
+            }
+        })();
+    }
+    async generateContent(req, userPromptId) {
+        const resp = await this.requestPost('generateContent', (0, converter_1.toGenerateContentRequest)(req, userPromptId, this.projectId, this.sessionId), req.config?.abortSignal);
+        return (0, converter_1.fromGenerateContentResponse)(resp);
+    }
+    async onboardUser(req) {
+        return await this.requestPost('onboardUser', req);
+    }
+    async loadCodeAssist(req) {
+        try {
+            return await this.requestPost('loadCodeAssist', req);
+        }
+        catch (e) {
+            if (isVpcScAffectedUser(e)) {
+                return {
+                    currentTier: { id: types_1.UserTierId.STANDARD },
+                };
+            }
+            else {
+                throw e;
+            }
+        }
+    }
+    async countTokens(req) {
+        const resp = await this.requestPost('countTokens', (0, converter_1.toCountTokenRequest)(req));
+        return (0, converter_1.fromCountTokenResponse)(resp);
+    }
+    async embedContent(_req) {
+        throw Error();
+    }
+    async requestPost(method, req, signal) {
+        const res = await this.client.request({
+            url: this.getMethodUrl(method),
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                ...this.httpOptions.headers,
+            },
+            responseType: 'json',
+            body: JSON.stringify(req),
+            signal,
+        });
+        return res.data;
+    }
+    async requestGet(method, signal) {
+        const res = await this.client.request({
+            url: this.getMethodUrl(method),
+            method: 'GET',
+            headers: {
+                'Content-Type': 'application/json',
+                ...this.httpOptions.headers,
+            },
+            responseType: 'json',
+            signal,
+        });
+        return res.data;
+    }
+    async requestStreamingPost(method, req, signal) {
+        const res = await this.client.request({
+            url: this.getMethodUrl(method),
+            method: 'POST',
+            params: {
+                alt: 'sse',
+            },
+            headers: {
+                'Content-Type': 'application/json',
+                ...this.httpOptions.headers,
+            },
+            responseType: 'stream',
+            body: JSON.stringify(req),
+            signal,
+        });
+        return (async function* () {
+            const rl = readline.createInterface({
+                input: res.data,
+                crlfDelay: Infinity,
+            });
+            let bufferedLines = [];
+            for await (const line of rl) {
+                if (line === '') {
+                    if (bufferedLines.length === 0) {
+                        continue;
+                    }
+                    yield JSON.parse(bufferedLines.join('\n'));
+                    bufferedLines = [];
+                }
+                else if (line.startsWith('data: ')) {
+                    bufferedLines.push(line.slice(6).trim());
+                }
+                else {
+                    throw new Error(`Unexpected line format in response: ${line}`);
+                }
+            }
+        })();
+    }
+    getMethodUrl(method) {
+        const endpoint = process.env['CODE_ASSIST_ENDPOINT'] ?? exports.CODE_ASSIST_ENDPOINT;
+        return `${endpoint}/${exports.CODE_ASSIST_API_VERSION}:${method}`;
+    }
+}
+exports.CodeAssistServer = CodeAssistServer;
+function isVpcScAffectedUser(error) {
+    if (error && typeof error === 'object' && 'response' in error) {
+        const gaxiosError = error;
+        const response = gaxiosError.response?.data;
+        if (Array.isArray(response?.error?.details)) {
+            return response.error.details.some((detail) => detail.reason === 'SECURITY_POLICY_VIOLATED');
+        }
+    }
+    return false;
+}
+//# sourceMappingURL=server.js.map

package/backend/dist/packages/gemini-core/src/code_assist/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../../../../packages/gemini-core/src/code_assist/server.ts"],"names":[],"mappings":";;;AAsBA,0CAA0C;AAE1C,mCAAqC;AAKrC,2CAKqB;AAQR,QAAA,oBAAoB,GAAG,qCAAqC,CAAC;AAC7D,QAAA,uBAAuB,GAAG,YAAY,CAAC;AAEpD,MAAa,gBAAgB;IAC3B,YACW,MAAoB,EACpB,SAAkB,EAClB,cAA2B,EAAE,EAC7B,SAAkB,EAClB,QAAqB;QAJrB,WAAM,GAAN,MAAM,CAAc;QACpB,cAAS,GAAT,SAAS,CAAS;QAClB,gBAAW,GAAX,WAAW,CAAkB;QAC7B,cAAS,GAAT,SAAS,CAAS;QAClB,aAAQ,GAAR,QAAQ,CAAa;IAC7B,CAAC;IAEJ,KAAK,CAAC,qBAAqB,CACzB,GAA8B,EAC9B,YAAoB;QAEpB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAC3C,uBAAuB,EACvB,IAAA,oCAAwB,EACtB,GAAG,EACH,YAAY,EACZ,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,SAAS,CACf,EACD,GAAG,CAAC,MAAM,EAAE,WAAW,CACxB,CAAC;QACF,OAAO,CAAC,KAAK,SAAS,CAAC;YACrB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBAC/B,MAAM,IAAA,uCAA2B,EAAC,IAAI,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;IACP,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,GAA8B,EAC9B,YAAoB;QAEpB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CACjC,iBAAiB,EACjB,IAAA,oCAAwB,EACtB,GAAG,EACH,YAAY,EACZ,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,SAAS,CACf,EACD,GAAG,CAAC,MAAM,EAAE,WAAW,CACxB,CAAC;QACF,OAAO,IAAA,uCAA2B,EAAC,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,WAAW,CACf,GAAuB;QAEvB,OAAO,MAAM,IAAI,CAAC,WAAW,CAC3B,aAAa,EACb,GAAG,CACJ,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,GAA0B;QAE1B,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,WAAW,CAC3B,gBAAgB,EAChB,GAAG,CACJ,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3B,OAAO;oBACL,WAAW,EAAE,EAAE,EAAE,EAAE,kBAAU,CAAC,QAAQ,EAAE;iBACzC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,CAAC;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,GAA0B;QAC1C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CACjC,aAAa,EACb,IAAA,+BAAmB,EAAC,GAAG,CAAC,CACzB,CAAC;QACF,OAAO,IAAA,kCAAsB,EAAC,IAAI,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,IAA4B;QAE5B,MAAM,KAAK,EAAE,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,WAAW,CACf,MAAc,EACd,GAAW,EACX,MAAoB;QAEpB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;YACpC,GAAG,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC;YAC9B,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO;aAC5B;YACD,YAAY,EAAE,MAAM;YACpB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;YACzB,MAAM;SACP,CAAC,CAAC;QACH,OAAO,GAAG,CAAC,IAAS,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,UAAU,CAAI,MAAc,EAAE,MAAoB;QACtD,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;YACpC,GAAG,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC;YAC9B,MAAM,EAAE,KAAK;YACb,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO;aAC5B;YACD,YAAY,EAAE,MAAM;YACpB,MAAM;SACP,CAAC,CAAC;QACH,OAAO,GAAG,CAAC,IAAS,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,MAAc,EACd,GAAW,EACX,MAAoB;QAEpB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;YACpC,GAAG,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC;YAC9B,MAAM,EAAE,MAAM;YACd,MAAM,EAAE;gBACN,GAAG,EAAE,KAAK;aACX;YACD,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO;aAC5B;YACD,YAAY,EAAE,QAAQ;YACtB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;YACzB,MAAM;SACP,CAAC,CAAC;QAEH,OAAO,CAAC,KAAK,SAAS,CAAC;YACrB,MAAM,EAAE,GAAG,QAAQ,CAAC,eAAe,CAAC;gBAClC,KAAK,EAAE,GAAG,CAAC,IAA6B;gBACxC,SAAS,EAAE,QAAQ;aACpB,CAAC,CAAC;YAEH,IAAI,aAAa,GAAa,EAAE,CAAC;YACjC,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,EAAE,EAAE,CAAC;gBAE5B,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;oBAChB,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBAC/B,SAAS;oBACX,CAAC;oBACD,MAAM,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAM,CAAC;oBAChD,aAAa,GAAG,EAAE,CAAC;gBACrB,CAAC;qBAAM,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACrC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC3C,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,KAAK,CAAC,uCAAuC,IAAI,EAAE,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;IACP,CAAC;IAED,YAAY,CAAC,MAAc;QACzB,MAAM,QAAQ,GACZ,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,IAAI,4BAAoB,CAAC;QAC9D,OAAO,GAAG,QAAQ,IAAI,+BAAuB,IAAI,MAAM,EAAE,CAAC;IAC5D,CAAC;CACF;AA3KD,4CA2KC;AAED,SAAS,mBAAmB,CAAC,KAAc;IACzC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAG,KAInB,CAAC;QACF,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,IAE1B,CAAC;QACd,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,EAAE,OAAO,CAAC,EAAE,CAAC;YAC5C,OAAO,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAChC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,0BAA0B,CACzD,CAAC;QACJ,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/backend/dist/packages/gemini-core/src/code_assist/setup.d.ts

@@ -0,0 +1,10 @@
+import { UserTierId } from './types';
+import type { OAuth2Client } from 'google-auth-library';
+export declare class ProjectIdRequiredError extends Error {
+    constructor();
+}
+export interface UserData {
+    projectId: string;
+    userTier: UserTierId;
+}
+export declare function setupUser(client: OAuth2Client): Promise<UserData>;

package/backend/dist/packages/gemini-core/src/code_assist/setup.js

@@ -0,0 +1,94 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ProjectIdRequiredError = void 0;
+exports.setupUser = setupUser;
+const types_1 = require("./types");
+const server_1 = require("./server");
+class ProjectIdRequiredError extends Error {
+    constructor() {
+        super('This account requires setting the GOOGLE_CLOUD_PROJECT env var. See https://goo.gle/gemini-cli-auth-docs#workspace-gca');
+    }
+}
+exports.ProjectIdRequiredError = ProjectIdRequiredError;
+async function setupUser(client) {
+    const projectId = process.env['GOOGLE_CLOUD_PROJECT'] || undefined;
+    const caServer = new server_1.CodeAssistServer(client, projectId, {}, '', undefined);
+    const coreClientMetadata = {
+        ideType: 'IDE_UNSPECIFIED',
+        platform: 'PLATFORM_UNSPECIFIED',
+        pluginType: 'GEMINI',
+    };
+    const loadRes = await caServer.loadCodeAssist({
+        cloudaicompanionProject: projectId,
+        metadata: {
+            ...coreClientMetadata,
+            duetProject: projectId,
+        },
+    });
+    if (loadRes.currentTier) {
+        if (!loadRes.cloudaicompanionProject) {
+            if (projectId) {
+                return {
+                    projectId,
+                    userTier: loadRes.currentTier.id,
+                };
+            }
+            throw new ProjectIdRequiredError();
+        }
+        return {
+            projectId: loadRes.cloudaicompanionProject,
+            userTier: loadRes.currentTier.id,
+        };
+    }
+    const tier = getOnboardTier(loadRes);
+    let onboardReq;
+    if (tier.id === types_1.UserTierId.FREE) {
+        onboardReq = {
+            tierId: tier.id,
+            cloudaicompanionProject: undefined,
+            metadata: coreClientMetadata,
+        };
+    }
+    else {
+        onboardReq = {
+            tierId: tier.id,
+            cloudaicompanionProject: projectId,
+            metadata: {
+                ...coreClientMetadata,
+                duetProject: projectId,
+            },
+        };
+    }
+    let lroRes = await caServer.onboardUser(onboardReq);
+    while (!lroRes.done) {
+        await new Promise((f) => setTimeout(f, 5000));
+        lroRes = await caServer.onboardUser(onboardReq);
+    }
+    if (!lroRes.response?.cloudaicompanionProject?.id) {
+        if (projectId) {
+            return {
+                projectId,
+                userTier: tier.id,
+            };
+        }
+        throw new ProjectIdRequiredError();
+    }
+    return {
+        projectId: lroRes.response.cloudaicompanionProject.id,
+        userTier: tier.id,
+    };
+}
+function getOnboardTier(res) {
+    for (const tier of res.allowedTiers || []) {
+        if (tier.isDefault) {
+            return tier;
+        }
+    }
+    return {
+        name: '',
+        description: '',
+        id: types_1.UserTierId.LEGACY,
+        userDefinedCloudaicompanionProject: true,
+    };
+}
+//# sourceMappingURL=setup.js.map