diff --git a/DevTools/HackStudio/AutoCompleteBox.cpp b/DevTools/HackStudio/AutoCompleteBox.cpp new file mode 100644 index 0000000000..82f0190894 --- /dev/null +++ b/DevTools/HackStudio/AutoCompleteBox.cpp @@ -0,0 +1,166 @@ +/* + * Copyright (c) 2020, Itamar S. + * 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 "AutoCompleteBox.h" +#include "Editor.h" +#include +#include +#include +#include + +namespace HackStudio { + +static RefPtr s_cplusplus_icon; + +class AutoCompleteSuggestionModel final : public GUI::Model { +public: + explicit AutoCompleteSuggestionModel(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) { + // TODO: Have separate icons for fields, functions, methods etc + if (suggestion.ends_with(".cpp")) + return *s_cplusplus_icon; + return *s_cplusplus_icon; + } + } + return {}; + } + virtual void update() override {}; + +private: + Vector m_suggestions; +}; + +AutoCompleteBox::~AutoCompleteBox() {} + +AutoCompleteBox::AutoCompleteBox(WeakPtr editor) + : m_editor(move(editor)) +{ + if (!s_cplusplus_icon) { + s_cplusplus_icon = Gfx::Bitmap::load_from_file("/res/icons/16x16/filetype-cplusplus.png"); + } + + m_popup_window = GUI::Window::construct(); + m_popup_window->set_window_type(GUI::WindowType::Tooltip); + m_popup_window->set_rect(0, 0, 200, 100); + + m_suggestion_view = m_popup_window->set_main_widget(); + m_suggestion_view->set_column_headers_visible(false); +} + +void AutoCompleteBox::update_suggestions(String partial_input, Vector&& suggestions) +{ + m_partial_input = partial_input; + if (suggestions.is_empty()) + return; + + bool has_suggestions = !suggestions.is_empty(); + m_suggestion_view->set_model(adopt(*new AutoCompleteSuggestionModel(move(suggestions)))); + + if (!has_suggestions) + m_suggestion_view->selection().clear(); + else + m_suggestion_view->selection().set(m_suggestion_view->model()->index(0)); +} + +void AutoCompleteBox::show(Gfx::IntPoint suggstion_box_location) +{ + m_popup_window->move_to(suggstion_box_location); + m_popup_window->show(); +} + +void AutoCompleteBox::close() +{ + m_partial_input.empty(); + m_popup_window->hide(); +} + +void AutoCompleteBox::next_suggestion() +{ + 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); + } +} + +void AutoCompleteBox::previous_suggestion() +{ + 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); + } +} + +void AutoCompleteBox::apply_suggestion() +{ + if (m_editor.is_null()) + return; + + auto selected_index = m_suggestion_view->selection().first(); + if (!selected_index.is_valid()) + return; + + auto suggestion_index = m_suggestion_view->model()->index(selected_index.row(), AutoCompleteSuggestionModel::Column::Name); + auto suggestion = suggestion_index.data().to_string(); + + ASSERT(!m_partial_input.is_null()); + ASSERT(suggestion.starts_with(m_partial_input)); + + auto completion = suggestion.substring(m_partial_input.length(), suggestion.length() - m_partial_input.length()); + m_editor->insert_at_cursor_or_replace_selection(completion); +} + +}; diff --git a/DevTools/HackStudio/AutoCompleteBox.h b/DevTools/HackStudio/AutoCompleteBox.h new file mode 100644 index 0000000000..0fe715aa12 --- /dev/null +++ b/DevTools/HackStudio/AutoCompleteBox.h @@ -0,0 +1,56 @@ +/* + * Copyright (c) 2020, Itamar S. + * 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. + */ + +#pragma once + +#include +#include + +namespace HackStudio { + +class Editor; +class AutoCompleteBox final { +public: + explicit AutoCompleteBox(WeakPtr); + ~AutoCompleteBox(); + + void update_suggestions(const String partial_input, Vector&& suggestions); + void show(Gfx::IntPoint suggstion_box_location); + void close(); + + void next_suggestion(); + void previous_suggestion(); + void apply_suggestion(); + +private: + void complete_suggestion(const GUI::ModelIndex&); + + WeakPtr m_editor; + RefPtr m_popup_window; + RefPtr m_suggestion_view; + String m_partial_input; +}; +} diff --git a/DevTools/HackStudio/CMakeLists.txt b/DevTools/HackStudio/CMakeLists.txt index 684de9fdf5..356de87621 100644 --- a/DevTools/HackStudio/CMakeLists.txt +++ b/DevTools/HackStudio/CMakeLists.txt @@ -1,4 +1,5 @@ set(SOURCES + AutoCompleteBox.cpp CodeDocument.cpp CppAutoComplete.cpp CursorTool.cpp