SylFilter

/* SylFilter - a message filter

 *

 * Copyright (C) 2011 Hiroyuki Yamamoto

 * Copyright (C) 2011 Sylpheed Development Team

 */

#include "config.h"

#include <stdio.h>

#include <string.h>

#include <locale.h>

#include "filter.h"

#include "filter-manager.h"

#include "filter-utils.h"

#include "filter-kvs.h"

#ifdef USE_QDBM

#  include "filter-kvs-qdbm.h"

#elif defined(USE_SQLITE)

#  include "filter-kvs-sqlite.h"

#endif

#include "textcontent-filter.h"

#include "blacklist-filter.h"

#include "whitelist-filter.h"

#include "wordsep-filter.h"

#include "ngram-filter.h"

#include "bayes-filter.h"

int main(int argc, char *argv[])

{

        XFilterManager *mgr;

        XMessageData *msgdata;

        XMessageData *resdata;

        XFilterResult *res;

        XFilterStatus status;

        int retval = 0;

        int i;

        int is_junk = 1;

        XFilterConstructorFunc learn_junk_ctors[] = {

                xfilter_textcontent_new,

                xfilter_wordsep_new,

                xfilter_ngram_new,

                xfilter_bayes_learn_junk_new,

                NULL

        };

        XFilterConstructorFunc learn_nojunk_ctors[] = {

                xfilter_textcontent_new,

                xfilter_wordsep_new,

                xfilter_ngram_new,

                xfilter_bayes_learn_nojunk_new,

                NULL

        };

        XFilterConstructorFunc ctors[] = {

                xfilter_textcontent_new,

                xfilter_blacklist_new,

                xfilter_whitelist_new,

                xfilter_wordsep_new,

                xfilter_ngram_new,

                xfilter_bayes_new,

                NULL

        };

        const char *rfc822content =

                "From: foo@bar\n"

                "To: fuga@hoge\n"

                "Subject: test message\n"

                "\n"

                "This is a test message.\n"

                "...\n"

                "\n"

                "-- \n"

                "Foo Bar <foo@bar>\n";

        const char *rfc822_file = NULL;

        setlocale(LC_ALL, "");

        xfilter_init();

#ifdef USE_QDBM

        xfilter_kvs_qdbm_set_engine();

#elif defined(USE_SQLITE)

        xfilter_kvs_sqlite_set_engine();

#endif

        if (xfilter_bayes_db_init(NULL) < 0) {

                fprintf(stderr, "Database initialization error\n");

                return 127;

        }

        // set message files

        for (i = 1; i < argc; i++) {

                if (!strcmp(argv[i], "-j"))

                        is_junk = 1;

                else if (!strcmp(argv[i], "-c"))

                        is_junk = 0;

                else

                        rfc822_file = argv[i];

        }

        // learning

        mgr = xfilter_manager_new();

        if (is_junk)

                xfilter_manager_add_filters(mgr, learn_junk_ctors);

        else

                xfilter_manager_add_filters(mgr, learn_nojunk_ctors);

        if (rfc822_file)

                msgdata = xfilter_message_data_read_file(rfc822_file, "message/rfc822");

        else

                msgdata = xfilter_message_data_new(rfc822content, "message/rfc822");

        res = xfilter_manager_run(mgr, msgdata);

        xfilter_result_print(res);

        status = xfilter_result_get_status(res);

        resdata = xfilter_result_get_message_data(res);

        if (resdata) {

                const char *content;

                content = xfilter_message_data_get_content(resdata);

                printf("content: %s\n", content);

        }

        xfilter_result_free(res);

        xfilter_message_data_free(msgdata);

        xfilter_manager_free(mgr);

        printf("--------\n");

        mgr = xfilter_manager_new();

        xfilter_manager_add_filters(mgr, ctors);

        // test 1

        msgdata = xfilter_message_data_new("test", "text/plain");

        res = xfilter_manager_run(mgr, msgdata);

        xfilter_result_print(res);

        status = xfilter_result_get_status(res);

        resdata = xfilter_result_get_message_data(res);

        if (resdata) {

                const char *content;

                content = xfilter_message_data_get_content(resdata);

                printf("content: %s\n", content);

        }

        xfilter_result_free(res);

        xfilter_message_data_free(msgdata);

        printf("--------\n");

        // test 2

        if (rfc822_file)

                msgdata = xfilter_message_data_read_file(rfc822_file, "message/rfc822");

        else

                msgdata = xfilter_message_data_new(rfc822content, "message/rfc822");

        res = xfilter_manager_run(mgr, msgdata);

        xfilter_result_print(res);

        status = xfilter_result_get_status(res);

        if (status == XF_JUNK || status == XF_REWRITTEN)

                retval = 0;

        else if (status == XF_UNCERTAIN)

                retval = 2;

        else if (status == XF_UNSUPPORTED_TYPE || status == XF_ERROR)

                retval = 127;

        else

                retval = 1;

        resdata = xfilter_result_get_message_data(res);

        if (resdata) {

                const char *content;

                content = xfilter_message_data_get_content(resdata);

                printf("content: %s\n", content);

        }

        xfilter_result_free(res);

        xfilter_message_data_free(msgdata);

        xfilter_manager_free(mgr);

        xfilter_bayes_db_done();

        xfilter_done();

        return retval;

}