github-copilot-sdk 1.0.0-beta.8

Rust SDK for programmatic control of the GitHub Copilot CLI via JSON-RPC. Technical preview, pre-1.0.
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use github_copilot_sdk::generated::api_types::{ShellExecRequest, ShellKillRequest};

use super::support::{wait_for_condition, with_e2e_context};

#[tokio::test]
async fn should_execute_shell_command() {
    with_e2e_context(
        "rpc_shell_and_fleet",
        "should_execute_shell_command",
        |ctx| {
            Box::pin(async move {
                ctx.set_default_copilot_user();
                let marker_path = ctx.work_dir().join("shell-rpc-marker.txt");
                let client = ctx.start_client().await;
                let session = client
                    .create_session(ctx.approve_all_session_config())
                    .await
                    .expect("create session");

                let result = session
                    .rpc()
                    .shell()
                    .exec(ShellExecRequest {
                        command: write_file_command(&marker_path, "copilot-sdk-shell-rpc"),
                        cwd: Some(ctx.work_dir().display().to_string()),
                        timeout: None,
                    })
                    .await
                    .expect("execute shell command");

                assert!(!result.process_id.trim().is_empty());
                wait_for_file_text(&marker_path, "copilot-sdk-shell-rpc").await;

                session.disconnect().await.expect("disconnect session");
                client.stop().await.expect("stop client");
            })
        },
    )
    .await;
}

#[tokio::test]
async fn should_kill_shell_process() {
    with_e2e_context("rpc_shell_and_fleet", "should_kill_shell_process", |ctx| {
        Box::pin(async move {
            ctx.set_default_copilot_user();
            let client = ctx.start_client().await;
            let session = client
                .create_session(ctx.approve_all_session_config())
                .await
                .expect("create session");

            let exec = session
                .rpc()
                .shell()
                .exec(ShellExecRequest {
                    command: long_running_command(),
                    cwd: Some(ctx.work_dir().display().to_string()),
                    timeout: None,
                })
                .await
                .expect("start shell process");
            assert!(!exec.process_id.trim().is_empty());

            let killed = session
                .rpc()
                .shell()
                .kill(ShellKillRequest {
                    process_id: exec.process_id,
                    signal: None,
                })
                .await
                .expect("kill shell process");
            assert!(killed.killed);

            session.disconnect().await.expect("disconnect session");
            client.stop().await.expect("stop client");
        })
    })
    .await;
}

async fn wait_for_file_text(path: &std::path::Path, expected: &'static str) {
    wait_for_condition("shell command output file", || async {
        match std::fs::read_to_string(path) {
            Ok(content) => content.contains(expected),
            Err(_) => false,
        }
    })
    .await;
}

#[cfg(windows)]
fn write_file_command(path: &std::path::Path, marker: &str) -> String {
    format!(
        "powershell -NoLogo -NoProfile -Command \"Set-Content -LiteralPath '{}' -Value '{}'\"",
        path.display(),
        marker
    )
}

#[cfg(not(windows))]
fn write_file_command(path: &std::path::Path, marker: &str) -> String {
    format!("sh -c \"printf '%s' '{}' > '{}'\"", marker, path.display())
}

#[cfg(windows)]
fn long_running_command() -> String {
    "powershell -NoLogo -NoProfile -Command \"Start-Sleep -Seconds 30\"".to_string()
}

#[cfg(not(windows))]
fn long_running_command() -> String {
    "sleep 30".to_string()
}