path: root/camel/providers/nntp/camel-nntp-utils.c

/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/* camel-nntp-utils.c : utilities used by the nntp code. */

/* 
 * Author : Chris Toshok <toshok@helixcode.com> 
 *
 * Copyright (C) 2000 Helix Code .
 *
 * 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 2 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 * USA
 */

#include "camel-folder-summary.h"
#include "camel-nntp-resp-codes.h"
#include "camel-nntp-folder.h"
#include "camel-nntp-store.h"
#include "camel-nntp-utils.h"
#include "camel-stream-mem.h"
#include "camel-exception.h"

#include <stdlib.h>
#include <string.h>

static gboolean
get_XOVER_headers(CamelNNTPStore *nntp_store, CamelFolder *folder,
          int first_message, int last_message, CamelException *ex)
{
    int status;
    CamelNNTPFolder *nntp_folder = CAMEL_NNTP_FOLDER (folder);

    status = camel_nntp_command (nntp_store, ex, NULL,
                     "XOVER %d-%d",
                     first_message,
                     last_message);

    if (status == NNTP_DATA_FOLLOWS) {
        gboolean done = FALSE;

        while (!done) {
            char *line;

            if (camel_remote_store_recv_line (CAMEL_REMOTE_STORE (nntp_store), &line, ex) < 0) {
                g_warning ("failed to recv_line while building OVER header list\n");
                break;
            }

            if (*line == '.') {
                done = TRUE;
                g_print ("done\n");
            }
            else {
                CamelMessageInfo *new_info = g_new0(CamelMessageInfo, 1);
                char **split_line = g_strsplit (line, "\t", 7);
                char *subject, *from, *date, *message_id, *bytes;

                subject = split_line [nntp_store->overview_field [CAMEL_NNTP_OVER_SUBJECT].index];
                from = split_line [nntp_store->overview_field [CAMEL_NNTP_OVER_FROM].index];
                date = split_line [nntp_store->overview_field [CAMEL_NNTP_OVER_DATE].index];
                message_id = split_line [nntp_store->overview_field [CAMEL_NNTP_OVER_MESSAGE_ID].index];
                bytes = split_line [nntp_store->overview_field [CAMEL_NNTP_OVER_BYTES].index];

                /* if the overview format flagged this
                                   field as "full", skip over the
                                   preceding field name and colon */
                if (nntp_store->overview_field [ CAMEL_NNTP_OVER_SUBJECT ].full)
                    subject += strlen ("Subject:");
                if (nntp_store->overview_field [ CAMEL_NNTP_OVER_FROM ].full)
                    from += strlen ("From:");
                if (nntp_store->overview_field [ CAMEL_NNTP_OVER_DATE ].full)
                    date += strlen ("Date:");
                if (nntp_store->overview_field [ CAMEL_NNTP_OVER_MESSAGE_ID ].full)
                    message_id += strlen ("Message-ID:");
                if (nntp_store->overview_field [ CAMEL_NNTP_OVER_BYTES ].full)
                    bytes += strlen ("Bytes:");

                new_info->subject = g_strdup(subject);
                new_info->from = g_strdup(from);
                new_info->to = g_strdup(folder->name);
                new_info->date_sent = header_decode_date(date, NULL);
#if 0
                /* XXX do we need to fill in both dates? */
                new_info->headers.date_received = g_strdup(date);
#endif
                new_info->size = atoi(bytes);
                new_info->uid = g_strdup_printf ("%s,%s", split_line[0], message_id);
                new_info->message_id = g_strdup(message_id);

                if (camel_nntp_newsrc_article_is_read (nntp_store->newsrc,
                                       folder->name,
                                       atoi (split_line[0])))
                    new_info->flags |= CAMEL_MESSAGE_SEEN;

                camel_folder_summary_add (nntp_folder->summary, new_info);
                g_strfreev (split_line);
            }
            g_free (line);
        }

        return TRUE;
    }
    else {
        /* XXX */
        g_warning ("weird nntp response for XOVER: %d\n", status);
        return FALSE;
    }
}

#if 0
static GArray*
get_HEAD_headers(CamelNNTPStore *nntp_store, CamelFolder *folder,
         int first_message, int last_message, CamelException *ex)
{
    int i;
    int status;

    for (i = first_message; i < last_message; i ++) {
        status = camel_nntp_command (nntp_store, ex, NULL,
                         "HEAD %d", i);

        if (status == NNTP_HEAD_FOLLOWS) {
            gboolean done = FALSE;
            char *buf;
            int buf_len;
            int buf_alloc;
            int h;
            CamelStream *header_stream;
            GArray *header_array;
            CamelStream *nntp_istream;
            CamelMessageInfo *new_info = g_new0(CamelMessageInfo, 1);

            buf_alloc = 2048;
            buf_len = 0;
            buf = g_malloc(buf_alloc);
            done = FALSE;

            buf[0] = 0;

            nntp_istream = nntp_store->istream;

            while (!done) {
                char *line;
                int line_length;

                line = camel_stream_buffer_read_line ( 
                              CAMEL_STREAM_BUFFER ( nntp_istream ));
                line_length = strlen ( line );

                if (*line == '.') {
                    done = TRUE;
                }
                else {
                    if (buf_len + line_length > buf_alloc) {
                        buf_alloc *= 2;
                        buf = g_realloc (buf, buf_alloc);
                    }
                    strcat(buf, line);
                    strcat(buf, "\n");
                    buf_len += strlen(line);
                    g_free (line);
                }
            }

            /* create a stream from which to parse the headers */
            header_stream = camel_stream_mem_new_with_buffer(buf,
                                 buf_len,
                                 CAMEL_STREAM_MEM_READ);

            header_array = get_header_array_from_stream (header_stream);

            memset (&info, 0, sizeof(info));

            for (h = 0; h < header_array->len; h ++) {
                Rfc822Header *header = &((Rfc822Header*)header_array->data)[h];
                if (!g_strcasecmp(header->name, "From"))
                    new_info->from = g_strdup(header->value);
                else if (!g_strcasecmp(header->name, "To"))
                    new_info->to = g_strdup(header->value);
                else if (!g_strcasecmp(header->name, "Subject"))
                    new_info->subject = g_strdup(header->value);
                else if (!g_strcasecmp(header->name, "Message-ID")) {
                    new_info->uid = g_strdup_printf("%d,%s", i, header->value);
                    new_info->message_id = g_strdup(header->value);
                }
                else if (!g_strcasecmp(header->name, "Date")) {
                    new_info->date_sent = header_decode_date (header->value);
#if 0
                    new_info->date_sent = g_strdup(header->value);
                    new_info->date_received = g_strdup(header->value);
#endif
                }
            }

            camel_folder_summary_add (nntp_folder->summary, new_info);
        }
        else if (status == CAMEL_NNTP_FAIL) {
            /* nasty things are afoot */
            g_warning ("failure doing HEAD\n");
            break;
        }
    }
}
#endif

void
camel_nntp_get_headers (CamelStore *store,
            CamelNNTPFolder *nntp_folder,
            CamelException *ex)
{
    CamelNNTPStore *nntp_store = CAMEL_NNTP_STORE (store);
    CamelFolder *folder = CAMEL_FOLDER (nntp_folder);
    char *ret;
    int first_message, nb_message, last_message;
    int status;

    status = camel_nntp_command (nntp_store, ex, &ret,
                     "GROUP %s", folder->name);

    sscanf (ret, "%d %d %d", &nb_message, &first_message, &last_message);
    g_free (ret);

    if (status == NNTP_NO_SUCH_GROUP) {
        /* XXX throw invalid group exception */
        camel_exception_setv (ex, 
                      CAMEL_EXCEPTION_FOLDER_INVALID,
                      "group %s not found on server",
                      folder->name);
        return;
    }


    if (!get_XOVER_headers (nntp_store, folder, first_message, last_message, ex)) {
        g_warning ("need to fix get_HEAD_headers\n");
#if 0
        get_HEAD_headers (nntp_store, folder, first_message, last_message, ex);
#endif
    }       
}