helix-plus/helix-view/src/editor.rs

use crate::theme::Theme;
use crate::tree::Tree;
use crate::{Document, View};

use std::path::PathBuf;

use slotmap::DefaultKey as Key;

use anyhow::Error;

pub struct Editor {
    pub tree: Tree,
    // pub documents: Vec<Document>,
    pub count: Option<usize>,
    pub theme: Theme,
    pub language_servers: helix_lsp::Registry,
}

impl Editor {
    pub fn new(mut area: tui::layout::Rect) -> Self {
        let theme = Theme::default();
        let language_servers = helix_lsp::Registry::new();

        // HAXX: offset the render area height by 1 to account for prompt/commandline
        area.height -= 1;

        Self {
            tree: Tree::new(area),
            count: None,
            theme,
            language_servers,
        }
    }

    pub fn open(&mut self, path: PathBuf, executor: &smol::Executor) -> Result<(), Error> {
        let mut doc = Document::load(path, self.theme.scopes())?;

        // try to find a language server based on the language name
        let language_server = doc
            .language()
            .and_then(|language| self.language_servers.get(language, &executor));

        if let Some(language_server) = language_server {
            // TODO: do this everywhere
            doc.set_language_server(Some(language_server.clone()));

            smol::block_on(language_server.text_document_did_open(
                doc.url().unwrap(),
                doc.version(),
                doc.text(),
            ))
            .unwrap();
        }

        let view = View::new(doc)?;
        self.tree.insert(view);
        Ok(())
    }

    pub fn close(&mut self, id: Key) {
        self.tree.remove(id)
    }

    pub fn should_close(&mut self) -> bool {
        self.tree.is_empty()
    }

    pub fn view(&self) -> &View {
        self.tree.get(self.tree.focus)
    }

    pub fn view_mut(&mut self) -> &mut View {
        self.tree.get_mut(self.tree.focus)
    }
}