################################################################################
#
#    Copyright 2015-2020 Félix Brezo and Yaiza Rubio
#
#    This program is part of OSRFramework. You can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as published by
#    the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
################################################################################

import argparse
import json
import sys

import osrframework.searchengines.google as google
import osrframework.thirdparties.infobel_com.processing as processing

def checkPhoneDetails(query=None):
    '''
        Method that checks if the given hash is stored in the md5crack.com website.

        :param query:    query to verify.

        :return:    Python structure.
    '''
    results = []

    #TO-DO executing the query against Google and grab the results
    #   The query should be something like "<QUERY> site:infobel.com"
    search = query + " site:infobel.com"

    # This will return a list of i3visio.uri objects
    uriObjects = google.processSearch(search)

    #TO-DO: grabbing the phone details for the QUERY
    for uri in uriObjects:
        # Recovering the website
        # Accessing the resources
        textResults = processing.getResults(uri["value"])
        # Iterating over every result to process it and recover a person object
        for r in textResults:
            person = {}
            fullname = ""
            person["type"]="com.i3visio.Person"
            person["value"] = "FULLNAME_NOT_FOUND"
            person["attributes"] = processing.extractFieldsFromResult(r)

            for entity in person["attributes"]:
                if entity["type"] == "com.i3visio.Name":
                    person["value"] = entity["value"]
                    break

            # Appending the Uri of the infobel record:
            aux = {}
            aux["type"]= "com.i3visio.URI"
            aux["value"] = uri["value"]
            aux["attributes"] = []
            person["attributes"].append(aux)

            # Appending the platform of the infobel record:
            aux = {}
            aux["type"]= "com.i3visio.Platform"
            aux["value"] = "Infobel"
            aux["attributes"] = []
            person["attributes"].append(aux)

            # Appending to the results
            results.append(person)

    return results

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='A library that wraps a search onto infobel.com via Google.', prog='checkPhoneDetails.py', add_help=False)
    # Adding the main options
    # Defining the mutually exclusive group for the main options
    parser.add_argument('-q', '--query', metavar='<query>', action='store', help='query to be performed to infobel.com.', required=True)

    groupAbout = parser.add_argument_group('About arguments', 'Showing additional information about this program.')
    groupAbout.add_argument('-h', '--help', action='help', help='shows this help and exists.')
    groupAbout.add_argument('--version', action='version', version='%(prog)s 0.1.0', help='shows the version of the program and exists.')

    args = parser.parse_args()

    print(json.dumps(checkPhoneDetails(query=args.query), indent=2))