podium-driver 0.2.0

Async Rust library for driving Android devices in tests via Maestro's gRPC server
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
68
69
70
71
72
73
74
75
76
77
78

use std::path::PathBuf;

// Podium version — APKs are published as assets on the matching GitHub release.
const PODIUM_VERSION: &str = env!("CARGO_PKG_VERSION");
const MAESTRO_VERSION: &str = "2.6.1";

const APKS: &[&str] = &["maestro-app", "maestro-server"];

fn main() {
    // ── proto codegen ────────────────────────────────────────────────────────
    let protoc = protoc_bin_vendored::protoc_bin_path().expect("vendored protoc not found");
    std::env::set_var("PROTOC", protoc);
    tonic_build::configure()
        .build_server(false)
        .compile_protos(&["proto/maestro_android.proto"], &["proto"])
        .expect("tonic_build failed");

    // ── APK resolution ───────────────────────────────────────────────────────
    // Priority:
    //   1. $PODIUM_APK_DIR/<stem>-<maestro_version>.apk  (explicit override, used by CI publish)
    //   2. vendor/<stem>-<maestro_version>.apk            (repo checkout)
    //   3. GitHub release asset download                  (crates.io consumers)
    let out_dir = PathBuf::from(std::env::var("OUT_DIR").unwrap());
    let manifest_dir = PathBuf::from(std::env::var("CARGO_MANIFEST_DIR").unwrap());
    let vendor_dir = manifest_dir.parent().unwrap().join("vendor");
    let apk_dir_override = std::env::var("PODIUM_APK_DIR").ok().map(PathBuf::from);

    for stem in APKS {
        let filename = format!("{stem}-{MAESTRO_VERSION}.apk");
        let dest = out_dir.join(&filename);

        if dest.exists() {
            continue;
        }

        // 1. Explicit override dir (set during cargo publish in CI)
        if let Some(ref dir) = apk_dir_override {
            let src = dir.join(&filename);
            if src.exists() {
                std::fs::copy(&src, &dest).unwrap_or_else(|e| {
                    panic!(
                        "failed to copy {} -> {}: {e}",
                        src.display(),
                        dest.display()
                    )
                });
                continue;
            }
        }

        // 2. Check vendor/ relative to workspace root
        let vendor = vendor_dir.join(&filename);
        if vendor.exists() {
            std::fs::copy(&vendor, &dest).unwrap_or_else(|e| {
                panic!(
                    "failed to copy {} -> {}: {e}",
                    vendor.display(),
                    dest.display()
                )
            });
            continue;
        }

        // 3. Download from GitHub release
        let url = format!(
            "https://github.com/mohitk05/podium/releases/download/v{PODIUM_VERSION}/{filename}"
        );
        eprintln!("podium build: downloading {url}");
        let bytes = reqwest::blocking::get(&url)
            .unwrap_or_else(|e| panic!("failed to fetch {url}: {e}"))
            .error_for_status()
            .unwrap_or_else(|e| panic!("HTTP error fetching {url}: {e}"))
            .bytes()
            .unwrap_or_else(|e| panic!("failed to read response for {url}: {e}"));
        std::fs::write(&dest, &bytes)
            .unwrap_or_else(|e| panic!("failed to write {}: {e}", dest.display()));
    }
}