File size: 1,603 Bytes
ea39c0e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
//! Test support for running the Windows exec-server under Wine.

use std::future::Future;
use std::path::PathBuf;

use anyhow::Context;
use anyhow::Result;
use tokio::io::AsyncBufReadExt;
use tokio::io::BufReader;
use wine_test_support::WineTestCommand;

/// Runs the Windows exec-server under Wine for the duration of a scoped operation.
pub struct WineExecServer;

impl WineExecServer {
    /// Starts the server, passes its WebSocket URL and Wine prefix to `operation`, and tears it
    /// down afterward.
    pub async fn scope<T, F, Fut>(self, operation: F) -> Result<T>
    where
        F: FnOnce(String, PathBuf) -> Fut,
        Fut: Future<Output = Result<T>>,
    {
        let executable = codex_utils_cargo_bin::cargo_bin("wine-windows-exec-server")?;
        let mut exec_server = WineTestCommand::new(executable)
            .env("CODEX_HOME", r"C:\codex-home")
            .spawn()?;
        let wine_prefix = exec_server.prefix_path().to_path_buf();
        let stdout = exec_server.take_stdout();

        exec_server
            .scope(async move {
                let mut lines = BufReader::new(stdout).lines();
                let exec_server_url = loop {
                    let line = lines
                        .next_line()
                        .await?
                        .context("Wine exec-server exited before reporting its URL")?;
                    if line.starts_with("ws://") {
                        break line;
                    }
                };
                operation(exec_server_url, wine_prefix).await
            })
            .await
    }
}