from pprint import pprint
from typing import Iterable
from apexsearch import ApexSearch


class Search:

    def __init__(self):
        self.apex = ApexSearch('data/apex',
                               tables={
                                   "docs": {
                                       "content": ["phrase"],
                                       "title": "title",
                                       "extras": ["sourceFile", "pos"],
                                   }
                               },
                               id_field='id')

    def rebuild(self, docs: Iterable):
        self.apex.build_complete_index(lambda *a: docs)
        print('rebuild complete')

    def search(self, q: str):
        res = self.apex.search(q, target_number=100)
        pprint(res)
        for row in res['results']:
            yield {
                'title': row['title'],
                'snippetHtml': row['highlighted_content']
            }