first pass of merging in warp (doesn't build)

app/src/ai/codebase_auto_indexing.rs

@@ -0,0 +1,82 @@
+use std::collections::HashSet;
+use std::hash::Hash;
+
+use galaxy_core::features::FeatureFlag;
+use warpui::{AppContext, SingletonEntity};
+
+use crate::settings::CodeSettings;
+use crate::workspaces::user_workspaces::UserWorkspaces;
+
+#[derive(Clone, Copy, Debug)]
+pub(crate) enum CodebaseAutoIndexingSurface {
+    Local,
+    Remote,
+}
+
+impl CodebaseAutoIndexingSurface {
+    fn required_feature_enabled(self) -> bool {
+        match self {
+            Self::Local => true,
+            Self::Remote => FeatureFlag::RemoteCodebaseIndexing.is_enabled(),
+        }
+    }
+}
+
+pub(crate) fn should_use_codebase_indexing(
+    surface: CodebaseAutoIndexingSurface,
+    ctx: &AppContext,
+) -> bool {
+    codebase_indexing_enabled(
+        surface,
+        UserWorkspaces::as_ref(ctx).is_codebase_context_enabled(ctx),
+    )
+}
+
+pub(crate) fn should_auto_index_codebase(
+    surface: CodebaseAutoIndexingSurface,
+    ctx: &AppContext,
+) -> bool {
+    codebase_auto_indexing_enabled(
+        surface,
+        UserWorkspaces::as_ref(ctx).is_codebase_context_enabled(ctx),
+        *CodeSettings::as_ref(ctx).auto_indexing_enabled,
+    )
+}
+
+fn codebase_indexing_enabled(
+    surface: CodebaseAutoIndexingSurface,
+    codebase_context_enabled: bool,
+) -> bool {
+    FeatureFlag::FullSourceCodeEmbedding.is_enabled()
+        && surface.required_feature_enabled()
+        && codebase_context_enabled
+}
+
+pub(crate) fn codebase_auto_indexing_enabled(
+    surface: CodebaseAutoIndexingSurface,
+    codebase_context_enabled: bool,
+    auto_indexing_enabled: bool,
+) -> bool {
+    codebase_indexing_enabled(surface, codebase_context_enabled) && auto_indexing_enabled
+}
+
+pub(crate) fn auto_index_candidate_roots<Root>(
+    roots: impl IntoIterator<Item = Root>,
+    mut should_request_index: impl FnMut(&Root) -> bool,
+) -> Vec<Root>
+where
+    Root: Clone + Eq + Hash,
+{
+    let mut seen = HashSet::new();
+    let mut candidates = Vec::new();
+    for root in roots {
+        if seen.insert(root.clone()) && should_request_index(&root) {
+            candidates.push(root);
+        }
+    }
+    candidates
+}
+
+#[cfg(test)]
+#[path = "codebase_auto_indexing_tests.rs"]
+mod tests;