SylFilter

#include <stdio.h>

#include <locale.h>

#include "filter.h"

#include "filter-manager.h"

#include "filter-utils.h"

#include "filter-kvs.h"

#define USE_QDBM 1

#ifdef USE_QDBM

#  include "filter-kvs-qdbm.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;

        XFilterConstructorFunc learn_ctors[] = {

                xfilter_textcontent_new,

                xfilter_wordsep_new,

                xfilter_ngram_new,

                xfilter_bayes_learn_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();

#endif

        xfilter_bayes_db_init();

        // set message files

        if (argc > 1)

                rfc822_file = argv[1];

        // learning

        mgr = xfilter_manager_new();

        xfilter_manager_add_filters(mgr, learn_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;

}