mirror of
https://github.com/RGBCube/serenity
synced 2025-05-22 07:25:07 +00:00
fc6d051dfd
The LexicalPath instance methods dirname(), basename(), title() and extension() will be changed to return StringView const& in a further commit. Due to this, users creating temporary LexicalPath objects just to call one of those getters will recieve a StringView const& pointing to a possible freed buffer. To avoid this, static methods for those APIs have been added, which will return a String by value to avoid those problems. All cases where temporary LexicalPath objects have been used as described above haven been changed to use the static APIs.
283 lines
11 KiB
C++
283 lines
11 KiB
C++
/*
|
|
* Copyright (c) 2020-2021, the SerenityOS developers.
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#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/StackWidget.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");
|
|
m_data_preview_error_label = m_page->body_widget().find_descendant_of_type_named<GUI::Label>("data_preview_error_label");
|
|
m_data_preview_widget = m_page->body_widget().find_descendant_of_type_named<GUI::StackWidget>("data_preview_widget");
|
|
|
|
m_quote_escape_combo_box->set_model(GUI::ItemListModel<String>::create(m_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 = [&] {
|
|
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 = [&] {
|
|
if (m_quote_other_radio->is_checked())
|
|
update_preview();
|
|
};
|
|
m_quote_escape_combo_box->on_change = [&](auto&, 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() | Reader::ParserBehaviour::Lenient;
|
|
|
|
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, move(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);
|
|
m_data_preview_error_label->set_text("Could not read the given file");
|
|
m_data_preview_widget->set_active_widget(m_data_preview_error_label);
|
|
return;
|
|
}
|
|
|
|
auto& reader = *m_previously_made_reader;
|
|
if (reader.has_error()) {
|
|
m_data_preview_table_view->set_model(nullptr);
|
|
m_data_preview_error_label->set_text(String::formatted("XSV parse error:\n{}", reader.error_string()));
|
|
m_data_preview_widget->set_active_widget(m_data_preview_error_label);
|
|
return;
|
|
}
|
|
|
|
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_widget->set_active_widget(m_data_preview_table_view);
|
|
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()) {
|
|
reader->parse();
|
|
if (reader.value().has_error())
|
|
return String::formatted("CSV Import failed: {}", reader.value().error_string());
|
|
|
|
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;
|
|
|
|
if (auto sheet = Sheet::from_json(sheet_json.as_object(), workbook))
|
|
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::basename(file.filename())));
|
|
|
|
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();
|
|
}
|
|
}
|
|
|
|
};
|