Merge pull request #2 from wacker-dev/client

Merge wasi-http-client into waki

Author: iawia002

.github/workflows/ci.yml

@@ -15,8 +15,16 @@ jobs:
       - name: Install Rust
         uses: dtolnay/rust-toolchain@1.78.0
         with:
+          targets: wasm32-wasi
           components: clippy, rustfmt
       - name: cargo fmt
         run: cargo fmt --all -- --check
       - name: cargo clippy
         run: cargo clippy --all-targets --all-features -- -D warnings
+      - name: Install wasmtime
+        uses: bytecodealliance/actions/wasmtime/setup@v1
+        with:
+          github_token: ${{ secrets.GITHUB_TOKEN }}
+          version: "v21.0.1"
+      - name: cargo test
+        run: cargo test

Cargo.toml

@@ -3,18 +3,22 @@ resolver = "2"
 members = [
     "waki",
     "waki-macros",
+    "test-programs",
 ]
 
 [workspace.package]
-version = "0.1.0"
+version = "0.2.0"
 authors = ["Xinzhao Xu"]
 edition = "2021"
 categories = ["wasm"]
 keywords = ["webassembly", "wasm", "wasi"]
 repository = "https://github.com/wacker-dev/waki"
 license = "Apache-2.0"
-description = "An HTTP library for building Web apps with WASI API"
+description = "HTTP client and server library for WASI"
 readme = "README.md"
 
 [workspace.dependencies]
-waki-macros = { path = "waki-macros", version = "0.1.0" }
+waki-macros = { path = "waki-macros", version = "0.2.0" }
+
+anyhow = "1.0.86"
+serde = "1.0.201"

README.md

@@ -1,12 +1,25 @@
 # waki
 
-An HTTP library for building Web apps with WASI API.
+HTTP client and server library for WASI.
+
+Send a request:
+
+```rust
+let resp = Client::new()
+    .post("https://httpbin.org/post")
+    .connect_timeout(Duration::from_secs(5))
+    .send()?;
+
+println!("status code: {}", resp.status_code());
+```
+
+Writing an HTTP component:
 
 ```rust
-use waki::{handler, Request, Response};
+use waki::{handler, ErrorCode, Request, Response};
 
 #[handler]
-fn hello(req: Request) -> Response {
-    Response::new().body(b"Hello, WASI!")
+fn hello(req: Request) -> Result<Response, ErrorCode> {
+    Response::builder().body(b"Hello, WASI!").build()
 }
 ```

test-programs/Cargo.toml

@@ -0,0 +1,9 @@
+[package]
+name = "test-programs"
+version = "0.0.0"
+edition = "2021"
+publish = false
+
+[dependencies]
+waki = { path = "../waki", features = ["json"] }
+serde = { workspace = true, features = ["derive"] }

test-programs/artifacts/Cargo.toml

@@ -0,0 +1,11 @@
+[package]
+name = "test-programs-artifacts"
+version = "0.0.0"
+edition = "2021"
+publish = false
+
+[build-dependencies]
+anyhow.workspace = true
+cargo_metadata = "0.18.1"
+wit-component = "0.208.1"
+heck = "0.5.0"

test-programs/artifacts/build.rs

@@ -0,0 +1,69 @@
+use anyhow::Result;
+use cargo_metadata::MetadataCommand;
+use heck::*;
+use std::path::{Path, PathBuf};
+use std::process::Command;
+use std::{env, fs};
+use wit_component::ComponentEncoder;
+
+fn main() -> Result<()> {
+    let out_dir = PathBuf::from(env::var_os("OUT_DIR").unwrap());
+
+    println!("cargo::rerun-if-changed=../src");
+
+    let status = Command::new("cargo")
+        .arg("build")
+        .arg("--package=test-programs")
+        .arg("--target=wasm32-wasi")
+        .env("CARGO_TARGET_DIR", &out_dir)
+        .env("CARGO_PROFILE_DEV_DEBUG", "1")
+        .status()?;
+    assert!(status.success());
+
+    let meta = MetadataCommand::new().exec()?;
+    let targets = meta
+        .packages
+        .iter()
+        .find(|p| p.name == "test-programs")
+        .unwrap()
+        .targets
+        .iter()
+        .filter(|t| t.kind == ["bin"])
+        .map(|t| &t.name)
+        .collect::<Vec<_>>();
+
+    let mut generated_code = String::new();
+
+    for target in targets {
+        let camel = target.to_shouty_snake_case();
+        let wasm = out_dir
+            .join("wasm32-wasi")
+            .join("debug")
+            .join(format!("{target}.wasm"));
+
+        let path = compile_component(&wasm)?;
+        generated_code += &format!("pub const {camel}_COMPONENT: &str = {path:?};\n");
+    }
+
+    fs::write(out_dir.join("gen.rs"), generated_code)?;
+
+    Ok(())
+}
+
+// Compile a component, return the path of the binary
+fn compile_component(wasm: &Path) -> Result<PathBuf> {
+    let module = fs::read(wasm)?;
+    let component = ComponentEncoder::default()
+        .module(module.as_slice())?
+        .validate(true)
+        .adapter(
+            "wasi_snapshot_preview1",
+            include_bytes!("wasi_snapshot_preview1.command.wasm"),
+        )?
+        .encode()?;
+    let out_dir = wasm.parent().unwrap();
+    let stem = wasm.file_stem().unwrap().to_str().unwrap();
+    let component_path = out_dir.join(format!("{stem}.component.wasm"));
+    fs::write(&component_path, component)?;
+    Ok(component_path)
+}

test-programs/artifacts/src/lib.rs

@@ -0,0 +1 @@
+include!(concat!(env!("OUT_DIR"), "/gen.rs"));

test-programs/artifacts/wasi_snapshot_preview1.command.wasm

@@ -0,0 +1,14 @@
+use waki::Client;
+
+fn main() {
+    let resp = Client::new()
+        .get("https://httpbin.org/range/20")
+        .query(&[("duration", "5"), ("chunk_size", "10")])
+        .send()
+        .unwrap();
+    assert_eq!(resp.status_code(), 200);
+
+    while let Some(chunk) = resp.chunk(1024).unwrap() {
+        assert_eq!(String::from_utf8(chunk).unwrap().len(), 10);
+    }
+}

test-programs/src/bin/get_chunk.rs

@@ -0,0 +1,20 @@
+use serde::Deserialize;
+use std::collections::HashMap;
+use waki::Client;
+
+#[derive(Deserialize)]
+struct Data {
+    args: HashMap<String, String>,
+}
+
+fn main() {
+    let resp = Client::new()
+        .get("https://httpbin.org/get?a=b")
+        .headers([("Content-Type", "application/json"), ("Accept", "*/*")])
+        .send()
+        .unwrap();
+    assert_eq!(resp.status_code(), 200);
+
+    let data = resp.json::<Data>().unwrap();
+    assert_eq!(data.args.get("a").unwrap(), "b");
+}