Automatically enable Rovo Dev for internal users

Author: marcomura

src/container.ts

@@ -11,6 +11,7 @@ import { BitbucketCheckoutHelper } from './bitbucket/checkoutHelper';
 import { CheckoutHelper } from './bitbucket/interfaces';
 import { PullRequest, WorkspaceRepo } from './bitbucket/model';
 import { BitbucketCloudPullRequestLinkProvider } from './bitbucket/terminal-link/createPrLinkProvider';
+import { CommandContext, setCommandContext } from './commandContext';
 import { openPullRequest } from './commands/bitbucket/pullRequest';
 import { configuration, IConfig } from './config/configuration';
 import { Commands } from './constants';
@@ -78,6 +79,15 @@ export class Container {
     private static _assignedWorkItemsView: AssignedWorkItemsViewProvider;
 
     static async initialize(context: ExtensionContext, version: string) {
+        canFetchInternalUrl().then((success) => {
+            this._isInternalUser = success;
+            this._isRovoDevEnabled = success;
+
+            if (success) {
+                this.enableRovoDev(context);
+            }
+        });
+
         const analyticsEnv: string = this.isDebugging ? 'staging' : 'prod';
 
         this._analyticsClient = analyticsClient({
@@ -211,26 +221,30 @@ export class Container {
         context.subscriptions.push(new CustomJQLViewProvider());
         context.subscriptions.push((this._assignedWorkItemsView = new AssignedWorkItemsViewProvider()));
 
-        if (!!process.env.ROVODEV_ENABLED) {
-            context.subscriptions.push(new RovoDevDecorator());
-            context.subscriptions.push(
-                languages.registerCodeActionsProvider({ scheme: 'file' }, new RovoDevCodeActionProvider(), {
-                    providedCodeActionKinds: [vscode.CodeActionKind.QuickFix],
-                }),
-            );
-            context.subscriptions.push(
-                (this._rovodevWebviewProvider = new RovoDevWebviewProvider(
-                    context,
-                    context.extensionPath,
-                    context.globalState,
-                )),
-            );
-            this.configureRovodevSettingsCommands(context);
-        }
-
         this._onboardingProvider = new OnboardingProvider();
     }
 
+    static enableRovoDev(context: ExtensionContext) {
+        // this enables the Rovo Dev activity bar
+        setCommandContext(CommandContext.RovoDevEnabled, true);
+
+        context.subscriptions.push(new RovoDevDecorator());
+        context.subscriptions.push(
+            languages.registerCodeActionsProvider({ scheme: 'file' }, new RovoDevCodeActionProvider(), {
+                providedCodeActionKinds: [vscode.CodeActionKind.QuickFix],
+            }),
+        );
+        context.subscriptions.push(
+            (this._rovodevWebviewProvider = new RovoDevWebviewProvider(
+                context,
+                context.extensionPath,
+                context.globalState,
+            )),
+        );
+
+        this.configureRovodevSettingsCommands(context);
+    }
+
     static configureRovodevSettingsCommands(context: ExtensionContext) {
         async function openConfigFile(subPath: string, friendlyName: string) {
             const home = process.env.HOME || process.env.USERPROFILE;
@@ -330,6 +344,16 @@ export class Container {
         this._context.globalState.update(ConfigTargetKey, target);
     }
 
+    private static _isInternalUser: boolean;
+    public static get isInternalUser() {
+        return this._isInternalUser;
+    }
+
+    private static _isRovoDevEnabled: boolean;
+    public static get isRovoDevEnabled() {
+        return this._isRovoDevEnabled;
+    }
+
     private static _version: string;
     public static get version() {
         return this._version;
@@ -470,3 +494,17 @@ export class Container {
         return this._rovodevWebviewProvider;
     }
 }
+
+async function canFetchInternalUrl(): Promise<boolean> {
+    try {
+        const result = await fetch('http://github.internal.atlassian.com/', {
+            method: 'GET',
+            signal: AbortSignal.timeout(5000),
+        });
+
+        const statusKind = Math.floor((result.status || 0) / 100);
+        return statusKind === 2 || statusKind === 3;
+    } catch {
+        return false;
+    }
+}

src/extension.ts

@@ -51,8 +51,6 @@ export async function activate(context: ExtensionContext) {
 
     // this disables the main Atlassian activity bar when we are in BBY
     setCommandContext(CommandContext.BbyEnvironmentActive, !!process.env.ROVODEV_BBY);
-    // this disables the Rovo Dev activity bar unless it's explicitely enabled
-    setCommandContext(CommandContext.RovoDevEnabled, !!process.env.ROVODEV_ENABLED);
 
     // Mark ourselves as the PID in charge of refreshing credentials and start listening for pings.
     context.globalState.update('rulingPid', pid);
@@ -113,7 +111,7 @@ export async function activate(context: ExtensionContext) {
         // icon to appear in the activity bar
         activateBitbucketFeatures();
         activateYamlFeatures(context);
-    } else if (!!process.env.ROVODEV_ENABLED) {
+    } else {
         commands.executeCommand('workbench.view.extension.atlascode-rovo-dev');
     }
 
@@ -215,7 +213,7 @@ async function sendAnalytics(version: string, globalState: Memento) {
 // this method is called when your extension is deactivated
 export function deactivate() {
     if (!process.env.ROVODEV_BBY) {
-        if (!!process.env.ROVODEV_ENABLED) {
+        if (Container.isRovoDevEnabled) {
             deactivateRovoDevProcessManager();
         }
 

src/rovo-dev/rovoDevWebviewProvider.ts

@@ -153,7 +153,7 @@ export class RovoDevWebviewProvider extends Disposable implements WebviewViewPro
 
         // Register the initialization of the process when the webview is ready
         this.onWebviewResolved(() => {
-            if (!!process.env.ROVODEV_ENABLED && !process.env.ROVODEV_BBY) {
+            if (!process.env.ROVODEV_BBY) {
                 initializeRovoDevProcessManager(context);
             } else {
                 this.signalProcessStarted();
@@ -292,7 +292,7 @@ export class RovoDevWebviewProvider extends Disposable implements WebviewViewPro
             }
         });
 
-        if (!!process.env.ROVODEV_ENABLED && !process.env.ROVODEV_BBY) {
+        if (!process.env.ROVODEV_BBY) {
             workspace.onDidChangeWorkspaceFolders(() => {
                 if (!this._disabled) {
                     webview.postMessage({

webpack.extension.dev.js

@@ -82,7 +82,6 @@ module.exports = [
                     process.env.ATLASCODE_EXP_OVERRIDES_STRING,
                 ),
                 'process.env.ROVODEV_BBY': JSON.stringify(process.env.ROVODEV_BBY),
-                'process.env.ROVODEV_ENABLED': JSON.stringify(process.env.ROVODEV_ENABLED),
             }),
         ],
     },

webpack.extension.prod.js

@@ -108,7 +108,6 @@ module.exports = [
                     process.env.ATLASCODE_EXP_OVERRIDES_STRING,
                 ),
                 'process.env.ROVODEV_BBY': JSON.stringify(process.env.ROVODEV_BBY),
-                'process.env.ROVODEV_ENABLED': JSON.stringify(process.env.ROVODEV_ENABLED),
             }),
         ],
         externals: ['vscode'],

webpack.react.dev.js

@@ -66,7 +66,6 @@ module.exports = {
             'process.env.ATLASCODE_EXP_OVERRIDES_BOOL': JSON.stringify(process.env.ATLASCODE_EXP_OVERRIDES_BOOL),
             'process.env.ATLASCODE_EXP_OVERRIDES_STRING': JSON.stringify(process.env.ATLASCODE_EXP_OVERRIDES_STRING),
             'process.env.ROVODEV_BBY': JSON.stringify(process.env.ROVODEV_BBY),
-            'process.env.ROVODEV_ENABLED': JSON.stringify(process.env.ROVODEV_ENABLED),
         }),
     ],
     module: {

webpack.react.prod.js

@@ -104,7 +104,6 @@ module.exports = {
             'process.env.ATLASCODE_EXP_OVERRIDES_BOOL': JSON.stringify(process.env.ATLASCODE_EXP_OVERRIDES_BOOL),
             'process.env.ATLASCODE_EXP_OVERRIDES_STRING': JSON.stringify(process.env.ATLASCODE_EXP_OVERRIDES_STRING),
             'process.env.ROVODEV_BBY': JSON.stringify(process.env.ROVODEV_BBY),
-            'process.env.ROVODEV_ENABLED': JSON.stringify(process.env.ROVODEV_ENABLED),
         }),
     ],
     performance: {