/**
 * Project SearchEngine
 */

#ifndef _WEBPAGEQUERY_H
#define _WEBPAGEQUERY_H

#include "SplitTool.h"
#include "WordSegmentation.h"
#include <fstream>
#include <iostream>
#include <ostream>
#include <set>
#include <sstream>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
using std::cin;
using std::cout;
using std::endl;
using std::pair;
using std::set;
using std::string;
using std::unordered_map;
using std::unordered_set;
using std::vector;

#include "WebPage.h"

class WebPageQuery {
public:
 WebPageQuery();

 string doQuery(string key);

private:
 void loadLibrary();

private:
 WordSegmentation _jieba;
 unordered_map<int, WebPage> _pageLib;
 unordered_map<int, pair<int, int>> _offsetLib;
 unordered_map<string, set<pair<int, double>>> _invertIndexTable;
};

#endif //_WEBPAGEQUERY_H