Merge pull request #3 from kauche/rust

all: Rewritten in Rust

Author: 110y

.gitignore

@@ -1,3 +1,4 @@
 /vendor
 /.bin
 /.env
+/target

Cargo.lock

@@ -0,0 +1,12 @@
+[package]
+name = "proxy-wasm-http-header-rename"
+version = "0.3.0"
+edition = "2021"
+
+[lib]
+crate-type = ["cdylib"]
+
+[dependencies]
+proxy-wasm = { git = "https://github.com/proxy-wasm/proxy-wasm-rust-sdk.git", branch = "main" }
+serde = { version = "1.0.216", features = ["derive"] }
+serde_json = "1.0.134"

Cargo.toml

@@ -0,0 +1,5 @@
+ARG RUST_VERSION=1.85.1
+
+FROM rust:${RUST_VERSION}
+
+RUN rustup target add wasm32-wasip1 --toolchain "${RUST_VERSION}"

Dockerfile

@@ -0,0 +1 @@
+*

Dockerfile.dockerignore

@@ -37,6 +37,11 @@ test-docker:
 build: $(TINYGO)
 	@$(TINYGO) build -o $(BIN_DIR)/proxy-wasm-http-header-rename.wasm -scheduler=none -target=wasi .
 
+.PHONY: build-rust
+build-rust:
+	@cargo build --target wasm32-wasip1 --release
+	@cp ./target/wasm32-wasip1/release/proxy_wasm_http_header_rename.wasm $(BIN_DIR)/proxy-wasm-http-header-rename.wasm
+
 .PHONY: build-docker
 build-docker:
 	@docker run \
@@ -47,3 +52,14 @@ build-docker:
 		--workdir /workspace \
 		golang:1.21.7-bullseye \
 		make build
+
+.PHONY: build-docker-rust
+build-docker-rust:
+	@docker build -t proxy-wasm-http-header-rename-rust-builder -f ./Dockerfile .
+	@docker run \
+		--rm \
+		--volume "$(shell pwd):/workspace" \
+		--user "$(shell id -u):$(shell id -g)" \
+		--workdir /workspace \
+		proxy-wasm-http-header-rename-rust-builder \
+		make build-rust

Makefile

@@ -4,6 +4,7 @@ services:
     image: envoyproxy/envoy:v1.29.1
     ports:
       - ${PORT-8080}:8080
+      - ${ADMIN_PORT-15000}:15000
     volumes:
       - ./test/envoy.yaml:/etc/envoy/envoy.yaml
       - ./.bin/proxy-wasm-http-header-rename.wasm:/etc/envoy/proxy-wasm-http-header-rename.wasm

compose.yaml

@@ -0,0 +1,81 @@
+use proxy_wasm::traits::{Context, HttpContext, RootContext};
+use proxy_wasm::types::{Action, ContextType, LogLevel};
+use serde::{Deserialize, Serialize};
+
+#[cfg(not(all(target_arch = "wasm32", target_os = "unknown")))]
+
+proxy_wasm::main! {{
+    proxy_wasm::set_log_level(LogLevel::Trace);
+        proxy_wasm::set_root_context(|_| -> Box<dyn RootContext> { Box::new(HTTPHeaderRenameRoot{
+            config: Config { request_headers_to_rename: vec![] }
+        })
+    });
+}}
+
+struct HTTPHeaderRenameRoot {
+    config: Config,
+}
+
+impl Context for HTTPHeaderRenameRoot {}
+
+impl RootContext for HTTPHeaderRenameRoot {
+    fn get_type(&self) -> Option<ContextType> {
+        Some(ContextType::HttpContext)
+    }
+
+    fn on_configure(&mut self, _: usize) -> bool {
+        if let Some(config_bytes) = self.get_plugin_configuration() {
+            self.config = serde_json::from_reader(config_bytes.as_slice()).unwrap();
+        }
+        true
+    }
+
+    fn create_http_context(&self, _: u32) -> Option<Box<dyn HttpContext>> {
+        Some(Box::new(HTTPHeadeRename {
+            config: self.config.clone(),
+        }))
+    }
+}
+
+struct HTTPHeadeRename {
+    config: Config,
+}
+
+impl Context for HTTPHeadeRename {}
+
+impl HttpContext for HTTPHeadeRename {
+    fn on_http_request_headers(&mut self, _: usize, _: bool) -> Action {
+        for header in &self.config.request_headers_to_rename {
+            let old_header = header.header.key.as_str();
+            let header_value = self.get_http_request_header(old_header);
+            if let Some(header_value) = header_value {
+                let mut value = header_value.to_string();
+                if let Some(prefix) = &header.header.prefix {
+                    value = format!("{}{}", prefix, header_value);
+                }
+
+                self.set_http_request_header(header.header.value.as_str(), Some(&value));
+                self.remove_http_request_header(old_header);
+            }
+        }
+
+        Action::Continue
+    }
+}
+
+#[derive(Serialize, Deserialize, Clone)]
+struct Config {
+    request_headers_to_rename: Vec<HTTPHeaderRenameRule>,
+}
+
+#[derive(Serialize, Deserialize, Clone)]
+struct HTTPHeaderRenameRule {
+    header: HTTPHeader,
+}
+
+#[derive(Serialize, Deserialize, Clone)]
+struct HTTPHeader {
+    key: String,
+    value: String,
+    prefix: Option<String>,
+}

src/lib.rs

@@ -101,5 +101,5 @@ static_resources:
 admin:
   address:
     socket_address:
-      address: 127.0.0.1
+      address: 0.0.0.0
       port_value: 15000