Merge pull request #27 from G-Core/feat/secret

ruslanti · web-flow · commit e9781068b78c · 2024-10-21T14:56:58.000+03:00
Secret feature
diff --git a/.cargo/config.toml b/.cargo/config.toml
@@ -1,2 +1,2 @@
 [build]
-target = "wasm32-wasi"
+target = "wasm32-wasip1"
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -19,7 +19,7 @@ jobs:
       - name: Setup Rust
         uses: dtolnay/rust-toolchain@stable
         with:
-          target: wasm32-wasi
+          target: wasm32-wasip1
           components: rustfmt, clippy
 
       - name: Run cargo-audit binary crate
@@ -49,7 +49,7 @@ jobs:
       - name: Setup Rust
         uses: dtolnay/rust-toolchain@stable
         with:
-          target: wasm32-wasi
+          target: wasm32-wasip1
           components: rustfmt, clippy
 
       - name: Run Release PR
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/README.md b/README.md
@@ -26,7 +26,7 @@ The table below summarizes the feature support for language SDKs.
 | Env Variables | Supported | Supported  |
 
 ## Rust toolchain setup:
-- `rustup target add wasm32-wasi`
+- `rustup target add wasm32-wasip1`
 
 # The FastEdge Rust SDK
 
diff --git a/examples/secret/Cargo.toml b/examples/secret/Cargo.toml
@@ -0,0 +1,12 @@
+[package]
+name = "secret"
+version = {workspace = true}
+edition = "2021"
+publish = false
+
+[lib]
+crate-type = ["cdylib"]
+
+[dependencies]
+fastedge = { path = "../../" }
+anyhow = "1.0"
diff --git a/examples/secret/src/lib.rs b/examples/secret/src/lib.rs
@@ -0,0 +1,44 @@
+
+use anyhow::{Error, Result};
+
+use fastedge::body::Body;
+use fastedge::http::{Request, Response, StatusCode};
+use fastedge::secret;
+
+#[allow(dead_code)]
+#[fastedge::http]
+fn main(_req: Request<Body>) -> Result<Response<Body>> {
+    let value = match secret::get("SECRET") {
+        Ok(value) => value,
+        Err(secret::Error::AccessDenied) => {
+            return Response::builder()
+                .status(StatusCode::FORBIDDEN)
+                .body(Body::empty())
+                .map_err(Error::msg);
+        },
+        Err(secret::Error::Other(msg)) => {
+            return Response::builder()
+                .status(StatusCode::FORBIDDEN)
+                .body(Body::from(msg))
+                .map_err(Error::msg);
+        },
+        Err(secret::Error::DecryptError) => {
+            return Response::builder()
+                .status(StatusCode::INTERNAL_SERVER_ERROR)
+                .body(Body::empty())
+                .map_err(Error::msg);
+        }
+    };
+
+    if value.is_none() {
+        return Response::builder()
+            .status(StatusCode::NOT_FOUND)
+            .body(Body::empty())
+            .map_err(Error::msg);
+    }
+
+    Response::builder()
+        .status(StatusCode::OK)
+        .body(Body::from(value.unwrap_or_default()))
+        .map_err(Error::msg)
+}
diff --git a/src/lib.rs b/src/lib.rs
@@ -11,6 +11,9 @@ pub use http_client::send_request;
 #[doc(hidden)]
 pub use crate::exports::gcore::fastedge::http_handler;
 use crate::gcore::fastedge::http::{Error as HttpError, Method, Request, Response};
+#[doc(hidden)]
+pub use crate::gcore::fastedge::secret;
+
 
 /// Implementation of Outbound HTTP component
 mod http_client;
@@ -24,7 +27,7 @@ pub mod wasi_nn {
 }
 
 wit_bindgen::generate!({
-    world: "http-reactor",
+    world: "reactor",
     path: "wit",
     pub_export_macro: true,
 });
diff --git a/wit/secret.wit b/wit/secret.wit
@@ -0,0 +1,16 @@
+interface secret {
+  /// Get the secret associated with the specified `key`
+  /// Returns `ok(none)` if the key does not exist.
+  get: func(key: string) -> result<option<string>, error>;
+
+  /// The set of errors which may be raised by functions in this interface
+  variant error {
+    /// The requesting component does not have access to the specified key
+    /// (which may or may not exist).
+    access-denied,
+    /// Decryption error.
+    decrypt-error,
+    /// Some implementation-specific error has occurred (e.g. I/O)
+    other(string)
+  }
+}
diff --git a/wit/world.wit b/wit/world.wit
@@ -1,10 +1,11 @@
 package gcore:fastedge;
 
-world http-reactor {
+world reactor {
     import http;
     import http-client;
 
     import dictionary;
+    import secret;
 
     export http-handler;
 }

Original file line number	Diff line number	Diff line change
`@@ -1,2 +1,2 @@`
`1`	`1`	`[build]`
`2`		`-target = "wasm32-wasi"`
	`2`	`+target = "wasm32-wasip1"`