sysreq 0.1.0

Simple, virtually-zero-dependencies HTTP client wrapping a system client
Documentation 
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

use super::*;

pub struct PowerShell;
impl SystemHTTPClient for PowerShell {
	#[cfg(windows)]
	const COMMAND: &'static str = "powershell";

	#[cfg(not(windows))]
	const COMMAND: &'static str = "pwsh";

	fn installed_spawn() -> Command {
		let mut cmd = Command::new(Self::COMMAND);
		cmd.arg("-help");
		cmd
	}

	fn get(&self, uri: &str) -> Result<Vec<u8>, Error> {
		#[inline]
		fn format_script(uri: &str) -> String {
			format!(r#"
				$data = (Invoke-WebRequest '{uri}').Content;
				$Writer = New-Object System.IO.BinaryWriter([console]::OpenStandardOutput());
				$Writer.Write($data, 0, $data.length)
				$Writer.Flush()
				$Writer.Close()
			"#)
		}

		let uri = if uri.find('\'').is_some() {
			format_script(&uri.replace('\'', "''"))
		} else {
			format_script(uri)
		};

		let mut output = Command::new(Self::COMMAND).arg("-command").arg(&uri).output()?;

		// Remove the trailing CRLF PowerShell adds...
		if output.stdout.len() >= 2 && &output.stdout[output.stdout.len() - 2..] == b"\r\n" {
			output.stdout.pop();
			output.stdout.pop();
		}

		if output.status.success() {
			Ok(output.stdout)
		} else {
			Err(Error::CommandFailed {
				status: output.status,
				stdout: CommandFailedOutput(output.stdout),
				stderr: CommandFailedOutput(output.stderr)
			})
		}
	}
}