#include "Xapian_search.h"
#include <QDebug>

Xapian_search::Xapian_search(QObject *parent) : QObject(parent)
{
#ifndef NO_OPENCC
    if(!QFile::exists("opencc/"))
    {
        QDir opencc;
        opencc.mkpath("opencc/");
    }
    if(!QFile::exists("opencc/s2t.json"))
    {
        QFile::copy(":/opencc/s2t.json","opencc/s2t.json");
    }
    if(!QFile::exists("opencc/STCharacters.ocd"))
    {
        QFile::copy(":/opencc/STCharacters.ocd","opencc/STCharacters.ocd");
    }
    if(!QFile::exists("opencc/STPhrases.ocd"))
    {
        QFile::copy(":/opencc/STPhrases.ocd","opencc/STPhrases.ocd");
    }
    if(!QFile::exists("opencc/t2s.json"))
    {
        QFile::copy(":/opencc/t2s.json","opencc/t2s.json");
    }
    if(!QFile::exists("opencc/TSCharacters.ocd"))
    {
        QFile::copy(":/opencc/TSCharacters.ocd","opencc/TSCharacters.ocd");
    }
    if(!QFile::exists("opencc/TSPhrases.ocd"))
    {
        QFile::copy(":/opencc/TSPhrases.ocd","opencc/TSPhrases.ocd");
    }
    enable_opencc = false;
    try
    {
        opencc_t2s = new opencc::SimpleConverter("opencc/t2s.json");
        opencc_s2t = new opencc::SimpleConverter("opencc/s2t.json");
        enable_opencc = true;
    }
    catch(...)
    {
        qDebug()<<"opencc start fail";
        enable_opencc = false;
    }
#endif
    qDebug()<<"start xapian";
#ifdef HAVE__PUTENV_S
    _putenv_s("XAPIAN_CJK_NGRAM", "1");
#elif defined HAVE_SETENV
    setenv("XAPIAN_CJK_NGRAM", "1", 1);
#else
    putenv(const_cast<char*>("XAPIAN_CJK_NGRAM=1"));
#endif
    enstem = Xapian::Stem("en");
    termgen.set_flags(Xapian::TermGenerator::FLAG_CJK_NGRAM);
    termgen.set_stemmer(enstem);
}

void Xapian_search::on_init(QStringList dir, int batch)
{
    qDebug()<<"init search in xapian";
    try{
        db = Xapian::Database();
		for(int i = 0; i < dir.size(); ++i)
        {
            if(QFile::exists(dir.at(i)+"/data.idx"))
                db.add_database(Xapian::Database(dir.at(i).toStdString()+"/data.idx"));
            else
                db.add_database(Xapian::Database(dir.at(i).toStdString()));
        }
        //enquire = shared_ptr<Xapian::Enquire>(new Xapian::Enquire(db));
        Q_EMIT init_finish(batch);
    }
    catch(exception& e)
    {
        stringstream e2;
        e2<<e.what();
        qDebug()<<e.what();
        Q_EMIT init_finish(-1);
    }
    catch(...)
    {
        Q_EMIT init_finish(-2);
    }
}

void Xapian_search::on_search(QString str, int batch, QStringList type)
{
    QStringList urls;
    QStringList key_words;
    QString runing = "";
    runing = "Strat try!";

    try{
        Xapian::Document doc;
        Xapian::Enquire enquire(db);
		termgen.set_document(doc);
        termgen.set_stemming_strategy(Xapian::TermGenerator::STEM_SOME);
        string str_o = str.toStdString();
        string str_c = "";
#ifndef NO_OPENCC
        if(enable_opencc)
        {
            string str_t = opencc_s2t->Convert(str_o);
            string str_s = opencc_t2s->Convert(str_o);
            if(str_t != str_o)
                str_c += str_t;
            if(str_s != str_o)
                str_c += str_s;
            str_o += str_c;
        }
#endif
        if(str_c == ""||str_c.empty())
            str_c = str_o;
        termgen.index_text_without_positions(str_o,1,"C");
		runing = "Add Term!";
		Xapian::TermIterator it;
        QList<Xapian::Query> querys;
		for (it = doc.termlist_begin(); it != doc.termlist_end(); ++it)
		{
            querys.append(Xapian::Query(*it,it.get_wdf()));
		}
        Xapian::Document doc2;
        termgen.set_document(doc2);
        termgen.set_stemming_strategy(Xapian::TermGenerator::STEM_NONE);
        termgen.index_text_without_positions(str_c+enstem(str.toStdString()),1);
        Xapian::TermIterator it2;
        for (it2 = doc2.termlist_begin(); it2 != doc2.termlist_end(); ++it2)
        {
            querys.append(Xapian::Query(*it2,it2.get_wdf()));
            key_words.append(QString::fromStdString(*it2));
        }
        Xapian::Query query(Xapian::Query::OP_ELITE_SET, querys.begin(), querys.end(), 20);
        if (query.empty() || querys.empty())
        {
            qDebug()<<"query is empty"<<str<<batch<<type;
            Q_EMIT search_result(urls,QStringList(),batch);
        }
        Xapian::Query name_query;
		for(int i1 = 0; i1 < type.size(); ++i1)
        {
            if (name_query.empty())
                name_query = Xapian::Query(Xapian::Query::OP_VALUE_RANGE,1,type.at(i1).toStdString(),type.at(i1).toStdString());
			else
                name_query = Xapian::Query(Xapian::Query::OP_OR,name_query,Xapian::Query(Xapian::Query::OP_VALUE_RANGE,1,type.at(i1).toStdString(),type.at(i1).toStdString()));
        }
        query = Xapian::Query(Xapian::Query::OP_AND,query,name_query);
		runing = "Strat query!";
        enquire.set_query(query);
        Xapian::MSet matches = enquire.get_mset(0, 25);
		for (Xapian::MSetIterator i = matches.begin(); i != matches.end(); ++i)
		{
			urls.append(QString::fromStdString(i.get_document().get_value(1)+":/"+i.get_document().get_data()));
        }
        if(urls.count() < 1)
        {
            urls.append(tr("对不起,没找到相匹配的内容,请换其他的关键词试试.\n提示:最好不要只打数学公式或单个字哦!"));
            key_words.clear();
        }
        //qDebug()<<"search result(url):"<<urls<<batch;
        Q_EMIT search_result(urls,key_words,batch);
    }
    catch(exception& e)
    {
        stringstream e2;
        e2<<e.what();
        urls.append(QString::fromStdString(e2.str()));
        urls.append("发生错误:"+runing);
        Q_EMIT search_result(urls,QStringList(),-1);
        qDebug()<<"search3"<<batch<<str<<type<<urls<<runing;
    }
    catch(...)
    {
        urls.append("未知错误");
        Q_EMIT search_result(urls,QStringList(),-2);
        qDebug()<<"search4"<<batch<<str<<type;
    }
}
/*
QString Xapian_search::segment_2(QString src)
{
    QString a = src.remove(QRegExp("[^\\x31C0-\\x9FCC]"));
    QString c;
    int b = a.length()-1;
    for(int i = 0;i<b;++i)
    {
        c.append(" "+a.mid(i,2));
    }
    return c;
}
*/