/* * Copyright (c) 2018-2020, Andreas Kling * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * 1. Redistributions of source code must retain the above copyright notice, this * list of conditions and the following disclaimer. * * 2. Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include "Locator.h" #include "HackStudio.h" #include "Project.h" #include #include #include #include namespace HackStudio { static RefPtr s_file_icon; static RefPtr s_cplusplus_icon; static RefPtr s_header_icon; static RefPtr s_form_icon; static RefPtr s_hackstudio_icon; class LocatorSuggestionModel final : public GUI::Model { public: explicit LocatorSuggestionModel(Vector&& suggestions) : m_suggestions(move(suggestions)) { } enum Column { Icon, Name, __Column_Count, }; virtual int row_count(const GUI::ModelIndex& = GUI::ModelIndex()) const override { return m_suggestions.size(); } virtual int column_count(const GUI::ModelIndex& = GUI::ModelIndex()) const override { return Column::__Column_Count; } virtual GUI::Variant data(const GUI::ModelIndex& index, GUI::ModelRole role) const override { auto& suggestion = m_suggestions.at(index.row()); if (role == GUI::ModelRole::Display) { if (index.column() == Column::Name) return suggestion; if (index.column() == Column::Icon) { if (suggestion.ends_with(".cpp")) return *s_cplusplus_icon; if (suggestion.ends_with(".frm")) return *s_form_icon; if (suggestion.ends_with(".h")) return *s_header_icon; if (suggestion.ends_with(".hsp")) return *s_hackstudio_icon; return *s_file_icon; } } return {}; } virtual void update() override {}; private: Vector m_suggestions; }; Locator::Locator() { if (!s_cplusplus_icon) { s_file_icon = Gfx::Bitmap::load_from_file("/res/icons/16x16/filetype-unknown.png"); s_cplusplus_icon = Gfx::Bitmap::load_from_file("/res/icons/16x16/filetype-cplusplus.png"); s_header_icon = Gfx::Bitmap::load_from_file("/res/icons/16x16/filetype-header.png"); s_form_icon = Gfx::Bitmap::load_from_file("/res/icons/16x16/filetype-form.png"); s_hackstudio_icon = Gfx::Bitmap::load_from_file("/res/icons/16x16/filetype-hackstudio.png"); } set_layout(); set_size_policy(GUI::SizePolicy::Fill, GUI::SizePolicy::Fixed); set_preferred_size(0, 20); m_textbox = add(); m_textbox->on_change = [this] { update_suggestions(); }; m_textbox->on_escape_pressed = [this] { m_popup_window->hide(); }; m_textbox->on_up_pressed = [this] { GUI::ModelIndex new_index = m_suggestion_view->selection().first(); if (new_index.is_valid()) new_index = m_suggestion_view->model()->index(new_index.row() - 1); else new_index = m_suggestion_view->model()->index(0); if (m_suggestion_view->model()->is_valid(new_index)) { m_suggestion_view->selection().set(new_index); m_suggestion_view->scroll_into_view(new_index, Orientation::Vertical); } }; m_textbox->on_down_pressed = [this] { GUI::ModelIndex new_index = m_suggestion_view->selection().first(); if (new_index.is_valid()) new_index = m_suggestion_view->model()->index(new_index.row() + 1); else new_index = m_suggestion_view->model()->index(0); if (m_suggestion_view->model()->is_valid(new_index)) { m_suggestion_view->selection().set(new_index); m_suggestion_view->scroll_into_view(new_index, Orientation::Vertical); } }; m_textbox->on_return_pressed = [this] { auto selected_index = m_suggestion_view->selection().first(); if (!selected_index.is_valid()) return; open_suggestion(selected_index); }; m_popup_window = GUI::Window::construct(); // FIXME: This is obviously not a tooltip window, but it's the closest thing to what we want atm. m_popup_window->set_window_type(GUI::WindowType::Tooltip); m_popup_window->set_rect(0, 0, 500, 200); m_suggestion_view = m_popup_window->set_main_widget(); m_suggestion_view->set_column_headers_visible(false); m_suggestion_view->on_activation = [this](auto& index) { open_suggestion(index); }; } Locator::~Locator() { } void Locator::open_suggestion(const GUI::ModelIndex& index) { auto filename_index = m_suggestion_view->model()->index(index.row(), LocatorSuggestionModel::Column::Name); auto filename = filename_index.data().to_string(); open_file(filename); close(); } void Locator::open() { m_textbox->set_focus(true); if (!m_textbox->text().is_empty()) { m_textbox->select_all(); m_popup_window->show(); } } void Locator::close() { m_popup_window->hide(); } void Locator::update_suggestions() { auto typed_text = m_textbox->text(); Vector suggestions; project().for_each_text_file([&](auto& file) { if (file.name().contains(typed_text)) suggestions.append(file.name()); }); dbg() << "I have " << suggestions.size() << " suggestion(s):"; for (auto& s : suggestions) { dbg() << " " << s; } bool has_suggestions = !suggestions.is_empty(); m_suggestion_view->set_model(adopt(*new LocatorSuggestionModel(move(suggestions)))); if (!has_suggestions) m_suggestion_view->selection().clear(); else m_suggestion_view->selection().set(m_suggestion_view->model()->index(0)); m_popup_window->move_to(screen_relative_rect().top_left().translated(0, -m_popup_window->height())); dbg() << "Popup rect: " << m_popup_window->rect(); m_popup_window->show(); } }