feat(terminal): Rewrite GJS plugin in Rust
This commit is contained in:
parent
3869e35a80
commit
2aa86e4ff7
8 changed files with 171 additions and 114 deletions
2
Makefile
2
Makefile
|
|
@ -87,7 +87,7 @@ install:
|
|||
ln -sf $(BIN) $(PLUGIN_DIR)/pulse/pulse
|
||||
|
||||
# Terminal plugin
|
||||
install -Dm0755 plugins/src/terminal/terminal.js $(PLUGIN_DIR)/terminal
|
||||
ln -sf $(BIN) $(PLUGIN_DIR)/terminal/terminal
|
||||
|
||||
# Scripts
|
||||
mkdir -p $(SCRIPTS_DIR)
|
||||
|
|
|
|||
|
|
@ -23,6 +23,7 @@ fn main() {
|
|||
"pop-shell" => block_on(plugins::pop_shell::main()),
|
||||
"pulse" => block_on(plugins::pulse::main()),
|
||||
"scripts" => block_on(plugins::scripts::main()),
|
||||
"terminal" => block_on(plugins::terminal::main()),
|
||||
"web" => block_on(plugins::web::main()),
|
||||
unknown => {
|
||||
eprintln!("unknown cmd: {}", unknown);
|
||||
|
|
|
|||
1
debian/pop-launcher.links
vendored
1
debian/pop-launcher.links
vendored
|
|
@ -5,4 +5,5 @@
|
|||
/usr/bin/pop-launcher /usr/lib/pop-launcher/plugins/pop_shell/pop-shell
|
||||
/usr/bin/pop-launcher /usr/lib/pop-launcher/plugins/pulse/pulse
|
||||
/usr/bin/pop-launcher /usr/lib/pop-launcher/plugins/scripts/scripts
|
||||
/usr/bin/pop-launcher /usr/lib/pop-launcher/plugins/terminal/terminal
|
||||
/usr/bin/pop-launcher /usr/lib/pop-launcher/plugins/web/web
|
||||
|
|
|
|||
|
|
@ -5,6 +5,7 @@ pub mod find;
|
|||
pub mod pop_shell;
|
||||
pub mod pulse;
|
||||
pub mod scripts;
|
||||
pub mod terminal;
|
||||
pub mod web;
|
||||
|
||||
use futures_lite::{AsyncWrite, AsyncWriteExt};
|
||||
|
|
|
|||
44
plugins/src/recent/mod.rs
Normal file
44
plugins/src/recent/mod.rs
Normal file
|
|
@ -0,0 +1,44 @@
|
|||
use futures_lite::prelude::*;
|
||||
use pop_launcher::*;
|
||||
use smol::Unblock;
|
||||
use std::io;
|
||||
|
||||
pub struct App {
|
||||
out: Unblock<io::Stdout>,
|
||||
}
|
||||
|
||||
impl Default for App {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
out: async_stdout(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn main() {
|
||||
let mut requests = json_input_stream(async_stdin());
|
||||
|
||||
let mut app = App::default();
|
||||
|
||||
while let Some(result) = requests.next().await {
|
||||
match result {
|
||||
Ok(request) => match request {
|
||||
Request::Activate(id) => app.activate(id).await,
|
||||
Request::Search(query) => app.search(query).await,
|
||||
Request::Exit => break,
|
||||
_ => (),
|
||||
},
|
||||
Err(why) => {
|
||||
tracing::error!("malformed JSON input: {}", why);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl App {
|
||||
async fn activate(&mut self, id: u32) {}
|
||||
|
||||
async fn search(&mut self, query: String) {
|
||||
|
||||
}
|
||||
}
|
||||
122
plugins/src/terminal/mod.rs
Normal file
122
plugins/src/terminal/mod.rs
Normal file
|
|
@ -0,0 +1,122 @@
|
|||
use futures_lite::prelude::*;
|
||||
use pop_launcher::*;
|
||||
use smol::Unblock;
|
||||
use std::{io, path::PathBuf};
|
||||
|
||||
pub struct App {
|
||||
last_query: Option<String>,
|
||||
out: Unblock<io::Stdout>,
|
||||
shell_only: bool,
|
||||
}
|
||||
|
||||
impl Default for App {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
last_query: None,
|
||||
out: async_stdout(),
|
||||
shell_only: false,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn main() {
|
||||
let mut requests = json_input_stream(async_stdin());
|
||||
|
||||
let mut app = App::default();
|
||||
|
||||
while let Some(result) = requests.next().await {
|
||||
match result {
|
||||
Ok(request) => match request {
|
||||
Request::Activate(id) => app.activate(id).await,
|
||||
Request::Search(query) => app.search(query).await,
|
||||
Request::Exit => break,
|
||||
_ => (),
|
||||
},
|
||||
Err(why) => {
|
||||
tracing::error!("malformed JSON input: {}", why);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl App {
|
||||
async fn activate(&mut self, _id: u32) {
|
||||
let exec = match self.last_query.take() {
|
||||
Some(cmd) => format!("{}; echo \"Press Enter to exit\"; read t", cmd),
|
||||
None => return,
|
||||
};
|
||||
|
||||
use fork::{daemon, Fork};
|
||||
|
||||
crate::send(&mut self.out, PluginResponse::Close).await;
|
||||
|
||||
if let Ok(Fork::Child) = daemon(true, true) {
|
||||
use std::os::unix::process::CommandExt;
|
||||
use std::process::Command;
|
||||
|
||||
let mut cmd;
|
||||
|
||||
if self.shell_only {
|
||||
cmd = Command::new("sh");
|
||||
cmd.args(&["-c", &exec]);
|
||||
} else {
|
||||
let (terminal, arg) = detect_terminal();
|
||||
cmd = Command::new(terminal);
|
||||
cmd.args(&[arg, "sh", "-c", &exec]);
|
||||
}
|
||||
|
||||
let _ = cmd.exec();
|
||||
std::process::exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
async fn search(&mut self, query: String) {
|
||||
self.splice_input(&query).await;
|
||||
crate::send(&mut self.out, PluginResponse::Finished).await;
|
||||
}
|
||||
|
||||
async fn splice_input(&mut self, mut query: &str) {
|
||||
if let Some(q) = query.strip_prefix(':') {
|
||||
self.shell_only = true;
|
||||
query = q.trim();
|
||||
self.last_query = Some(query.to_owned());
|
||||
} else {
|
||||
self.shell_only = false;
|
||||
|
||||
let query = if let Some(query) = query.strip_prefix("t:") {
|
||||
query.trim()
|
||||
} else if let Some(pos) = query.find(' ') {
|
||||
query[pos + 1..].trim()
|
||||
} else {
|
||||
return;
|
||||
};
|
||||
|
||||
self.last_query = Some(query.to_owned());
|
||||
}
|
||||
|
||||
crate::send(
|
||||
&mut self.out,
|
||||
PluginResponse::Append(PluginSearchResult {
|
||||
id: 0,
|
||||
name: query.to_owned(),
|
||||
description: String::from("run command in terminal"),
|
||||
..Default::default()
|
||||
}),
|
||||
)
|
||||
.await;
|
||||
}
|
||||
}
|
||||
|
||||
fn detect_terminal() -> (PathBuf, &'static str) {
|
||||
use std::fs::read_link;
|
||||
|
||||
const SYMLINK: &str = "/usr/bin/x-terminal-emulator";
|
||||
|
||||
if let Ok(found) = read_link(SYMLINK) {
|
||||
if let Ok(found) = read_link(&found) {
|
||||
return (found, "-e");
|
||||
}
|
||||
}
|
||||
|
||||
(PathBuf::from("/usr/bin/gnome-terminal"), "--")
|
||||
}
|
||||
|
|
@ -6,6 +6,6 @@
|
|||
help: "run ",
|
||||
isolate: true,
|
||||
),
|
||||
bin: (path: "terminal.js"),
|
||||
bin: (path: "terminal"),
|
||||
icon: Name("utilities-terminal"),
|
||||
)
|
||||
|
|
@ -1,112 +0,0 @@
|
|||
#!/usr/bin/gjs
|
||||
|
||||
const { GLib, Gio } = imports.gi;
|
||||
|
||||
const STDIN = new Gio.DataInputStream({ base_stream: new Gio.UnixInputStream({ fd: 0 }) })
|
||||
const STDOUT = new Gio.DataOutputStream({ base_stream: new Gio.UnixOutputStream({ fd: 1 }) })
|
||||
|
||||
class App {
|
||||
constructor() {
|
||||
this.last_query = ""
|
||||
this.shell_only = false
|
||||
}
|
||||
|
||||
/** @param {string} input */
|
||||
query(input) {
|
||||
if (input.startsWith(':')) {
|
||||
this.shell_only = true
|
||||
this.last_query = input.substr(1).trim()
|
||||
} else {
|
||||
this.shell_only = false
|
||||
this.last_query = input.startsWith('t:')
|
||||
? input.substr(2).trim()
|
||||
: input.substr(input.indexOf(" ") + 1).trim()
|
||||
}
|
||||
|
||||
this.send({ "Append": {
|
||||
id: 0,
|
||||
name: this.last_query,
|
||||
description: "run command in terminal"
|
||||
}})
|
||||
|
||||
this.send("Finished")
|
||||
}
|
||||
|
||||
/** @param {number} _id */
|
||||
submit(_id) {
|
||||
try {
|
||||
let runner
|
||||
if (this.shell_only) {
|
||||
runner = ""
|
||||
} else {
|
||||
let path = GLib.find_program_in_path('x-terminal-emulator');
|
||||
let [terminal, splitter] = path ? [path, "-e"] : ["gnome-terminal", "--"];
|
||||
runner = `${terminal} ${splitter} `
|
||||
}
|
||||
|
||||
GLib.spawn_command_line_async(`${runner}sh -c '${this.last_query}; echo "Press to exit"; read t'`);
|
||||
} catch (e) {
|
||||
log(`command launch error: ${e}`)
|
||||
}
|
||||
|
||||
this.send("Close")
|
||||
}
|
||||
|
||||
/** @param {Object} object */
|
||||
send(object) {
|
||||
try {
|
||||
STDOUT.write_bytes(new GLib.Bytes(JSON.stringify(object) + "\n"), null)
|
||||
} catch (e) {
|
||||
log(`failed to send response to Pop Shell: ${e}`)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function main() {
|
||||
/** @type {null | ByteArray} */
|
||||
let input_array
|
||||
|
||||
/** @type {string} */
|
||||
let input_str
|
||||
|
||||
/** @type {null | LauncherRequest} */
|
||||
let event
|
||||
|
||||
let app = new App()
|
||||
|
||||
mainloop:
|
||||
while (true) {
|
||||
try {
|
||||
[input_array,] = STDIN.read_line(null)
|
||||
} catch (e) {
|
||||
break
|
||||
}
|
||||
|
||||
input_str = imports.byteArray.toString(input_array)
|
||||
if ((event = parse_event(input_str)) !== null) {
|
||||
if ("Search" in event) {
|
||||
app.query(event.Search)
|
||||
} else if ("Activate" in event) {
|
||||
app.submit(event.Activate)
|
||||
} else if ("Exit" === event) {
|
||||
break mainloop
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Parses an IPC event received from STDIN
|
||||
* @param {string} input
|
||||
* @returns {null | LauncherRequest}
|
||||
*/
|
||||
function parse_event(input) {
|
||||
try {
|
||||
return JSON.parse(input)
|
||||
} catch (e) {
|
||||
log(`Input not valid JSON`)
|
||||
return null
|
||||
}
|
||||
}
|
||||
|
||||
main()
|
||||
Loading…
Add table
Add a link
Reference in a new issue