use crate::DriverCallback;
use crate::DriverContext;
use crate::types::{Driver, DriverParameter};
use crate::{DriverError, DriverResult};
use serde_json::{Value, json};
use sha2::{Digest, Sha256};
use std::collections::HashMap;
use tracing::{debug, info, warn};
#[derive(Debug)]
pub struct HashSha256TextDriver;
#[async_trait::async_trait]
impl Driver for HashSha256TextDriver {
fn name(&self) -> &str {
"hash_sha256_text"
}
fn description(&self) -> &str {
"Calculate SHA256 hash of a text string"
}
fn usage_hint(&self) -> &str {
"Use this skill when you need to compute SHA256 hash for a text string. For file hashing, use file/hash_sha256."
}
fn parameters(&self) -> Vec<DriverParameter> {
return vec![DriverParameter {
name: "input".to_string(),
param_type: "string".to_string(),
description: "Input string to hash".to_string(),
required: true,
default: None,
example: Some(Value::String("Hello World".to_string())),
enum_values: None,
}];
}
fn example_call(&self) -> DriverResult<Value> {
return Ok(json!({
"action": "hash_sha256_text",
"parameters": {
"input": "Hello World"
}
}));
}
fn example_output(&self) -> String {
return "SHA256: a591a6d40bf420404a011733cfb7b190d62c65bf0bcda32b57b277d9ad9f146e".to_string();
}
fn category(&self) -> crate::DriverCategory {
return crate::DriverCategory::Cryptography;
}
async fn execute(
&self,
parameters: &HashMap<String, Value>,
callback: Option<&dyn DriverCallback>,
context: Option<&DriverContext>,
) -> DriverResult<String> {
debug!("Executing hash_sha256_text driver");
let input = parameters.get("input").and_then(|v| v.as_str()).ok_or_else(|| {
debug!("Missing 'input' parameter");
DriverError::missing_parameter("input")
})?;
debug!("Input length: {} characters", input.len());
let mut hasher = Sha256::new();
hasher.update(input.as_bytes());
let result = hasher.finalize();
info!("SHA256 hash computed successfully");
return Ok(format!("SHA256: {}", hex::encode(result)));
}
fn validate(&self, parameters: &HashMap<String, Value>) -> DriverResult<()> {
debug!("Validating hash_sha256_text parameters");
if parameters.get("input").is_none() {
return Err(DriverError::missing_parameter("input"));
}
debug!("Parameter validation passed");
return Ok(());
}
}