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']}