// This file is part of ICU4X. For terms of use, please see the file
// called LICENSE at the top level of the ICU4X source tree
// (online at: https://github.com/unicode-org/icu4x/blob/main/LICENSE ).

#include "../../include/ICU4XCaseMapper.hpp"
#include "../../include/ICU4XLogger.hpp"
#include "../../include/ICU4XDataProvider.hpp"
#include "../../include/ICU4XCodePointSetBuilder.hpp"
#include "../../include/ICU4XTitlecaseOptionsV1.hpp"

#include <iostream>

int main() {
    ICU4XLogger::init_simple_logger();
    ICU4XLocale und = ICU4XLocale::create_from_string("und").ok().value();
    ICU4XLocale greek = ICU4XLocale::create_from_string("el").ok().value();
    ICU4XLocale turkish = ICU4XLocale::create_from_string("tr").ok().value();
    ICU4XDataProvider dp = ICU4XDataProvider::create_compiled();

    ICU4XCaseMapper cm = ICU4XCaseMapper::create(dp).ok().value();

    ICU4XTitlecaseOptionsV1 tc_options = ICU4XTitlecaseOptionsV1::default_options();

    std::string out = cm.lowercase("hEllO WorLd", und).ok().value();
    std::cout << "Lowercased value is " << out << std::endl;
    if (out != "hello world") {
        std::cout << "Output does not match expected output" << std::endl;
        return 1;
    }
    out = cm.uppercase("hEllO WorLd", und).ok().value();
    std::cout << "Uppercased value is " << out << std::endl;
    if (out != "HELLO WORLD") {
        std::cout << "Output does not match expected output" << std::endl;
        return 1;
    }

    out = cm.titlecase_segment_with_only_case_data_v1("hEllO WorLd", und, tc_options).ok().value();
    std::cout << "Titlecased value is " << out << std::endl;
    if (out != "Hello world") {
        std::cout << "Output does not match expected output" << std::endl;
        return 1;
    }


    // locale-specific behavior

    out = cm.uppercase("Γειά σου Κόσμε", und).ok().value();
    std::cout << "Uppercased value is " << out << std::endl;
    if (out != "ΓΕΙΆ ΣΟΥ ΚΌΣΜΕ") {
        std::cout << "Output does not match expected output" << std::endl;
        return 1;
    }

    out = cm.uppercase("Γειά σου Κόσμε", greek).ok().value();
    std::cout << "Uppercased value is " << out << std::endl;
    if (out != "ΓΕΙΑ ΣΟΥ ΚΟΣΜΕ") {
        std::cout << "Output does not match expected output" << std::endl;
        return 1;
    }

    out = cm.uppercase("istanbul", und).ok().value();
    std::cout << "Uppercased value is " << out << std::endl;
    if (out != "ISTANBUL") {
        std::cout << "Output does not match expected output" << std::endl;
        return 1;
    }

    out = cm.uppercase("istanbul", turkish).ok().value();
    std::cout << "Uppercased value is " << out << std::endl;
    if (out != "İSTANBUL") {
        std::cout << "Output does not match expected output" << std::endl;
        return 1;
    }


    out = cm.fold("ISTANBUL").ok().value();
    std::cout << "Folded value is " << out << std::endl;
    if (out != "istanbul") {
        std::cout << "Output does not match expected output" << std::endl;
        return 1;
    }

    out = cm.fold_turkic("ISTANBUL").ok().value();
    std::cout << "Turkic-folded value is " << out << std::endl;
    if (out != "ıstanbul") {
        std::cout << "Output does not match expected output" << std::endl;
        return 1;
    }

    ICU4XCodePointSetBuilder builder = ICU4XCodePointSetBuilder::create();

    cm.add_case_closure_to('s', builder);

    auto set = builder.build();

    if (set.contains('s')) {
        std::cout << "Set contains 's', which was not expected" << std::endl;
        return 1;
    } 
    if (!set.contains('S')) {
        std::cout << "Set does not 'S', which was not expected" << std::endl;
        return 1;
    } 
    if (!set.contains(U'ſ')) {
        std::cout << "Set does not 'S', which was not expected" << std::endl;
        return 1;
    } 
    return 0;
}