import json

import click
from flask import jsonify, make_response, render_template

from app import app, mongo
from app.suggestions.suggest_cluster import ClusterSuggestor


def add_cluster_json_file_to_mongo(json_file):

    with open(json_file) as f:
        clust_list = json.load(f)

    for clust in clust_list:
        mongo.db.clusters.replace_one(
            {'_id': clust['_id']}, clust, upsert=True)


def add_ctep_json_file_to_mongo(json_file):

    with open(json_file) as f:
        ctep_list = json.load(f)

    for ctep in ctep_list:
        mongo.db.ctep.replace_one(
            {'_id': ctep['_id']}, ctep, upsert=True)


@app.errorhandler(404)
def not_found(error):
    return make_response(jsonify({'error': 'Not found'}), 404)


@app.route('/')
def index():
    return render_template('index.html')


@app.route('/get-suggestion/')
@app.route('/get-suggestion/<input_json>')
def get_suggestions(input_json=None, methods=['POST']):

    if not input_json:
        return 'No input text.'

    try:
        input_data = json.loads(input_json)
    except ValueError:
        input_data = json.loads(json.dumps(input_json))

    cluster_suggestor = ClusterSuggestor()
    suggestion = cluster_suggestor.suggest(input_data)

    return jsonify(suggestion)


@click.group()
def cli():
    pass


@cli.command()
def run_server():
    app.run(host='0.0.0.0', port=4000)


@cli.command()
@click.argument('json_file')
def add_cluster_json(json_file):
    add_cluster_json_file_to_mongo(json_file)


@cli.command()
@click.argument('json_file')
def add_ctep_json(json_file):
    add_ctep_json_file_to_mongo(json_file)


if __name__ == '__main__':
    cli()