@codingame/monaco-vscode-api 21.6.0 → 22.0.0

package/vscode/src/vs/workbench/services/chat/common/chatEntitlementService.js

@@ -0,0 +1,861 @@
+
+import { __decorate, __param } from '../../../../../../../external/tslib/tslib.es6.js';
+import product from '../../../../platform/product/common/product.js';
+import { Barrier } from '../../../../base/common/async.js';
+import { CancellationTokenSource, CancellationToken } from '../../../../base/common/cancellation.js';
+import { Emitter, Event } from '../../../../base/common/event.js';
+import { Lazy } from '../../../../base/common/lazy.js';
+import { Disposable, MutableDisposable } from '../../../../base/common/lifecycle.js';
+import { localize } from '../../../../nls.js';
+import { IConfigurationService } from '../../../../platform/configuration/common/configuration.service.js';
+import { RawContextKey } from '../../../../platform/contextkey/common/contextkey.js';
+import { IContextKeyService } from '../../../../platform/contextkey/common/contextkey.service.js';
+import { IDialogService } from '../../../../platform/dialogs/common/dialogs.service.js';
+import { IInstantiationService } from '../../../../platform/instantiation/common/instantiation.js';
+import { ILogService } from '../../../../platform/log/common/log.service.js';
+import { IProductService } from '../../../../platform/product/common/productService.service.js';
+import { asText } from '../../../../platform/request/common/request.js';
+import { IRequestService } from '../../../../platform/request/common/request.service.js';
+import { StorageScope, StorageTarget } from '../../../../platform/storage/common/storage.js';
+import { IStorageService } from '../../../../platform/storage/common/storage.service.js';
+import { TelemetryLevel } from '../../../../platform/telemetry/common/telemetry.js';
+import { ITelemetryService } from '../../../../platform/telemetry/common/telemetry.service.js';
+import { IAuthenticationService, IAuthenticationExtensionsService } from '../../authentication/common/authentication.service.js';
+import { IOpenerService } from '../../../../platform/opener/common/opener.service.js';
+import { URI } from '../../../../base/common/uri.js';
+import Severity$1 from '../../../../base/common/severity.js';
+import { IWorkbenchEnvironmentService } from '../../environment/common/environmentService.service.js';
+import { isWeb } from '../../../../base/common/platform.js';
+import { ILifecycleService } from '../../lifecycle/common/lifecycle.service.js';
+import { distinct } from '../../../../base/common/arrays.js';
+import '../../../../platform/instantiation/common/extensions.js';
+import '../../../../base/common/observableInternal/index.js';
+import { observableFromEvent } from '../../../../base/common/observableInternal/observables/observableFromEvent.js';
+
+var ChatEntitlementRequests_1, ChatEntitlementContext_1;
+var ChatEntitlementContextKeys;
+(function (ChatEntitlementContextKeys) {
+    ChatEntitlementContextKeys.Setup = {
+        hidden: ( new RawContextKey('chatSetupHidden', false, true)),
+        installed: ( new RawContextKey('chatSetupInstalled', false, true)),
+        disabled: ( new RawContextKey('chatSetupDisabled', false, true)),
+        untrusted: ( new RawContextKey('chatSetupUntrusted', false, true)),
+        later: ( new RawContextKey('chatSetupLater', false, true)),
+        registered: ( new RawContextKey('chatSetupRegistered', false, true))
+    };
+    ChatEntitlementContextKeys.Entitlement = {
+        signedOut: ( new RawContextKey('chatEntitlementSignedOut', false, true)),
+        canSignUp: ( new RawContextKey('chatPlanCanSignUp', false, true)),
+        planFree: ( new RawContextKey('chatPlanFree', false, true)),
+        planPro: ( new RawContextKey('chatPlanPro', false, true)),
+        planProPlus: ( new RawContextKey('chatPlanProPlus', false, true)),
+        planBusiness: ( new RawContextKey('chatPlanBusiness', false, true)),
+        planEnterprise: ( new RawContextKey('chatPlanEnterprise', false, true)),
+        organisations: ( new RawContextKey('chatEntitlementOrganisations', undefined, true)),
+        internal: ( new RawContextKey('chatEntitlementInternal', false, true)),
+        sku: ( new RawContextKey('chatEntitlementSku', undefined, true)),
+    };
+    ChatEntitlementContextKeys.chatQuotaExceeded = ( new RawContextKey('chatQuotaExceeded', false, true));
+    ChatEntitlementContextKeys.completionsQuotaExceeded = ( new RawContextKey('completionsQuotaExceeded', false, true));
+    ChatEntitlementContextKeys.chatAnonymous = ( new RawContextKey('chatAnonymous', false, true));
+})(ChatEntitlementContextKeys || (ChatEntitlementContextKeys = {}));
+var ChatEntitlement;
+(function (ChatEntitlement) {
+    ChatEntitlement[ChatEntitlement["Unknown"] = 1] = "Unknown";
+    ChatEntitlement[ChatEntitlement["Unresolved"] = 2] = "Unresolved";
+    ChatEntitlement[ChatEntitlement["Available"] = 3] = "Available";
+    ChatEntitlement[ChatEntitlement["Unavailable"] = 4] = "Unavailable";
+    ChatEntitlement[ChatEntitlement["Free"] = 5] = "Free";
+    ChatEntitlement[ChatEntitlement["Pro"] = 6] = "Pro";
+    ChatEntitlement[ChatEntitlement["ProPlus"] = 7] = "ProPlus";
+    ChatEntitlement[ChatEntitlement["Business"] = 8] = "Business";
+    ChatEntitlement[ChatEntitlement["Enterprise"] = 9] = "Enterprise";
+})(ChatEntitlement || (ChatEntitlement = {}));
+function isProUser(chatEntitlement) {
+    return chatEntitlement === ChatEntitlement.Pro ||
+        chatEntitlement === ChatEntitlement.ProPlus ||
+        chatEntitlement === ChatEntitlement.Business ||
+        chatEntitlement === ChatEntitlement.Enterprise;
+}
+const defaultChat = {
+    extensionId: product.defaultChatAgent?.extensionId ?? '',
+    chatExtensionId: product.defaultChatAgent?.chatExtensionId ?? '',
+    upgradePlanUrl: product.defaultChatAgent?.upgradePlanUrl ?? '',
+    provider: product.defaultChatAgent?.provider ?? { default: { id: '' }, enterprise: { id: '' } },
+    providerUriSetting: product.defaultChatAgent?.providerUriSetting ?? '',
+    providerScopes: product.defaultChatAgent?.providerScopes ?? [[]],
+    entitlementUrl: product.defaultChatAgent?.entitlementUrl ?? '',
+    entitlementSignupLimitedUrl: product.defaultChatAgent?.entitlementSignupLimitedUrl ?? '',
+    completionsAdvancedSetting: product.defaultChatAgent?.completionsAdvancedSetting ?? '',
+    chatQuotaExceededContext: product.defaultChatAgent?.chatQuotaExceededContext ?? '',
+    completionsQuotaExceededContext: product.defaultChatAgent?.completionsQuotaExceededContext ?? ''
+};
+const CHAT_ALLOW_ANONYMOUS_CONFIGURATION_KEY = 'chat.allowAnonymousAccess';
+function isAnonymous(configurationService, entitlement, sentiment) {
+    if (configurationService.getValue(CHAT_ALLOW_ANONYMOUS_CONFIGURATION_KEY) !== true) {
+        return false;
+    }
+    if (entitlement !== ChatEntitlement.Unknown) {
+        return false;
+    }
+    if (sentiment.hidden || sentiment.disabled) {
+        return false;
+    }
+    return true;
+}
+let ChatEntitlementService = class ChatEntitlementService extends Disposable {
+    constructor(instantiationService, productService, environmentService, contextKeyService, configurationService) {
+        super();
+        this.contextKeyService = contextKeyService;
+        this.configurationService = configurationService;
+        this._onDidChangeQuotaExceeded = this._register(( new Emitter()));
+        this.onDidChangeQuotaExceeded = this._onDidChangeQuotaExceeded.event;
+        this._onDidChangeQuotaRemaining = this._register(( new Emitter()));
+        this.onDidChangeQuotaRemaining = this._onDidChangeQuotaRemaining.event;
+        this._quotas = {};
+        this.ExtensionQuotaContextKeys = {
+            chatQuotaExceeded: defaultChat.chatQuotaExceededContext,
+            completionsQuotaExceeded: defaultChat.completionsQuotaExceededContext,
+        };
+        this._onDidChangeAnonymous = this._register(( new Emitter()));
+        this.onDidChangeAnonymous = this._onDidChangeAnonymous.event;
+        this.anonymousObs = observableFromEvent(this.onDidChangeAnonymous, () => this.anonymous);
+        this.chatQuotaExceededContextKey = ChatEntitlementContextKeys.chatQuotaExceeded.bindTo(this.contextKeyService);
+        this.completionsQuotaExceededContextKey = ChatEntitlementContextKeys.completionsQuotaExceeded.bindTo(this.contextKeyService);
+        this.anonymousContextKey = ChatEntitlementContextKeys.chatAnonymous.bindTo(this.contextKeyService);
+        this.onDidChangeEntitlement = ( Event.map(
+            Event.filter(this.contextKeyService.onDidChangeContext, e => e.affectsSome(( new Set([
+                ChatEntitlementContextKeys.Entitlement.planPro.key,
+                ChatEntitlementContextKeys.Entitlement.planBusiness.key,
+                ChatEntitlementContextKeys.Entitlement.planEnterprise.key,
+                ChatEntitlementContextKeys.Entitlement.planProPlus.key,
+                ChatEntitlementContextKeys.Entitlement.planFree.key,
+                ChatEntitlementContextKeys.Entitlement.canSignUp.key,
+                ChatEntitlementContextKeys.Entitlement.signedOut.key,
+                ChatEntitlementContextKeys.Entitlement.organisations.key,
+                ChatEntitlementContextKeys.Entitlement.internal.key,
+                ChatEntitlementContextKeys.Entitlement.sku.key
+            ]))), this._store),
+            () => { },
+            this._store
+        ));
+        this.entitlementObs = observableFromEvent(this.onDidChangeEntitlement, () => this.entitlement);
+        this.onDidChangeSentiment = ( Event.map(
+            Event.filter(this.contextKeyService.onDidChangeContext, e => e.affectsSome(( new Set([
+                ChatEntitlementContextKeys.Setup.hidden.key,
+                ChatEntitlementContextKeys.Setup.disabled.key,
+                ChatEntitlementContextKeys.Setup.untrusted.key,
+                ChatEntitlementContextKeys.Setup.installed.key,
+                ChatEntitlementContextKeys.Setup.later.key,
+                ChatEntitlementContextKeys.Setup.registered.key
+            ]))), this._store),
+            () => { },
+            this._store
+        ));
+        this.sentimentObs = observableFromEvent(this.onDidChangeSentiment, () => this.sentiment);
+        if ((
+        (isWeb &&
+            !environmentService.remoteAuthority && !configurationService.getValue('chat.experimental.serverlessWebEnabled')))) {
+            ChatEntitlementContextKeys.Setup.hidden.bindTo(this.contextKeyService).set(true);
+            return;
+        }
+        if (!productService.defaultChatAgent) {
+            return;
+        }
+        const context = this.context = ( new Lazy(
+            () => this._register(instantiationService.createInstance(ChatEntitlementContext))
+        ));
+        this.requests = ( new Lazy(
+            () => this._register(instantiationService.createInstance(ChatEntitlementRequests, context.value, {
+                clearQuotas: () => this.clearQuotas(),
+                acceptQuotas: quotas => this.acceptQuotas(quotas)
+            }))
+        ));
+        this.registerListeners();
+    }
+    get entitlement() {
+        if (this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Entitlement.planPro.key) === true) {
+            return ChatEntitlement.Pro;
+        }
+        else if (this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Entitlement.planBusiness.key) === true) {
+            return ChatEntitlement.Business;
+        }
+        else if (this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Entitlement.planEnterprise.key) === true) {
+            return ChatEntitlement.Enterprise;
+        }
+        else if (this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Entitlement.planProPlus.key) === true) {
+            return ChatEntitlement.ProPlus;
+        }
+        else if (this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Entitlement.planFree.key) === true) {
+            return ChatEntitlement.Free;
+        }
+        else if (this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Entitlement.canSignUp.key) === true) {
+            return ChatEntitlement.Available;
+        }
+        else if (this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Entitlement.signedOut.key) === true) {
+            return ChatEntitlement.Unknown;
+        }
+        return ChatEntitlement.Unresolved;
+    }
+    get isInternal() {
+        return this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Entitlement.internal.key) === true;
+    }
+    get organisations() {
+        return this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Entitlement.organisations.key);
+    }
+    get sku() {
+        return this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Entitlement.sku.key);
+    }
+    get quotas() { return this._quotas; }
+    registerListeners() {
+        const quotaExceededSet = ( new Set(
+            [this.ExtensionQuotaContextKeys.chatQuotaExceeded, this.ExtensionQuotaContextKeys.completionsQuotaExceeded]
+        ));
+        const cts = this._register(( new MutableDisposable()));
+        this._register(this.contextKeyService.onDidChangeContext(e => {
+            if (e.affectsSome(quotaExceededSet)) {
+                if (cts.value) {
+                    cts.value.cancel();
+                }
+                cts.value = ( new CancellationTokenSource());
+                this.update(cts.value.token);
+            }
+        }));
+        let anonymousUsage = this.anonymous;
+        const updateAnonymousUsage = () => {
+            const newAnonymousUsage = this.anonymous;
+            if (newAnonymousUsage !== anonymousUsage) {
+                anonymousUsage = newAnonymousUsage;
+                this.anonymousContextKey.set(newAnonymousUsage);
+                this._onDidChangeAnonymous.fire();
+            }
+        };
+        this._register(this.configurationService.onDidChangeConfiguration(e => {
+            if (e.affectsConfiguration(CHAT_ALLOW_ANONYMOUS_CONFIGURATION_KEY)) {
+                updateAnonymousUsage();
+            }
+        }));
+        this._register(this.onDidChangeEntitlement(() => updateAnonymousUsage()));
+        this._register(this.onDidChangeSentiment(() => updateAnonymousUsage()));
+    }
+    acceptQuotas(quotas) {
+        const oldQuota = this._quotas;
+        this._quotas = quotas;
+        this.updateContextKeys();
+        const { changed: chatChanged } = this.compareQuotas(oldQuota.chat, quotas.chat);
+        const { changed: completionsChanged } = this.compareQuotas(oldQuota.completions, quotas.completions);
+        const { changed: premiumChatChanged } = this.compareQuotas(oldQuota.premiumChat, quotas.premiumChat);
+        if (chatChanged.exceeded || completionsChanged.exceeded || premiumChatChanged.exceeded) {
+            this._onDidChangeQuotaExceeded.fire();
+        }
+        if (chatChanged.remaining || completionsChanged.remaining || premiumChatChanged.remaining) {
+            this._onDidChangeQuotaRemaining.fire();
+        }
+    }
+    compareQuotas(oldQuota, newQuota) {
+        return {
+            changed: {
+                exceeded: (oldQuota?.percentRemaining === 0) !== (newQuota?.percentRemaining === 0),
+                remaining: oldQuota?.percentRemaining !== newQuota?.percentRemaining
+            }
+        };
+    }
+    clearQuotas() {
+        this.acceptQuotas({});
+    }
+    updateContextKeys() {
+        this.chatQuotaExceededContextKey.set(this._quotas.chat?.percentRemaining === 0);
+        this.completionsQuotaExceededContextKey.set(this._quotas.completions?.percentRemaining === 0);
+    }
+    get sentiment() {
+        return {
+            installed: this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Setup.installed.key) === true,
+            hidden: this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Setup.hidden.key) === true,
+            disabled: this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Setup.disabled.key) === true,
+            untrusted: this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Setup.untrusted.key) === true,
+            later: this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Setup.later.key) === true,
+            registered: this.contextKeyService.getContextKeyValue(ChatEntitlementContextKeys.Setup.registered.key) === true
+        };
+    }
+    get anonymous() {
+        return isAnonymous(this.configurationService, this.entitlement, this.sentiment);
+    }
+    async update(token) {
+        await this.requests?.value.forceResolveEntitlement(undefined, token);
+    }
+};
+ChatEntitlementService = ( __decorate([
+    ( __param(0, IInstantiationService)),
+    ( __param(1, IProductService)),
+    ( __param(2, IWorkbenchEnvironmentService)),
+    ( __param(3, IContextKeyService)),
+    ( __param(4, IConfigurationService))
+], ChatEntitlementService));
+let ChatEntitlementRequests = ChatEntitlementRequests_1 = class ChatEntitlementRequests extends Disposable {
+    static providerId(configurationService) {
+        if (configurationService.getValue(`${defaultChat.completionsAdvancedSetting}.authProvider`) === defaultChat.provider.enterprise.id) {
+            return defaultChat.provider.enterprise.id;
+        }
+        return defaultChat.provider.default.id;
+    }
+    constructor(context, chatQuotasAccessor, telemetryService, authenticationService, logService, requestService, dialogService, openerService, configurationService, authenticationExtensionsService, lifecycleService) {
+        super();
+        this.context = context;
+        this.chatQuotasAccessor = chatQuotasAccessor;
+        this.telemetryService = telemetryService;
+        this.authenticationService = authenticationService;
+        this.logService = logService;
+        this.requestService = requestService;
+        this.dialogService = dialogService;
+        this.openerService = openerService;
+        this.configurationService = configurationService;
+        this.authenticationExtensionsService = authenticationExtensionsService;
+        this.lifecycleService = lifecycleService;
+        this.pendingResolveCts = ( new CancellationTokenSource());
+        this.didResolveEntitlements = false;
+        this.state = { entitlement: this.context.state.entitlement };
+        this.registerListeners();
+        this.resolve();
+    }
+    registerListeners() {
+        this._register(this.authenticationService.onDidChangeDeclaredProviders(() => this.resolve()));
+        this._register(this.authenticationService.onDidChangeSessions(e => {
+            if (e.providerId === ChatEntitlementRequests_1.providerId(this.configurationService)) {
+                this.resolve();
+            }
+        }));
+        this._register(this.authenticationService.onDidRegisterAuthenticationProvider(e => {
+            if (e.id === ChatEntitlementRequests_1.providerId(this.configurationService)) {
+                this.resolve();
+            }
+        }));
+        this._register(this.authenticationService.onDidUnregisterAuthenticationProvider(e => {
+            if (e.id === ChatEntitlementRequests_1.providerId(this.configurationService)) {
+                this.resolve();
+            }
+        }));
+        this._register(this.context.onDidChange(() => {
+            if (!this.context.state.installed || this.context.state.disabled || this.context.state.entitlement === ChatEntitlement.Unknown) {
+                this.state = { entitlement: this.state.entitlement, quotas: undefined };
+                this.chatQuotasAccessor.clearQuotas();
+            }
+        }));
+    }
+    async resolve() {
+        this.pendingResolveCts.dispose(true);
+        const cts = this.pendingResolveCts = ( new CancellationTokenSource());
+        const session = await this.findMatchingProviderSession(cts.token);
+        if (cts.token.isCancellationRequested) {
+            return;
+        }
+        let state = undefined;
+        if (session) {
+            if (this.state.entitlement === ChatEntitlement.Unknown) {
+                state = { entitlement: ChatEntitlement.Unresolved };
+            }
+        }
+        else {
+            this.didResolveEntitlements = false;
+            state = { entitlement: ChatEntitlement.Unknown };
+        }
+        if (state) {
+            this.update(state);
+        }
+        if (session && !this.didResolveEntitlements) {
+            await this.resolveEntitlement(session, cts.token);
+        }
+    }
+    async findMatchingProviderSession(token) {
+        const sessions = await this.doGetSessions(ChatEntitlementRequests_1.providerId(this.configurationService));
+        if (token.isCancellationRequested) {
+            return undefined;
+        }
+        const matchingSessions = ( new Set());
+        for (const session of sessions) {
+            for (const scopes of defaultChat.providerScopes) {
+                if (this.includesScopes(session.scopes, scopes)) {
+                    matchingSessions.add(session);
+                }
+            }
+        }
+        return matchingSessions.size > 0 ? Array.from(matchingSessions) : undefined;
+    }
+    async doGetSessions(providerId) {
+        const preferredAccountName = this.authenticationExtensionsService.getAccountPreference(defaultChat.chatExtensionId, providerId) ?? this.authenticationExtensionsService.getAccountPreference(defaultChat.extensionId, providerId);
+        let preferredAccount;
+        for (const account of await this.authenticationService.getAccounts(providerId)) {
+            if (account.label === preferredAccountName) {
+                preferredAccount = account;
+                break;
+            }
+        }
+        try {
+            return await this.authenticationService.getSessions(providerId, undefined, { account: preferredAccount });
+        }
+        catch (error) {
+        }
+        return [];
+    }
+    includesScopes(scopes, expectedScopes) {
+        return expectedScopes.every(scope => scopes.includes(scope));
+    }
+    async resolveEntitlement(sessions, token) {
+        const entitlements = await this.doResolveEntitlement(sessions, token);
+        if (typeof entitlements?.entitlement === 'number' && !token.isCancellationRequested) {
+            this.didResolveEntitlements = true;
+            this.update(entitlements);
+        }
+        return entitlements;
+    }
+    async doResolveEntitlement(sessions, token) {
+        if (token.isCancellationRequested) {
+            return undefined;
+        }
+        const response = await this.request(this.getEntitlementUrl(), 'GET', undefined, sessions, token);
+        if (token.isCancellationRequested) {
+            return undefined;
+        }
+        if (!response) {
+            this.logService.trace('[chat entitlement]: no response');
+            return { entitlement: ChatEntitlement.Unresolved };
+        }
+        if (response.res.statusCode && response.res.statusCode !== 200) {
+            this.logService.trace(`[chat entitlement]: unexpected status code ${response.res.statusCode}`);
+            return ((
+            response.res.statusCode === 401 || response.res.statusCode === 404)
+            ) ? { entitlement: ChatEntitlement.Unknown  } : { entitlement: ChatEntitlement.Unresolved };
+        }
+        let responseText = null;
+        try {
+            responseText = await asText(response);
+        }
+        catch (error) {
+        }
+        if (token.isCancellationRequested) {
+            return undefined;
+        }
+        if (!responseText) {
+            this.logService.trace('[chat entitlement]: response has no content');
+            return { entitlement: ChatEntitlement.Unresolved };
+        }
+        let entitlementsResponse;
+        try {
+            entitlementsResponse = JSON.parse(responseText);
+            this.logService.trace(`[chat entitlement]: parsed result is ${JSON.stringify(entitlementsResponse)}`);
+        }
+        catch (err) {
+            this.logService.trace(`[chat entitlement]: error parsing response (${err})`);
+            return { entitlement: ChatEntitlement.Unresolved };
+        }
+        let entitlement;
+        if (entitlementsResponse.access_type_sku === 'free_limited_copilot') {
+            entitlement = ChatEntitlement.Free;
+        }
+        else if (entitlementsResponse.can_signup_for_limited) {
+            entitlement = ChatEntitlement.Available;
+        }
+        else if (entitlementsResponse.copilot_plan === 'individual') {
+            entitlement = ChatEntitlement.Pro;
+        }
+        else if (entitlementsResponse.copilot_plan === 'individual_pro') {
+            entitlement = ChatEntitlement.ProPlus;
+        }
+        else if (entitlementsResponse.copilot_plan === 'business') {
+            entitlement = ChatEntitlement.Business;
+        }
+        else if (entitlementsResponse.copilot_plan === 'enterprise') {
+            entitlement = ChatEntitlement.Enterprise;
+        }
+        else if (entitlementsResponse.chat_enabled) {
+            entitlement = ChatEntitlement.Pro;
+        }
+        else {
+            entitlement = ChatEntitlement.Unavailable;
+        }
+        const entitlements = {
+            entitlement,
+            organisations: entitlementsResponse.organization_login_list,
+            quotas: this.toQuotas(entitlementsResponse),
+            sku: entitlementsResponse.access_type_sku
+        };
+        this.logService.trace(`[chat entitlement]: resolved to ${entitlements.entitlement}, quotas: ${JSON.stringify(entitlements.quotas)}`);
+        this.telemetryService.publicLog2('chatInstallEntitlement', {
+            entitlement: entitlements.entitlement,
+            tid: entitlementsResponse.analytics_tracking_id,
+            sku: entitlements.sku,
+            quotaChat: entitlements.quotas?.chat?.remaining,
+            quotaPremiumChat: entitlements.quotas?.premiumChat?.remaining,
+            quotaCompletions: entitlements.quotas?.completions?.remaining,
+            quotaResetDate: entitlements.quotas?.resetDate
+        });
+        return entitlements;
+    }
+    getEntitlementUrl() {
+        if (ChatEntitlementRequests_1.providerId(this.configurationService) === defaultChat.provider.enterprise.id) {
+            try {
+                const enterpriseUrl = ( new URL(this.configurationService.getValue(defaultChat.providerUriSetting)));
+                return `${enterpriseUrl.protocol}//api.${enterpriseUrl.hostname}${enterpriseUrl.port ? ':' + enterpriseUrl.port : ''}/copilot_internal/user`;
+            }
+            catch (error) {
+                this.logService.error(error);
+            }
+        }
+        return defaultChat.entitlementUrl;
+    }
+    toQuotas(response) {
+        const quotas = {
+            resetDate: response.quota_reset_date_utc ?? response.quota_reset_date ?? response.limited_user_reset_date,
+            resetDateHasTime: typeof response.quota_reset_date_utc === 'string',
+        };
+        if (response.monthly_quotas?.chat && typeof response.limited_user_quotas?.chat === 'number') {
+            quotas.chat = {
+                total: response.monthly_quotas.chat,
+                remaining: response.limited_user_quotas.chat,
+                percentRemaining: Math.min(100, Math.max(0, (response.limited_user_quotas.chat / response.monthly_quotas.chat) * 100)),
+                overageEnabled: false,
+                overageCount: 0,
+                unlimited: false
+            };
+        }
+        if (response.monthly_quotas?.completions && typeof response.limited_user_quotas?.completions === 'number') {
+            quotas.completions = {
+                total: response.monthly_quotas.completions,
+                remaining: response.limited_user_quotas.completions,
+                percentRemaining: Math.min(100, Math.max(0, (response.limited_user_quotas.completions / response.monthly_quotas.completions) * 100)),
+                overageEnabled: false,
+                overageCount: 0,
+                unlimited: false
+            };
+        }
+        if (response.quota_snapshots) {
+            for (const quotaType of ['chat', 'completions', 'premium_interactions']) {
+                const rawQuotaSnapshot = response.quota_snapshots[quotaType];
+                if (!rawQuotaSnapshot) {
+                    continue;
+                }
+                const quotaSnapshot = {
+                    total: rawQuotaSnapshot.entitlement,
+                    remaining: rawQuotaSnapshot.remaining,
+                    percentRemaining: Math.min(100, Math.max(0, rawQuotaSnapshot.percent_remaining)),
+                    overageEnabled: rawQuotaSnapshot.overage_permitted,
+                    overageCount: rawQuotaSnapshot.overage_count,
+                    unlimited: rawQuotaSnapshot.unlimited
+                };
+                switch (quotaType) {
+                    case 'chat':
+                        quotas.chat = quotaSnapshot;
+                        break;
+                    case 'completions':
+                        quotas.completions = quotaSnapshot;
+                        break;
+                    case 'premium_interactions':
+                        quotas.premiumChat = quotaSnapshot;
+                        break;
+                }
+            }
+        }
+        return quotas;
+    }
+    async request(url, type, body, sessions, token) {
+        let lastRequest;
+        for (const session of sessions) {
+            if (token.isCancellationRequested) {
+                return lastRequest;
+            }
+            try {
+                const response = await this.requestService.request({
+                    type,
+                    url,
+                    data: type === 'POST' ? JSON.stringify(body) : undefined,
+                    disableCache: true,
+                    headers: {
+                        'Authorization': `Bearer ${session.accessToken}`
+                    }
+                }, token);
+                const status = response.res.statusCode;
+                if (status && status !== 200) {
+                    lastRequest = response;
+                    continue;
+                }
+                return response;
+            }
+            catch (error) {
+                if (!token.isCancellationRequested) {
+                    this.logService.error(`[chat entitlement] request: error ${error}`);
+                }
+            }
+        }
+        return lastRequest;
+    }
+    update(state) {
+        this.state = state;
+        this.context.update({ entitlement: this.state.entitlement, organisations: this.state.organisations, sku: this.state.sku });
+        if (state.quotas) {
+            this.chatQuotasAccessor.acceptQuotas(state.quotas);
+        }
+    }
+    async forceResolveEntitlement(sessions, token = CancellationToken.None) {
+        if (!sessions) {
+            sessions = await this.findMatchingProviderSession(token);
+        }
+        if (!sessions || sessions.length === 0) {
+            return undefined;
+        }
+        return this.resolveEntitlement(sessions, token);
+    }
+    async signUpFree(sessions) {
+        const body = {
+            restricted_telemetry: this.telemetryService.telemetryLevel === TelemetryLevel.NONE ? 'disabled' : 'enabled',
+            public_code_suggestions: 'enabled'
+        };
+        const response = await this.request(defaultChat.entitlementSignupLimitedUrl, 'POST', body, sessions, CancellationToken.None);
+        if (!response) {
+            const retry = await this.onUnknownSignUpError(( localize(13212, "No response received.")), '[chat entitlement] sign-up: no response');
+            return retry ? this.signUpFree(sessions) : { errorCode: 1 };
+        }
+        if (response.res.statusCode && response.res.statusCode !== 200) {
+            if (response.res.statusCode === 422) {
+                try {
+                    const responseText = await asText(response);
+                    if (responseText) {
+                        const responseError = JSON.parse(responseText);
+                        if (typeof responseError.message === 'string' && responseError.message) {
+                            this.onUnprocessableSignUpError(`[chat entitlement] sign-up: unprocessable entity (${responseError.message})`, responseError.message);
+                            return { errorCode: response.res.statusCode };
+                        }
+                    }
+                }
+                catch (error) {
+                }
+            }
+            const retry = await this.onUnknownSignUpError(( localize(13213, "Unexpected status code {0}.", response.res.statusCode)), `[chat entitlement] sign-up: unexpected status code ${response.res.statusCode}`);
+            return retry ? this.signUpFree(sessions) : { errorCode: response.res.statusCode };
+        }
+        let responseText = null;
+        try {
+            responseText = await asText(response);
+        }
+        catch (error) {
+        }
+        if (!responseText) {
+            const retry = await this.onUnknownSignUpError(( localize(13214, "Response has no contents.")), '[chat entitlement] sign-up: response has no content');
+            return retry ? this.signUpFree(sessions) : { errorCode: 2 };
+        }
+        let parsedResult = undefined;
+        try {
+            parsedResult = JSON.parse(responseText);
+            this.logService.trace(`[chat entitlement] sign-up: response is ${responseText}`);
+        }
+        catch (err) {
+            const retry = await this.onUnknownSignUpError(( localize(13215, "Invalid response contents.")), `[chat entitlement] sign-up: error parsing response (${err})`);
+            return retry ? this.signUpFree(sessions) : { errorCode: 3 };
+        }
+        this.update({ entitlement: ChatEntitlement.Free });
+        return Boolean(parsedResult?.subscribed);
+    }
+    async onUnknownSignUpError(detail, logMessage) {
+        this.logService.error(logMessage);
+        if (!this.lifecycleService.willShutdown) {
+            const { confirmed } = await this.dialogService.confirm({
+                type: Severity$1.Error,
+                message: ( localize(
+                    13216,
+                    "An error occurred while signing up for the GitHub Copilot Free plan. Would you like to try again?"
+                )),
+                detail,
+                primaryButton: ( localize(13217, "Retry"))
+            });
+            return confirmed;
+        }
+        return false;
+    }
+    onUnprocessableSignUpError(logMessage, logDetails) {
+        this.logService.error(logMessage);
+        if (!this.lifecycleService.willShutdown) {
+            this.dialogService.prompt({
+                type: Severity$1.Error,
+                message: ( localize(
+                    13218,
+                    "An error occurred while signing up for the GitHub Copilot Free plan."
+                )),
+                detail: logDetails,
+                buttons: [
+                    {
+                        label: ( localize(13219, "OK")),
+                        run: () => { }
+                    },
+                    {
+                        label: ( localize(13220, "Learn More")),
+                        run: () => this.openerService.open(( URI.parse(defaultChat.upgradePlanUrl)))
+                    }
+                ]
+            });
+        }
+    }
+    async signIn(options) {
+        const providerId = ChatEntitlementRequests_1.providerId(this.configurationService);
+        const scopes = options?.additionalScopes ? distinct([...defaultChat.providerScopes[0], ...options.additionalScopes]) : defaultChat.providerScopes[0];
+        const session = await this.authenticationService.createSession(providerId, scopes, {
+            extraAuthorizeParameters: { get_started_with: 'copilot-vscode' },
+            provider: options?.useSocialProvider
+        });
+        this.authenticationExtensionsService.updateAccountPreference(defaultChat.extensionId, providerId, session.account);
+        this.authenticationExtensionsService.updateAccountPreference(defaultChat.chatExtensionId, providerId, session.account);
+        const entitlements = await this.forceResolveEntitlement([session]);
+        return { session, entitlements };
+    }
+    dispose() {
+        this.pendingResolveCts.dispose(true);
+        super.dispose();
+    }
+};
+ChatEntitlementRequests = ChatEntitlementRequests_1 = ( __decorate([
+    ( __param(2, ITelemetryService)),
+    ( __param(3, IAuthenticationService)),
+    ( __param(4, ILogService)),
+    ( __param(5, IRequestService)),
+    ( __param(6, IDialogService)),
+    ( __param(7, IOpenerService)),
+    ( __param(8, IConfigurationService)),
+    ( __param(9, IAuthenticationExtensionsService)),
+    ( __param(10, ILifecycleService))
+], ChatEntitlementRequests));
+let ChatEntitlementContext = class ChatEntitlementContext extends Disposable {
+    static { ChatEntitlementContext_1 = this; }
+    static { this.CHAT_ENTITLEMENT_CONTEXT_STORAGE_KEY = 'chat.setupContext'; }
+    static { this.CHAT_DISABLED_CONFIGURATION_KEY = 'chat.disableAIFeatures'; }
+    get state() { return this.withConfiguration(this.suspendedState ?? this._state); }
+    constructor(contextKeyService, storageService, logService, configurationService, telemetryService) {
+        super();
+        this.storageService = storageService;
+        this.logService = logService;
+        this.configurationService = configurationService;
+        this.telemetryService = telemetryService;
+        this.suspendedState = undefined;
+        this._onDidChange = this._register(( new Emitter()));
+        this.onDidChange = this._onDidChange.event;
+        this.updateBarrier = undefined;
+        this.canSignUpContextKey = ChatEntitlementContextKeys.Entitlement.canSignUp.bindTo(contextKeyService);
+        this.signedOutContextKey = ChatEntitlementContextKeys.Entitlement.signedOut.bindTo(contextKeyService);
+        this.freeContextKey = ChatEntitlementContextKeys.Entitlement.planFree.bindTo(contextKeyService);
+        this.proContextKey = ChatEntitlementContextKeys.Entitlement.planPro.bindTo(contextKeyService);
+        this.proPlusContextKey = ChatEntitlementContextKeys.Entitlement.planProPlus.bindTo(contextKeyService);
+        this.businessContextKey = ChatEntitlementContextKeys.Entitlement.planBusiness.bindTo(contextKeyService);
+        this.enterpriseContextKey = ChatEntitlementContextKeys.Entitlement.planEnterprise.bindTo(contextKeyService);
+        this.organisationsContextKey = ChatEntitlementContextKeys.Entitlement.organisations.bindTo(contextKeyService);
+        this.isInternalContextKey = ChatEntitlementContextKeys.Entitlement.internal.bindTo(contextKeyService);
+        this.skuContextKey = ChatEntitlementContextKeys.Entitlement.sku.bindTo(contextKeyService);
+        this.hiddenContext = ChatEntitlementContextKeys.Setup.hidden.bindTo(contextKeyService);
+        this.laterContext = ChatEntitlementContextKeys.Setup.later.bindTo(contextKeyService);
+        this.installedContext = ChatEntitlementContextKeys.Setup.installed.bindTo(contextKeyService);
+        this.disabledContext = ChatEntitlementContextKeys.Setup.disabled.bindTo(contextKeyService);
+        this.untrustedContext = ChatEntitlementContextKeys.Setup.untrusted.bindTo(contextKeyService);
+        this.registeredContext = ChatEntitlementContextKeys.Setup.registered.bindTo(contextKeyService);
+        this._state = this.storageService.getObject(ChatEntitlementContext_1.CHAT_ENTITLEMENT_CONTEXT_STORAGE_KEY, StorageScope.PROFILE) ?? { entitlement: ChatEntitlement.Unknown, organisations: undefined, sku: undefined };
+        this.updateContextSync();
+        this.registerListeners();
+    }
+    registerListeners() {
+        this._register(this.configurationService.onDidChangeConfiguration(e => {
+            if (e.affectsConfiguration(ChatEntitlementContext_1.CHAT_DISABLED_CONFIGURATION_KEY)) {
+                this.updateContext();
+            }
+        }));
+    }
+    withConfiguration(state) {
+        if (this.configurationService.getValue(ChatEntitlementContext_1.CHAT_DISABLED_CONFIGURATION_KEY) === true) {
+            return {
+                ...state,
+                hidden: true
+            };
+        }
+        return state;
+    }
+    async update(context) {
+        this.logService.trace(`[chat entitlement context] update(): ${JSON.stringify(context)}`);
+        const oldState = JSON.stringify(this._state);
+        if (typeof context.installed === 'boolean' && typeof context.disabled === 'boolean' && typeof context.untrusted === 'boolean') {
+            this._state.installed = context.installed;
+            this._state.disabled = context.disabled;
+            this._state.untrusted = context.untrusted;
+            if (context.installed && !context.disabled) {
+                context.hidden = false;
+            }
+        }
+        if (typeof context.hidden === 'boolean') {
+            this._state.hidden = context.hidden;
+        }
+        if (typeof context.later === 'boolean') {
+            this._state.later = context.later;
+        }
+        if (typeof context.entitlement === 'number') {
+            this._state.entitlement = context.entitlement;
+            this._state.organisations = context.organisations;
+            this._state.sku = context.sku;
+            if (this._state.entitlement === ChatEntitlement.Free || isProUser(this._state.entitlement)) {
+                this._state.registered = true;
+            }
+            else if (this._state.entitlement === ChatEntitlement.Available) {
+                this._state.registered = false;
+            }
+        }
+        if (oldState === JSON.stringify(this._state)) {
+            return;
+        }
+        this.storageService.store(ChatEntitlementContext_1.CHAT_ENTITLEMENT_CONTEXT_STORAGE_KEY, {
+            ...this._state,
+            later: undefined
+        }, StorageScope.PROFILE, StorageTarget.MACHINE);
+        return this.updateContext();
+    }
+    async updateContext() {
+        await this.updateBarrier?.wait();
+        this.updateContextSync();
+    }
+    updateContextSync() {
+        const state = this.withConfiguration(this._state);
+        this.signedOutContextKey.set(state.entitlement === ChatEntitlement.Unknown);
+        this.canSignUpContextKey.set(state.entitlement === ChatEntitlement.Available);
+        this.freeContextKey.set(state.entitlement === ChatEntitlement.Free);
+        this.proContextKey.set(state.entitlement === ChatEntitlement.Pro);
+        this.proPlusContextKey.set(state.entitlement === ChatEntitlement.ProPlus);
+        this.businessContextKey.set(state.entitlement === ChatEntitlement.Business);
+        this.enterpriseContextKey.set(state.entitlement === ChatEntitlement.Enterprise);
+        this.organisationsContextKey.set(state.organisations);
+        this.isInternalContextKey.set(Boolean(state.organisations?.some(org => org === 'github' || org === 'microsoft' || org === 'ms-copilot' || org === 'MicrosoftCopilot')));
+        this.skuContextKey.set(state.sku);
+        this.hiddenContext.set(!!state.hidden);
+        this.laterContext.set(!!state.later);
+        this.installedContext.set(!!state.installed);
+        this.disabledContext.set(!!state.disabled);
+        this.untrustedContext.set(!!state.untrusted);
+        this.registeredContext.set(!!state.registered);
+        this.logService.trace(`[chat entitlement context] updateContext(): ${JSON.stringify(state)}`);
+        this.telemetryService.publicLog2('chatEntitlements', {
+            chatHidden: Boolean(state.hidden),
+            chatDisabled: Boolean(state.disabled),
+            chatEntitlement: state.entitlement,
+            chatRegistered: Boolean(state.registered),
+            chatAnonymous: isAnonymous(this.configurationService, state.entitlement, state)
+        });
+        this._onDidChange.fire();
+    }
+    suspend() {
+        this.suspendedState = { ...this._state };
+        this.updateBarrier = ( new Barrier());
+    }
+    resume() {
+        this.suspendedState = undefined;
+        this.updateBarrier?.open();
+        this.updateBarrier = undefined;
+    }
+};
+ChatEntitlementContext = ChatEntitlementContext_1 = ( __decorate([
+    ( __param(0, IContextKeyService)),
+    ( __param(1, IStorageService)),
+    ( __param(2, ILogService)),
+    ( __param(3, IConfigurationService)),
+    ( __param(4, ITelemetryService))
+], ChatEntitlementContext));
+
+export { ChatEntitlement, ChatEntitlementContext, ChatEntitlementContextKeys, ChatEntitlementRequests, ChatEntitlementService, isProUser };