lib.rs

  1use editor::Editor;
  2use encoding::Encoding;
  3use encoding::all::{
  4    BIG5_2003, EUC_JP, GB18030, GBK, HZ, IBM866, ISO_2022_JP, ISO_8859_1, ISO_8859_2, ISO_8859_3,
  5    ISO_8859_4, ISO_8859_5, ISO_8859_6, ISO_8859_7, ISO_8859_8, ISO_8859_10, ISO_8859_13,
  6    ISO_8859_14, ISO_8859_15, ISO_8859_16, KOI8_R, KOI8_U, MAC_CYRILLIC, MAC_ROMAN, UTF_8,
  7    UTF_16BE, UTF_16LE, WINDOWS_874, WINDOWS_949, WINDOWS_1250, WINDOWS_1251, WINDOWS_1252,
  8    WINDOWS_1253, WINDOWS_1254, WINDOWS_1255, WINDOWS_1256, WINDOWS_1257, WINDOWS_1258,
  9};
 10use gpui::{ClickEvent, Entity, Subscription, WeakEntity};
 11use ui::{Button, ButtonCommon, Context, LabelSize, Render, Tooltip, Window, div};
 12use ui::{Clickable, ParentElement};
 13use workspace::{ItemHandle, StatusItemView, Workspace};
 14
 15use crate::selectors::save_or_reopen::{EncodingSaveOrReopenSelector, get_current_encoding};
 16
 17pub struct EncodingIndicator {
 18    pub encoding: Option<&'static dyn Encoding>,
 19    pub workspace: WeakEntity<Workspace>,
 20    observe: Option<Subscription>,
 21}
 22
 23pub mod selectors;
 24
 25impl Render for EncodingIndicator {
 26    fn render(&mut self, _window: &mut Window, cx: &mut Context<Self>) -> impl ui::IntoElement {
 27        let status_element = div();
 28
 29        status_element.child(
 30            Button::new("encoding", encoding_name(self.encoding.unwrap_or(UTF_8)))
 31                .label_size(LabelSize::Small)
 32                .tooltip(Tooltip::text("Select Encoding"))
 33                .on_click(cx.listener(|indicator, _: &ClickEvent, window, cx| {
 34                    if let Some(workspace) = indicator.workspace.upgrade() {
 35                        workspace.update(cx, |workspace, cx| {
 36                            EncodingSaveOrReopenSelector::toggle(workspace, window, cx)
 37                        })
 38                    } else {
 39                    }
 40                })),
 41        )
 42    }
 43}
 44
 45impl EncodingIndicator {
 46    pub fn get_current_encoding(&self, cx: &mut Context<Self>, editor: WeakEntity<Editor>) {}
 47
 48    pub fn new(
 49        encoding: Option<&'static dyn encoding::Encoding>,
 50        workspace: WeakEntity<Workspace>,
 51        observe: Option<Subscription>,
 52    ) -> EncodingIndicator {
 53        EncodingIndicator {
 54            encoding,
 55            workspace,
 56            observe,
 57        }
 58    }
 59
 60    pub fn update(
 61        &mut self,
 62        editor: Entity<Editor>,
 63        _: &mut Window,
 64        cx: &mut Context<EncodingIndicator>,
 65    ) {
 66        let editor = editor.read(cx);
 67        if let Some((_, buffer, _)) = editor.active_excerpt(cx) {
 68            let encoding = buffer.read(cx).encoding;
 69            self.encoding = Some(encoding);
 70        }
 71
 72        cx.notify();
 73    }
 74}
 75
 76impl StatusItemView for EncodingIndicator {
 77    fn set_active_pane_item(
 78        &mut self,
 79        active_pane_item: Option<&dyn ItemHandle>,
 80        window: &mut Window,
 81        cx: &mut Context<Self>,
 82    ) {
 83        match active_pane_item.and_then(|item| item.downcast::<Editor>()) {
 84            Some(editor) => {
 85                self.observe = Some(cx.observe_in(&editor, window, Self::update));
 86                self.update(editor, window, cx);
 87            }
 88            None => {
 89                self.encoding = None;
 90                self.observe = None;
 91            }
 92        }
 93    }
 94}
 95
 96pub fn encoding_name(encoding: &'static dyn Encoding) -> String {
 97    let name = encoding.name();
 98
 99    match () {
100        () if name == UTF_8.name() => "UTF-8",
101        () if name == UTF_16LE.name() => "UTF-16 LE",
102        () if name == UTF_16BE.name() => "UTF-16 BE",
103        () if name == IBM866.name() => "IBM866",
104        () if name == ISO_8859_1.name() => "ISO 8859-1",
105        () if name == ISO_8859_2.name() => "ISO 8859-2",
106        () if name == ISO_8859_3.name() => "ISO 8859-3",
107        () if name == ISO_8859_4.name() => "ISO 8859-4",
108        () if name == ISO_8859_5.name() => "ISO 8859-5",
109        () if name == ISO_8859_6.name() => "ISO 8859-6",
110        () if name == ISO_8859_7.name() => "ISO 8859-7",
111        () if name == ISO_8859_8.name() => "ISO 8859-8",
112        () if name == ISO_8859_10.name() => "ISO 8859-10",
113        () if name == ISO_8859_13.name() => "ISO 8859-13",
114        () if name == ISO_8859_14.name() => "ISO 8859-14",
115        () if name == ISO_8859_15.name() => "ISO 8859-15",
116        () if name == ISO_8859_16.name() => "ISO 8859-16",
117        () if name == KOI8_R.name() => "KOI8-R",
118        () if name == KOI8_U.name() => "KOI8-U",
119        () if name == MAC_ROMAN.name() => "MacRoman",
120        () if name == MAC_CYRILLIC.name() => "Mac Cyrillic",
121        () if name == WINDOWS_874.name() => "Windows-874",
122        () if name == WINDOWS_1250.name() => "Windows-1250",
123        () if name == WINDOWS_1251.name() => "Windows-1251",
124        () if name == WINDOWS_1252.name() => "Windows-1252",
125        () if name == WINDOWS_1253.name() => "Windows-1253",
126        () if name == WINDOWS_1254.name() => "Windows-1254",
127        () if name == WINDOWS_1255.name() => "Windows-1255",
128        () if name == WINDOWS_1256.name() => "Windows-1256",
129        () if name == WINDOWS_1257.name() => "Windows-1257",
130        () if name == WINDOWS_1258.name() => "Windows-1258",
131        () if name == WINDOWS_949.name() => "Windows-949",
132        () if name == EUC_JP.name() => "EUC-JP",
133        () if name == ISO_2022_JP.name() => "ISO 2022-JP",
134        () if name == GBK.name() => "GBK",
135        () if name == GB18030.name() => "GB18030",
136        () if name == BIG5_2003.name() => "Big5",
137        () if name == HZ.name() => "HZ-GB-2312",
138        _ => "",
139    }
140    .to_string()
141}
142
143pub fn encoding_from_index(index: usize) -> &'static dyn Encoding {
144    match index {
145        0 => UTF_8,
146        1 => UTF_16LE,
147        2 => UTF_16BE,
148        3 => IBM866,
149        4 => ISO_8859_1,
150        5 => ISO_8859_2,
151        6 => ISO_8859_3,
152        7 => ISO_8859_4,
153        8 => ISO_8859_5,
154        9 => ISO_8859_6,
155        10 => ISO_8859_7,
156        11 => ISO_8859_8,
157        12 => ISO_8859_10,
158        13 => ISO_8859_13,
159        14 => ISO_8859_14,
160        15 => ISO_8859_15,
161        16 => ISO_8859_16,
162        17 => KOI8_R,
163        18 => KOI8_U,
164        19 => MAC_ROMAN,
165        20 => MAC_CYRILLIC,
166        21 => WINDOWS_874,
167        22 => WINDOWS_1250,
168        23 => WINDOWS_1251,
169        24 => WINDOWS_1252,
170        25 => WINDOWS_1253,
171        26 => WINDOWS_1254,
172        27 => WINDOWS_1255,
173        28 => WINDOWS_1256,
174        29 => WINDOWS_1257,
175        30 => WINDOWS_1258,
176        31 => WINDOWS_949,
177        32 => EUC_JP,
178        33 => ISO_2022_JP,
179        34 => GBK,
180        35 => GB18030,
181        36 => BIG5_2003,
182        37 => HZ,
183        _ => UTF_8,
184    }
185}