import requests
import bs4
from colorama import Fore
from plugin import plugin, require


@require(network=True)
@plugin('google')
class Scraper():
    """
    Runs a basic search and returns the default answer provided by google.
    Google's default answers all have a class contaning BNeawe, s3v9rd, or
    orAP7Wnd, so we can just run a google query and parse the results

    example: google What is the Large Hadron Collider?
    """

    def __call__(self, jarvis, s):
        jarvis.say(self.search(s), Fore.BLUE)

    def search(self, question):
        query = '+'.join(question.split(' '))
        url = f'https://www.google.com/search?q={query}&ie=utf-8&oe=utf-8'
        response = requests.get(url)
        soup = bs4.BeautifulSoup(response.text, 'html.parser')

        google_answers = soup.find_all("div", {"class": "BNeawe iBp4i AP7Wnd"})
        for answer in google_answers:
            if answer.text:
                return self.parse_result(answer.text)

        first_answers = soup.find_all("div", {"class": "BNeawe s3v9rd AP7Wnd"})
        for answer in first_answers:
            if answer.text:
                return self.parse_result(answer.text)

        all_div = soup.find_all("div")
        google_answer_keys = ['BNeawe', 's3v9rd', 'AP7Wnd', 'iBp4i']

        for div in all_div:
            for key in google_answer_keys:
                if key in div:
                    return self.parse_result(div.text)

        return 'No Answers Found'

    def parse_result(self, result):
        result = result.split('\n')[0]
        if result.endswith('Wikipedia'):
            result = result[:result.find('Wikipedia')]
        result += '\n'

        return result