import json
from _ransomcoindb import ransomcoindb
import pprint

copyright = """
  Copyright 2019 (C) by Aaron Kaplan <aaron@lo-res.org>, all rights reserved.
  This file is part of the ransomwarecoindDB project and licensed under the AGPL 3.0 license
"""

__version__ = 0.1


debug=False

misperrors = {'error': 'Error'}
# mispattributes = {'input': ['sha1', 'sha256', 'md5', 'btc', 'xmr', 'dash' ], 'output': ['btc', 'sha1', 'sha256', 'md5', 'freetext']}
mispattributes = {'input': ['sha1', 'sha256', 'md5', 'btc'], 'output': ['btc', 'sha1', 'sha256', 'md5', 'freetext']}
moduleinfo = {'version': __version__, 'author': 'Aaron Kaplan', 'description': 'Module to access the ransomcoinDB (ransomcoindb.metadata.li)', 'module-type': ['expansion', 'hover']}
moduleconfig = ['api-key']


def handler(q=False):
    """ the main handler function which gets a JSON dict as input and returns a results dict """

    if q is False:
        return False

    q = json.loads(q)
    api_key = q["config"]["api-key"]
    r = {"results": []}

    """ the "q" query coming in should look something like this:
        {'config': {'api-key': '<api key here>'},
         'md5': 'md5 or sha1 or sha256 or btc',
         'module': 'metadatali_ransomcoindb',
         'persistent': 1}
    """

    for key in ['md5', 'sha1', 'sha256', 'btc']:        # later: xmr, dash
        if key in q:
            answer = ransomcoindb.get_data_by('BTC', key, q[key], api_key)
            """ The results data type should be:
              r =  { 'results': [ {'types': 'md5', 'values': [ a list of all md5s or all binaries related to this btc address ]  } ] }
            """
            if key in ['md5', 'sha1', 'sha256']:
                r['results'].append({'types': 'btc', 'values': [ a['btc'] for a in answer ]})
            elif key == 'btc':
                # better: create a MISP object
                r['results'].append({ 'types': 'sha1', 'values': [ a['sha1'] for a in answer ]})
                r['results'].append({ 'types': 'md5', 'values': [ a['md5'] for a in answer ]})
                r['results'].append({ 'types': 'sha256', 'values': [ a['sha256'] for a in answer ]})

    return r


def introspection():
    return mispattributes


def version():
    moduleinfo['config'] = moduleconfig
    return moduleinfo