use std::env;
use std::fs;
use std::io;

const PROTO_FILES: &[&str] = &[
    // Types
    "vendor/github.com/containerd/containerd/api/types/descriptor.proto",
    "vendor/github.com/containerd/containerd/api/types/metrics.proto",
    "vendor/github.com/containerd/containerd/api/types/mount.proto",
    "vendor/github.com/containerd/containerd/api/types/platform.proto",
    "vendor/github.com/containerd/containerd/api/types/task/task.proto",
    // Services
    "vendor/github.com/containerd/containerd/api/services/containers/v1/containers.proto",
    "vendor/github.com/containerd/containerd/api/services/content/v1/content.proto",
    "vendor/github.com/containerd/containerd/api/services/diff/v1/diff.proto",
    "vendor/github.com/containerd/containerd/api/services/events/v1/events.proto",
    "vendor/github.com/containerd/containerd/api/services/images/v1/images.proto",
    "vendor/github.com/containerd/containerd/api/services/introspection/v1/introspection.proto",
    "vendor/github.com/containerd/containerd/api/services/leases/v1/leases.proto",
    "vendor/github.com/containerd/containerd/api/services/namespaces/v1/namespace.proto",
    "vendor/github.com/containerd/containerd/api/services/snapshots/v1/snapshots.proto",
    "vendor/github.com/containerd/containerd/api/services/version/v1/version.proto",
    // Events
    "vendor/github.com/containerd/containerd/api/events/container.proto",
    "vendor/github.com/containerd/containerd/api/events/content.proto",
    "vendor/github.com/containerd/containerd/api/events/image.proto",
    "vendor/github.com/containerd/containerd/api/events/namespace.proto",
    "vendor/github.com/containerd/containerd/api/events/snapshot.proto",
    "vendor/github.com/containerd/containerd/api/events/task.proto",
];

const FIXUP_MODULES: &[&str] = &[
    "containerd.services.diff.v1",
    "containerd.services.images.v1",
    "containerd.services.introspection.v1",
    "containerd.services.snapshots.v1",
];

fn main() {
    tonic_build::configure()
        .build_server(false)
        .compile(PROTO_FILES, &["vendor/"])
        .expect("Failed to generate GRPC bindings");

    for module in FIXUP_MODULES {
        fixup_imports(module).expect("Failed to fixup module");
    }
}

// Original containerd's protobuf files contain Go style imports:
// import "github.com/containerd/containerd/api/types/mount.proto";
//
// Tonic produces invalid code for these imports:
// error[E0433]: failed to resolve: there are too many leading `super` keywords
//   --> /containerd-rust-extensions/target/debug/build/containerd-client-protos-0a328c0c63f60cd0/out/containerd.services.diff.v1.rs:47:52
//    |
// 47 |     pub diff: ::core::option::Option<super::super::super::types::Descriptor>,
//    |                                                    ^^^^^ there are too many leading `super` keywords
//
// This func aims to fix these imports to crate level ones, like `crate::types::Mount`
fn fixup_imports(path: &str) -> Result<(), io::Error> {
    let out_dir = env::var("OUT_DIR").unwrap();
    let path = format!("{}/{}.rs", out_dir, path);

    let contents = fs::read_to_string(&path)?
        .replace("super::super::super::types", "crate::types")
        .replace("super::super::super::super::google", "crate::google");
    fs::write(path, contents)?;
    Ok(())
}