lib.rs 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. mod bash;
  2. mod bootstrap;
  3. mod compact;
  4. mod config;
  5. mod conversation;
  6. mod file_ops;
  7. mod hooks;
  8. mod json;
  9. mod mcp;
  10. mod mcp_client;
  11. mod mcp_stdio;
  12. mod oauth;
  13. mod permissions;
  14. mod prompt;
  15. mod remote;
  16. pub mod sandbox;
  17. mod session;
  18. mod usage;
  19. pub use bash::{execute_bash, BashCommandInput, BashCommandOutput};
  20. pub use bootstrap::{BootstrapPhase, BootstrapPlan};
  21. pub use compact::{
  22. compact_session, estimate_session_tokens, format_compact_summary,
  23. get_compact_continuation_message, should_compact, CompactionConfig, CompactionResult,
  24. };
  25. pub use config::{
  26. ConfigEntry, ConfigError, ConfigLoader, ConfigSource, McpManagedProxyServerConfig,
  27. McpConfigCollection, McpOAuthConfig, McpRemoteServerConfig, McpSdkServerConfig,
  28. McpServerConfig, McpStdioServerConfig, McpTransport, McpWebSocketServerConfig, OAuthConfig,
  29. ResolvedPermissionMode, RuntimeConfig, RuntimeFeatureConfig, RuntimeHookConfig,
  30. RuntimePluginConfig, ScopedMcpServerConfig, CLAW_SETTINGS_SCHEMA_NAME,
  31. };
  32. pub use conversation::{
  33. auto_compaction_threshold_from_env, ApiClient, ApiRequest, AssistantEvent, AutoCompactionEvent,
  34. ConversationRuntime, RuntimeError, StaticToolExecutor, ToolError, ToolExecutor, TurnSummary,
  35. };
  36. pub use file_ops::{
  37. edit_file, glob_search, grep_search, read_file, write_file, EditFileOutput, GlobSearchOutput,
  38. GrepSearchInput, GrepSearchOutput, ReadFileOutput, StructuredPatchHunk, TextFilePayload,
  39. WriteFileOutput,
  40. };
  41. pub use hooks::{HookEvent, HookRunResult, HookRunner};
  42. pub use mcp::{
  43. mcp_server_signature, mcp_tool_name, mcp_tool_prefix, normalize_name_for_mcp,
  44. scoped_mcp_config_hash, unwrap_ccr_proxy_url,
  45. };
  46. pub use mcp_client::{
  47. McpManagedProxyTransport, McpClientAuth, McpClientBootstrap, McpClientTransport,
  48. McpRemoteTransport, McpSdkTransport, McpStdioTransport,
  49. };
  50. pub use mcp_stdio::{
  51. spawn_mcp_stdio_process, JsonRpcError, JsonRpcId, JsonRpcRequest, JsonRpcResponse,
  52. ManagedMcpTool, McpInitializeClientInfo, McpInitializeParams, McpInitializeResult,
  53. McpInitializeServerInfo, McpListResourcesParams, McpListResourcesResult, McpListToolsParams,
  54. McpListToolsResult, McpReadResourceParams, McpReadResourceResult, McpResource,
  55. McpResourceContents, McpServerManager, McpServerManagerError, McpStdioProcess, McpTool,
  56. McpToolCallContent, McpToolCallParams, McpToolCallResult, UnsupportedMcpServer,
  57. };
  58. pub use oauth::{
  59. clear_oauth_credentials, code_challenge_s256, credentials_path, generate_pkce_pair,
  60. generate_state, load_oauth_credentials, loopback_redirect_uri, parse_oauth_callback_query,
  61. parse_oauth_callback_request_target, save_oauth_credentials, OAuthAuthorizationRequest,
  62. OAuthCallbackParams, OAuthRefreshRequest, OAuthTokenExchangeRequest, OAuthTokenSet,
  63. PkceChallengeMethod, PkceCodePair,
  64. };
  65. pub use permissions::{
  66. PermissionMode, PermissionOutcome, PermissionPolicy, PermissionPromptDecision,
  67. PermissionPrompter, PermissionRequest,
  68. };
  69. pub use prompt::{
  70. load_system_prompt, prepend_bullets, ContextFile, ProjectContext, PromptBuildError,
  71. SystemPromptBuilder, FRONTIER_MODEL_NAME, SYSTEM_PROMPT_DYNAMIC_BOUNDARY,
  72. };
  73. pub use remote::{
  74. inherited_upstream_proxy_env, no_proxy_list, read_token, upstream_proxy_ws_url,
  75. RemoteSessionContext, UpstreamProxyBootstrap, UpstreamProxyState, DEFAULT_REMOTE_BASE_URL,
  76. DEFAULT_SESSION_TOKEN_PATH, DEFAULT_SYSTEM_CA_BUNDLE, NO_PROXY_HOSTS, UPSTREAM_PROXY_ENV_KEYS,
  77. };
  78. pub use session::{ContentBlock, ConversationMessage, MessageRole, Session, SessionError};
  79. pub use usage::{
  80. format_usd, pricing_for_model, ModelPricing, TokenUsage, UsageCostEstimate, UsageTracker,
  81. };
  82. #[cfg(test)]
  83. pub(crate) fn test_env_lock() -> std::sync::MutexGuard<'static, ()> {
  84. static LOCK: std::sync::OnceLock<std::sync::Mutex<()>> = std::sync::OnceLock::new();
  85. LOCK.get_or_init(|| std::sync::Mutex::new(()))
  86. .lock()
  87. .unwrap_or_else(std::sync::PoisonError::into_inner)
  88. }