import os

from whoosh.fields import *
from whoosh.index import open_dir
from whoosh.qparser import QueryParser

from csvparser.csvparser import getfullpublication

SCRIPT_DIR = os.path.dirname(__file__)
DATA_DIR = os.path.abspath(os.path.join(SCRIPT_DIR, "data"))


def search(searchinput):
    """search and get search result titles and return them as book ids"""
    ix = open_dir(DATA_DIR)
    with ix.searcher() as searcher:
        query = QueryParser("content", ix.schema).parse(searchinput)
        search_results = searcher.search(query)
        searched_book_ids = []
        for book in search_results:
            searched_book_ids.append(book["title"])
        return searched_book_ids