from bs4 import BeautifulSoup
import re


class Parser:

    def __init__(self, resultstoparse):
        self.ipaddresses = []
        self.souphosts = BeautifulSoup(resultstoparse.total_resultshosts, features='html.parser')
        self.soupcerts = BeautifulSoup(resultstoparse.total_resultscerts, features='html.parser')
        self.hostnames = []
        self.hostnamesfromcerts = []
        self.urls = []
        self.numberofpageshosts = 0
        self.numberofpagescerts = 0
        self.domain = resultstoparse.word

    def search_hostnamesfromcerts(self):
        try:
            hostnamelist = self.soupcerts.findAll('i', 'fa fa-fw fa-home')
            for hostnameitem in hostnamelist:
                hostitems = hostnameitem.next_sibling
                hostnames = str(hostitems)
                hostnamesclean = re.sub('[ \'\[\]]', '', hostnames)
                hostnamesclean = re.sub(r'\.\.\.', r'', hostnamesclean)
                self.hostnamesfromcerts.extend(hostnamesclean.split(','))
            self.hostnamesfromcerts = list(filter(None, self.hostnamesfromcerts))
            matchingdomains = [s for s in self.hostnamesfromcerts if str(self.domain) in s]  # filter out domains issued to other sites
            self.hostnamesfromcerts = matchingdomains
            return self.hostnamesfromcerts
        except Exception as e:
            print('Error occurred in the Censys module: certificate hostname parser: ' + str(e))

    def search_ipaddresses(self):
        try:
            ipaddresslist = self.souphosts.findAll('a', 'SearchResult__title-text')
            for ipaddressitem in ipaddresslist:
                self.ipaddresses.append(ipaddressitem.text.strip())
            return self.ipaddresses
        except Exception as e:
            print('Error occurred in the Censys module: IP address parser: ' + str(e))

    def search_totalpageshosts(self):
        try:
            items = self.souphosts.findAll('span', 'SearchResultSectionHeader__statistic')
            if items == [] or items is None:
                self.numberofpageshosts = 0
                return self.numberofpageshosts
            numbers = re.findall(r"/\d*", items[0].text)
            pagenumber = numbers[0].replace('/', '')
            self.numberofpageshosts = int(pagenumber)
            return self.numberofpageshosts
        except Exception as e:
            print('Error occurred in the Censys module IP search: page parser: ' + str(e))

    def search_totalpagescerts(self):
        try:
            items = self.soupcerts.findAll('span', 'SearchResultSectionHeader__statistic')
            if items == [] or items is None:
                self.numberofpageshosts = 0
                return self.numberofpageshosts
            numbers = re.findall(r"/\d*", items[0].text)
            pagenumber = numbers[0].replace('/', '')
            self.numberofpagescerts = int(pagenumber)
            return self.numberofpagescerts
        except Exception as e:
            print('Error occurred in the Censys module IP search: page parser: ' + str(e))