|
@@ -0,0 +1,294 @@
|
|
|
+/*
|
|
|
+ * Copyright (c) 2020-2021, the SerenityOS developers.
|
|
|
+ * 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 "ImportDialog.h"
|
|
|
+#include "Spreadsheet.h"
|
|
|
+#include <AK/JsonArray.h>
|
|
|
+#include <AK/JsonObject.h>
|
|
|
+#include <AK/JsonParser.h>
|
|
|
+#include <AK/LexicalPath.h>
|
|
|
+#include <Applications/Spreadsheet/CSVImportGML.h>
|
|
|
+#include <Applications/Spreadsheet/FormatSelectionPageGML.h>
|
|
|
+#include <LibCore/File.h>
|
|
|
+#include <LibGUI/Application.h>
|
|
|
+#include <LibGUI/CheckBox.h>
|
|
|
+#include <LibGUI/ComboBox.h>
|
|
|
+#include <LibGUI/ItemListModel.h>
|
|
|
+#include <LibGUI/RadioButton.h>
|
|
|
+#include <LibGUI/TableView.h>
|
|
|
+#include <LibGUI/TextBox.h>
|
|
|
+#include <LibGUI/Wizards/AbstractWizardPage.h>
|
|
|
+#include <LibGUI/Wizards/WizardDialog.h>
|
|
|
+#include <LibGUI/Wizards/WizardPage.h>
|
|
|
+
|
|
|
+namespace Spreadsheet {
|
|
|
+
|
|
|
+CSVImportDialogPage::CSVImportDialogPage(StringView csv)
|
|
|
+ : m_csv(csv)
|
|
|
+{
|
|
|
+ m_page = GUI::WizardPage::construct(
|
|
|
+ "CSV Import Options",
|
|
|
+ "Please select the options for the csv file you wish to import");
|
|
|
+
|
|
|
+ m_page->body_widget().load_from_gml(csv_import_gml);
|
|
|
+ m_page->set_is_final_page(true);
|
|
|
+
|
|
|
+ m_delimiter_comma_radio = m_page->body_widget().find_descendant_of_type_named<GUI::RadioButton>("delimiter_comma_radio");
|
|
|
+ m_delimiter_semicolon_radio = m_page->body_widget().find_descendant_of_type_named<GUI::RadioButton>("delimiter_semicolon_radio");
|
|
|
+ m_delimiter_tab_radio = m_page->body_widget().find_descendant_of_type_named<GUI::RadioButton>("delimiter_tab_radio");
|
|
|
+ m_delimiter_space_radio = m_page->body_widget().find_descendant_of_type_named<GUI::RadioButton>("delimiter_space_radio");
|
|
|
+ m_delimiter_other_radio = m_page->body_widget().find_descendant_of_type_named<GUI::RadioButton>("delimiter_other_radio");
|
|
|
+ m_delimiter_other_text_box = m_page->body_widget().find_descendant_of_type_named<GUI::TextBox>("delimiter_other_text_box");
|
|
|
+ m_quote_single_radio = m_page->body_widget().find_descendant_of_type_named<GUI::RadioButton>("quote_single_radio");
|
|
|
+ m_quote_double_radio = m_page->body_widget().find_descendant_of_type_named<GUI::RadioButton>("quote_double_radio");
|
|
|
+ m_quote_other_radio = m_page->body_widget().find_descendant_of_type_named<GUI::RadioButton>("quote_other_radio");
|
|
|
+ m_quote_other_text_box = m_page->body_widget().find_descendant_of_type_named<GUI::TextBox>("quote_other_text_box");
|
|
|
+ m_quote_escape_combo_box = m_page->body_widget().find_descendant_of_type_named<GUI::ComboBox>("quote_escape_combo_box");
|
|
|
+ m_read_header_check_box = m_page->body_widget().find_descendant_of_type_named<GUI::CheckBox>("read_header_check_box");
|
|
|
+ m_trim_leading_field_spaces_check_box = m_page->body_widget().find_descendant_of_type_named<GUI::CheckBox>("trim_leading_field_spaces_check_box");
|
|
|
+ m_trim_trailing_field_spaces_check_box = m_page->body_widget().find_descendant_of_type_named<GUI::CheckBox>("trim_trailing_field_spaces_check_box");
|
|
|
+ m_data_preview_table_view = m_page->body_widget().find_descendant_of_type_named<GUI::TableView>("data_preview_table_view");
|
|
|
+
|
|
|
+ Vector<String> quote_escape_items {
|
|
|
+ // Note: Keep in sync with Reader::ParserTraits::QuoteEscape.
|
|
|
+ "Repeat",
|
|
|
+ "Backslash",
|
|
|
+ };
|
|
|
+ m_quote_escape_combo_box->set_model(GUI::ItemListModel<String>::create(quote_escape_items));
|
|
|
+
|
|
|
+ // By default, use commas, double quotes with repeat, and disable headers.
|
|
|
+ m_delimiter_comma_radio->set_checked(true);
|
|
|
+ m_quote_double_radio->set_checked(true);
|
|
|
+ m_quote_escape_combo_box->set_selected_index(0); // Repeat
|
|
|
+
|
|
|
+ m_delimiter_comma_radio->on_checked = [&](auto) { update_preview(); };
|
|
|
+ m_delimiter_semicolon_radio->on_checked = [&](auto) { update_preview(); };
|
|
|
+ m_delimiter_tab_radio->on_checked = [&](auto) { update_preview(); };
|
|
|
+ m_delimiter_space_radio->on_checked = [&](auto) { update_preview(); };
|
|
|
+ m_delimiter_other_radio->on_checked = [&](auto) { update_preview(); };
|
|
|
+ m_delimiter_other_text_box->on_change = [&](auto&) {
|
|
|
+ if (m_delimiter_other_radio->is_checked())
|
|
|
+ update_preview();
|
|
|
+ };
|
|
|
+ m_quote_single_radio->on_checked = [&](auto) { update_preview(); };
|
|
|
+ m_quote_double_radio->on_checked = [&](auto) { update_preview(); };
|
|
|
+ m_quote_other_radio->on_checked = [&](auto) { update_preview(); };
|
|
|
+ m_quote_other_text_box->on_change = [&](auto&) {
|
|
|
+ if (m_quote_other_radio->is_checked())
|
|
|
+ update_preview();
|
|
|
+ };
|
|
|
+ m_quote_escape_combo_box->on_change = [&](auto&) { update_preview(); };
|
|
|
+ m_read_header_check_box->on_checked = [&](auto) { update_preview(); };
|
|
|
+ m_trim_leading_field_spaces_check_box->on_checked = [&](auto) { update_preview(); };
|
|
|
+ m_trim_trailing_field_spaces_check_box->on_checked = [&](auto) { update_preview(); };
|
|
|
+
|
|
|
+ update_preview();
|
|
|
+}
|
|
|
+
|
|
|
+auto CSVImportDialogPage::make_reader() -> Optional<Reader::XSV>
|
|
|
+{
|
|
|
+ String delimiter;
|
|
|
+ String quote;
|
|
|
+ Reader::ParserTraits::QuoteEscape quote_escape;
|
|
|
+
|
|
|
+ // Delimiter
|
|
|
+ if (m_delimiter_other_radio->is_checked())
|
|
|
+ delimiter = m_delimiter_other_text_box->text();
|
|
|
+ else if (m_delimiter_comma_radio->is_checked())
|
|
|
+ delimiter = ",";
|
|
|
+ else if (m_delimiter_semicolon_radio->is_checked())
|
|
|
+ delimiter = ";";
|
|
|
+ else if (m_delimiter_tab_radio->is_checked())
|
|
|
+ delimiter = "\t";
|
|
|
+ else if (m_delimiter_space_radio->is_checked())
|
|
|
+ delimiter = " ";
|
|
|
+ else
|
|
|
+ return {};
|
|
|
+
|
|
|
+ // Quote separator
|
|
|
+ if (m_quote_other_radio->is_checked())
|
|
|
+ quote = m_quote_other_text_box->text();
|
|
|
+ else if (m_quote_single_radio->is_checked())
|
|
|
+ quote = "'";
|
|
|
+ else if (m_quote_double_radio->is_checked())
|
|
|
+ quote = "\"";
|
|
|
+ else
|
|
|
+ return {};
|
|
|
+
|
|
|
+ // Quote escape
|
|
|
+ auto index = m_quote_escape_combo_box->selected_index();
|
|
|
+ if (index == 0)
|
|
|
+ quote_escape = Reader::ParserTraits::Repeat;
|
|
|
+ else if (index == 1)
|
|
|
+ quote_escape = Reader::ParserTraits::Backslash;
|
|
|
+ else
|
|
|
+ return {};
|
|
|
+
|
|
|
+ auto should_read_headers = m_read_header_check_box->is_checked();
|
|
|
+ auto should_trim_leading = m_trim_leading_field_spaces_check_box->is_checked();
|
|
|
+ auto should_trim_trailing = m_trim_trailing_field_spaces_check_box->is_checked();
|
|
|
+
|
|
|
+ if (quote.is_empty() || delimiter.is_empty())
|
|
|
+ return {};
|
|
|
+
|
|
|
+ Reader::ParserTraits traits {
|
|
|
+ move(delimiter),
|
|
|
+ move(quote),
|
|
|
+ quote_escape,
|
|
|
+ };
|
|
|
+
|
|
|
+ auto behaviours = Reader::default_behaviours();
|
|
|
+
|
|
|
+ if (should_read_headers)
|
|
|
+ behaviours = behaviours | Reader::ParserBehaviour::ReadHeaders;
|
|
|
+ if (should_trim_leading)
|
|
|
+ behaviours = behaviours | Reader::ParserBehaviour::TrimLeadingFieldSpaces;
|
|
|
+ if (should_trim_trailing)
|
|
|
+ behaviours = behaviours | Reader::ParserBehaviour::TrimTrailingFieldSpaces;
|
|
|
+
|
|
|
+ return Reader::XSV(m_csv, traits, behaviours);
|
|
|
+};
|
|
|
+
|
|
|
+void CSVImportDialogPage::update_preview()
|
|
|
+
|
|
|
+{
|
|
|
+ m_previously_made_reader = make_reader();
|
|
|
+ if (!m_previously_made_reader.has_value()) {
|
|
|
+ m_data_preview_table_view->set_model(nullptr);
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ auto& reader = *m_previously_made_reader;
|
|
|
+ auto headers = reader.headers();
|
|
|
+
|
|
|
+ m_data_preview_table_view->set_model(
|
|
|
+ GUI::ItemListModel<Reader::XSV::Row, Reader::XSV, Vector<String>>::create(reader, headers, min(8ul, reader.size())));
|
|
|
+ m_data_preview_table_view->update();
|
|
|
+}
|
|
|
+
|
|
|
+Result<NonnullRefPtrVector<Sheet>, String> ImportDialog::make_and_run_for(StringView mime, Core::File& file, Workbook& workbook)
|
|
|
+{
|
|
|
+ auto wizard = GUI::WizardDialog::construct(GUI::Application::the()->active_window());
|
|
|
+ wizard->set_title("File Import Wizard");
|
|
|
+ wizard->set_icon(GUI::Icon::default_icon("app-spreadsheet").bitmap_for_size(16));
|
|
|
+
|
|
|
+ auto import_xsv = [&]() -> Result<NonnullRefPtrVector<Sheet>, String> {
|
|
|
+ auto contents = file.read_all();
|
|
|
+ CSVImportDialogPage page { contents };
|
|
|
+ wizard->replace_page(page.page());
|
|
|
+ auto result = wizard->exec();
|
|
|
+
|
|
|
+ if (result == GUI::Dialog::ExecResult::ExecOK) {
|
|
|
+ auto& reader = page.reader();
|
|
|
+
|
|
|
+ NonnullRefPtrVector<Sheet> sheets;
|
|
|
+
|
|
|
+ if (reader.has_value()) {
|
|
|
+ auto sheet = Sheet::from_xsv(reader.value(), workbook);
|
|
|
+ if (sheet)
|
|
|
+ sheets.append(sheet.release_nonnull());
|
|
|
+ }
|
|
|
+
|
|
|
+ return sheets;
|
|
|
+ } else {
|
|
|
+ return String { "CSV Import was cancelled" };
|
|
|
+ }
|
|
|
+ };
|
|
|
+
|
|
|
+ auto import_worksheet = [&]() -> Result<NonnullRefPtrVector<Sheet>, String> {
|
|
|
+ auto json_value_option = JsonParser(file.read_all()).parse();
|
|
|
+ if (!json_value_option.has_value()) {
|
|
|
+ StringBuilder sb;
|
|
|
+ sb.append("Failed to parse ");
|
|
|
+ sb.append(file.filename());
|
|
|
+
|
|
|
+ return sb.to_string();
|
|
|
+ }
|
|
|
+
|
|
|
+ auto& json_value = json_value_option.value();
|
|
|
+ if (!json_value.is_array()) {
|
|
|
+ StringBuilder sb;
|
|
|
+ sb.append("Did not find a spreadsheet in ");
|
|
|
+ sb.append(file.filename());
|
|
|
+
|
|
|
+ return sb.to_string();
|
|
|
+ }
|
|
|
+
|
|
|
+ NonnullRefPtrVector<Sheet> sheets;
|
|
|
+
|
|
|
+ auto& json_array = json_value.as_array();
|
|
|
+ json_array.for_each([&](auto& sheet_json) {
|
|
|
+ if (!sheet_json.is_object())
|
|
|
+ return IterationDecision::Continue;
|
|
|
+
|
|
|
+ auto sheet = Sheet::from_json(sheet_json.as_object(), workbook);
|
|
|
+ if (!sheet)
|
|
|
+ return IterationDecision::Continue;
|
|
|
+
|
|
|
+ sheets.append(sheet.release_nonnull());
|
|
|
+
|
|
|
+ return IterationDecision::Continue;
|
|
|
+ });
|
|
|
+
|
|
|
+ return sheets;
|
|
|
+ };
|
|
|
+
|
|
|
+ if (mime == "text/csv") {
|
|
|
+ return import_xsv();
|
|
|
+ } else if (mime == "text/plain" && file.filename().ends_with(".sheets")) {
|
|
|
+ return import_worksheet();
|
|
|
+ } else {
|
|
|
+ auto page = GUI::WizardPage::construct(
|
|
|
+ "Import File Format",
|
|
|
+ String::formatted("Select the format you wish to import '{}' as", LexicalPath { file.filename() }.basename()));
|
|
|
+
|
|
|
+ page->on_next_page = [] { return nullptr; };
|
|
|
+
|
|
|
+ page->body_widget().load_from_gml(select_format_page_gml);
|
|
|
+ auto format_combo_box = page->body_widget().find_descendant_of_type_named<GUI::ComboBox>("select_format_page_format_combo_box");
|
|
|
+
|
|
|
+ Vector<String> supported_formats {
|
|
|
+ "CSV (text/csv)",
|
|
|
+ "Spreadsheet Worksheet",
|
|
|
+ };
|
|
|
+ format_combo_box->set_model(GUI::ItemListModel<String>::create(supported_formats));
|
|
|
+
|
|
|
+ wizard->push_page(page);
|
|
|
+
|
|
|
+ if (wizard->exec() != GUI::Dialog::ExecResult::ExecOK)
|
|
|
+ return String { "Import was cancelled" };
|
|
|
+
|
|
|
+ if (format_combo_box->selected_index() == 0)
|
|
|
+ return import_xsv();
|
|
|
+
|
|
|
+ if (format_combo_box->selected_index() == 1)
|
|
|
+ return import_worksheet();
|
|
|
+
|
|
|
+ VERIFY_NOT_REACHED();
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+};
|