codemp/src/ffi/java/workspace.rs

use jni::{objects::{JClass, JObject, JString, JValueGen}, sys::{jlong, jobject, jobjectArray, jstring}, JNIEnv};
use crate::Workspace;

use super::{JExceptable, RT};

/// Gets the workspace id.
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_get_1workspace_1id<'local>(
	mut env: JNIEnv<'local>,
	_class: JClass<'local>,
	self_ptr: jlong
) -> jstring {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	env.new_string(workspace.id()).jexcept(&mut env).as_raw()
}

/// Gets a cursor controller by name and returns a pointer to it.
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_get_1cursor<'local>(
	mut env: JNIEnv<'local>,
	_class: JClass<'local>,
	self_ptr: jlong
) -> jobject {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	env.find_class("mp/code/CursorController").and_then(|class|
		env.new_object(class, "(J)V", &[JValueGen::Long(Box::into_raw(Box::new(workspace.cursor())) as jlong)])
	).jexcept(&mut env).as_raw()
}

/// Gets a buffer controller by name and returns a pointer to it.
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_get_1buffer<'local>(
	mut env: JNIEnv<'local>,
	_class: JClass<'local>,
	self_ptr: jlong,
	input: JString<'local>
) -> jobject {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	let path = unsafe { env.get_string_unchecked(&input) }
		.map(|path| path.to_string_lossy().to_string())
		.jexcept(&mut env);
	
	workspace.buffer_by_name(&path).map(|buf| {
		env.find_class("mp/code/BufferController").and_then(|class|
			env.new_object(class, "(J)V", &[JValueGen::Long(Box::into_raw(Box::new(buf)) as jlong)])
		).jexcept(&mut env)
	}).unwrap_or_default().as_raw()
}

/// Creates a new buffer.
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_create_1buffer<'local>(
	mut env: JNIEnv,
	_class: JClass<'local>,
	self_ptr: jlong,
	input: JString<'local>
) {
	let ws = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	let path = unsafe { env.get_string_unchecked(&input) }
		.map(|path| path.to_string_lossy().to_string())
		.jexcept(&mut env);
	RT.block_on(ws.create(&path))
		.jexcept(&mut env);
}

/// Gets the filetree.
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_get_1file_1tree(
	mut env: JNIEnv,
	_class: JClass,
	self_ptr: jlong,
) -> jobjectArray {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	let file_tree = workspace.filetree();
	env.find_class("java/lang/String")
		.and_then(|class| env.new_object_array(file_tree.len() as i32, class, JObject::null()))
		.map(|arr| {
			for (idx, path) in file_tree.iter().enumerate() {
				env.new_string(path)
					.and_then(|path| env.set_object_array_element(&arr, idx as i32, path))
					.jexcept(&mut env)
			}
			arr
		}).jexcept(&mut env).as_raw()
}

/// Attaches to a buffer and returns a pointer to its [crate::buffer::Controller].
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_attach_1to_1buffer<'local>(
	mut env: JNIEnv,
	_class: JClass<'local>,
	self_ptr: jlong,
	input: JString<'local>
) -> jobject {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	let path = unsafe { env.get_string_unchecked(&input) }
		.map(|path| path.to_string_lossy().to_string())
		.jexcept(&mut env);
	RT.block_on(workspace.attach(&path))
		.map(|buffer| Box::into_raw(Box::new(buffer)) as jlong)
		.map(|ptr| {
			env.find_class("mp/code/BufferController")
				.and_then(|class| env.new_object(class, "(J)V", &[JValueGen::Long(ptr)]))
				.jexcept(&mut env)
		}).jexcept(&mut env).as_raw()
}

#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_detach_1from_1buffer<'local>(
	mut env: JNIEnv,
	_class: JClass<'local>,
	self_ptr: jlong,
	input: JString<'local>
) -> jobject {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	let path = unsafe { env.get_string_unchecked(&input) }
		.map(|path| path.to_string_lossy().to_string())
		.jexcept(&mut env);
	let name = match workspace.detach(&path) {
		crate::workspace::worker::DetachResult::NotAttached => "NOT_ATTACHED",
		crate::workspace::worker::DetachResult::Detaching => "DETACHED",
		crate::workspace::worker::DetachResult::AlreadyDetached => "ALREADY_DETACHED"
	};

	env.find_class("mp/code/data/DetachResult")
		.and_then(|class| env.get_static_field(class, name, "Lmp/code/data/DetachResult;"))
		.and_then(|res| res.l())
		.jexcept(&mut env)
		.as_raw()
}

/// Updates the local buffer list.
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_fetch_1buffers(
	mut env: JNIEnv,
	_class: JClass,
	self_ptr: jlong,
) {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	RT.block_on(workspace.fetch_buffers()).jexcept(&mut env);
}

/// Updates the local user list.
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_fetch_1users(
	mut env: JNIEnv,
	_class: JClass,
	self_ptr: jlong,
) {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	RT.block_on(workspace.fetch_users()).jexcept(&mut env);
}

/// Lists users attached to a buffer.
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_list_1buffer_1users<'local>(
	mut env: JNIEnv,
	_class: JClass<'local>,
	self_ptr: jlong,
	input: JString<'local>,
) -> jobjectArray {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	let buffer = unsafe { env.get_string_unchecked(&input) }
		.map(|buffer| buffer.to_string_lossy().to_string())
		.jexcept(&mut env);
	let users = RT.block_on(workspace.list_buffer_users(&buffer))
		.jexcept(&mut env);

	env.find_class("java/lang/String")
		.and_then(|class| env.new_object_array(users.len() as i32, class, JObject::null()))
		.map(|arr| {
			for (idx, user) in users.iter().enumerate() {
				env.new_string(&user.id)
					.and_then(|id| env.set_object_array_element(&arr, idx as i32, id))
					.jexcept(&mut env);
			}
			arr
		}).jexcept(&mut env).as_raw()
}

/// Deletes a buffer.
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_delete_1buffer<'local>(
	mut env: JNIEnv,
	_class: JClass<'local>,
	self_ptr: jlong,
	input: JString<'local>,
) {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	let buffer = unsafe { env.get_string_unchecked(&input) }
		.map(|buffer| buffer.to_string_lossy().to_string())
		.jexcept(&mut env);
	RT.block_on(workspace.delete(&buffer)).jexcept(&mut env);
}

/// Receives a workspace event if present.
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_event(
	mut env: JNIEnv,
	_class: JClass,
	self_ptr: jlong
) -> jobject {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	RT.block_on(workspace.event())
		.map(|event| {
			let (name, arg) = match event {
				crate::api::Event::FileTreeUpdated => ("FILE_TREE_UPDATED", None),
				crate::api::Event::UserJoin(arg) => ("USER_JOIN", Some(arg)),
				crate::api::Event::UserLeave(arg) => ("USER_LEAVE", Some(arg)),
			};
			let event_type = env.find_class("mp/code/Workspace$Event$Type")
				.and_then(|class| env.get_static_field(class, name, "Lmp/code/Workspace/Event/Type;"))
				.and_then(|f| f.l())
				.jexcept(&mut env);
			let arg = arg.map(|s| env.new_string(s).jexcept(&mut env))
				.unwrap_or_default();

			env.find_class("mp/code/Workspace$Event").and_then(|class|
				env.new_object(
					class,
					"(Lmp/code/Workspace/Event/Type;Ljava/lang/String;)V",
					&[
						JValueGen::Object(&event_type),
						JValueGen::Object(&arg)
					]
				)
			).jexcept(&mut env)
		}).jexcept(&mut env).as_raw()
}

/// Polls a list of buffers, returning the first ready one.
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_select_1buffer(
	mut env: JNIEnv,
	_class: JClass,
	self_ptr: jlong,
	timeout: jlong
) -> jobject {
	let workspace = unsafe { Box::leak(Box::from_raw(self_ptr as *mut Workspace)) };
	let buffers = workspace.buffer_list();
	let mut controllers = Vec::default();
	for buffer in buffers {
		if let Some(controller) = workspace.buffer_by_name(&buffer) {
			controllers.push(controller);
		}
	}

	RT.block_on(crate::ext::select_buffer(
		&controllers,
		Some(std::time::Duration::from_millis(timeout as u64)),
		&RT,
	)).jexcept(&mut env)
		.map(|buf| {
			env.find_class("mp/code/BufferController").and_then(|class|
				env.new_object(class, "(J)V", &[JValueGen::Long(Box::into_raw(Box::new(buf)) as jlong)])
			).jexcept(&mut env)
		}).unwrap_or_default().as_raw()
}

/// Called by the Java GC to drop a [Workspace].
#[no_mangle]
pub extern "system" fn Java_mp_code_Workspace_free(_env: JNIEnv, _class: JClass, input: jlong) {
	let _ = unsafe { Box::from_raw(input as *mut Workspace) };
}