61 lines
2.5 KiB
Rust
61 lines
2.5 KiB
Rust
|
use super::models::*;
|
|||
|
|
use axum::{extract::{Multipart, State}, http::StatusCode, Json};
|
|||
|
|
use std::{path::Path, sync::atomic::Ordering};
|
|||
|
|
use tempfile::NamedTempFile;
|
|||
|
|
use tokio::fs;
|
|||
|
|
use tracing::info;
|
|||
|
|
|
|||
|
|
pub async fn create_print(
|
|||
|
|
State(store): State<JobStore>,
|
|||
|
|
mut multipart: Multipart,
|
|||
|
|
) -> Result<Json<JobView>, (StatusCode, String)> {
|
|||
|
|
info!("📥 Получен запрос на печать");
|
|||
|
|
let mut settings_list: Vec<Option<PrintSettings>> = Vec::new();
|
|||
|
|
let mut files_data: Vec<(String, bytes::Bytes)> = Vec::new();
|
|||
|
|
|
|||
|
|
while let Some(field) = multipart.next_field().await.unwrap_or(None) {
|
|||
|
|
let field_name = field.name().unwrap_or("").to_string();
|
|||
|
|
if field_name == "settings" {
|
|||
|
|
let parsed = match field.text().await {
|
|||
|
|
Ok(text) => serde_json::from_str::<PrintSettings>(&text).ok(),
|
|||
|
|
Err(_) => None,
|
|||
|
|
};
|
|||
|
|
settings_list.push(parsed);
|
|||
|
|
} else if field_name == "files" {
|
|||
|
|
let file_name = field.file_name().unwrap_or("unknown").to_string();
|
|||
|
|
if let Ok(data) = field.bytes().await { files_data.push((file_name, data)); }
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if files_data.is_empty() { return Err((StatusCode::BAD_REQUEST, "Нет файлов".into())); }
|
|||
|
|
|
|||
|
|
let mut job_files: Vec<JobFile> = Vec::new();
|
|||
|
|
for (i, (file_name, data)) in files_data.into_iter().enumerate() {
|
|||
|
|
let ext = Path::new(&file_name).extension().and_then(|e| e.to_str()).unwrap_or("").to_lowercase();
|
|||
|
|
let temp = if ext.is_empty() { NamedTempFile::new() } else { NamedTempFile::with_suffix(&format!(".{}", ext)) };
|
|||
|
|
|
|||
|
|
if let Ok(temp) = temp {
|
|||
|
|
let path = temp.into_temp_path();
|
|||
|
|
if fs::write(&*path, &data).await.is_ok() {
|
|||
|
|
let settings = settings_list.get(i).cloned().flatten();
|
|||
|
|
job_files.push(JobFile {
|
|||
|
|
name: file_name, path, settings,
|
|||
|
|
status: FileStatus::Queued, error: None,
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if job_files.is_empty() { return Err((StatusCode::BAD_REQUEST, "Не удалось сохранить файлы".into())); }
|
|||
|
|
|
|||
|
|
let job_id = format!("job-{}", JOB_COUNTER.fetch_add(1, Ordering::SeqCst));
|
|||
|
|
let job = Job {
|
|||
|
|
id: job_id.clone(), files: job_files, current: None,
|
|||
|
|
phase: JobPhase::WaitingStart, cups_id: None, duplex_first_pass_done: false,
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
let view = job.view();
|
|||
|
|
store.lock().unwrap().insert(job_id.clone(), job);
|
|||
|
|
Ok(Json(view))
|
|||
|
|
}
|