tenant.openagents/omega

1use gpui::{App, Context, Entity, IntoElement, ParentElement, Render, Subscription, div};
2use project::image_store::{ImageFormat, ImageMetadata};
3use settings::Settings;
4use ui::prelude::*;
5use util::size::format_file_size;
6use workspace::{HideStatusItem, ItemHandle, StatusItemView, Workspace};
7
8use crate::{ImageFileSizeUnit, ImageView, ImageViewerSettings};
9
10pub struct ImageInfo {
11    metadata: Option<ImageMetadata>,
12    _observe_active_image: Option<Subscription>,
13    observe_image_item: Option<Subscription>,
14}
15
16impl ImageInfo {
17    pub fn new(_workspace: &Workspace) -> Self {
18        Self {
19            metadata: None,
20            _observe_active_image: None,
21            observe_image_item: None,
22        }
23    }
24
25    fn update_metadata(&mut self, image_view: &Entity<ImageView>, cx: &mut Context<Self>) {
26        let image_item = image_view.read(cx).image_item.clone();
27        let current_metadata = image_item.read(cx).image_metadata;
28        if current_metadata.is_some() {
29            self.metadata = current_metadata;
30            cx.notify();
31        } else {
32            self.observe_image_item = Some(cx.observe(&image_item, |this, item, cx| {
33                this.metadata = item.read(cx).image_metadata;
34                cx.notify();
35            }));
36        }
37    }
38}
39
40fn format_image_size(size: u64, image_unit_type: ImageFileSizeUnit) -> String {
41    let use_decimal = matches!(image_unit_type, ImageFileSizeUnit::Decimal);
42    format_file_size(size, use_decimal)
43}
44
45impl Render for ImageInfo {
46    fn render(&mut self, _window: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {
47        let settings = ImageViewerSettings::get_global(cx);
48
49        let Some(metadata) = self.metadata.as_ref() else {
50            return div().hidden();
51        };
52
53        let mut components = Vec::new();
54        components.push(format!("{}x{}", metadata.width, metadata.height));
55        components.push(format_image_size(metadata.file_size, settings.unit));
56
57        if let Some(colors) = metadata.colors {
58            components.push(format!(
59                "{} channels, {} bits per pixel",
60                colors.channels,
61                colors.bits_per_pixel()
62            ));
63        }
64
65        components.push(
66            match metadata.format {
67                ImageFormat::Png => "PNG",
68                ImageFormat::Jpeg => "JPEG",
69                ImageFormat::Gif => "GIF",
70                ImageFormat::WebP => "WebP",
71                ImageFormat::Tiff => "TIFF",
72                ImageFormat::Bmp => "BMP",
73                ImageFormat::Ico => "ICO",
74                ImageFormat::Avif => "Avif",
75                _ => "Unknown",
76            }
77            .to_string(),
78        );
79
80        div().child(Label::new(components.join(" • ")).size(LabelSize::Small))
81    }
82}
83
84impl StatusItemView for ImageInfo {
85    fn set_active_pane_item(
86        &mut self,
87        active_pane_item: Option<&dyn ItemHandle>,
88        _window: &mut Window,
89        cx: &mut Context<Self>,
90    ) {
91        self._observe_active_image = None;
92        self.observe_image_item = None;
93
94        if let Some(image_view) = active_pane_item.and_then(|item| item.act_as::<ImageView>(cx)) {
95            self.update_metadata(&image_view, cx);
96
97            self._observe_active_image = Some(cx.observe(&image_view, |this, view, cx| {
98                this.update_metadata(&view, cx);
99            }));
100        } else {
101            self.metadata = None;
102        }
103        cx.notify();
104    }
105
106    fn hide_setting(&self, _: &App) -> Option<HideStatusItem> {
107        // The image info is only visible when an image viewer item is active.
108        None
109    }
110}
111