#ifndef TGBOT_UTILS_H
#define TGBOT_UTILS_H

#include <string>
#include <vector>
#include <sstream>
#include <map>
#include <unordered_map>

#include <boost/lexical_cast.hpp>

template<typename T>
inline std::string diff(const T& test, const T& expected, std::string (*toStringFunc)(const typename T::value_type&)) {
    std::string result;
    result += "\n*** BEGIN ***	 Count: t=";
    result += boost::lexical_cast<std::string>(test.size());
    result += " e=";
    result += boost::lexical_cast<std::string>(expected.size());
    result += '\n';

    typename T::const_iterator iter1 = test.begin();
    typename T::const_iterator end1 = test.end();
    typename T::const_iterator iter2 = expected.begin();
    typename T::const_iterator end2 = expected.end();
    bool r1, r2;
    std::string s1, s2;
    std::size_t i = 0;
    do {
        r1 = iter1 != end1;
        r2 = iter2 != end2;
        if (r1) {
            s1 = toStringFunc(*iter1++);
        }
        if (r2) {
            s2 = toStringFunc(*iter2++);
        }
        if (r1 && r2 && s1 == s2) {
            result += boost::lexical_cast<std::string>(i);
            result += " [=] ";
            result += s1;
            result += "\n";
        } else {
            if (r1) {
                result += boost::lexical_cast<std::string>(i);
                result += " [t] ";
                result += s1;
                result += "\n";
            }
            if (r2) {
                result += boost::lexical_cast<std::string>(i);
                result += " [e] ";
                result += s2;
                result += "\n";
            }
        }
        ++i;
    } while (r1 || r2);

    result += "*** END ***\n";
    return result;
}

inline std::string diffS(const std::string& test, const std::string& expected) {
    std::vector<std::string> v1, v2;
    std::istringstream ss1(test);
    std::istringstream ss2(expected);
    std::string s1, s2;
    bool r1, r2;
    do {
        r1 = std::getline(ss1, s1) ? true : false;
        r2 = std::getline(ss2, s2) ? true : false;
        if (r1) {
            v1.push_back(s1);
        }
        if (r2) {
            v2.push_back(s2);
        }
    } while (r1 || r2);

    std::string (*toStringFunc)(const std::string&) =
            [](const std::string& item) -> std::string { return item; };

    return diff(v1, v2, toStringFunc);
}

inline std::string diffMSS(const std::unordered_map<std::string, std::string>& test, const std::unordered_map<std::string, std::string>& expected) {
    std::map<std::string, std::string> v1(test.begin(), test.end());
    std::map<std::string, std::string> v2(expected.begin(), expected.end());

    std::string (*toStringFunc)(const std::pair<const std::string, std::string>&) =
            [](const std::pair<const std::string, std::string>& item) -> std::string { return item.first + '=' + item.second; };

    return diff(v1, v2, toStringFunc);
}

#endif //TGBOT_UTILS_H