unity-reference-server 0.1.5

An application and server for resolving references within Unity projects
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

use std::sync::Arc;

use api::method::rpc_method_handler;
use api::status::rpc_status_handler;
use args::CliArgs;
use clap::Parser;
use crawler::Crawler;
use jsonlogger::JsonLogger;
use jsonrpsee::server::Server;
use jsonrpsee::RpcModule;
use simplelog::{ConfigBuilder, TermLogger};

mod api;
mod args;
mod crawler;
mod jsonlogger;
mod yamlparser;

#[tokio::main]
async fn main() {
    let args = CliArgs::parse();

    if args.json_logs {
        JsonLogger::init(args.verbosity.into(), std::io::stderr()).unwrap();
    } else {
        TermLogger::init(
            args.verbosity.into(),
            ConfigBuilder::new().build(),
            simplelog::TerminalMode::Stderr,
            simplelog::ColorChoice::Auto,
        )
        .unwrap();
    }

    log::info!("Watching folder: {}", args.folder.to_string_lossy());
    log::info!(
        "Starting server on address and port: {}:{}",
        args.addr,
        args.port
    );

    let full_addr = format!("{}:{}", args.addr, args.port);

    let server = Server::builder().build(full_addr).await.unwrap();
    let actual_addr = server.local_addr().expect("Could not get server address");

    // Print the port to stdout
    println!("{}", actual_addr.port());

    let crawler = Arc::new(Crawler::new(args.folder).await);
    crawler.start().await;

    let mut module = RpcModule::from_arc(crawler);
    module
        .register_async_method("status", rpc_status_handler)
        .unwrap();

    module
        .register_async_method("method", rpc_method_handler)
        .unwrap();

    let handle = server.start(module);

    log::info!("Started server");

    tokio::spawn(handle.stopped()).await.unwrap();
}