Skip to main content

aether_lspd/
lib.rs

1#![doc = include_str!("../README.md")]
2
3#[cfg(feature = "testing")]
4pub mod testing;
5
6mod client_connection;
7mod daemonize;
8mod diagnostics_store;
9mod document_lifecycle;
10mod file_watcher;
11pub mod language_catalog;
12mod pid_lockfile;
13mod process_transport;
14mod refresh_queue;
15mod workspace_registry;
16mod workspace_session;
17
18mod client;
19pub mod daemon;
20pub mod error;
21pub mod lsp_utils;
22pub mod protocol;
23pub mod socket_path;
24pub mod uri;
25
26use std::path::PathBuf;
27use std::time::Duration;
28use tracing_subscriber::EnvFilter;
29use tracing_subscriber::fmt::writer::MakeWriterExt;
30
31pub use client::{ClientError, ClientResult, LspClient};
32pub use daemon::{LspDaemon, run_daemon};
33pub use error::{DaemonError, DaemonResult};
34pub use language_catalog::LanguageId;
35pub use language_catalog::{
36    LANGUAGE_METADATA, LanguageMetadata, LspConfig, extensions_for_alias, from_lsp_id, get_config_for_language,
37    metadata_for,
38};
39pub use lsp_utils::symbol_kind_to_string;
40pub use socket_path::{ensure_socket_dir, lockfile_path, log_file_path, socket_path};
41
42pub use protocol::{
43    DaemonRequest, DaemonResponse, InitializeRequest, LspErrorResponse, MAX_MESSAGE_SIZE, ProtocolError,
44};
45pub use uri::{path_to_uri, uri_to_path};
46
47#[derive(clap::Args)]
48pub struct LspdArgs {
49    /// Socket path to listen on
50    #[arg(long)]
51    pub socket: PathBuf,
52
53    /// Idle timeout in seconds (0 = no timeout)
54    #[arg(long, default_value = "300")]
55    pub idle_timeout: u64,
56
57    /// Log level (trace, debug, info, warn, error)
58    #[arg(long, default_value = "info")]
59    pub log_level: String,
60
61    /// Log file path. Required for daemon mode since stderr is redirected to /dev/null.
62    #[arg(long)]
63    pub log_file: Option<PathBuf>,
64}
65
66pub fn run_lspd(args: LspdArgs) -> Result<(), String> {
67    let idle_timeout = if args.idle_timeout == 0 { None } else { Some(Duration::from_secs(args.idle_timeout)) };
68
69    daemonize::daemonize()?;
70
71    let runtime = tokio::runtime::Runtime::new().map_err(|e| format!("Failed to create tokio runtime: {e}"))?;
72
73    runtime.block_on(async {
74        let filter = EnvFilter::try_from_default_env().unwrap_or_else(|_| EnvFilter::new(&args.log_level));
75
76        if let Some(ref log_file) = args.log_file {
77            if let Some(parent) = log_file.parent() {
78                let _ = std::fs::create_dir_all(parent);
79            }
80            let file = std::fs::OpenOptions::new()
81                .create(true)
82                .append(true)
83                .open(log_file)
84                .map_err(|e| format!("Failed to open log file: {e}"))?;
85
86            tracing_subscriber::fmt()
87                .with_env_filter(filter)
88                .with_target(true)
89                .with_ansi(false)
90                .with_writer(file.with_max_level(tracing::Level::TRACE))
91                .init();
92        } else {
93            tracing_subscriber::fmt().with_env_filter(filter).with_target(true).init();
94        }
95
96        tracing::info!("Starting LSP daemon on socket: {:?}", args.socket);
97        run_daemon(args.socket, idle_timeout).await.map_err(|e| format!("Daemon error: {e}"))
98    })
99}