#include "../inc/text_query_.h"
#include <iostream>
#include <string>
#include <istream>
#include <sstream>

using namespace std;
void runQueries(ifstream &infile)
{
    //根据输入文件构建一个TextQuery对象
    TextQuery tq(infile);
    //根据输入的单词返回查询结果
    while (true)
    {
        cout << "enter word to look for, or q to quit" << endl;
        //输入单词查询结果，或者输入q退出循环,或者遇到终止退出。
        string s;
        if (!(cin >> s) || s == "q")
            break;
        print(cout, tq.query(s)) << endl;
    }
}

TextQuery::TextQuery(ifstream &infile) : file(make_shared<vector<string>>())
{
    string text;
    //按行读取文件内容写入text
    while (getline(infile, text))
    {
        //将一行的内容写入vector file中
        file->push_back(text);
        //统计行号，vector的大小-1就是行号
        int n = file->size() - 1;
        //将字符串构造为string流
        istringstream line(text);
        string word;
        //将字符串中的单词依次写入word
        while (line >> word)
        {
            auto &lineset = wm[word];
            //如果单词不在wm中，返回的就是空指针
            if (!lineset)
            {
                //重新构造一个set，让智能指针绑定他
                lineset.reset(new set<line_no>());
            }
            lineset->insert(n);
        }
    }
}

QueryResult TextQuery::query(const string &str) const
{
    //根据单词去map中查找
    auto findval = wm.find(str);
    //构造一个指向空的set的指针
    static auto nodata = make_shared<set<line_no>>();
    if (findval == wm.end())
    {
        return QueryResult(str, nodata, file);
    }

    return QueryResult(str, findval->second, file);
}

std::ostream &print(std::ostream &os, const QueryResult &qr)
{
    //依次输出单词，出现次数
    os << qr.sought << " occurs " << qr.lines->size() << " times" << endl;
    //打印出现的每一行
    for (auto num : *(qr.lines))
    {
        os << "\t line( " << num + 1 << " ) " << *(qr.file->begin() + num) << endl;
    }
    return os;
}