"""
ergal.utils
~~~~~~~~~~~

This module implements the utility methods used by the
Profile interface.

:author: Elliott Maguire
:copyright: (c) 2019 by Elliott Maguire
"""

import json
import types
import sqlite3

import xmltodict


def get_db(test=False):
    """ Get/create a database connection.

    If a local ergal.db file exists, a connection
    is established and returned, otherwise a new
    database is created and the connection is returned.

    :param test: (optional) determines whether or not a test database
                            should be created.
    """
    file = 'ergal_test.db' if test else 'ergal.db'
    db = sqlite3.connect(file)
    cursor = db.cursor()

    db.execute("""
        CREATE TABLE IF NOT EXISTS Profile (
            id          TEXT    NOT NULL,
            name        TEXT    NOT NULL,
            base        TEXT    NOT NULL,
            auth        TEXT,
            endpoints   TEXT,

            PRIMARY KEY(id))""")

    return db, cursor


async def parse(response, targets=None):
    """ Parse response data.

    :param response: a requests.Response object
    :param targets: a list of data targets
    """
    try:
        data = json.loads(response.text)
    except json.JSONDecodeError:
        data = xmltodict.parse(response.text)

    if type(data) is list:
        data = {'data': data}

    output = {}
    async def search(d):
        for k, v in d.items():
            if k in targets:
                output[k] = None
                yield v
            elif type(v) is dict:
                async for i in search(v):
                    output[i] = None
                    yield i

    for k, v in zip(output, [i async for i in search(data)]):
        output[k] = v

    return output

