Initial commit

This commit is contained in:
maru committed 2026-10-15 16:22:00 +03:00
commit 296b987e3b
14 files changed
+4035

No files matched your search

+150
View File
@@ -0,0 +1,150 @@
use std::{
collections::HashMap,
ffi::OsStr,
fs::File,
io::{BufReader, Read},
path::{Component, Path, PathBuf},
};
use anyhow::{Context, Result};
use backhand::{FilesystemReader, InnerNode, SquashfsFileReader};
use crate::desktop::DesktopEntry;
pub struct Icon {
pub data: Vec<u8>,
pub ext: &'static str,
}
pub struct Embedded {
pub desktop: Option<DesktopEntry>,
pub icon: Option<Icon>,
}
pub fn read(path: &Path, offset: u64) -> Result<Embedded> {
let file = BufReader::new(File::open(path)?);
let fs = FilesystemReader::from_reader_with_offset(file, offset).with_context(|| {
format!(
"cannot read the squashfs image embedded in {}",
path.display()
)
})?;
let image = Image::new(&fs);
let desktop = image.root_desktop();
let icon_name = desktop.as_ref().and_then(|d| d.get("Icon"));
let icon = image.icon(icon_name.as_deref());
Ok(Embedded { desktop, icon })
}
struct Image<'a, 'b> {
fs: &'a FilesystemReader<'b>,
nodes: HashMap<PathBuf, &'a InnerNode<SquashfsFileReader>>,
}
impl<'a, 'b> Image<'a, 'b> {
fn new(fs: &'a FilesystemReader<'b>) -> Self {
let nodes = fs
.files()
.map(|node| (normalize(&node.fullpath), &node.inner))
.collect();
Self { fs, nodes }
}
fn read(&self, path: &Path) -> Option<Vec<u8>> {
let mut path = normalize(path);
for _ in 0..16 {
match self.nodes.get(&path)? {
InnerNode::File(file) => {
let mut data = Vec::new();
self.fs.file(file).reader().read_to_end(&mut data).ok()?;
return Some(data);
}
InnerNode::Symlink(symlink) => {
let parent = path.parent().unwrap_or(Path::new(""));
path = normalize(&parent.join(&symlink.link));
}
_ => return None,
}
}
None
}
fn root_desktop(&self) -> Option<DesktopEntry> {
let mut candidates: Vec<&PathBuf> = self
.nodes
.keys()
.filter(|p| p.components().count() == 1 && p.extension() == Some(OsStr::new("desktop")))
.collect();
candidates.sort();
let data = self.read(candidates.first()?)?;
Some(DesktopEntry::parse(&String::from_utf8_lossy(&data)))
}
fn icon(&self, name: Option<&str>) -> Option<Icon> {
let mut candidates = Vec::new();
if let Some(name) = name.filter(|n| !n.is_empty()) {
let name = Path::new(name)
.file_name()
.and_then(OsStr::to_str)
.unwrap_or(name);
candidates.push(PathBuf::from(name));
for ext in ["svg", "png", "xpm"] {
candidates.push(PathBuf::from(format!("{name}.{ext}")));
}
candidates.extend(self.hicolor(name));
}
candidates.push(PathBuf::from(".DirIcon"));
candidates
.iter()
.find_map(|path| self.read(path).and_then(icon_from))
}
fn hicolor(&self, name: &str) -> Vec<PathBuf> {
let mut found: Vec<(u32, PathBuf)> = self
.nodes
.keys()
.filter_map(|path| {
let rest = path.strip_prefix("usr/share/icons/hicolor").ok()?;
if path.file_stem()? != OsStr::new(name) {
return None;
}
let size = rest.components().next()?.as_os_str().to_str()?;
let rank = if size == "scalable" {
u32::MAX
} else {
size.split('x').next()?.parse().ok()?
};
Some((rank, path.clone()))
})
.collect();
found.sort_by_key(|(rank, _)| std::cmp::Reverse(*rank));
found.into_iter().map(|(_, path)| path).collect()
}
}
fn icon_from(data: Vec<u8>) -> Option<Icon> {
let ext = if data.starts_with(b"\x89PNG") {
"png"
} else if data.starts_with(b"/* XPM */") {
"xpm"
} else if String::from_utf8_lossy(&data[..data.len().min(4096)]).contains("<svg") {
"svg"
} else {
return None;
};
Some(Icon { data, ext })
}
fn normalize(path: &Path) -> PathBuf {
let mut out = PathBuf::new();
for component in path.components() {
match component {
Component::Normal(part) => out.push(part),
Component::ParentDir => {
out.pop();
}
_ => {}
}
}
out
}
+69
View File
@@ -0,0 +1,69 @@
use std::{fs, path::PathBuf};
use anyhow::{Context, Result};
use serde::{Deserialize, Serialize};
#[derive(Serialize, Deserialize)]
#[serde(default)]
pub struct Config {
pub apps_dir: String,
}
impl Default for Config {
fn default() -> Self {
Self {
apps_dir: "~/Applications".into(),
}
}
}
impl Config {
pub fn path() -> Result<PathBuf> {
Ok(dirs::config_dir()
.context("cannot determine the config directory")?
.join("ratbin/config.toml"))
}
pub fn load() -> Result<Self> {
let path = Self::path()?;
if !path.exists() {
let config = Self::default();
if let Some(parent) = path.parent() {
fs::create_dir_all(parent)?;
}
fs::write(&path, toml::to_string_pretty(&config)?)
.with_context(|| format!("cannot write {}", path.display()))?;
return Ok(config);
}
let text =
fs::read_to_string(&path).with_context(|| format!("cannot read {}", path.display()))?;
toml::from_str(&text).with_context(|| format!("invalid config in {}", path.display()))
}
pub fn apps_dir(&self) -> Result<PathBuf> {
expand_home(&self.apps_dir)
}
}
pub fn applications_dir() -> Result<PathBuf> {
Ok(data_dir()?.join("applications"))
}
pub fn icons_dir() -> Result<PathBuf> {
Ok(data_dir()?.join("ratbin/icons"))
}
fn data_dir() -> Result<PathBuf> {
dirs::data_dir().context("cannot determine the data directory")
}
fn expand_home(path: &str) -> Result<PathBuf> {
let home = || dirs::home_dir().context("cannot determine the home directory");
if path == "~" {
return home();
}
if let Some(rest) = path.strip_prefix("~/") {
return Ok(home()?.join(rest));
}
Ok(PathBuf::from(path))
}
+232
View File
@@ -0,0 +1,232 @@
use std::{fmt, path::Path};
const MAIN: &str = "Desktop Entry";
#[derive(Clone, Debug)]
pub struct DesktopEntry {
groups: Vec<Group>,
}
#[derive(Clone, Debug)]
struct Group {
name: Option<String>,
lines: Vec<Line>,
}
#[derive(Clone, Debug)]
enum Line {
Pair(String, String),
Raw(String),
}
impl DesktopEntry {
pub fn new() -> Self {
Self {
groups: vec![Group {
name: Some(MAIN.into()),
lines: Vec::new(),
}],
}
}
pub fn parse(text: &str) -> Self {
let mut groups = vec![Group {
name: None,
lines: Vec::new(),
}];
for line in text.lines() {
let trimmed = line.trim();
let current = groups.last_mut().expect("groups is never empty");
if trimmed.starts_with('[') && trimmed.ends_with(']') {
groups.push(Group {
name: Some(trimmed[1..trimmed.len() - 1].to_string()),
lines: Vec::new(),
});
} else if !trimmed.starts_with('#')
&& let Some((key, value)) = line.split_once('=')
{
current
.lines
.push(Line::Pair(key.trim().into(), value.trim_start().into()));
} else {
current.lines.push(Line::Raw(line.into()));
}
}
if groups[0].lines.is_empty() {
groups.remove(0);
}
Self { groups }
}
pub fn get(&self, key: &str) -> Option<String> {
self.group(MAIN)?.lines.iter().find_map(|line| match line {
Line::Pair(k, v) if k == key => Some(unescape(v)),
_ => None,
})
}
pub fn set(&mut self, key: &str, value: &str) {
let value = escape(value);
let group = self.main_mut();
for line in &mut group.lines {
if let Line::Pair(k, v) = line
&& k == key
{
*v = value;
return;
}
}
group.lines.push(Line::Pair(key.into(), value));
}
pub fn remove(&mut self, key: &str) {
self.main_mut()
.lines
.retain(|line| !matches!(line, Line::Pair(k, _) if k == key));
}
pub fn remove_localized(&mut self, key: &str) {
self.main_mut().lines.retain(|line| match line {
Line::Pair(k, _) => !(k.starts_with(key) && k[key.len()..].starts_with('[')),
Line::Raw(_) => true,
});
}
pub fn map_values(&mut self, key: &str, f: impl Fn(&str) -> String) {
for group in &mut self.groups {
for line in &mut group.lines {
if let Line::Pair(k, v) = line
&& k == key
{
*v = escape(&f(&unescape(v)));
}
}
}
}
fn group(&self, name: &str) -> Option<&Group> {
self.groups.iter().find(|g| g.name.as_deref() == Some(name))
}
fn main_mut(&mut self) -> &mut Group {
let index = match self
.groups
.iter()
.position(|g| g.name.as_deref() == Some(MAIN))
{
Some(index) => index,
None => {
let at = usize::from(self.groups.first().is_some_and(|g| g.name.is_none()));
self.groups.insert(
at,
Group {
name: Some(MAIN.into()),
lines: Vec::new(),
},
);
at
}
};
&mut self.groups[index]
}
}
impl fmt::Display for DesktopEntry {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
for group in &self.groups {
if let Some(name) = &group.name {
writeln!(f, "[{name}]")?;
}
for line in &group.lines {
match line {
Line::Pair(k, v) => writeln!(f, "{k}={v}")?,
Line::Raw(raw) => writeln!(f, "{raw}")?,
}
}
}
Ok(())
}
}
fn unescape(value: &str) -> String {
let mut out = String::with_capacity(value.len());
let mut chars = value.chars();
while let Some(c) = chars.next() {
if c != '\\' {
out.push(c);
continue;
}
match chars.next() {
Some('s') => out.push(' '),
Some('n') => out.push('\n'),
Some('t') => out.push('\t'),
Some('r') => out.push('\r'),
Some('\\') => out.push('\\'),
Some(other) => {
out.push('\\');
out.push(other);
}
None => out.push('\\'),
}
}
out
}
fn escape(value: &str) -> String {
value
.replace('\\', "\\\\")
.replace('\n', "\\n")
.replace('\t', "\\t")
.replace('\r', "\\r")
}
pub fn quote_exec_arg(arg: &str) -> String {
const RESERVED: &[char] = &[
' ', '\t', '\n', '"', '\'', '\\', '>', '<', '~', '|', '&', ';', '$', '*', '?', '#', '(',
')', '`',
];
let arg = arg.replace('%', "%%");
if !arg.is_empty() && !arg.contains(RESERVED) {
return arg;
}
let mut out = String::from('"');
for c in arg.chars() {
if matches!(c, '"' | '`' | '$' | '\\') {
out.push('\\');
}
out.push(c);
}
out.push('"');
out
}
pub fn exec_args(exec: &str) -> &str {
let exec = exec.trim_start();
let end = match exec.strip_prefix('"') {
Some(rest) => {
let mut escaped = false;
rest.char_indices()
.find_map(|(i, c)| {
match c {
_ if escaped => escaped = false,
'\\' => escaped = true,
'"' => return Some(i + 2),
_ => {}
}
None
})
.unwrap_or(exec.len())
}
None => exec.find(char::is_whitespace).unwrap_or(exec.len()),
};
exec[end..].trim()
}
pub fn build_exec(program: &Path, args: &str) -> String {
let program = quote_exec_arg(&program.to_string_lossy());
if args.is_empty() {
program
} else {
format!("{program} {args}")
}
}
+76
View File
@@ -0,0 +1,76 @@
use std::{fs::File, io::Read, path::Path};
use anyhow::{Context, Result, bail};
pub enum Kind {
AppImage(Option<u64>),
Elf,
Script,
}
impl Kind {
pub fn label(&self) -> &'static str {
match self {
Kind::AppImage(_) => "AppImage",
Kind::Elf => "ELF",
Kind::Script => "Script",
}
}
pub fn fallback_icon(&self) -> &'static str {
match self {
Kind::AppImage(_) | Kind::Elf => "application-x-executable",
Kind::Script => "text-x-script",
}
}
}
pub fn detect(path: &Path) -> Result<Kind> {
let mut header = Vec::with_capacity(64);
File::open(path)
.with_context(|| format!("cannot open {}", path.display()))?
.take(64)
.read_to_end(&mut header)?;
if header.starts_with(b"#!") {
return Ok(Kind::Script);
}
if !header.starts_with(b"\x7fELF") {
bail!(
"{} is not an AppImage, ELF binary or script",
path.display()
);
}
let named_appimage = path
.extension()
.is_some_and(|ext| ext.eq_ignore_ascii_case("appimage"));
Ok(match header.get(8..11) {
Some(b"AI\x02") => Kind::AppImage(elf_end(&header)),
Some(b"AI\x01") => Kind::AppImage(None),
_ if named_appimage => Kind::AppImage(elf_end(&header)),
_ => Kind::Elf,
})
}
fn elf_end(header: &[u8]) -> Option<u64> {
let little_endian = match header.get(5)? {
1 => true,
2 => false,
_ => return None,
};
let read = |offset: usize, len: usize| -> Option<u64> {
let bytes = header.get(offset..offset + len)?;
let fold = |acc: u64, &b: &u8| (acc << 8) | u64::from(b);
Some(if little_endian {
bytes.iter().rev().fold(0, fold)
} else {
bytes.iter().fold(0, fold)
})
};
let (shoff, shentsize, shnum) = match header.get(4)? {
1 => (read(0x20, 4)?, read(0x2E, 2)?, read(0x30, 2)?),
2 => (read(0x28, 8)?, read(0x3A, 2)?, read(0x3C, 2)?),
_ => return None,
};
Some(shoff + shentsize * shnum)
}
+56
View File
@@ -0,0 +1,56 @@
use std::{path::Path, process::Command};
use anyhow::Result;
use crate::store;
const SCANNED_DIRS: [&str; 2] = ["AppImages", "Applications"];
pub fn packages_line() -> Result<String> {
let integrations = store::list()?;
let home = dirs::home_dir().unwrap_or_default();
let counted_twice = integrations
.iter()
.filter(|i| counted_by_fastfetch(&i.source(), &home))
.count();
let mut parts: Vec<String> = fastfetch_packages()
.split(", ")
.filter(|part| !part.is_empty())
.filter_map(|part| match part.strip_suffix(" (appimage)") {
Some(count) => {
let count = count.parse::<usize>().ok()?.saturating_sub(counted_twice);
(count > 0).then(|| format!("{count} (appimage)"))
}
None => Some(part.to_string()),
})
.collect();
parts.push(format!("{} (ratbin)", integrations.len()));
Ok(parts.join(", "))
}
fn fastfetch_packages() -> String {
let Ok(output) = Command::new("fastfetch")
.args(["--config", "none", "--logo", "none", "--pipe", "true"])
.args(["--structure", "packages"])
.output()
else {
return String::new();
};
let text = String::from_utf8_lossy(&output.stdout);
let line = text.lines().next().unwrap_or_default();
line.split_once(": ")
.map_or(line, |(_, counts)| counts)
.trim()
.to_string()
}
fn counted_by_fastfetch(source: &Path, home: &Path) -> bool {
source.is_file()
&& source
.extension()
.is_some_and(|ext| ext.eq_ignore_ascii_case("appimage"))
&& source
.parent()
.is_some_and(|dir| SCANNED_DIRS.iter().any(|d| dir == home.join(d)))
}
+174
View File
@@ -0,0 +1,174 @@
mod appimage;
mod config;
mod desktop;
mod detect;
mod fastfetch;
mod store;
mod thunar;
mod tui;
use std::{
path::{Path, PathBuf},
process::{Command as Process, ExitCode},
};
use anyhow::Result;
use clap::{Parser, Subcommand};
use crate::config::Config;
#[derive(Parser)]
#[command(
version,
about = "Integrate AppImages, binaries and scripts into the application menu"
)]
struct Cli {
#[command(subcommand)]
command: Option<Command>,
}
#[derive(Subcommand)]
enum Command {
/// Move files into the applications folder and create menu entries
Integrate {
/// Report the result as a desktop notification
#[arg(long)]
notify: bool,
#[arg(required = true)]
files: Vec<PathBuf>,
},
/// Remove the menu entries of integrated files
Remove {
#[arg(long)]
notify: bool,
/// Also delete the files themselves
#[arg(long)]
delete: bool,
#[arg(required = true)]
files: Vec<PathBuf>,
},
/// List integrated files
List,
/// Open the interactive manager (default)
Tui,
/// Print fastfetch's package counts with ratbin integrations added
Fastfetch,
/// Manage the Thunar context menu actions
Thunar {
#[command(subcommand)]
action: ThunarAction,
},
}
#[derive(Subcommand)]
enum ThunarAction {
/// Add the ratbin actions to Thunar's context menu
Install,
/// Remove the ratbin actions from Thunar's context menu
Uninstall,
}
fn main() -> ExitCode {
match run(Cli::parse()) {
Ok(code) => code,
Err(e) => {
eprintln!("error: {e:#}");
ExitCode::FAILURE
}
}
}
fn run(cli: Cli) -> Result<ExitCode> {
match cli.command.unwrap_or(Command::Tui) {
Command::Integrate { notify, files } => Ok(batch("Integrated", notify, &files, |file| {
let apps_dir = Config::load()?.apps_dir()?;
store::integrate(&apps_dir, file).map(|i| i.name())
})),
Command::Remove {
notify,
delete,
files,
} => Ok(batch("Removed", notify, &files, |file| {
store::remove_path(file, delete)
})),
Command::List => {
for integration in store::list()? {
println!(
"{}\t{}\t{}",
integration.name(),
integration.kind(),
integration.source().display()
);
}
Ok(ExitCode::SUCCESS)
}
Command::Fastfetch => {
println!("{}", fastfetch::packages_line()?);
Ok(ExitCode::SUCCESS)
}
Command::Tui => {
tui::run()?;
Ok(ExitCode::SUCCESS)
}
Command::Thunar { action } => {
match action {
ThunarAction::Install => {
let path = thunar::install()?;
println!("Added the ratbin actions to {}", path.display());
}
ThunarAction::Uninstall => {
let path = thunar::uninstall()?;
println!("Removed the ratbin actions from {}", path.display());
}
}
println!("Restart Thunar (thunar -q) to apply.");
Ok(ExitCode::SUCCESS)
}
}
}
fn batch(
verb: &str,
notify: bool,
files: &[PathBuf],
op: impl Fn(&Path) -> Result<String>,
) -> ExitCode {
let mut done = Vec::new();
let mut failed = Vec::new();
for file in files {
match op(file) {
Ok(name) => {
println!("{verb}: {name}");
done.push(name);
}
Err(e) => {
eprintln!("error: {e:#}");
failed.push(format!("{e:#}"));
}
}
}
if notify {
if !done.is_empty() {
send_notification(&format!("{verb} {}", done.join(", ")), "", false);
}
if !failed.is_empty() {
send_notification("ratbin", &failed.join("\n"), true);
}
}
if failed.is_empty() {
ExitCode::SUCCESS
} else {
ExitCode::FAILURE
}
}
fn send_notification(summary: &str, body: &str, error: bool) {
let icon = if error {
"dialog-error"
} else {
"application-x-executable"
};
let _ = Process::new("notify-send")
.args(["--app-name=ratbin", "--icon", icon, summary, body])
.status();
}
+322
View File
@@ -0,0 +1,322 @@
use std::{
fs, io,
os::unix::fs::PermissionsExt,
path::{Path, PathBuf},
process::{Command, Stdio},
};
use anyhow::{Context, Result, bail};
use crate::{
appimage, config,
desktop::{self, DesktopEntry},
detect::{self, Kind},
};
const KEY_SOURCE: &str = "X-Ratbin-Source";
const KEY_ARGS: &str = "X-Ratbin-Args";
const KEY_KIND: &str = "X-Ratbin-Kind";
const PREFIX: &str = "ratbin-";
pub struct Integration {
pub desktop_path: PathBuf,
pub entry: DesktopEntry,
}
pub struct Fields {
pub name: String,
pub comment: String,
pub categories: String,
pub icon: String,
pub args: String,
pub terminal: bool,
}
impl Integration {
pub fn source(&self) -> PathBuf {
self.entry
.get(KEY_SOURCE)
.map(PathBuf::from)
.unwrap_or_default()
}
pub fn name(&self) -> String {
self.entry
.get("Name")
.unwrap_or_else(|| file_stem(&self.source()))
}
pub fn kind(&self) -> String {
self.entry.get(KEY_KIND).unwrap_or_default()
}
pub fn fields(&self) -> Fields {
let get = |key: &str| self.entry.get(key).unwrap_or_default();
Fields {
name: self.name(),
comment: get("Comment"),
categories: get("Categories"),
icon: get("Icon"),
args: get(KEY_ARGS),
terminal: get("Terminal") == "true",
}
}
pub fn apply(&mut self, fields: &Fields) {
let source = self.source();
let entry = &mut self.entry;
let name = fields.name.trim();
entry.remove_localized("Name");
if name.is_empty() {
entry.set("Name", &file_stem(&source));
} else {
entry.set("Name", name);
}
entry.remove_localized("Comment");
set_or_remove(entry, "Comment", fields.comment.trim());
let categories = fields.categories.trim();
if categories.is_empty() || categories.ends_with(';') {
set_or_remove(entry, "Categories", categories);
} else {
entry.set("Categories", &format!("{categories};"));
}
set_or_remove(entry, "Icon", fields.icon.trim());
let args = fields.args.trim();
entry.set(KEY_ARGS, args);
entry.set("Exec", &desktop::build_exec(&source, args));
entry.set("Terminal", if fields.terminal { "true" } else { "false" });
}
}
pub fn list() -> Result<Vec<Integration>> {
let mut out = Vec::new();
let Ok(entries) = fs::read_dir(config::applications_dir()?) else {
return Ok(out);
};
for dir_entry in entries.flatten() {
let file_name = dir_entry.file_name();
let file_name = file_name.to_string_lossy();
if !file_name.starts_with(PREFIX) || !file_name.ends_with(".desktop") {
continue;
}
let Ok(text) = fs::read_to_string(dir_entry.path()) else {
continue;
};
let entry = DesktopEntry::parse(&text);
if entry.get(KEY_SOURCE).is_some() {
out.push(Integration {
desktop_path: dir_entry.path(),
entry,
});
}
}
out.sort_by_cached_key(|i| i.name().to_lowercase());
Ok(out)
}
fn find(source: &Path) -> Result<Option<Integration>> {
Ok(list()?.into_iter().find(|i| i.source() == source))
}
pub fn integrate(apps_dir: &Path, path: &Path) -> Result<Integration> {
let src = path
.canonicalize()
.with_context(|| format!("cannot access {}", path.display()))?;
if !src.is_file() {
bail!("{} is not a file", src.display());
}
if let Some(existing) = find(&src)? {
bail!(
"{} is already integrated as \"{}\"",
src.display(),
existing.name()
);
}
let kind = detect::detect(&src)?;
let target = place(apps_dir, &src)?;
make_executable(&target)?;
let id = unique_id(&target)?;
let entry = build_entry(&kind, &target, &id)?;
let integration = Integration {
desktop_path: config::applications_dir()?.join(format!("{id}.desktop")),
entry,
};
save(&integration)?;
Ok(integration)
}
pub fn remove_path(path: &Path, delete_file: bool) -> Result<String> {
let src = path
.canonicalize()
.with_context(|| format!("cannot access {}", path.display()))?;
let integration =
find(&src)?.with_context(|| format!("{} is not integrated", src.display()))?;
remove(&integration, delete_file)?;
Ok(integration.name())
}
pub fn remove(integration: &Integration, delete_file: bool) -> Result<()> {
remove_if_exists(&integration.desktop_path)?;
if let Some(icon) = integration.entry.get("Icon").map(PathBuf::from)
&& icon.starts_with(config::icons_dir()?)
{
remove_if_exists(&icon)?;
}
if delete_file {
remove_if_exists(&integration.source())?;
}
refresh_menu();
Ok(())
}
pub fn save(integration: &Integration) -> Result<()> {
let path = &integration.desktop_path;
if let Some(dir) = path.parent() {
fs::create_dir_all(dir)?;
}
let tmp = path.with_extension("desktop.tmp");
fs::write(&tmp, integration.entry.to_string())
.with_context(|| format!("cannot write {}", tmp.display()))?;
fs::rename(&tmp, path).with_context(|| format!("cannot write {}", path.display()))?;
refresh_menu();
Ok(())
}
fn build_entry(kind: &Kind, target: &Path, id: &str) -> Result<DesktopEntry> {
let embedded = match kind {
Kind::AppImage(Some(offset)) => appimage::read(target, *offset)
.inspect_err(|e| eprintln!("warning: {e:#}"))
.ok(),
_ => None,
};
let (desktop, icon) = embedded.map(|e| (e.desktop, e.icon)).unwrap_or_default();
let mut entry = desktop.unwrap_or_else(DesktopEntry::new);
let args = entry
.get("Exec")
.map(|exec| desktop::exec_args(&exec).to_string())
.unwrap_or_default();
entry.map_values("Exec", |exec| {
desktop::build_exec(target, desktop::exec_args(exec))
});
entry.remove("TryExec");
entry.remove("DBusActivatable");
entry.set("Type", "Application");
if entry.get("Name").is_none() {
entry.set("Name", &file_stem(target));
}
if entry.get("Terminal").is_none() {
let terminal = matches!(kind, Kind::Script);
entry.set("Terminal", if terminal { "true" } else { "false" });
}
let icon = match icon {
Some(icon) => store_icon(&icon, id)?.to_string_lossy().into_owned(),
None => kind.fallback_icon().into(),
};
entry.set("Icon", &icon);
entry.set(KEY_SOURCE, &target.to_string_lossy());
entry.set(KEY_KIND, kind.label());
entry.set(KEY_ARGS, &args);
entry.set("Exec", &desktop::build_exec(target, &args));
Ok(entry)
}
fn place(apps_dir: &Path, src: &Path) -> Result<PathBuf> {
fs::create_dir_all(apps_dir)
.with_context(|| format!("cannot create {}", apps_dir.display()))?;
let apps_dir = apps_dir.canonicalize()?;
if src.starts_with(&apps_dir) {
return Ok(src.to_path_buf());
}
let dest = apps_dir.join(src.file_name().context("path has no file name")?);
if dest.exists() {
bail!("{} already exists", dest.display());
}
if fs::rename(src, &dest).is_err() {
fs::copy(src, &dest).with_context(|| format!("cannot copy to {}", dest.display()))?;
fs::remove_file(src).with_context(|| format!("cannot remove {}", src.display()))?;
}
Ok(dest)
}
fn make_executable(path: &Path) -> Result<()> {
let mut permissions = fs::metadata(path)?.permissions();
let mode = permissions.mode();
permissions.set_mode(mode | 0o100 | ((mode & 0o044) >> 2));
fs::set_permissions(path, permissions)
.with_context(|| format!("cannot make {} executable", path.display()))
}
fn unique_id(target: &Path) -> Result<String> {
let slug: String = file_stem(target)
.chars()
.map(|c| {
if c.is_ascii_alphanumeric() {
c.to_ascii_lowercase()
} else {
'-'
}
})
.collect();
let slug = slug
.split('-')
.filter(|part| !part.is_empty())
.collect::<Vec<_>>()
.join("-");
let slug = if slug.is_empty() { "app".into() } else { slug };
let apps = config::applications_dir()?;
let mut id = format!("{PREFIX}{slug}");
let mut n = 2;
while apps.join(format!("{id}.desktop")).exists() {
id = format!("{PREFIX}{slug}-{n}");
n += 1;
}
Ok(id)
}
fn store_icon(icon: &appimage::Icon, id: &str) -> Result<PathBuf> {
let dir = config::icons_dir()?;
fs::create_dir_all(&dir)?;
let path = dir.join(format!("{id}.{}", icon.ext));
fs::write(&path, &icon.data).with_context(|| format!("cannot write {}", path.display()))?;
Ok(path)
}
fn set_or_remove(entry: &mut DesktopEntry, key: &str, value: &str) {
if value.is_empty() {
entry.remove(key);
} else {
entry.set(key, value);
}
}
fn file_stem(path: &Path) -> String {
path.file_stem()
.map(|s| s.to_string_lossy().into_owned())
.unwrap_or_default()
}
fn remove_if_exists(path: &Path) -> Result<()> {
match fs::remove_file(path) {
Err(e) if e.kind() != io::ErrorKind::NotFound => {
Err(e).with_context(|| format!("cannot remove {}", path.display()))
}
_ => Ok(()),
}
}
fn refresh_menu() {
if let Ok(dir) = config::applications_dir() {
let _ = Command::new("update-desktop-database")
.arg(dir)
.stdout(Stdio::null())
.stderr(Stdio::null())
.status();
}
}
+131
View File
@@ -0,0 +1,131 @@
use std::{fs, path::PathBuf};
use anyhow::{Context, Result};
const SKELETON: &str = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n<actions>\n</actions>\n";
const IDS: [&str; 2] = ["ratbin-integrate", "ratbin-remove"];
pub fn install() -> Result<PathBuf> {
let exe = std::env::current_exe()?.canonicalize()?;
let exe = shell_quote(&exe.to_string_lossy());
let actions = [
action(
IDS[0],
"Integrate into menu",
"application-x-executable",
&format!("{exe} integrate --notify %F"),
"Move to the applications folder and add a menu entry",
),
action(
IDS[1],
"Remove menu integration",
"edit-delete",
&format!("{exe} remove --notify %F"),
"Remove the menu entry created by ratbin",
),
];
let mut doc = stripped()?;
let pos = doc
.rfind("</actions>")
.context("uca.xml has no closing </actions> tag")?;
doc.insert_str(pos, &actions.concat());
write(doc)
}
pub fn uninstall() -> Result<PathBuf> {
write(stripped()?)
}
fn uca_path() -> Result<PathBuf> {
Ok(dirs::config_dir()
.context("cannot determine the config directory")?
.join("Thunar/uca.xml"))
}
fn stripped() -> Result<String> {
let path = uca_path()?;
let doc = if path.exists() {
fs::read_to_string(&path).with_context(|| format!("cannot read {}", path.display()))?
} else {
SKELETON.into()
};
Ok(strip(&doc))
}
fn write(doc: String) -> Result<PathBuf> {
let path = uca_path()?;
if path.exists() {
fs::copy(&path, path.with_extension("xml.bak"))?;
} else if let Some(dir) = path.parent() {
fs::create_dir_all(dir)?;
}
fs::write(&path, doc).with_context(|| format!("cannot write {}", path.display()))?;
Ok(path)
}
fn strip(doc: &str) -> String {
const CLOSE: &str = "</action>";
let mut out = String::with_capacity(doc.len());
let mut rest = doc;
while let Some(start) = rest.find("<action>") {
let Some(len) = rest[start..].find(CLOSE) else {
break;
};
let mut end = start + len + CLOSE.len();
let block = &rest[start..end];
out.push_str(&rest[..start]);
if IDS
.iter()
.any(|id| block.contains(&format!("<unique-id>{id}</unique-id>")))
{
if rest[end..].starts_with('\n') {
end += 1;
}
} else {
out.push_str(block);
}
rest = &rest[end..];
}
out.push_str(rest);
out
}
fn action(id: &str, name: &str, icon: &str, command: &str, description: &str) -> String {
format!(
"<action>\n\
\t<icon>{icon}</icon>\n\
\t<name>{name}</name>\n\
\t<submenu></submenu>\n\
\t<unique-id>{id}</unique-id>\n\
\t<command>{command}</command>\n\
\t<description>{description}</description>\n\
\t<range></range>\n\
\t<patterns>*</patterns>\n\
\t<other-files/>\n\
\t<text-files/>\n\
</action>\n",
icon = xml_escape(icon),
name = xml_escape(name),
id = xml_escape(id),
command = xml_escape(command),
description = xml_escape(description),
)
}
fn xml_escape(s: &str) -> String {
s.replace('&', "&amp;")
.replace('<', "&lt;")
.replace('>', "&gt;")
.replace('"', "&quot;")
.replace('\'', "&apos;")
}
fn shell_quote(s: &str) -> String {
if s.chars()
.all(|c| c.is_ascii_alphanumeric() || "/._-+".contains(c))
{
s.into()
} else {
format!("'{}'", s.replace('\'', "'\\''"))
}
}
+463
View File
@@ -0,0 +1,463 @@
use std::{io, path::PathBuf};
use anyhow::Result;
use ratatui::{
DefaultTerminal, Frame,
crossterm::{
event::{self, Event, KeyCode, KeyEvent, KeyEventKind, KeyModifiers},
execute,
terminal::SetTitle,
},
layout::{Constraint, Layout, Rect},
style::{Color, Modifier, Style},
text::{Line, Span},
widgets::{Block, Clear, List, ListItem, ListState, Paragraph, Wrap},
};
use crate::store::{self, Fields, Integration};
const LABELS: [&str; 6] = [
"Name",
"Comment",
"Categories",
"Icon",
"Arguments",
"Terminal",
];
const TERMINAL_FIELD: usize = 5;
const LABEL_WIDTH: u16 = 12;
pub fn run() -> Result<()> {
let mut app = App::new()?;
let mut terminal = ratatui::init();
let _ = execute!(io::stdout(), SetTitle("Ratbin"));
let result = app.run(&mut terminal);
ratatui::restore();
result
}
enum Mode {
Browse,
Edit(Form),
Confirm,
}
struct App {
items: Vec<Integration>,
list: ListState,
mode: Mode,
status: Option<(String, bool)>,
quit: bool,
}
impl App {
fn new() -> Result<Self> {
let mut app = Self {
items: Vec::new(),
list: ListState::default(),
mode: Mode::Browse,
status: None,
quit: false,
};
app.reload(None)?;
Ok(app)
}
fn reload(&mut self, keep: Option<PathBuf>) -> Result<()> {
self.items = store::list()?;
let index = keep
.and_then(|path| self.items.iter().position(|i| i.desktop_path == path))
.or(self.list.selected())
.unwrap_or(0)
.min(self.items.len().saturating_sub(1));
self.list.select((!self.items.is_empty()).then_some(index));
Ok(())
}
fn selected(&self) -> Option<&Integration> {
self.items.get(self.list.selected()?)
}
fn run(&mut self, terminal: &mut DefaultTerminal) -> Result<()> {
while !self.quit {
terminal.draw(|frame| self.draw(frame))?;
if let Event::Key(key) = event::read()?
&& key.kind == KeyEventKind::Press
{
self.status = None;
self.on_key(key);
}
}
Ok(())
}
fn on_key(&mut self, key: KeyEvent) {
let ctrl = key.modifiers.contains(KeyModifiers::CONTROL);
match &mut self.mode {
Mode::Browse => self.on_browse(key),
Mode::Confirm => self.on_confirm(key),
Mode::Edit(form) => match key.code {
KeyCode::Esc => self.mode = Mode::Browse,
KeyCode::Enter => self.save(),
KeyCode::Char('s') if ctrl => self.save(),
_ => form.on_key(key),
},
}
}
fn on_browse(&mut self, key: KeyEvent) {
match key.code {
KeyCode::Char('q') | KeyCode::Esc => self.quit = true,
KeyCode::Down | KeyCode::Char('j') => self.list.select_next(),
KeyCode::Up | KeyCode::Char('k') => self.list.select_previous(),
KeyCode::Home | KeyCode::Char('g') => self.list.select_first(),
KeyCode::End | KeyCode::Char('G') => self.list.select_last(),
KeyCode::Enter | KeyCode::Char('e') => {
if let Some(item) = self.selected() {
self.mode = Mode::Edit(Form::new(item.fields()));
}
}
KeyCode::Delete | KeyCode::Char('d') => {
if self.selected().is_some() {
self.mode = Mode::Confirm;
}
}
KeyCode::Char('r') => {
let result = self.reload(None);
self.report(result.map(|()| "Reloaded".into()));
}
_ => {}
}
}
fn on_confirm(&mut self, key: KeyEvent) {
self.mode = Mode::Browse;
let delete_file = match key.code {
KeyCode::Char('e') => false,
KeyCode::Char('d') => true,
_ => return,
};
let Some(item) = self.selected() else {
return;
};
let name = item.name();
let result = store::remove(item, delete_file).and_then(|()| self.reload(None));
self.report(result.map(|()| format!("Removed {name}")));
}
fn save(&mut self) {
let Mode::Edit(form) = std::mem::replace(&mut self.mode, Mode::Browse) else {
return;
};
let Some(item) = self
.list
.selected()
.and_then(|index| self.items.get_mut(index))
else {
return;
};
item.apply(&form.fields());
let path = item.desktop_path.clone();
let name = item.name();
match store::save(item) {
Ok(()) => {
let result = self.reload(Some(path));
self.report(result.map(|()| format!("Saved {name}")));
}
Err(e) => {
self.status = Some((format!("{e:#}"), true));
self.mode = Mode::Edit(form);
}
}
}
fn report(&mut self, result: Result<String>) {
self.status = Some(match result {
Ok(message) => (message, false),
Err(e) => (format!("{e:#}"), true),
});
}
fn draw(&mut self, frame: &mut Frame) {
let [main, footer] =
Layout::vertical([Constraint::Min(0), Constraint::Length(1)]).areas(frame.area());
let [left, right] =
Layout::horizontal([Constraint::Percentage(40), Constraint::Percentage(60)])
.areas(main);
let items: Vec<ListItem> = self
.items
.iter()
.map(|item| {
let name_style = if item.source().exists() {
Style::default()
} else {
Style::default().fg(Color::Red)
};
ListItem::new(Line::from(vec![
Span::styled(item.name(), name_style),
Span::styled(
format!(" {}", item.kind()),
Style::default().fg(Color::DarkGray),
),
]))
})
.collect();
let list = List::new(items)
.block(Block::bordered().title(" ratbin "))
.highlight_style(Style::default().add_modifier(Modifier::REVERSED))
.highlight_symbol("› ");
frame.render_stateful_widget(list, left, &mut self.list);
frame.render_widget(self.details(), right);
frame.render_widget(self.footer(), footer);
match &self.mode {
Mode::Browse => {}
Mode::Edit(form) => form.draw(frame, main),
Mode::Confirm => self.draw_confirm(frame, main),
}
}
fn details(&self) -> Paragraph<'static> {
let block = Block::bordered().title(" Details ");
let Some(item) = self.selected() else {
return Paragraph::new(
"Nothing integrated yet.\n\nRight-click an AppImage, binary or script in Thunar \
and choose \"Integrate into menu\".",
)
.block(block)
.wrap(Wrap { trim: false });
};
let source = item.source();
let get = |key: &str| item.entry.get(key).unwrap_or_default();
let row = |label: &str, value: String| {
Line::from(vec![
Span::styled(
format!("{label:<width$}", width = LABEL_WIDTH as usize),
Style::default().fg(Color::Cyan),
),
Span::raw(value),
])
};
let mut lines = vec![
row("Name", item.name()),
row("Kind", item.kind()),
row("Comment", get("Comment")),
row("Categories", get("Categories")),
row("Icon", get("Icon")),
row("Terminal", get("Terminal")),
row("Exec", get("Exec")),
row("File", source.display().to_string()),
row("Entry", item.desktop_path.display().to_string()),
];
if !source.exists() {
lines.push(Line::default());
lines.push(Line::styled(
"The file no longer exists, so this menu entry is broken.",
Style::default().fg(Color::Red),
));
}
Paragraph::new(lines)
.block(block)
.wrap(Wrap { trim: false })
}
fn footer(&self) -> Line<'static> {
if let Some((message, error)) = &self.status {
let color = if *error { Color::Red } else { Color::Green };
return Line::styled(message.clone(), Style::default().fg(color));
}
let hints = match self.mode {
Mode::Browse => "↑↓ select enter edit d remove r reload q quit",
Mode::Edit(_) => "tab/↑↓ field space toggle ctrl+u clear enter save esc cancel",
Mode::Confirm => "e entry only d entry + file esc cancel",
};
Line::styled(hints, Style::default().fg(Color::DarkGray))
}
fn draw_confirm(&self, frame: &mut Frame, area: Rect) {
let Some(item) = self.selected() else {
return;
};
let file_name = item
.source()
.file_name()
.map(|n| n.to_string_lossy().into_owned())
.unwrap_or_default();
let rect = popup(area, 64, 6);
frame.render_widget(Clear, rect);
frame.render_widget(
Paragraph::new(vec![
Line::from(format!("Remove \"{}\"?", item.name())),
Line::default(),
Line::from("e remove the menu entry only"),
Line::from(format!("d also delete {file_name}")),
])
.block(
Block::bordered()
.title(" Remove ")
.border_style(Style::default().fg(Color::Red)),
),
rect,
);
}
}
struct Form {
inputs: [Input; 5],
terminal: bool,
focus: usize,
}
impl Form {
fn new(fields: Fields) -> Self {
Self {
inputs: [
fields.name,
fields.comment,
fields.categories,
fields.icon,
fields.args,
]
.map(Input::new),
terminal: fields.terminal,
focus: 0,
}
}
fn fields(&self) -> Fields {
let [name, comment, categories, icon, args] =
self.inputs.each_ref().map(|input| input.value.clone());
Fields {
name,
comment,
categories,
icon,
args,
terminal: self.terminal,
}
}
fn on_key(&mut self, key: KeyEvent) {
match key.code {
KeyCode::Tab | KeyCode::Down => self.focus = (self.focus + 1) % LABELS.len(),
KeyCode::BackTab | KeyCode::Up => {
self.focus = (self.focus + LABELS.len() - 1) % LABELS.len();
}
KeyCode::Char(' ') | KeyCode::Left | KeyCode::Right if self.focus == TERMINAL_FIELD => {
self.terminal = !self.terminal;
}
_ if self.focus == TERMINAL_FIELD => {}
_ => self.inputs[self.focus].on_key(key),
}
}
fn draw(&self, frame: &mut Frame, area: Rect) {
let rect = popup(area, 80, LABELS.len() as u16 + 2);
frame.render_widget(Clear, rect);
let block = Block::bordered().title(" Edit entry ");
let inner = block.inner(rect);
frame.render_widget(block, rect);
let value_width = inner.width.saturating_sub(LABEL_WIDTH) as usize;
for (row, label) in LABELS.iter().enumerate() {
let y = inner.y + row as u16;
if y >= inner.bottom() {
break;
}
let focused = row == self.focus;
let label_style = if focused {
Style::default()
.fg(Color::Cyan)
.add_modifier(Modifier::BOLD)
} else {
Style::default().fg(Color::DarkGray)
};
let (text, cursor) = if row == TERMINAL_FIELD {
let mark = if self.terminal { "[x]" } else { "[ ]" };
(format!("{mark} run in a terminal"), None)
} else {
let (text, cursor) = self.inputs[row].view(value_width);
(text, Some(cursor))
};
let line = Line::from(vec![
Span::styled(
format!("{label:<width$}", width = LABEL_WIDTH as usize),
label_style,
),
Span::raw(text),
]);
frame.render_widget(line, Rect::new(inner.x, y, inner.width, 1));
if focused && let Some(cursor) = cursor {
frame.set_cursor_position((inner.x + LABEL_WIDTH + cursor as u16, y));
}
}
}
}
struct Input {
value: String,
cursor: usize,
}
impl Input {
fn new(value: String) -> Self {
let cursor = value.chars().count();
Self { value, cursor }
}
fn byte_index(&self, char_index: usize) -> usize {
self.value
.char_indices()
.nth(char_index)
.map_or(self.value.len(), |(i, _)| i)
}
fn on_key(&mut self, key: KeyEvent) {
let ctrl = key.modifiers.contains(KeyModifiers::CONTROL);
let len = self.value.chars().count();
match key.code {
KeyCode::Char('u') if ctrl => {
self.value.clear();
self.cursor = 0;
}
KeyCode::Char(c) if !ctrl => {
let at = self.byte_index(self.cursor);
self.value.insert(at, c);
self.cursor += 1;
}
KeyCode::Backspace if self.cursor > 0 => {
self.cursor -= 1;
let at = self.byte_index(self.cursor);
self.value.remove(at);
}
KeyCode::Delete if self.cursor < len => {
let at = self.byte_index(self.cursor);
self.value.remove(at);
}
KeyCode::Left => self.cursor = self.cursor.saturating_sub(1),
KeyCode::Right => self.cursor = (self.cursor + 1).min(len),
KeyCode::Home => self.cursor = 0,
KeyCode::End => self.cursor = len,
_ => {}
}
}
fn view(&self, width: usize) -> (String, usize) {
let width = width.max(1);
let start = (self.cursor + 1).saturating_sub(width);
let text = self.value.chars().skip(start).take(width).collect();
(text, self.cursor - start)
}
}
fn popup(area: Rect, width: u16, height: u16) -> Rect {
let width = width.min(area.width);
let height = height.min(area.height);
Rect::new(
area.x + (area.width - width) / 2,
area.y + (area.height - height) / 2,
width,
height,
)
}