mroot/yazi
1
0
Fork 0
mirror of https://github.com/sxyazi/yazi.git synced 2026-07-25 08:41:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

..

Browse source
This commit is contained in:
sxyazi 2023-09-13 20:15:42 +08:00
parent 26031ab8c2
commit 0ecdd05482
No known key found for this signature in database
2 changed files with 44 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

62
adaptor/src/adaptor.rs
View file

@ -1,15 +1,17 @@
use std::{env, path::{Path, PathBuf}, sync::atomic::{AtomicBool, Ordering}}; use std::{env, path::{Path, PathBuf}, sync::atomic::{AtomicBool, Ordering}};
use anyhow::Result; use anyhow::{anyhow, Result};
use config::PREVIEW; use config::PREVIEW;
use ratatui::prelude::Rect; use ratatui::prelude::Rect;
use shared::RoCell; use shared::RoCell;
use tokio::{fs, sync::mpsc::UnboundedSender}; use tokio::{fs, sync::mpsc::UnboundedSender};
use super::{Iterm2, Kitty}; use super::{Iterm2, Kitty};
use crate::{ueberzug::Ueberzug, Sixel}; use crate::{ueberzug::Ueberzug, Sixel, TMUX};
static IMAGE_SHOWN: AtomicBool = AtomicBool::new(false); static IMAGE_SHOWN: AtomicBool = AtomicBool::new(false);
#[allow(clippy::type_complexity)]
static UEBERZUG: RoCell<Option<UnboundedSender<Option<(PathBuf, Rect)>>>> = RoCell::new(); static UEBERZUG: RoCell<Option<UnboundedSender<Option<(PathBuf, Rect)>>>> = RoCell::new();
#[derive(Clone, Copy)] #[derive(Clone, Copy)]
@ -24,8 +26,8 @@ pub enum Adaptor {
Chafa, Chafa,
} }
impl Default for Adaptor { impl Adaptor {
fn default() -> Self { pub(super) fn detect() -> Self {
let vars = [ let vars = [
("KITTY_WINDOW_ID", Self::Kitty), ("KITTY_WINDOW_ID", Self::Kitty),
("KONSOLE_VERSION", Self::Kitty), ("KONSOLE_VERSION", Self::Kitty),
@ -37,7 +39,8 @@ impl Default for Adaptor {
return var.1; return var.1;
} }
match env::var("TERM_PROGRAM").unwrap_or_default().as_str() { let (term, program) = Self::term_program();
match program.as_str() {
"iTerm.app" => return Self::Iterm2, "iTerm.app" => return Self::Iterm2,
"WezTerm" => return cfg!(windows).then_some(Self::Iterm2).unwrap_or(Self::Kitty), "WezTerm" => return cfg!(windows).then_some(Self::Iterm2).unwrap_or(Self::Kitty),
"BlackBox" => return Self::Sixel, "BlackBox" => return Self::Sixel,
@ -46,7 +49,7 @@ impl Default for Adaptor {
"mintty" => return Self::Iterm2, "mintty" => return Self::Iterm2,
_ => {} _ => {}
} }
match env::var("TERM").unwrap_or_default().as_str() { match term.as_str() {
"xterm-kitty" => return Self::Kitty, "xterm-kitty" => return Self::Kitty,
"foot" => return Self::Sixel, "foot" => return Self::Sixel,
_ => {} _ => {}
@ -57,17 +60,38 @@ impl Default for Adaptor {
_ => Self::Chafa, _ => Self::Chafa,
} }
} }
pub(super) fn term_program() -> (String, String) {
fn tmux_env(name: &str) -> Result<String> {
let output = std::process::Command::new("tmux").args(["show-environment", name]).output()?;
String::from_utf8(output.stdout)?
.trim()
.strip_prefix(&format!("{}=", name))
.map_or_else(|| Err(anyhow!("")), |s| Ok(s.to_string()))
}
let mut term = env::var("TERM").unwrap_or_default();
let mut program = env::var("TERM_PROGRAM").unwrap_or_default();
if *TMUX {
term = tmux_env("TERM").unwrap_or(term);
program = tmux_env("TERM_PROGRAM").unwrap_or(program);
}
(term, program)
}
} }
impl ToString for Adaptor { impl ToString for Adaptor {
fn to_string(&self) -> String { fn to_string(&self) -> String {
match self { match self {
Adaptor::Kitty => "kitty", Self::Kitty => "kitty",
Adaptor::Iterm2 => "iterm2", Self::Iterm2 => "iterm2",
Adaptor::Sixel => "sixel", Self::Sixel => "sixel",
Adaptor::X11 => "x11", Self::X11 => "x11",
Adaptor::Wayland => "wayland", Self::Wayland => "wayland",
Adaptor::Chafa => "chafa", Self::Chafa => "chafa",
} }
.to_string() .to_string()
} }
@ -88,9 +112,9 @@ impl Adaptor {
IMAGE_SHOWN.store(true, Ordering::Relaxed); IMAGE_SHOWN.store(true, Ordering::Relaxed);
match self { match self {
Adaptor::Kitty => Kitty::image_show(path, rect).await, Self::Kitty => Kitty::image_show(path, rect).await,
Adaptor::Iterm2 => Iterm2::image_show(path, rect).await, Self::Iterm2 => Iterm2::image_show(path, rect).await,
Adaptor::Sixel => Sixel::image_show(path, rect).await, Self::Sixel => Sixel::image_show(path, rect).await,
_ => Ok(if let Some(tx) = &*UEBERZUG { _ => Ok(if let Some(tx) = &*UEBERZUG {
tx.send(Some((path.to_path_buf(), rect)))?; tx.send(Some((path.to_path_buf(), rect)))?;
}), }),
@ -103,9 +127,9 @@ impl Adaptor {
} }
match self { match self {
Adaptor::Kitty => Kitty::image_hide(), Self::Kitty => Kitty::image_hide(),
Adaptor::Iterm2 => Iterm2::image_hide(rect), Self::Iterm2 => Iterm2::image_hide(rect),
Adaptor::Sixel => Sixel::image_hide(rect), Self::Sixel => Sixel::image_hide(rect),
_ => Ok(if let Some(tx) = &*UEBERZUG { _ => Ok(if let Some(tx) = &*UEBERZUG {
tx.send(None)?; tx.send(None)?;
}), }),
@ -114,6 +138,6 @@ impl Adaptor {
#[inline] #[inline]
pub(super) fn needs_ueberzug(self) -> bool { pub(super) fn needs_ueberzug(self) -> bool {
!matches!(self, Adaptor::Kitty | Adaptor::Iterm2 | Adaptor::Sixel) !matches!(self, Self::Kitty | Self::Iterm2 | Self::Sixel)
} }
} }

2
adaptor/src/lib.rs
View file

@ -29,6 +29,6 @@ pub fn init() {
CLOSE.init(if *TMUX { "\x1b\\" } else { "" }); CLOSE.init(if *TMUX { "\x1b\\" } else { "" });
ESCAPE.init(if *TMUX { "\x1b\x1b" } else { "\x1b" }); ESCAPE.init(if *TMUX { "\x1b\x1b" } else { "\x1b" });
ADAPTOR.with(Default::default); ADAPTOR.init(Adaptor::detect());
ADAPTOR.start(); ADAPTOR.start();
} }