pub fn cron(import_statement: &str, rust_function_name: &str, is_local: bool) -> String {
if is_local {
format!(
"{import_statement}
use kinetics::tools::config::Config as KineticsConfig;
#[tokio::main]\n\
async fn main() -> Result<(), tower::BoxError> {{\n\
let user_function = {rust_function_name};
let config = aws_config::load_defaults(aws_config::BehaviorVersion::latest()).await;
let kinetics_config = KineticsConfig::new(&config, None).await?;
let mut secrets = std::collections::HashMap::new();
for (k, v) in std::env::vars() {{
if k.starts_with(\"KINETICS_SECRET_\") {{
let key = k.replace(\"KINETICS_SECRET_\", \"\");
secrets.insert(key, v);
}}
}}
user_function(&secrets, &kinetics_config).await?;
Ok(())
}}\n\n"
)
} else {
format!(
"{import_statement}
use kinetics::tools::config::Config as KineticsConfig;
use lambda_runtime::{{LambdaEvent, Error, run, service_fn}};\n\
use aws_lambda_events::eventbridge::EventBridgeEvent;\n\
#[tokio::main]\n\
async fn main() -> Result<(), Error> {{\n\
let user_function = {rust_function_name};
let config = aws_config::load_defaults(aws_config::BehaviorVersion::latest()).await;
println!(\"Provisioning secrets\");
let secrets_client = aws_sdk_ssm::Client::new(&config);
let secrets_names_env = \"KINETICS_SECRETS_NAMES\";
let mut secrets = std::collections::HashMap::new();
for secret_name in std::env::var(secrets_names_env)?
.split(\",\")
.map(|s| s.trim().to_string())
.filter(|s| !s.is_empty())
{{
let desc = secrets_client
.get_parameter()
.name(secret_name.clone())
.with_decryption(true)
.send()
.await?;
let result = desc.parameter.unwrap();
let tags = secrets_client
.list_tags_for_resource()
.resource_type(aws_sdk_ssm::types::ResourceTypeForTagging::Parameter)
.resource_id(secret_name.clone())
.send()
.await?
.tag_list
.unwrap_or_default();
let name = match tags.iter().find(|t| t.key() == \"original_name\") {{
Some(tag) => tag.value(),
None => &secret_name.clone(),
}};
let secret_value = result.value().unwrap();
secrets.insert(name.into(), secret_value.to_string());
}}
let kinetics_config = KineticsConfig::new(&config, None).await?;
println!(\"Serving requests\");
run(service_fn(|_event: LambdaEvent<EventBridgeEvent<serde_json::Value>>| async {{
match user_function(&secrets, &kinetics_config).await {{
Ok(()) => Ok(()),
Err(err) => {{
eprintln!(\"Error occurred while handling request: {{:?}}\", err);
Err(err)
}}
}}
}}))
.await
}}\n\n"
)
}
}