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-nntp-folder.h"
#include "camel-nntp-store.h"
#include "camel-nntp-summary.h"
#include "camel-nntp-utils.h"
#include "camel-stream-buffer.h"
#include "camel-stream-mem.h"
#include "gmime-utils.h"

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

static GArray*
get_XOVER_headers(CamelNNTPStore *nntp_store, CamelFolder *folder,
          int first_message, int last_message)
{
    int status;

    status = camel_nntp_command (nntp_store, NULL,
                     "XOVER %d-%d",
                     first_message,
                     last_message);
        
    if (status == CAMEL_NNTP_OK) {
        CamelStream *nntp_istream = nntp_store->istream;
        GArray *array;
        gboolean done = FALSE;

        array = g_array_new(FALSE, FALSE, sizeof(CamelNNTPSummaryInformation));

        while (!done) {
            char *line;

            line = camel_stream_buffer_read_line ( 
                          CAMEL_STREAM_BUFFER ( nntp_istream ));

            if (*line == '.') {
                done = TRUE;
            }
            else {
                CamelNNTPSummaryInformation new_info;
                char **split_line = g_strsplit (line, "\t", 7);

                memset (&new_info, 0, sizeof(new_info));
                    
                new_info.headers.subject = g_strdup(split_line[1]);
                new_info.headers.sender = g_strdup(split_line[2]);
                new_info.headers.to = g_strdup(folder->name);
                new_info.headers.sent_date = g_strdup(split_line[3]);
                /* XXX do we need to fill in both dates? */
                new_info.headers.received_date = g_strdup(split_line[3]);
                new_info.headers.size = atoi(split_line[5]);
                new_info.headers.uid = g_strdup(split_line[4]);
                g_strfreev (split_line);

                g_array_append_val(array, new_info);
            }
            g_free (line);
        }

        return array;
    }

    return NULL;
}

static GArray*
get_HEAD_headers(CamelNNTPStore *nntp_store, CamelFolder *folder,
         int first_message, int last_message)
{
    int i;
    int status;
    GArray *array;
    CamelNNTPSummaryInformation info;

    array = g_array_new(FALSE, FALSE, sizeof(CamelNNTPSummaryInformation));

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

        if (status == CAMEL_NNTP_OK) {
            gboolean done = FALSE;
            char *buf;
            int buf_len;
            int buf_alloc;
            int h;
            CamelStream *header_stream;
            GArray *header_array;
            CamelStream *nntp_istream;

            buf_alloc = 2048;
            buf_len = 0;
            buf = 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 = 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 (!strcasecmp(header->name, "From"))
                    info.headers.sender = g_strdup(header->value);
                else if (!strcasecmp(header->name, "To"))
                    info.headers.to = g_strdup(header->value);
                else if (!strcasecmp(header->name, "Subject"))
                    info.headers.subject = g_strdup(header->value);
                else if (!strcasecmp(header->name, "Message-ID"))
                    info.headers.uid = g_strdup(header->value);
                else if (!strcasecmp(header->name, "Date")) {
                    info.headers.sent_date = g_strdup(header->value);
                    info.headers.received_date = g_strdup(header->value);
                }
            }
            g_array_append_val(array, info);
        }
        else if (status == CAMEL_NNTP_FAIL) {
            /* nasty things are afoot */
            g_warning ("failure doing HEAD\n");
            break;
        }
    }
    return array;
}

GArray *
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, &ret,
                     "GROUP %s", folder->name);

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

    if (status != CAMEL_NNTP_OK)
        return NULL;

    if (TRUE /* nntp_store->extensions & CAMEL_NNTP_EXT_XOVER */) {
        return get_XOVER_headers (nntp_store, folder, first_message, last_message);
    }
    else {
        return get_HEAD_headers (nntp_store, folder, first_message, last_message);
    }
}