/*
** Copyright (C) 2008-2020 Dirk-Jan C. Binnema <djcb@djcbsoftware.nl>
**
** This program is free software; you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation; either version 3 of the License, or
** (at your option) any later version.
**
** 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 General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with this program; if not, write to the Free Software Foundation,
** Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
**
*/

#include <stdlib.h>
#include <iostream>
#include <string.h>
#include <errno.h>
#include <xapian.h>

#include "mu-msg-fields.h"
#include "mu-msg-doc.hh"

#include "utils/mu-util.h"
#include "utils/mu-str.h"
#include "utils/mu-date.h"
#include "utils/mu-utils.hh"
#include "utils/mu-xapian-utils.hh"

using namespace Mu;

struct Mu::MuMsgDoc {
	MuMsgDoc(Xapian::Document* doc) : _doc(doc) {}
	~MuMsgDoc() { delete _doc; }
	const Xapian::Document doc() const { return *_doc; }

private:
	Xapian::Document* _doc;
};

MuMsgDoc*
Mu::mu_msg_doc_new(XapianDocument* doc, GError** err)
{
	g_return_val_if_fail(doc, NULL);
	MuMsgDoc* mdoc =
	    xapian_try([&] { return new MuMsgDoc((Xapian::Document*)doc); }, (MuMsgDoc*)nullptr);

	if (!mdoc)
		mu_util_g_set_error(err, MU_ERROR_INTERNAL, "failed to create message doc");
	return mdoc;
}

void
Mu::mu_msg_doc_destroy(MuMsgDoc* self)
{
	xapian_try([&] { delete self; });
}

gchar*
Mu::mu_msg_doc_get_str_field(MuMsgDoc* self, MuMsgFieldId mfid)
{
	g_return_val_if_fail(self, NULL);
	g_return_val_if_fail(mu_msg_field_id_is_valid(mfid), NULL);

	// disable this check:
	//    g_return_val_if_fail (mu_msg_field_is_string(mfid), NULL);
	// because it's useful to get numerical field as strings,
	// for example when sorting (which is much faster if don't
	// have to convert to numbers first, esp. when it's a date
	// time_t)

	return xapian_try(
	    [&] {
		    const std::string s(self->doc().get_value(mfid));
		    return s.empty() ? NULL : g_strdup(s.c_str());
	    },
	    (gchar*)nullptr);
}

GSList*
Mu::mu_msg_doc_get_str_list_field(MuMsgDoc* self, MuMsgFieldId mfid)
{
	g_return_val_if_fail(self, NULL);
	g_return_val_if_fail(mu_msg_field_id_is_valid(mfid), NULL);
	g_return_val_if_fail(mu_msg_field_is_string_list(mfid), NULL);

	return xapian_try(
	    [&] {
		    /* return a comma-separated string as a GSList */
		    const std::string s(self->doc().get_value(mfid));
		    return s.empty() ? NULL : mu_str_to_list(s.c_str(), ',', TRUE);
	    },
	    (GSList*)nullptr);
}

gint64
Mu::mu_msg_doc_get_num_field(MuMsgDoc* self, MuMsgFieldId mfid)
{
	g_return_val_if_fail(self, -1);
	g_return_val_if_fail(mu_msg_field_id_is_valid(mfid), -1);
	g_return_val_if_fail(mu_msg_field_is_numeric(mfid), -1);

	return xapian_try(
	    [&] {
		    const std::string s(self->doc().get_value(mfid));
		    if (s.empty())
			    return (gint64)0;
		    else if (mfid == MU_MSG_FIELD_ID_DATE || mfid == MU_MSG_FIELD_ID_SIZE)
			    return static_cast<gint64>(strtol(s.c_str(), NULL, 10));
		    else {
			    return static_cast<gint64>(Xapian::sortable_unserialise(s));
		    }
	    },
	    (gint64)-1);
}