Graphite/frontend/src-tauri/src/main.rs

#![cfg_attr(all(not(debug_assertions), target_os = "windows"), windows_subsystem = "windows")]

use graphite_editor::application::Editor;
use graphite_editor::messages::frontend::utility_types::FrontendImageData;
use graphite_editor::messages::prelude::*;

use axum::body::StreamBody;
use axum::extract::Path;
use axum::http;
use axum::response::IntoResponse;
use axum::routing::get;
use axum::Router;
use fern::colors::{Color, ColoredLevelConfig};
use http::{Response, StatusCode};
use std::collections::HashMap;
use std::sync::Arc;
use std::sync::Mutex;

static IMAGES: Mutex<Option<HashMap<String, FrontendImageData>>> = Mutex::new(None);
static EDITOR: Mutex<Option<Editor>> = Mutex::new(None);

async fn respond_to(id: Path<String>) -> impl IntoResponse {
	let builder = Response::builder().header("Access-Control-Allow-Origin", "*").status(StatusCode::OK);

	let guard = IMAGES.lock().unwrap();
	let images = guard;
	let image = images.as_ref().unwrap().get(&id.0).unwrap();

	println!("image: {:#?}", image.path);
	let result: Result<Vec<u8>, &str> = Ok((*image.image_data).clone());
	let stream = futures::stream::once(async move { result });
	builder.body(StreamBody::new(stream)).unwrap()
}

#[tokio::main]
async fn main() {
	println!("Starting server...");

	let colors = ColoredLevelConfig::new().debug(Color::Magenta).info(Color::Green).error(Color::Red);

	fern::Dispatch::new()
		.chain(std::io::stdout())
		.level(log::LevelFilter::Trace)
		.format(move |out, message, record| {
			out.finish(format_args!(
				"[{}]{} {}",
				// This will color the log level only, not the whole line. Just a touch.
				colors.color(record.level()),
				chrono::Utc::now().format("[%Y-%m-%d %H:%M:%S]"),
				message
			))
		})
		.apply()
		.unwrap();

	*(IMAGES.lock().unwrap()) = Some(HashMap::new());
	graphite_editor::application::set_uuid_seed(0);
	*(EDITOR.lock().unwrap()) = Some(Editor::new());
	let app = Router::new().route("/", get(|| async { "Hello, World!" })).route("/image/:id", get(respond_to));

	// run it with hyper on localhost:3000
	tauri::async_runtime::spawn(async {
		axum::Server::bind(&"0.0.0.0:3001".parse().unwrap()).serve(app.into_make_service()).await.unwrap();
	});

	tauri::Builder::default()
		.invoke_handler(tauri::generate_handler![set_random_seed, handle_message])
		.setup(|_app| {
			//use tauri::Manager;
			//_app.get_window("main").unwrap().open_devtools();
			Ok(())
		})
		.run(tauri::generate_context!())
		.expect("error while running tauri application");
}
#[tauri::command]
fn set_random_seed(seed: f64) {
	let seed = seed as u64;
	graphite_editor::application::set_uuid_seed(seed);
}

#[tauri::command]
fn handle_message(message: String) -> String {
	let Ok(message) = ron::from_str::<graphite_editor::messages::message::Message>(&message) else {
		panic!("Error parsing message: {}", message)
	};
	let mut guard = EDITOR.lock().unwrap();
	let editor = (*guard).as_mut().unwrap();
	let responses = editor.handle_message(message);

	// Sends a FrontendMessage to JavaScript
	fn send_frontend_message_to_js(message: FrontendMessage) -> FrontendMessage {
		// Special case for update image data to avoid serialization times.
		if let FrontendMessage::UpdateImageData { document_id, image_data } = message {
			let mut guard = IMAGES.lock().unwrap();
			let images = (*guard).as_mut().unwrap();
			let mut stub_data = Vec::with_capacity(image_data.len());
			for image in image_data {
				let path = image.path.clone();
				let mime = image.mime.clone();
				images.insert(format!("{:?}_{}", &image.path, document_id), image);
				stub_data.push(FrontendImageData {
					path,
					mime,
					image_data: Arc::new(Vec::new()),
				});
			}
			FrontendMessage::UpdateImageData { document_id, image_data: stub_data }
		} else {
			message
		}
	}

	for response in &responses {
		let serialized = ron::to_string(&send_frontend_message_to_js(response.clone())).unwrap();
		if let Err(error) = ron::from_str::<FrontendMessage>(&serialized) {
			log::error!("Error deserializing message: {}", error);
		}
	}

	// Process any `FrontendMessage` responses resulting from the backend processing the dispatched message
	let result: Vec<_> = responses.into_iter().map(send_frontend_message_to_js).collect();

	ron::to_string(&result).expect("Failed to serialize FrontendMessage")
}