/*  This file is part of epubgrep.
 *  Copyright © 2021 tastytea <tastytea@tastytea.de>
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Affero General Public License as published by
 *  the Free Software Foundation, version 3.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Affero General Public License for more details.
 *
 *  You should have received a copy of the GNU Affero General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "output.hpp"

#include <boost/locale/message.hpp>
#include <fmt/format.h>
#include <fmt/ostream.h> // For compatibility with fmt 4.
#include <nlohmann/json.hpp>
#include <termcolor/termcolor.hpp>

#include <iostream>
#include <sstream>

namespace epubgrep::output
{

using boost::locale::translate;
using fmt::format;
using std::cout;

void print_matches(const std::vector<search::match> &matches,
                   const options::options &opts, bool single_file)
{
    fs::path last_epub;
    for (const auto &match : matches)
    {
        if (!single_file && !opts.no_fn_fs)
        {
            if (match.filepath_epub != last_epub)
            {
                if (!opts.nocolor)
                {
                    cout << termcolor::yellow;
                }
                cout << format(translate("  In {0:s}: \n").str(),
                               fs::relative(match.filepath_epub));
                last_epub = match.filepath_epub;
                if (!opts.nocolor)
                {
                    cout << termcolor::reset;
                }
            }
        }

        std::vector<std::string> metadata;
        if (!opts.no_fn_epub)
        {
            metadata.emplace_back(match.filepath_inside);
        }
        if (!match.headline.empty())
        {
            // <https://github.com/ikalnytskyi/termcolor/issues/45>
            if (!opts.nocolor && termcolor::_internal::is_colorized(cout))
            {
                std::stringstream ss;
                ss << termcolor::colorize << termcolor::underline
                   << match.headline << termcolor::reset << termcolor::italic;
                metadata.emplace_back(ss.str());
            }
            else
            {
                metadata.emplace_back(match.headline);
            }
        }
        if (!match.page.empty())
        {
            metadata.emplace_back("page " + match.page);
        }
        if (!opts.nocolor)
        {
            cout << termcolor::italic;
        }
        for (const auto &part : metadata)
        {
            cout << part;
            if (part != *(metadata.rbegin()))
            {
                cout << ", ";
            }
        }
        cout << ": ";
        if (!opts.nocolor)
        {
            cout << termcolor::reset;
        }
        cout << match.context.first;
        if (!opts.nocolor)
        {
            cout << termcolor::bright_magenta;
        }
        cout << match.text;
        if (!opts.nocolor)
        {
            cout << termcolor::reset;
        }
        cout << match.context.second << '\n';
    }
}

void json(const std::vector<search::match> &matches)
{
    nlohmann::json json;

    for (const auto &match : matches)
    {
        json[match.filepath_epub].push_back(
            {{"filepath", match.filepath_inside},
             {"match", match.text},
             {"context", match.context},
             {"headline", match.headline},
             {"page", match.page}});
    }
    std::cout << json.dump(2) << '\n';
}

} // namespace epubgrep::output