Hide the internals menu behind a feature flag.

sourcegraph · Apr 18, 2024 · 945074e · 945074e
1 parent 316aa56
commit 945074e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/main/kotlin/com/sourcegraph/cody/internals/InternalsStatusBarActionGroup.kt b/src/main/kotlin/com/sourcegraph/cody/internals/InternalsStatusBarActionGroup.kt
@@ -12,7 +12,7 @@ class InternalsStatusBarActionGroup : DefaultActionGroup() {
 
   override fun update(e: AnActionEvent) {
     super.update(e)
-    e.presentation.isVisible = ConfigUtil.isCodyDebugEnabled()
+    e.presentation.isVisible = ConfigUtil.isFeatureFlagEnabled("cody.feature.internals-menu")
     removeAll()
     if (e.project != null) {
       addAll(

diff --git a/src/main/kotlin/com/sourcegraph/cody/internals/InternalsStatusBarWidgetFactory.kt b/src/main/kotlin/com/sourcegraph/cody/internals/InternalsStatusBarWidgetFactory.kt
@@ -2,14 +2,22 @@ package com.sourcegraph.cody.internals
 
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.util.Disposer
+import com.intellij.openapi.wm.StatusBar
 import com.intellij.openapi.wm.StatusBarWidget
-import com.intellij.openapi.wm.impl.status.widget.StatusBarEditorBasedWidgetFactory
+import com.intellij.openapi.wm.StatusBarWidgetFactory
+import com.sourcegraph.config.ConfigUtil
 
-class InternalsStatusBarWidgetFactory : StatusBarEditorBasedWidgetFactory() {
+class InternalsStatusBarWidgetFactory : StatusBarWidgetFactory {
   override fun getId(): String = ID
 
   override fun getDisplayName(): String = "⚠\uFE0F Cody Internals"
 
+  override fun isAvailable(project: Project): Boolean {
+    return ConfigUtil.isFeatureFlagEnabled("cody.feature.internals-menu")
+  }
+
+  override fun canBeEnabledOn(statusBar: StatusBar): Boolean = true
+
   override fun createWidget(project: Project): StatusBarWidget = InternalsStatusBarWidget(project)
 
   override fun disposeWidget(widget: StatusBarWidget) {