source: storage.c @ 0d581bd

/********************************************************************\
  * BitlBee -- An IRC to other IM-networks gateway                     *
  *                                                                    *
  * Copyright 2002-2004 Wilmer van der Gaast and others                *
  \********************************************************************/

/* Support for multiple storage backends */

/* Copyright (C) 2005 Jelmer Vernooij <jelmer@samba.org> */

/*
  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 with
  the Debian GNU/Linux distribution in /usr/share/common-licenses/GPL;
  if not, write to the Free Software Foundation, Inc., 51 Franklin St.,
  Fifth Floor, Boston, MA  02110-1301  USA
*/

#define BITLBEE_CORE
#include "bitlbee.h"

extern storage_t storage_xml;

static GList *storage_backends = NULL;

void register_storage_backend(storage_t *backend)
{
        storage_backends = g_list_append(storage_backends, backend);
}

static storage_t *storage_init_single(const char *name)
{
        GList *gl;
        storage_t *st = NULL;

        for (gl = storage_backends; gl; gl = gl->next) {
                st = gl->data;
                if (strcmp(st->name, name) == 0) {
                        break;
                }
        }

        if (gl == NULL) {
                return NULL;
        }

        if (st->init) {
                st->init();
        }

        return st;
}

GList *storage_init(const char *primary, char **migrate)
{
        GList *ret = NULL;
        int i;
        storage_t *storage;

        register_storage_backend(&storage_xml);

        storage = storage_init_single(primary);
        if (storage == NULL && storage->save == NULL) {
                return NULL;
        }

        ret = g_list_append(ret, storage);

        for (i = 0; migrate && migrate[i]; i++) {
                storage = storage_init_single(migrate[i]);

                if (storage) {
                        ret = g_list_append(ret, storage);
                }
        }

        return ret;
}

storage_status_t storage_check_pass(const char *nick, const char *password)
{
        GList *gl;

        /* Loop until we don't get NO_SUCH_USER */

        for (gl = global.storage; gl; gl = gl->next) {
                storage_t *st = gl->data;
                storage_status_t status;

                status = st->check_pass(nick, password);
                if (status != STORAGE_NO_SUCH_USER) {
                        return status;
                }
        }

        return STORAGE_NO_SUCH_USER;
}

storage_status_t storage_load(irc_t * irc, const char *password)
{
        GList *gl;

        if (irc && irc->status & USTATUS_IDENTIFIED) {
                return STORAGE_OTHER_ERROR;
        }

        /* Loop until we don't get NO_SUCH_USER */
        for (gl = global.storage; gl; gl = gl->next) {
                storage_t *st = gl->data;
                storage_status_t status;

                status = st->load(irc, password);
                if (status == STORAGE_OK) {
                        GSList *l;
                        for (l = irc_plugins; l; l = l->next) {
                                irc_plugin_t *p = l->data;
                                if (p->storage_load) {
                                        p->storage_load(irc);
                                }
                        }
                        return status;
                }

                if (status != STORAGE_NO_SUCH_USER) {
                        return status;
                }
        }

        return STORAGE_NO_SUCH_USER;
}

storage_status_t storage_save(irc_t *irc, char *password, int overwrite)
{
        storage_status_t st;
        GSList *l;

        if (password != NULL) {
                /* Should only use this in the "register" command. */
                if (irc->password || overwrite) {
                        return STORAGE_OTHER_ERROR;
                }

                irc_setpass(irc, password);
        } else if ((irc->status & USTATUS_IDENTIFIED) == 0) {
                return STORAGE_NO_SUCH_USER;
        }

        st = ((storage_t *) global.storage->data)->save(irc, overwrite);

        for (l = irc_plugins; l; l = l->next) {
                irc_plugin_t *p = l->data;
                if (p->storage_save) {
                        p->storage_save(irc);
                }
        }

        if (password != NULL) {
                irc_setpass(irc, NULL);
        }

        return st;
}

storage_status_t storage_remove(const char *nick, const char *password)
{
        GList *gl;
        storage_status_t ret = STORAGE_OK;
        gboolean ok = FALSE;
        GSList *l;

        /* Remove this account from all storage backends. If this isn't
         * done, the account will still be usable, it'd just be
         * loaded from a different backend. */
        for (gl = global.storage; gl; gl = gl->next) {
                storage_t *st = gl->data;
                storage_status_t status;

                status = st->remove(nick, password);
                ok |= status == STORAGE_OK;
                if (status != STORAGE_NO_SUCH_USER && status != STORAGE_OK) {
                        ret = status;
                }
        }

        /* If at least one succeeded, remove plugin data. */
        if (ok) {
                for (l = irc_plugins; l; l = l->next) {
                        irc_plugin_t *p = l->data;
                        if (p->storage_remove) {
                                p->storage_remove(nick);
                        }
                }
        }

        return ret;
}