provekit_nargo_cli 1.0.0-beta.20

Noir's package manager
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

use async_lsp::{
    concurrency::ConcurrencyLayer, panic::CatchUnwindLayer, server::LifecycleLayer,
    tracing::TracingLayer,
};
use bn254_blackbox_solver::Bn254BlackBoxSolver;
use clap::Args;
use noir_lsp::NargoLspService;
use tower::ServiceBuilder;

use crate::errors::CliError;

/// Starts the Noir LSP server
///
/// Starts an LSP server which allows IDEs such as VS Code to display diagnostics in Noir source.
///
/// VS Code Noir Language Support: <https://marketplace.visualstudio.com/items?itemName=noir-lang.vscode-noir>
#[derive(Debug, Clone, Args)]
pub(crate) struct LspCommand;

pub(crate) fn run() -> Result<(), CliError> {
    use tokio::runtime::Builder;

    let runtime = Builder::new_current_thread().enable_all().build().unwrap();

    runtime.block_on(async {
        let (server, _) = async_lsp::MainLoop::new_server(|client| {
            let router = NargoLspService::new(&client, Bn254BlackBoxSolver);

            ServiceBuilder::new()
                .layer(TracingLayer::default())
                .layer(LifecycleLayer::default())
                .layer(CatchUnwindLayer::default())
                .layer(ConcurrencyLayer::default())
                .service(router)
        });

        // Prefer truly asynchronous piped stdin/stdout without blocking tasks.
        #[cfg(unix)]
        let (stdin, stdout) = (
            async_lsp::stdio::PipeStdin::lock_tokio().unwrap(),
            async_lsp::stdio::PipeStdout::lock_tokio().unwrap(),
        );
        // Fallback to spawn blocking read/write otherwise.
        #[cfg(not(unix))]
        let (stdin, stdout) = (
            tokio_util::compat::TokioAsyncReadCompatExt::compat(tokio::io::stdin()),
            tokio_util::compat::TokioAsyncWriteCompatExt::compat_write(tokio::io::stdout()),
        );

        server.run_buffered(stdin, stdout).await.map_err(CliError::LspError)
    })
}