from .models import *


def get_action(jsn):
    if jsn['type'] == 'answer':
        # language switching action is not an explicit one, but
        # implicit when the action contains the "language" tag
        if "language" in jsn:
            if 'plan_delay' in jsn.keys() and 'plan_date' in jsn.keys():
                return LanguageSwitchAction(jsn['language'], jsn['expression'], jsn['plan_delay'], jsn['plan_date'])
            return LanguageSwitchAction(jsn['language'], jsn['expression'])
        else:
            if 'plan_delay' in jsn.keys() and 'plan_date' in jsn.keys():
                return AnswerAction(jsn['expression'], jsn['plan_delay'], jsn['plan_date'])
            return AnswerAction(jsn['expression'])
    elif jsn['type'] == 'table':
        return TableAction(jsn['columns'])
    elif jsn['type'] == 'map':
        return MapAction(jsn['latitude'], jsn['longitude'], jsn['zoom'])
    elif jsn['type'] == 'anchor':
        return AnchorAction(jsn['link'], jsn['text'])
    elif jsn['type'] == 'rss':
        return RssAction(jsn['count'], jsn['title'], jsn['description'], jsn['link'])
    elif jsn['type'] == 'video_play':
        return VideoAction(jsn['identifier'], jsn['identifier_type'])
    elif jsn['type'] == 'stop':
        return StopAction()
    elif jsn['type'] == 'audio_play':
        if 'plan_delay' in jsn.keys() and 'plan_date' in jsn.keys():
            return AudioAction(jsn['identifier'], jsn['identifier_type'], jsn['plan_delay'], jsn['plan_date'])
        return AudioAction(jsn['identifier'], jsn['identifier_type'])
    elif jsn['type'] == 'audio_volume':
        return VolumeAction(jsn['volume'])
    elif jsn['type'] == 'pause':
        return MediaAction(jsn['type'])
    elif jsn['type'] == 'restart':
        return MediaAction(jsn['type'])
    elif jsn['type'] == 'resume':
        return MediaAction(jsn['type'])
    elif jsn['type'] == 'next':
        return MediaAction(jsn['type'])
    elif jsn['type'] == 'previous':
        return MediaAction(jsn['type'])
    elif jsn['type'] == 'shuffle':
        return MediaAction(jsn['type'])
    else:
        return UnknownAction()


def get_query_response(parsed_dict):
    ans = parsed_dict['answers'][0]

    data = [Datum(jsn)
            for jsn in ans['data']]

    metadata = Metadata(ans['metadata'])

    actions = [get_action(jsn)
               for jsn in ans['actions']]
    answer = Answer(data, metadata, actions[::-1])

    try:
        identity_json = parsed_dict['session']['identity']
        identity = Identity(identity_json)
        session = Session(identity)
    except KeyError:
        session = None

    return QueryResponse(answer,parsed_dict, session)


def get_sign_in_response(parsed_dict):
    identity_json = parsed_dict['session']['identity']
    identity = Identity(identity_json)
    session = Session(identity)
    return LoginResponse(parsed_dict, session)


def get_memory_responses(parsed_dict):
    cognitions = parsed_dict['cognitions']

    susi_responses = []
    for cognition in cognitions:
        susi_responses.append(get_query_response(cognition))

    return susi_responses


def get_forgot_password_response(parsed_dict):
    return ForgotPasswordResponse(parsed_dict)


def get_sign_up_response(parsed_dict):
    identity_json = parsed_dict['session']['identity']
    identity = Identity(identity_json)
    session = Session(identity)
    return SignUpResponse(parsed_dict, session)