add code, iac, sbom, code and log4shell functions

fluent-ci-templates · Aug 6, 2024 · 73336d4 · 73336d4
1 parent f097562
commit 73336d4
Show file tree

Hide file tree

Showing 4 changed files with 93 additions and 3 deletions.
diff --git a/.github/workflows/example.yml b/.github/workflows/example.yml
@@ -17,7 +17,7 @@ jobs:
           pipeline: .
           args: |
             setup
-            iac_test --severity-threshold=medium
+            iac test --severity-threshold=medium
           working-directory: example
         env:
           SNYK_TOKEN: ${{ secrets.SNYK_TOKEN }}

diff --git a/plugin/Cargo.lock b/plugin/Cargo.lock
diff --git a/plugin/Cargo.toml b/plugin/Cargo.toml
@@ -1,7 +1,7 @@
 [package]
 edition = "2021"
 name = "snyk"
-version = "0.6.0"
+version = "0.6.1"
 
 [lib]
 crate-type = ["cdylib"]

diff --git a/plugin/src/lib.rs b/plugin/src/lib.rs
@@ -35,6 +35,96 @@ pub fn test(args: String) -> FnResult<String> {
     Ok(stdout)
 }
 
+#[plugin_fn]
+pub fn iac(args: String) -> FnResult<String> {
+    let stdout = dag()
+        .pipeline("iac test")?
+        .pkgx()?
+        .with_exec(vec![
+            "pkgx",
+            "+nodejs.org",
+            "+bun.sh",
+            "bunx",
+            "snyk",
+            "iac",
+            &args,
+        ])?
+        .stdout()?;
+    Ok(stdout)
+}
+
+#[plugin_fn]
+pub fn code(args: String) -> FnResult<String> {
+    let stdout = dag()
+        .pipeline("iac test")?
+        .pkgx()?
+        .with_exec(vec![
+            "pkgx",
+            "+nodejs.org",
+            "+bun.sh",
+            "bunx",
+            "snyk",
+            "code",
+            &args,
+        ])?
+        .stdout()?;
+    Ok(stdout)
+}
+
+#[plugin_fn]
+pub fn sbom(args: String) -> FnResult<String> {
+    let stdout = dag()
+        .pipeline("iac test")?
+        .pkgx()?
+        .with_exec(vec![
+            "pkgx",
+            "+nodejs.org",
+            "+bun.sh",
+            "bunx",
+            "snyk",
+            "sbom",
+            &args,
+        ])?
+        .stdout()?;
+    Ok(stdout)
+}
+
+#[plugin_fn]
+pub fn log4shell(args: String) -> FnResult<String> {
+    let stdout = dag()
+        .pipeline("iac test")?
+        .pkgx()?
+        .with_exec(vec![
+            "pkgx",
+            "+nodejs.org",
+            "+bun.sh",
+            "bunx",
+            "snyk",
+            "log4shell",
+            &args,
+        ])?
+        .stdout()?;
+    Ok(stdout)
+}
+
+#[plugin_fn]
+pub fn container(args: String) -> FnResult<String> {
+    let stdout = dag()
+        .pipeline("iac test")?
+        .pkgx()?
+        .with_exec(vec![
+            "pkgx",
+            "+nodejs.org",
+            "+bun.sh",
+            "bunx",
+            "snyk",
+            "container",
+            &args,
+        ])?
+        .stdout()?;
+    Ok(stdout)
+}
+
 #[plugin_fn]
 pub fn iac_test(args: String) -> FnResult<String> {
     let stdout = dag()