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