/* The Ruby interface to the notmuch mail library
 *
 * Copyright © 2010 Ali Polatel
 *
 * 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, see http://www.gnu.org/licenses/ .
 *
 * Author: Ali Polatel <alip@exherbo.org>
 */

#include "defs.h"

/*
 * call-seq: MESSAGE.message_id => String
 *
 * Get the message ID of 'message'.
 */
VALUE
notmuch_rb_message_get_message_id(VALUE self)
{
    const char *msgid;
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

    msgid = notmuch_message_get_message_id(message->nm_message);
    return msgid ? rb_str_new2(msgid) : Qnil;
}

/*
 * call-seq: MESSAGE.thread_id => String
 *
 * Get the thread ID of 'message'.
 */
VALUE
notmuch_rb_message_get_thread_id(VALUE self)
{
    const char *tid;
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

    tid = notmuch_message_get_thread_id(message->nm_message);
    return tid ? rb_str_new2(tid) : Qnil;
}

/*
 * call-seq: MESSAGE.replies => MESSAGES
 *
 * Get a Notmuch::Messages enumerable for all of the replies to 'message'.
 */
VALUE
notmuch_rb_message_get_replies(VALUE self)
{
    notmuch_rb_messages_t *messages;
    notmuch_rb_message_t *message;
    VALUE messagesv;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

    messagesv = Data_Make_Struct(notmuch_rb_cMessages, notmuch_rb_messages_t,
            notmuch_rb_messages_mark, notmuch_rb_messages_free, messages);
    messages->nm_messages = notmuch_message_get_replies(message->nm_message);
    messages->parent = self;

    return messages->nm_messages ? messagesv : Qnil;
}

/*
 * call-seq: MESSAGE.filename => String
 *
 * Get a filename for the email corresponding to 'message'
 */
VALUE
notmuch_rb_message_get_filename(VALUE self)
{
    const char *fname;
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

    fname = notmuch_message_get_filename(message->nm_message);
    return fname ? rb_str_new2(fname) : Qnil;
}

/*
 * call-seq: MESSAGE.get_flag(flag) => true or false
 *
 * Get a value of a flag for the email corresponding to 'message'
 */
VALUE
notmuch_rb_message_get_flag(VALUE self, VALUE flagv)
{
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

    if (!FIXNUM_P(flagv))
        rb_raise(rb_eTypeError, "Flag not a Fixnum");

    return notmuch_message_get_flag(message->nm_message, FIX2INT(flagv)) ? Qtrue : Qfalse;
}

/*
 * call-seq: MESSAGE.set_flag(flag, value) => nil
 *
 * Set a value of a flag for the email corresponding to 'message'
 */
VALUE
notmuch_rb_message_set_flag(VALUE self, VALUE flagv, VALUE valuev)
{
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

    if (!FIXNUM_P(flagv))
        rb_raise(rb_eTypeError, "Flag not a Fixnum");

    notmuch_message_set_flag(message->nm_message, FIX2INT(flagv), RTEST(valuev));
    return Qnil;
}

/*
 * call-seq: MESSAGE.date => Fixnum
 *
 * Get the date of 'message'
 */
VALUE
notmuch_rb_message_get_date(VALUE self)
{
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

    return UINT2NUM(notmuch_message_get_date(message->nm_message));
}

/*
 * call-seq: MESSAGE.header(name) => String
 *
 * Get the value of the specified header from 'message'
 */
VALUE
notmuch_rb_message_get_header(VALUE self, VALUE headerv)
{
    const char *header, *value;
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

#if !defined(RSTRING_PTR)
#define RSTRING_PTR(v) (RSTRING((v))->ptr)
#endif /* !defined(RSTRING_PTR) */

    SafeStringValue(headerv);
    header = RSTRING_PTR(headerv);

    value = notmuch_message_get_header(message->nm_message, header);
    if (!value)
        rb_raise(notmuch_rb_eMemoryError, "out of memory");

    return rb_str_new2(value);
}

/*
 * call-seq: MESSAGE.tags => TAGS
 *
 * Get a Notmuch::Tags enumerable for all of the tags of 'message'.
 */
VALUE
notmuch_rb_message_get_tags(VALUE self)
{
    notmuch_rb_message_t *message;
    notmuch_rb_tags_t *tags;
    VALUE tagsv;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

    tagsv = Data_Make_Struct(notmuch_rb_cTags, notmuch_rb_tags_t,
            notmuch_rb_tags_mark, notmuch_rb_tags_free, tags);
    tags->nm_tags = notmuch_message_get_tags(message->nm_message);
    tags->parent = self;
    if (!tags->nm_tags)
        rb_raise(notmuch_rb_eMemoryError, "out of memory");

    return tagsv;
}

/*
 * call-seq: MESSAGE.add_tag(tag) => true
 *
 * Add a tag to the 'message'
 */
VALUE
notmuch_rb_message_add_tag(VALUE self, VALUE tagv)
{
    const char *tag;
    notmuch_status_t ret;
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

#if !defined(RSTRING_PTR)
#define RSTRING_PTR(v) (RSTRING((v))->ptr)
#endif /* !defined(RSTRING_PTR) */

    SafeStringValue(tagv);
    tag = RSTRING_PTR(tagv);

    ret = notmuch_message_add_tag(message->nm_message, tag);
    notmuch_rb_status_raise(ret);
    return Qtrue;
}

/*
 * call-seq: MESSAGE.remove_tag(tag) => true
 *
 * Remove a tag from the 'message'
 */
VALUE
notmuch_rb_message_remove_tag(VALUE self, VALUE tagv)
{
    const char *tag;
    notmuch_status_t ret;
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

#if !defined(RSTRING_PTR)
#define RSTRING_PTR(v) (RSTRING((v))->ptr)
#endif /* !defined(RSTRING_PTR) */

    SafeStringValue(tagv);
    tag = RSTRING_PTR(tagv);

    ret = notmuch_message_remove_tag(message->nm_message, tag);
    notmuch_rb_status_raise(ret);
    return Qtrue;
}

/*
 * call-seq: MESSAGE.remove_all_tags => true
 *
 * Remove all tags of the 'message'
 */
VALUE
notmuch_rb_message_remove_all_tags(VALUE self)
{
    notmuch_status_t ret;
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

    ret = notmuch_message_remove_all_tags(message->nm_message);
    notmuch_rb_status_raise(ret);
    return Qtrue;
}

/*
 * call-seq: MESSAGE.freeze => true
 *
 * Freeze the 'message'
 */
VALUE
notmuch_rb_message_freeze(VALUE self)
{
    notmuch_status_t ret;
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

    ret = notmuch_message_freeze(message->nm_message);
    notmuch_rb_status_raise(ret);
    return Qtrue;
}

/*
 * call-seq: MESSAGE.thaw => true
 *
 * Thaw a 'message'
 */
VALUE
notmuch_rb_message_thaw(VALUE self)
{
    notmuch_status_t ret;
    notmuch_rb_message_t *message;

    Data_Get_Struct(self, notmuch_rb_message_t, message);

    ret = notmuch_message_thaw(message->nm_message);
    notmuch_rb_status_raise(ret);
    return Qtrue;
}