53 lines
1.5 KiB
Rust
53 lines
1.5 KiB
Rust
|
use serde_json::Value;
|
||
|
|
|
||
|
|
use crate::types::{CreateInstanceRequest, CreateResponse, InstanceInfo};
|
||
|
|
|
||
|
|
/// Create one vast.ai instance from a fully generic payload.
|
||
|
|
pub async fn create_instance(
|
||
|
|
client: &reqwest::Client,
|
||
|
|
base_url: &str,
|
||
|
|
api_key: &str,
|
||
|
|
req: &CreateInstanceRequest,
|
||
|
|
) -> Result<InstanceInfo, String> {
|
||
|
|
let url = format!("{base_url}/api/v0/asks/{}/", req.offer_id);
|
||
|
|
let env = req
|
||
|
|
.env
|
||
|
|
.iter()
|
||
|
|
.map(|(k, v)| (k.clone(), Value::String(v.clone())))
|
||
|
|
.collect::<serde_json::Map<String, Value>>();
|
||
|
|
|
||
|
|
let mut body = serde_json::json!({
|
||
|
|
"image": req.image,
|
||
|
|
"env": env,
|
||
|
|
"disk": req.disk_gb,
|
||
|
|
});
|
||
|
|
if let Some(onstart) = req.onstart.as_deref() {
|
||
|
|
body["onstart"] = Value::String(onstart.to_string());
|
||
|
|
}
|
||
|
|
if let Some(label) = req.label.as_deref() {
|
||
|
|
body["label"] = Value::String(label.to_string());
|
||
|
|
}
|
||
|
|
|
||
|
|
let resp = client
|
||
|
|
.put(&url)
|
||
|
|
.header("Authorization", format!("Bearer {api_key}"))
|
||
|
|
.json(&body)
|
||
|
|
.send()
|
||
|
|
.await
|
||
|
|
.map_err(|e| format!("create_instance request failed: {e}"))?;
|
||
|
|
|
||
|
|
if !resp.status().is_success() {
|
||
|
|
let status = resp.status();
|
||
|
|
let body = resp.text().await.unwrap_or_default();
|
||
|
|
return Err(format!("create_instance HTTP {status}: {body}"));
|
||
|
|
}
|
||
|
|
|
||
|
|
let parsed: CreateResponse = resp
|
||
|
|
.json()
|
||
|
|
.await
|
||
|
|
.map_err(|e| format!("create_instance parse failed: {e}"))?;
|
||
|
|
Ok(InstanceInfo {
|
||
|
|
contract_id: parsed.new_contract,
|
||
|
|
})
|
||
|
|
}
|