path: root/mail/mail-session.c

/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/* mail-session.c: handles the session information and resource manipulation */
/*
 *  Copyright 2001 Ximian, Inc. (www.ximian.com)
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of version 2 of the GNU General Public
 * License as published by the Free Software Foundation.
 *
 * 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 <config.h>

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

#include <glib.h>
#include <glib/gstdio.h>

#include <gtk/gtk.h>

#include <gconf/gconf-client.h>

#include <libgnome/gnome-config.h>
#include <libgnome/gnome-sound.h>

#include <libedataserverui/e-passwords.h>
#include <libedataserver/e-msgport.h>

#include <camel/camel.h>    /* FIXME: this is where camel_init is defined, it shouldn't include everything else */
#include <camel/camel-filter-driver.h>
#include <camel/camel-i18n.h>

#include "e-util/e-error.h"
#include "e-util/e-util-private.h"

#include "em-filter-context.h"
#include "em-filter-rule.h"
#include "mail-component.h"
#include "mail-config.h"
#include "mail-mt.h"
#include "mail-ops.h"
#include "mail-session.h"
#include "mail-tools.h"

#define d(x)

CamelSession *session;
static int session_check_junk_notify_id = -1;

#define MAIL_SESSION_TYPE     (mail_session_get_type ())
#define MAIL_SESSION(obj)     (CAMEL_CHECK_CAST((obj), MAIL_SESSION_TYPE, MailSession))
#define MAIL_SESSION_CLASS(k) (CAMEL_CHECK_CLASS_CAST ((k), MAIL_SESSION_TYPE, MailSessionClass))
#define MAIL_IS_SESSION(o)    (CAMEL_CHECK_TYPE((o), MAIL_SESSION_TYPE))

typedef struct _MailSession {
    CamelSession parent_object;

    gboolean interactive;
    FILE *filter_logfile;
    GList *junk_plugins;

    MailAsyncEvent *async;
} MailSession;

typedef struct _MailSessionClass {
    CamelSessionClass parent_class;

} MailSessionClass;

static CamelSessionClass *ms_parent_class;

static char *get_password(CamelSession *session, CamelService *service, const char *domain, const char *prompt, const char *item, guint32 flags, CamelException *ex);
static void forget_password(CamelSession *session, CamelService *service, const char *domain, const char *item, CamelException *ex);
static gboolean alert_user(CamelSession *session, CamelSessionAlertType type, const char *prompt, gboolean cancel);
static CamelFilterDriver *get_filter_driver(CamelSession *session, const char *type, CamelException *ex);

static void ms_thread_status(CamelSession *session, CamelSessionThreadMsg *msg, const char *text, int pc);
static void *ms_thread_msg_new(CamelSession *session, CamelSessionThreadOps *ops, unsigned int size);
static void ms_thread_msg_free(CamelSession *session, CamelSessionThreadMsg *m);

static void
init (MailSession *session)
{
    session->async = mail_async_event_new();
    session->junk_plugins = NULL;
}

static void
finalise (MailSession *session)
{
    if (session_check_junk_notify_id != -1)
        gconf_client_notify_remove (mail_config_get_gconf_client (), session_check_junk_notify_id);

    mail_async_event_destroy(session->async);
}

static void
class_init (MailSessionClass *mail_session_class)
{
    CamelSessionClass *camel_session_class = CAMEL_SESSION_CLASS (mail_session_class);
    
    /* virtual method override */
    camel_session_class->get_password = get_password;
    camel_session_class->forget_password = forget_password;
    camel_session_class->alert_user = alert_user;
    camel_session_class->get_filter_driver = get_filter_driver;

    camel_session_class->thread_msg_new = ms_thread_msg_new;
    camel_session_class->thread_msg_free = ms_thread_msg_free;
    camel_session_class->thread_status = ms_thread_status;
}

static CamelType
mail_session_get_type (void)
{
    static CamelType mail_session_type = CAMEL_INVALID_TYPE;
    
    if (mail_session_type == CAMEL_INVALID_TYPE) {
        ms_parent_class = (CamelSessionClass *)camel_session_get_type();
        mail_session_type = camel_type_register (
            camel_session_get_type (),
            "MailSession",
            sizeof (MailSession),
            sizeof (MailSessionClass),
            (CamelObjectClassInitFunc) class_init,
            NULL,
            (CamelObjectInitFunc) init,
            (CamelObjectFinalizeFunc) finalise);
    }
    
    return mail_session_type;
}


static char *
make_key (CamelService *service, const char *item)
{
    char *key;
    
    if (service)
        key = camel_url_to_string (service->url, CAMEL_URL_HIDE_PASSWORD | CAMEL_URL_HIDE_PARAMS);
    else
        key = g_strdup (item);
    
    return key;
}

/* ********************************************************************** */

static char *
get_password (CamelSession *session, CamelService *service, const char *domain,
          const char *prompt, const char *item, guint32 flags, CamelException *ex)
{
    char *url;
    char *ret = NULL;
    EAccount *account = NULL;

    url = service?camel_url_to_string(service->url, CAMEL_URL_HIDE_ALL):NULL;

    if (!strcmp(item, "popb4smtp_uri")) {
        /* not 100% mt safe, but should be ok */
        if (url
            && (account = mail_config_get_account_by_transport_url(url)))
            ret = g_strdup(account->source->url);
    } else {
        char *key = make_key(service, item);
        EAccountService *config_service = NULL;

        if (domain == NULL)
            domain = "Mail";

        ret = e_passwords_get_password(domain, key);
        if (ret == NULL || (flags & CAMEL_SESSION_PASSWORD_REPROMPT)) {
            gboolean remember;

            if (url) {
                if  ((account = mail_config_get_account_by_source_url(url)))
                    config_service = account->source;
                else if ((account = mail_config_get_account_by_transport_url(url)))
                    config_service = account->transport;
            }

            remember = config_service?config_service->save_passwd:FALSE;

            if (!config_service || (config_service && !config_service->get_password_canceled)) {
                guint32 eflags;
                char *title;

                if (flags & CAMEL_SESSION_PASSPHRASE) {
                    if (account)
                        title = g_strdup_printf (_("Enter Passphrase for %s"), account->name);
                    else
                        title = g_strdup (_("Enter Passphrase"));
                } else {
                    if (account)
                        title = g_strdup_printf (_("Enter Password for %s"), account->name);
                    else
                        title = g_strdup (_("Enter Password"));             
                }
                if ((flags & CAMEL_SESSION_PASSWORD_STATIC) != 0)
                    eflags = E_PASSWORDS_REMEMBER_NEVER;
                else if (config_service == NULL)
                    eflags = E_PASSWORDS_REMEMBER_SESSION;
                else
                    eflags = E_PASSWORDS_REMEMBER_FOREVER;

                if (flags & CAMEL_SESSION_PASSWORD_REPROMPT)
                    eflags |= E_PASSWORDS_REPROMPT;

                if (flags & CAMEL_SESSION_PASSWORD_SECRET)
                    eflags |= E_PASSWORDS_SECRET;

                if (flags & CAMEL_SESSION_PASSPHRASE)
                    eflags |= E_PASSWORDS_PASSPHRASE;

                /* HACK: breaks abstraction ...
                   e_account_writable doesn't use the eaccount, it also uses the same writable key for
                   source and transport */
                if (!e_account_writable(NULL, E_ACCOUNT_SOURCE_SAVE_PASSWD))
                    eflags |= E_PASSWORDS_DISABLE_REMEMBER;

                ret = e_passwords_ask_password(title, domain, key, prompt, eflags, &remember, NULL);

                g_free(title);

                if (ret && config_service)
                    mail_config_service_set_save_passwd(config_service, remember);

                if (config_service)
                    config_service->get_password_canceled = ret == NULL;
            }
        }

        g_free(key);
    }

    g_free(url);

    if (ret == NULL)
        camel_exception_set(ex, CAMEL_EXCEPTION_USER_CANCEL, _("User canceled operation."));

    return ret;
}

static void
forget_password (CamelSession *session, CamelService *service, const char *domain, const char *item, CamelException *ex)
{
    char *key = make_key (service, item);

    e_passwords_forget_password (domain?domain:"Mail", key);
    g_free (key);
}

/* ********************************************************************** */

static GtkDialog *message_dialog;
static EDList message_list = E_DLIST_INITIALISER(message_list);

struct _user_message_msg {
    struct _mail_msg msg;

    CamelSessionAlertType type;
    char *prompt;

    unsigned int allow_cancel:1;
    unsigned int result:1;
    unsigned int ismain:1;
};

static void do_user_message (struct _mail_msg *mm);

/* clicked, send back the reply */
static void
user_message_response (GtkDialog *dialog, int button, struct _user_message_msg *m)
{
    gtk_widget_destroy ((GtkWidget *) dialog);
    
    message_dialog = NULL;
    
    /* if !allow_cancel, then we've already replied */
    if (m->allow_cancel) {
        m->result = button == GTK_RESPONSE_OK;
        e_msgport_reply((EMsg *)m);
    }
    
    /* check for pendings */
    if ((m = (struct _user_message_msg *)e_dlist_remhead(&message_list)))
        do_user_message((struct _mail_msg *)m);
}

static void
user_message_destroy_notify (struct _user_message_msg *m, GObject *deadbeef)
{
    message_dialog = NULL;
}

/* This is kinda ugly/inefficient, but oh well, it works */
static const char *error_type[] = {
    "mail:session-message-info", "mail:session-message-warning", "mail:session-message-error",
    "mail:session-message-info-cancel", "mail:session-message-warning-cancel", "mail:session-message-error-cancel"
};

static void
do_user_message (struct _mail_msg *mm)
{
    struct _user_message_msg *m = (struct _user_message_msg *)mm;
    int type;
    
    if (!m->ismain && message_dialog != NULL) {
        e_dlist_addtail (&message_list, (EDListNode *)m);
        return;
    }
    
    switch (m->type) {
    case CAMEL_SESSION_ALERT_INFO:
        type = 0;
        break;
    case CAMEL_SESSION_ALERT_WARNING:
        type = 1;
        break;
    case CAMEL_SESSION_ALERT_ERROR:
        type = 2;
        break;
    default:
        type = 0;
    }

    if (m->allow_cancel)
        type += 3;
    
    message_dialog = (GtkDialog *)e_error_new(NULL, error_type[type], m->prompt, NULL);
    g_object_set ((GObject *) message_dialog, "allow_shrink", TRUE, "allow_grow", TRUE, NULL);
    
    /* We only need to wait for the result if we allow cancel otherwise show but send result back instantly */
    if (m->allow_cancel) {
        if (m->ismain) {
            user_message_response(message_dialog, gtk_dialog_run (message_dialog), m);
        } else {
            g_signal_connect (message_dialog, "response", G_CALLBACK (user_message_response), m);
            gtk_widget_show ((GtkWidget *) message_dialog);
        }
    } else {
        g_signal_connect (message_dialog, "response", G_CALLBACK (gtk_widget_destroy), message_dialog);
        g_object_weak_ref ((GObject *) message_dialog, (GWeakNotify) user_message_destroy_notify, m);
        gtk_widget_show ((GtkWidget *) message_dialog);
        mail_msg_free(m);
    }
}

static void
free_user_message(struct _mail_msg *mm)
{
    struct _user_message_msg *m = (struct _user_message_msg *)mm;

    g_free(m->prompt);
}

static struct _mail_msg_op user_message_op = { NULL, do_user_message, NULL, free_user_message };

static gboolean
alert_user(CamelSession *session, CamelSessionAlertType type, const char *prompt, gboolean cancel)
{
    MailSession *mail_session = MAIL_SESSION (session);
    struct _user_message_msg *m, *r;
    EMsgPort *user_message_reply = NULL;
    gboolean ret;

    if (!mail_session->interactive)
        return FALSE;

    if (cancel)
        user_message_reply = e_msgport_new ();  
    m = mail_msg_new (&user_message_op, user_message_reply, sizeof (*m));
    m->ismain = pthread_equal(pthread_self(), mail_gui_thread);
    m->type = type;
    m->prompt = g_strdup(prompt);
    m->allow_cancel = cancel;

    if (m->ismain)
        do_user_message((struct _mail_msg *)m);
    else {
        extern EMsgPort *mail_gui_port2;

        e_msgport_put(mail_gui_port2, (EMsg *)m);
    }

    if (cancel) {
        r = (struct _user_message_msg *)e_msgport_wait(user_message_reply);
        g_return_val_if_fail (m == r, FALSE);

        ret = m->result;
        mail_msg_free(m);
        e_msgport_destroy(user_message_reply);
    } else
        ret = TRUE;

    return ret;
}

static CamelFolder *
get_folder (CamelFilterDriver *d, const char *uri, void *data, CamelException *ex)
{
    return mail_tool_uri_to_folder(uri, 0, ex);
}

static void
main_play_sound (CamelFilterDriver *driver, char *filename, gpointer user_data)
{
    if (filename && *filename)
        gnome_sound_play (filename);
    else
        gdk_beep ();
    
    g_free (filename);
    camel_object_unref (session);
}

static void
session_play_sound (CamelFilterDriver *driver, const char *filename, gpointer user_data)
{
    MailSession *ms = (MailSession *) session;
    
    camel_object_ref (session);
    
    mail_async_event_emit (ms->async, MAIL_ASYNC_GUI, (MailAsyncFunc) main_play_sound,
                   driver, g_strdup (filename), user_data);
}

static void
main_system_beep (CamelFilterDriver *driver, gpointer user_data)
{
    gdk_beep ();
    camel_object_unref (session);
}

static void
session_system_beep (CamelFilterDriver *driver, gpointer user_data)
{
    MailSession *ms = (MailSession *) session;
    
    camel_object_ref (session);
    
    mail_async_event_emit (ms->async, MAIL_ASYNC_GUI, (MailAsyncFunc) main_system_beep,
                   driver, user_data, NULL);
}

static CamelFilterDriver *
main_get_filter_driver (CamelSession *session, const char *type, CamelException *ex)
{
    CamelFilterDriver *driver;
    FilterRule *rule = NULL;
    char *user, *system;
    GConfClient *gconf;
    RuleContext *fc;
    
    gconf = mail_config_get_gconf_client ();
    
    user = g_strdup_printf ("%s/mail/filters.xml", mail_component_peek_base_directory (mail_component_peek ()));
    system = g_build_filename (EVOLUTION_PRIVDATADIR, "filtertypes.xml", NULL);
    fc = (RuleContext *) em_filter_context_new ();
    rule_context_load (fc, system, user);
    g_free (system);
    g_free (user);
    
    driver = camel_filter_driver_new (session);
    camel_filter_driver_set_folder_func (driver, get_folder, NULL);
    
    if (gconf_client_get_bool (gconf, "/apps/evolution/mail/filters/log", NULL)) {
        MailSession *ms = (MailSession *) session;
        
        if (ms->filter_logfile == NULL) {
            char *filename;
            
            filename = gconf_client_get_string (gconf, "/apps/evolution/mail/filters/logfile", NULL);
            if (filename) {
                ms->filter_logfile = g_fopen (filename, "a+");
                g_free (filename);
            }
        }
        
        if (ms->filter_logfile)
            camel_filter_driver_set_logfile (driver, ms->filter_logfile);
    }
    
    camel_filter_driver_set_shell_func (driver, mail_execute_shell_command, NULL);
    camel_filter_driver_set_play_sound_func (driver, session_play_sound, NULL);
    camel_filter_driver_set_system_beep_func (driver, session_system_beep, NULL);

    if ((!strcmp (type, FILTER_SOURCE_INCOMING) || !strcmp (type, FILTER_SOURCE_JUNKTEST))
        && camel_session_check_junk (session)) {
        /* implicit junk check as 1st rule */
        camel_filter_driver_add_rule (driver, "Junk check", "(junk-test)", "(begin (set-system-flag \"junk\")(set-system-flag \"seen\"))");
    }

    if (strcmp (type, FILTER_SOURCE_JUNKTEST) != 0) {
        GString *fsearch, *faction;
        
        fsearch = g_string_new ("");
        faction = g_string_new ("");
        
        if (!strcmp (type, FILTER_SOURCE_DEMAND))
            type = FILTER_SOURCE_INCOMING;
        
        /* add the user-defined rules next */
        while ((rule = rule_context_next_rule (fc, rule, type))) {
            g_string_truncate (fsearch, 0);
            g_string_truncate (faction, 0);
        
            filter_rule_build_code (rule, fsearch);
            em_filter_rule_build_action ((EMFilterRule *) rule, faction);
            camel_filter_driver_add_rule (driver, rule->name, fsearch->str, faction->str);
        }
        
        g_string_free (fsearch, TRUE);
        g_string_free (faction, TRUE);
    }
    
    g_object_unref (fc);
    
    return driver;
}

static CamelFilterDriver *
get_filter_driver (CamelSession *session, const char *type, CamelException *ex)
{
    return (CamelFilterDriver *) mail_call_main (MAIL_CALL_p_ppp, (MailMainFunc) main_get_filter_driver,
                             session, type, ex);
}

/* TODO: This is very temporary, until we have a better way to do the progress reporting,
   we just borrow a dummy mail-mt thread message and hook it onto out camel thread message */

static mail_msg_op_t ms_thread_ops_dummy = { NULL };

static void *ms_thread_msg_new(CamelSession *session, CamelSessionThreadOps *ops, unsigned int size)
{
    CamelSessionThreadMsg *msg = ms_parent_class->thread_msg_new(session, ops, size);

    /* We create a dummy mail_msg, and then copy its cancellation port over to ours, so
       we get cancellation and progress in common with hte existing mail code, for free */
    if (msg) {
        struct _mail_msg *m = mail_msg_new(&ms_thread_ops_dummy, NULL, sizeof(struct _mail_msg));

        msg->data = m;
        camel_operation_unref(msg->op);
        msg->op = m->cancel;
        camel_operation_ref(msg->op);
    }

    return msg;
}

static void ms_thread_msg_free(CamelSession *session, CamelSessionThreadMsg *m)
{
    mail_msg_free(m->data);
    ms_parent_class->thread_msg_free(session, m);
}

static void ms_thread_status(CamelSession *session, CamelSessionThreadMsg *msg, const char *text, int pc)
{
    /* This should never be called since we bypass it in alloc! */
    printf("Thread status '%s' %d%%\n", text, pc);
}

char *
mail_session_get_password (const char *url_string)
{
    CamelURL *url;
    char *simple_url;
    char *passwd;
    
    url = camel_url_new (url_string, NULL);
    simple_url = camel_url_to_string (url, CAMEL_URL_HIDE_PASSWORD | CAMEL_URL_HIDE_PARAMS);
    camel_url_free (url);
    
    passwd = e_passwords_get_password ("Mail", simple_url);
    
    g_free (simple_url);
    
    return passwd;
}

void
mail_session_add_password (const char *url_string,
               const char *passwd)
{
    CamelURL *url;
    char *simple_url;
    
    url = camel_url_new (url_string, NULL);
    simple_url = camel_url_to_string (url, CAMEL_URL_HIDE_PASSWORD | CAMEL_URL_HIDE_PARAMS);
    camel_url_free (url);
    
    e_passwords_add_password (simple_url, passwd);
    
    g_free (simple_url);
}

void
mail_session_remember_password (const char *url_string)
{
    CamelURL *url;
    char *simple_url;
    
    url = camel_url_new (url_string, NULL);
    simple_url = camel_url_to_string (url, CAMEL_URL_HIDE_PASSWORD | CAMEL_URL_HIDE_PARAMS);
    camel_url_free (url);
    
    e_passwords_remember_password ("Mail", simple_url);
    
    g_free (simple_url);
}

void
mail_session_forget_password (const char *key)
{
    e_passwords_forget_password ("Mail", key);
}

static void
mail_session_check_junk_notify (GConfClient *gconf, guint id, GConfEntry *entry, CamelSession *session)
{
    gchar *key;

    g_return_if_fail (gconf_entry_get_key (entry) != NULL);
    g_return_if_fail (gconf_entry_get_value (entry) != NULL);

    key = strrchr (gconf_entry_get_key (entry), '/');
    if (key) {
        key ++;
        if (!strcmp (key, "check_incoming"))
            camel_session_set_check_junk (session, gconf_value_get_bool (gconf_entry_get_value (entry)));
    }
}

void
mail_session_init (const char *base_directory)
{
    char *camel_dir;
    GConfClient *gconf;
    
    if (camel_init (base_directory, TRUE) != 0)
        exit (0);

    camel_provider_init();

    session = CAMEL_SESSION (camel_object_new (MAIL_SESSION_TYPE));
    
    camel_dir = g_strdup_printf ("%s/mail", base_directory);
    camel_session_construct (session, camel_dir);

    gconf = mail_config_get_gconf_client ();
    gconf_client_add_dir (gconf, "/apps/evolution/mail/junk", GCONF_CLIENT_PRELOAD_ONELEVEL, NULL);
    camel_session_set_check_junk (session, gconf_client_get_bool (gconf, "/apps/evolution/mail/junk/check_incoming", NULL));
    session_check_junk_notify_id = gconf_client_notify_add (gconf, "/apps/evolution/mail/junk",
                                (GConfClientNotifyFunc) mail_session_check_junk_notify,
                                session, NULL, NULL);
    session->junk_plugin = NULL;

    /* The shell will tell us to go online. */
    camel_session_set_online ((CamelSession *) session, FALSE);
    
    g_free (camel_dir);
}

gboolean
mail_session_get_interactive (void)
{
    return MAIL_SESSION (session)->interactive;
}

void
mail_session_set_interactive (gboolean interactive)
{
    MAIL_SESSION (session)->interactive = interactive;
    
    if (!interactive) {
        struct _user_message_msg *um;
        
        d(printf ("Gone non-interactive, checking for outstanding interactive tasks\n"));

        e_passwords_cancel();
        
        /* flush/cancel pending user messages */
        while ((um = (struct _user_message_msg *) e_dlist_remhead (&message_list))) {
            d(printf ("Flusing message request: %s\n", um->prompt));
            e_msgport_reply((EMsg *) um);
        }
        
        /* and the current */
        if (message_dialog) {
            d(printf("Destroying message dialogue\n"));
            gtk_widget_destroy ((GtkWidget *) message_dialog);
        }
    }
}

void
mail_session_forget_passwords (BonoboUIComponent *uih, void *user_data,
                   const char *path)
{
    e_passwords_forget_passwords ();
}

void
mail_session_flush_filter_log (void)
{
    MailSession *ms = (MailSession *) session;
    
    if (ms->filter_logfile)
        fflush (ms->filter_logfile);
}

void
mail_session_add_junk_plugin (const char *plugin_name, CamelJunkPlugin *junk_plugin)
{
    MailSession *ms = (MailSession *) session;
    GConfClient *gconf;
    char *def_plugin;
    
    gconf = mail_config_get_gconf_client ();
    def_plugin = gconf_client_get_string (gconf, "/apps/evolution/mail/junk/default_plugin", NULL);
    
    ms->junk_plugins = g_list_append(ms->junk_plugins, junk_plugin);
    if (def_plugin && plugin_name) {
        if (!strcmp(def_plugin, plugin_name)) {
            printf("Loading %s as the default junk plugin\n", def_plugin);
            session->junk_plugin = junk_plugin;
            camel_junk_plugin_init (junk_plugin);
        }
    }

    g_free (def_plugin);
}

const GList *
mail_session_get_junk_plugins (void)
{
    MailSession *ms = (MailSession *) session;  
    return ms->junk_plugins;
}