1 files changed, 157 insertions, 0 deletions
diff --git a/policy/_policies/src/lib.rs.tmpl b/policy/_policies/src/lib.rs.tmpl
new file mode 100644
index 0000000..db0dc0f
--- /dev/null
+++ b/policy/_policies/src/lib.rs.tmpl
@@ -0,0 +1,157 @@
+// Auto generated dependencies
+// If you find issues with the dependencies, please
+//   1. Delete all code after this comment
+//   2. Clear the auto generated part in `policy/_policies/Cargo.toml`
+//   3. Run `cargo run --bin butckrepo-refresh` in the Butchunker root directory
+pub mod error;
+pub mod stream_read;
+
+use error::ChunkFailed;
+use std::{collections::HashMap, path::Path};
+
+use crate::stream_read::chunk_stream_process;
+
+/// Chunks the specified raw data using the specified chunking policy
+///
+/// # Parameters
+/// - `policy_name`: Chunking policy name, currently supports <<<policy_count>>> policies
+/// - `raw_data`: Raw data byte slice
+/// - `params`: Hashmap of parameters required by the chunking policy
+pub async fn chunk_with(
+    policy_name: &str,
+    raw_data: &[u8],
+    params: &HashMap<&str, &str>,
+) -> Result<Vec<u32>, ChunkFailed> {
+    match policy_name {
+>>>>>>>>>> match_arms
+        _ => Err(ChunkFailed::PolicyNotFound),
+    }
+}
+
+pub async fn chunk_stream_with<F>(
+    policy_name: &str,
+    size: u32,
+    path: &Path,
+    callback: F,
+    params: &HashMap<&str, &str>,
+) -> Result<(), ChunkFailed>
+where
+    F: FnMut(Vec<u8>) -> std::pin::Pin<Box<dyn std::future::Future<Output = Result<(), std::io::Error>> + Send>>,
+{
+    match policy_name {
+>>>>>>>>>> match_arms_stream
+        _ => Err(ChunkFailed::PolicyNotFound),
+    }
+}
+
+pub async fn chunk_stream_display_boundaries(
+    policy_name: &str,
+    size: u32,
+    path: &Path,
+    params: &HashMap<&str, &str>,
+) -> Result<Vec<u32>, ChunkFailed> {
+    match policy_name {
+>>>>>>>>>> match_arms_stream_display
+        _ => Err(ChunkFailed::PolicyNotFound),
+    }
+}
+
+pub fn policies() -> Vec<&'static str> {
+    vec![
+>>>>>>>>>> policy_names
+    ]
+}
+
+pub fn stream_policies() -> Vec<&'static str> {
+    vec![
+>>>>>>>>>> stream_policy_names
+    ]
+}
+
+>>>>>>>>>> exports_simple
+>>>>>>>>>> exports_stream
+>>>>>>>>>> exports_both
+
+@@@ >>> match_arms
+        "<<<crate_name>>>" => Ok(<<<crate_name>>>::chunk(raw_data, params).await),
+@@@ <<<
+
+@@@ >>> match_arms_stream
+        "<<<crate_name>>>" => {
+            let mut stream = <<<stream_struct_id>>>::default();
+            chunk_stream_process(
+                path, &mut stream, size, callback, params,
+                async |current_data, len, stream, params| {
+                    <<<crate_name>>>::chunk_stream(current_data, len, stream, params).await
+                },
+            )
+            .await
+        }
+@@@ <<<
+
+@@@ >>> match_arms_stream_display
+        "<<<crate_name>>>" => {
+            let mut stream = <<<stream_struct_id>>>::default();
+            crate::stream_read::chunk_stream_display_boundaries(
+                path, &mut stream, size, params,
+                async |current_data, len, stream, params| {
+                    <<<crate_name>>>::chunk_stream(current_data, len, stream, params).await
+                },
+            )
+            .await
+        }
+@@@ <<<
+
+@@@ >>> policy_names
+        // <<<name>>>
+        "<<<name>>>",
+@@@ <<<
+
+@@@ >>> stream_policy_names
+        // <<<name>>>
+        "<<<name>>>",
+@@@ <<<
+
+@@@ >>> exports_simple
+pub mod <<<crate_name>>> {
+    use std::collections::HashMap;
+    pub async fn chunk(raw_data: &[u8], params: &HashMap<&str, &str>) -> Vec<u32> {
+        <<<crate_name>>>::<<<matched_func>>>(raw_data, params)<<<has_await>>>
+    }
+}
+@@@ <<<
+
+@@@ >>> exports_stream
+pub mod <<<crate_name>>> {
+    pub use <<<stream_struct_id>>>;
+
+    pub async fn chunk_stream(
+        current_data: &[u8],
+        len: u32,
+        stream: &mut <<<stream_struct_id>>>,
+        params: &std::collections::HashMap<&str, &str>,
+    ) -> Option<u32> {
+        <<<crate_name>>>::<<<matched_func_stream>>>(current_data, len, stream, params)<<<has_await_stream>>>
+    }
+}
+@@@ <<<
+
+@@@ >>> exports_both
+pub mod <<<crate_name>>> {
+    pub use <<<stream_struct_id>>>;
+    use std::collections::HashMap;
+
+    pub async fn chunk(raw_data: &[u8], params: &HashMap<&str, &str>) -> Vec<u32> {
+        <<<crate_name>>>::<<<matched_func>>>(raw_data, params)<<<has_await>>>
+    }
+
+    pub async fn chunk_stream(
+        current_data: &[u8],
+        len: u32,
+        stream: &mut <<<stream_struct_id>>>,
+        params: &HashMap<&str, &str>,
+    ) -> Option<u32> {
+        <<<crate_name>>>::<<<matched_func_stream>>>(current_data, len, stream, params)<<<has_await_stream>>>
+    }
+}
+@@@ <<<