File size: 3,407 Bytes
afa0cbf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 | use super::*;
use codex_core::config::ConfigOverrides;
use codex_core_plugins::PluginMarketplaceContext;
use codex_core_plugins::PluginMarketplaceScope;
use codex_core_plugins::marketplace::MarketplaceListError;
use futures::StreamExt;
const CONFIG_LOAD_CONCURRENCY: usize = 5;
impl PluginRequestProcessor {
pub(super) async fn load_catalog_config(
&self,
cwds: &[AbsolutePathBuf],
) -> Result<Config, JSONRPCErrorError> {
let config = if cwds.is_empty() {
self.config_manager
.load_non_project_config()
.await
.map_err(|err| internal_error(format!("failed to reload config: {err}")))
} else {
self.load_latest_config(/*fallback_cwd*/ None).await
}?;
// TODO(sites-migration): Remove this initial migration wait after bundled Sites is retired.
let auth = self.auth_manager.auth().await;
match Box::pin(
self.thread_manager
.plugins_manager()
.ensure_sites_migration_ready(&config.plugins_config_input(), auth.as_ref()),
)
.await
{
Ok(Some(change)) => (self.effective_plugins_changed_callback())(change),
Ok(None) => {}
Err(err) => {
warn!(error = %err, "Sites migration refresh failed; preserving local plugins")
}
}
Ok(config)
}
pub(super) async fn load_marketplace_context(
&self,
cwds: Vec<AbsolutePathBuf>,
catalog_config: &Config,
) -> PluginMarketplaceContext {
let global_config = catalog_config.plugins_config_input();
if cwds.is_empty() {
return PluginMarketplaceContext {
global_config: global_config.clone(),
scopes: vec![PluginMarketplaceScope {
cwd: None,
config: global_config,
}],
load_errors: Vec::new(),
};
}
let outcomes = futures::stream::iter(cwds)
.map(|cwd| async move {
match self
.config_manager
.load_with_overrides(
/*request_overrides*/ None,
ConfigOverrides {
cwd: Some(cwd.to_path_buf()),
..Default::default()
},
)
.await
{
Ok(config) => Ok(PluginMarketplaceScope {
cwd: Some(cwd),
config: config.plugins_config_input(),
}),
Err(err) => Err(MarketplaceListError {
path: cwd,
message: format!("failed to reload config: {err}"),
}),
}
})
.buffered(CONFIG_LOAD_CONCURRENCY)
.collect::<Vec<_>>()
.await;
let mut scopes = Vec::new();
let mut load_errors = Vec::new();
for outcome in outcomes {
match outcome {
Ok(scope) => scopes.push(scope),
Err(error) => load_errors.push(error),
}
}
PluginMarketplaceContext {
global_config,
scopes,
load_errors,
}
}
}
|