#pragma once

#include <iostream>
#include <cxxtest/TestSuite.h>

#include <helpers/path_helper.hpp>
#include <xlnt/utils/exceptions.hpp>
#include <xlnt/workbook/workbook.hpp>

class test_write_workbook : public CxxTest::TestSuite
{
public:
    void test_write_auto_filter()
    {
        xlnt::workbook wb;
        auto ws = wb.get_active_sheet();

        ws.get_cell("F42").set_value("hello");
        ws.auto_filter("A1:F1");

		TS_ASSERT(xml_helper::file_matches_workbook_part(
			path_helper::get_data_directory("writer/expected/workbook_auto_filter.xml"),
			wb, xlnt::path("xl/workbook.xml")));
    }
    
    void test_write_hidden_worksheet()
    {
		xlnt::workbook wb;
        auto ws = wb.get_active_sheet();

        ws.set_sheet_state(xlnt::sheet_state::hidden);
        wb.create_sheet();

        std::string expected_string =
        "<workbook xmlns=\"http://schemas.openxmlformats.org/spreadsheetml/2006/main\" xmlns:r=\"http://schemas.openxmlformats.org/officeDocument/2006/relationships\">"
        "    <workbookPr/>"
        "    <bookViews>"
        "        <workbookView activeTab=\"0\"/>"
        "    </bookViews>"
        "    <sheets>"
        "        <sheet name=\"Sheet\" sheetId=\"1\" state=\"hidden\" r:id=\"rId1\"/>"
        "        <sheet name=\"Sheet1\" sheetId=\"2\" r:id=\"rId2\"/>"
        "    </sheets>"
        "    <definedNames/>"
        "    <calcPr calcId=\"124519\" fullCalcOnLoad=\"1\"/>"
        "</workbook>";

		TS_ASSERT(xml_helper::string_matches_workbook_part(expected_string,
			wb, xlnt::path("xl/workbook.xml")));
    }
    
    void test_write_hidden_single_worksheet()
    {
        xlnt::workbook wb;
        auto ws = wb.get_active_sheet();
        ws.set_sheet_state(xlnt::sheet_state::hidden);
		std::vector<std::uint8_t> trash;
        TS_ASSERT_THROWS(wb.save(trash), xlnt::no_visible_worksheets);
    }
    
    void test_write_empty_workbook()
    {
        xlnt::workbook wb;
        temporary_file file;

		TS_ASSERT(!file.get_path().exists())
		wb.save(file.get_path());
		TS_ASSERT(file.get_path().exists());
    }
    
    void test_write_virtual_workbook()
    {
        xlnt::workbook old_wb, new_wb;

        std::vector<std::uint8_t> wb_bytes;
		old_wb.save(wb_bytes);
		new_wb.load(wb_bytes);

		// TODO more tests!
    }
    
    void test_write_workbook_rels()
    {
        xlnt::workbook wb;
        auto ws = wb.get_active_sheet();
        ws.get_cell("A1").set_value("string");

		TS_ASSERT(xml_helper::file_matches_workbook_part(
			path_helper::get_data_directory("writer/expected/workbook.xml.rels"),
			wb, xlnt::path("xl/_rels/workbook.xml.rels")));
    }
    
    void test_write_workbook_part()
    {
        xlnt::workbook wb;

        auto filename = path_helper::get_data_directory("writer/expected/workbook.xml");
		TS_ASSERT(xml_helper::file_matches_workbook_part(filename, wb, xlnt::path("xl/_rels/workbook.xml.rels")));
    }
    
    void _test_write_named_range()
    {
		/*
        xlnt::workbook wb;
        auto ws = wb.get_active_sheet();
        wb.create_named_range("test_range", ws, "$A$1:$B$5");
        xlnt::workbook_serializer serializer(wb);
        pugi::xml_document xml;
        auto root = xml.root().append_child("root");
        serializer.write_named_ranges(root);
        std::string expected =
        "<root>"
            "<s:definedName xmlns:s=\"http://schemas.openxmlformats.org/spreadsheetml/2006/main\" name=\"test_range\">'Sheet'!$A$1:$B$5</s:definedName>"
        "</root>";

        TS_ASSERT(xml_helper::string_matches_document(expected, xml));
		*/
    }
    
    void test_read_workbook_code_name()
    {
//        with open(tmpl, "rb") as expected:
//        TS_ASSERT(read_workbook_code_name(expected.read()) == code_name
    }
    
    void test_write_workbook_code_name()
    {
        xlnt::workbook wb;
        wb.set_code_name("MyWB");

        std::string expected =
        "<workbook xmlns=\"http://schemas.openxmlformats.org/spreadsheetml/2006/main\" xmlns:r=\"http://schemas.openxmlformats.org/officeDocument/2006/relationships\">"
        "    <workbookPr codeName=\"MyWB\"/>"
        "    <bookViews>"
        "        <workbookView activeTab=\"0\"/>"
        "    </bookViews>"
        "    <sheets>"
        "        <sheet name=\"Sheet\" sheetId=\"1\" r:id=\"rId1\"/>"
        "    </sheets>"
        "    <definedNames/>"
        "    <calcPr calcId=\"124519\" fullCalcOnLoad=\"1\"/>"
        "</workbook>";

		TS_ASSERT(xml_helper::string_matches_workbook_part(expected, wb, xlnt::path("xl/workbook.xml")));
    }
    
    void test_write_root_rels()
    {
        xlnt::workbook wb;
        
        std::string expected =
        "<Relationships xmlns=\"http://schemas.openxmlformats.org/package/2006/relationships\">"
        "    <Relationship Id=\"rId1\" Target=\"xl/workbook.xml\" Type=\"http://schemas.openxmlformats.org/officeDocument/2006/relationships/officeDocument\"/>"
        "    <Relationship Id=\"rId2\" Target=\"docProps/core.xml\" Type=\"http://schemas.openxmlformats.org/package/2006/relationships/metadata/core-properties\"/>"
        "    <Relationship Id=\"rId3\" Target=\"docProps/app.xml\" Type=\"http://schemas.openxmlformats.org/officeDocument/2006/relationships/extended-properties\"/>"
        "</Relationships>";
        
        TS_ASSERT(xml_helper::string_matches_workbook_part(expected, wb, xlnt::path("_rels/.rels")));
    }

    void test_write_shared_strings_with_runs()
    {
        xlnt::workbook wb;
        auto ws = wb.get_active_sheet();
        auto cell = ws.get_cell("A1");
        xlnt::text_run run;
        run.set_color("color");
        run.set_family(12);
        run.set_font("font");
        run.set_scheme("scheme");
        run.set_size(13);
        run.set_string("string");
        xlnt::text text;
        text.add_run(run);
        cell.set_value(text);

        std::string expected =
        "<sst xmlns=\"http://schemas.openxmlformats.org/spreadsheetml/2006/main\" count=\"1\" uniqueCount=\"1\">"
        "  <si>"
        "    <r>"
        "      <rPr>"
        "        <sz val=\"13\"/>"
        "        <color rgb=\"color\"/>"
        "        <rFont val=\"font\"/>"
        "        <family val=\"12\"/>"
        "        <scheme val=\"scheme\"/>"
        "      </rPr>"
        "      <t>string</t>"
        "    </r>"
        "  </si>"
        "</sst>";

        TS_ASSERT(xml_helper::string_matches_workbook_part(expected, wb, 
			xlnt::constants::part_shared_strings()));
    }

    void test_write_worksheet_order()
    {
        auto path = path_helper::get_data_directory("genuine/tab_order.xlsx");

        // Load an original workbook produced by Excel
        xlnt::workbook wb_src;
		wb_src.load(path);

        // Save it to a new file, unmodified
        temporary_file file;
		wb_src.save(file.get_path());
        TS_ASSERT(file.get_path().exists());

        // Load it again
        xlnt::workbook wb_dst;
		wb_dst.load(file.get_path());

		TS_ASSERT_EQUALS(wb_src.get_sheet_titles(), wb_dst.get_sheet_titles());
    }

private:
    xlnt::workbook wb_;
};