build.rs

use std::io;
use std::io::ErrorKind;
use std::path::Path;

const PROTO_FILES: &[&str] = &[
    "./vendor2/v2ray-core/app/observatory/command/command.proto",
    "./vendor2/v2ray-core/app/subscription/subscriptionmanager/command/command.proto",
    "./vendor2/v2ray-core/app/router/command/command.proto",
];

const INCLUDE_DIRS: &[&str] = &["./vendor2/protoc/include", "./vendor2/v2ray-core"];

fn require_path(path: &str, description: &str) -> io::Result<()> {
    if !Path::new(path).exists() {
        return Err(io::Error::new(
            ErrorKind::NotFound,
            format!(
                "missing {description}: {path}. Populate vendor2/protoc and vendor2/v2ray-core before building"
            ),
        ));
    }

    Ok(())
}

fn main() -> io::Result<()> {
    for include_dir in INCLUDE_DIRS {
        require_path(include_dir, "protobuf include directory")?;
    }

    for proto_file in PROTO_FILES {
        require_path(proto_file, "V2Ray protobuf file")?;
    }

    tonic_prost_build::configure()
        .build_server(false)
        .build_client(true)
        .build_transport(false)
        .compile_protos(PROTO_FILES, INCLUDE_DIRS)
        .expect("failed to compile V2Ray protobuf definitions");
    Ok(())
}