feat(gui): add input parameter handling for GUI mode

uni-helper · Jan 8, 2025 · 4bb9833 · 4bb9833
1 parent 504d3c0
commit 4bb9833
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/packages/core/src/command/actions/gui.ts b/packages/core/src/command/actions/gui.ts
@@ -25,8 +25,7 @@ export function actionGuiCLI() {
   })
 
   const [command, ..._args] = fullCustomCommand.split(' ')
-  const { error, stdout } = sync(command, [..._args], {
-    input,
+  const { error, stdout } = sync(command, [..._args, '--input', input], {
     stdio: 'pipe',
   })
 
@@ -36,6 +35,7 @@ export function actionGuiCLI() {
   let data: any
   if (stdout.length > 0) {
     const data_string = stdout.toString()
+    console.log(data_string)
     try {
       const _data = JSON.parse(data_string)
       if (_data.useTemplate) {

diff --git a/packages/gui/src/lib.rs b/packages/gui/src/lib.rs
@@ -29,7 +29,12 @@ pub fn create_webview() -> Result<()> {
   let current_dir: PathBuf = env::current_dir().expect("Unable to get current working directory");
   let mut input = String::new();
 
-  io::stdin().read_to_string(&mut input).unwrap();
+  let args: Vec<String> = env::args().collect();
+  if let Some(input_value) = args.iter().position(|x| x == "--input").and_then(|i| args.get(i + 1)) {
+      input = input_value.to_string();
+  } else {
+      println!("No input provided.");
+  }
 
   let current_dir_str = current_dir.to_str().unwrap_or("");
   let escaped_current_dir_str = current_dir_str.replace("\\", "\\\\");