lune-org/lune
1
1
Fork 0
mirror of https://github.com/lune-org/lune.git synced 2024-12-12 13:00:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Potential fix for process.spawn yield issue

Browse source
This commit is contained in:
Filip Tibell 2023-01-24 15:36:09 -05:00
parent 7eff6e4555
commit d3046194bc
No known key found for this signature in database
1 changed files with 23 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
src/lib/globals/process.rs
View file

@ -1,8 +1,12 @@
use std::{env, process::Stdio, sync::Weak};
use std::{
env,
process::{Command, Stdio},
sync::Weak,
};
use mlua::prelude::*;
use os_str_bytes::RawOsString;
use smol::{channel::Sender, process::Command, LocalExecutor};
use smol::{channel::Sender, unblock};
use crate::{utils::table::TableBuilder, LuneMessage};
@ -116,30 +120,24 @@ async fn process_spawn(
lua: &Lua,
(program, args): (String, Option<Vec<String>>),
) -> LuaResult<LuaTable> {
let exec = lua
.app_data_ref::<Weak<LocalExecutor>>()
.unwrap()
.upgrade()
.unwrap();
// Create and spawn a child process in a new task to prevent
// Create and spawn a **blocking** child process to prevent
// issues with yielding across the metamethod/c-call boundary
let output = exec
.spawn(async move {
let mut cmd = Command::new(program);
if let Some(args) = args {
cmd.args(args);
}
let child = cmd
.current_dir(env::current_dir().map_err(LuaError::external)?)
.stdin(Stdio::null())
.stdout(Stdio::piped())
.stderr(Stdio::piped())
.spawn()
.map_err(LuaError::external)?;
let output = child.output().await.map_err(LuaError::external)?;
Ok::<_, LuaError>(output)
})
.await?;
let pwd = env::current_dir()?;
let output = unblock(move || {
let mut cmd = Command::new(program);
if let Some(args) = args {
cmd.args(args);
}
let child = cmd
.current_dir(pwd)
.stdin(Stdio::null())
.stdout(Stdio::piped())
.stderr(Stdio::piped())
.spawn()?;
child.wait_with_output()
})
.await
.map_err(LuaError::external)?;
// NOTE: If an exit code was not given by the child process,
// we default to 1 if it yielded any error output, otherwise 0
let code = output