"""
Project:    PARTS Website
Author:     Edward Middleton-Smith
            Precision And Research Technology Systems Limited

Technology: App Routing
Feature:    Dog - Assessment Routes

Description:
Dog Assessments Page Controller.
"""

# IMPORTS
# internal
from business_objects.api import API
from business_objects.dog.command import Command
from business_objects.dog.assessment import Assessment
from business_objects.dog.assessment_command_modality_link import Assessment_Command_Modality_Link
from business_objects.dog.assessment_response import Assessment_Response
from business_objects.dog.distraction import Distraction
from datastores.datastore_dog import DataStore_Dog
from forms.dog.assessment import Filters_Assessment
from helpers.helper_app import Helper_App
from models.model_view_dog_assessment import Model_View_Dog_Assessment
from models.model_view_home import Model_View_Home
import lib.argument_validation as av
# external
from flask import Flask, render_template, jsonify, request,  render_template_string, send_from_directory, redirect, url_for, session, Blueprint, current_app, flash
from flask_mail import Mail, Message
from extensions import db, oauth, mail
from urllib.parse import quote_plus, urlencode
from authlib.integrations.flask_client import OAuth
from authlib.integrations.base_client import OAuthError
from urllib.parse import quote, urlparse, parse_qs
import json
import base64
import hmac
import hashlib
import datetime
from altcha import ChallengeOptions, create_challenge, verify_solution


routes_dog_assessment = Blueprint('routes_dog_assessment', __name__)


@routes_dog_assessment.route(Model_View_Dog_Assessment.HASH_PAGE_DOG_ASSESSMENTS, methods=['GET'])
def assessments():
    Helper_App.console_log('assessments')
    Helper_App.console_log(f'request_args: {request.args}')
    try:
        form_filters = Filters_Assessment.from_json(request.args)
    except Exception as e:
        Helper_App.console_log(f'Error: {e}')
        form_filters = Filters_Assessment()
    Helper_App.console_log(f'form_filters={form_filters}')
    model = Model_View_Dog_Assessment(form_filters_old = form_filters)
    if not model.is_user_logged_in:
        return redirect(url_for('routes_dog_home.home'))
    Helper_App.console_log(f'form_filters={form_filters}')
    return render_template('pages/dog/_assessments.html', model = model)

@routes_dog_assessment.route(Model_View_Dog_Assessment.HASH_SAVE_DOG_ASSESSMENT_DISTRACTION_AND_RESPONSE, methods=['POST'])
def save_assessment():
    Helper_App.console_log('save_assessment')
    data = Helper_App.get_request_data(request)
    try:
        token_received = data.get(Model_View_Dog_Assessment.FLAG_CSRF_TOKEN)
        token_expected = session.get(Model_View_Dog_Assessment.FLAG_CSRF_TOKEN) # 'csrf_token'
        if (not token_received) or not hmac.compare_digest(token_received, token_expected):
            Helper_App.console_log(f'token_received: {token_received}\ntoken_expected: {token_expected}')
            Helper_App.console_log(f'session: {session}\nkey: {Model_View_Dog_Assessment.FLAG_CSRF_TOKEN}')
            return jsonify({
                Model_View_Dog_Assessment.FLAG_STATUS: Model_View_Dog_Assessment.FLAG_FAILURE, 
                Model_View_Dog_Assessment.FLAG_MESSAGE: f'Invalid token received.'
            })
        form_filters = Filters_Assessment.from_json(data[Model_View_Dog_Assessment.FLAG_FORM_FILTERS])
        """
        if not form_filters.validate_on_submit():
            return jsonify({
                Model_View_Dog_Assessment.FLAG_STATUS: Model_View_Dog_Assessment.FLAG_FAILURE, 
                Model_View_Dog_Assessment.FLAG_MESSAGE: f'Filters form invalid.\n{form_filters.errors}'
            })
        """
        model_permissions = Model_View_Dog_Assessment(form_filters_old=form_filters)
        Helper_App.console_log('made model')
        if not model_permissions.is_user_logged_in:
            raise Exception('User not logged in')
        
        assessments = data[Model_View_Dog_Assessment.FLAG_ASSESSMENT]
        if len(assessments) == 0:
            return jsonify({
                Model_View_Dog_Assessment.FLAG_STATUS: Model_View_Dog_Assessment.FLAG_FAILURE, 
                Model_View_Dog_Assessment.FLAG_MESSAGE: f'No assessments.'
            })
        objs_assessment = []
        objs_distraction = []
        objs_assessment_command_modality_link = []
        objs_assessment_responses = []
        for assessment in assessments:
            objs_assessment.append(Assessment.from_json(assessment))
            objs_distraction.extend([Distraction.from_json(link_json) for link_json in assessment.get(Distraction.FLAG_DISTRACTION, [])])
            for link_json in assessment.get(Assessment_Command_Modality_Link.FLAG_ASSESSMENT_COMMAND_MODALITY_LINK, []):
                objs_assessment_command_modality_link.append(Assessment_Command_Modality_Link.from_json(link_json))
                objs_assessment_responses.extend([Assessment_Response.from_json(response_json) for response_json in link_json.get(Assessment_Response.FLAG_ASSESSMENT_RESPONSE, [])])
        """
        errors = []
        Helper_App.console_log(f'objs_assessment={objs_assessment}')
        if len(objs_assessment) > 0:
            errors.extend(DataStore_Dog.save_assessments(data.get('comment', 'No comment'), objs_assessment))
        Helper_App.console_log(f'objs_distraction={objs_distraction}')
        if len(objs_distraction) > 0:
            errors.extend(DataStore_Dog.save_distractions(data.get('comment', 'No comment'), objs_distraction))
        Helper_App.console_log(f'objs_assessment_command_modality_link={objs_assessment_command_modality_link}')
        if len(objs_assessment_command_modality_link) > 0:
            errors.extend(DataStore_Dog.save_assessment_command_modality_links(data.get('comment', 'No comment'), objs_assessment_command_modality_link))
        Helper_App.console_log(f'objs_assessment_responses={objs_assessment_responses}')
        if len(objs_assessment_responses) > 0:
            errors.extend(DataStore_Dog.save_assessment_responses(data.get('comment', 'No comment'), objs_assessment_responses))
        """
        errors = DataStore_Dog.save_assessments_distactions_and_responses(
              comment = data.get('comment', 'No comment')
            , assessments = objs_assessment
            , distractions = objs_distraction
            , assessment_command_modality_links = objs_assessment_command_modality_link
            , assessment_responses = objs_assessment_responses
        )
        # model_return = Model_View_Dog_Assessment(form_filters_old=form_filters)
        
        if (len(errors) > 0):
            return jsonify({
                Model_View_Dog_Assessment.FLAG_STATUS: Model_View_Dog_Assessment.FLAG_FAILURE, 
                Model_View_Dog_Assessment.FLAG_MESSAGE: f'Error saving assessments.\n{model_permissions.convert_list_objects_to_json(errors)}'
            })
        return jsonify({
            Model_View_Dog_Assessment.FLAG_STATUS: Model_View_Dog_Assessment.FLAG_SUCCESS, 
            Model_View_Dog_Assessment.FLAG_DATA: None # Model_View_Dog_Assessment.convert_list_objects_to_json(model_permissions.assessments)
        })
    except Exception as e:
        return jsonify({
            Model_View_Dog_Assessment.FLAG_STATUS: Model_View_Dog_Assessment.FLAG_FAILURE, 
            Model_View_Dog_Assessment.FLAG_MESSAGE: f'Bad data received by controller.\n{e}'
        })

@routes_dog_assessment.route(Model_View_Dog_Assessment.HASH_PAGE_DOG_ASSESSMENT, methods=['GET'])
def assessment():
    Helper_App.console_log('assessment:')
    Helper_App.console_log(f'request_args: {request.args}')
    try:
        form_filters = Filters_Assessment.from_json(request.args)
    except Exception as e:
        Helper_App.console_log(f'Error: {e}')
        form_filters = Filters_Assessment()
    Helper_App.console_log(f'form_filters={form_filters}')
    temp_id_assessment = request.args.get(Model_View_Dog_Assessment.ATTR_ID_ASSESSMENT, -1)
    id_assessment = None if (temp_id_assessment is None or temp_id_assessment == '') else int(temp_id_assessment)
    model = Model_View_Dog_Assessment(form_filters_old = form_filters, id_assessment = id_assessment, hash_page_current = Model_View_Dog_Assessment.HASH_PAGE_DOG_ASSESSMENT)
    model._title = 'Assessment'
    if model.assessments is None or len(model.assessments) == 0:
        if id_assessment is not None and id_assessment > 0:
            return assessments()
        else:
            model.assessments = [Assessment()]
    if not model.is_user_logged_in:
        return redirect(url_for('routes_dog_home.home'))
    session[Model_View_Dog_Assessment.FLAG_CSRF_TOKEN] = model.form_filters.csrf_token.current_token #.hidden_tag()
    Helper_App.console_log(f'form_filters={form_filters}')
    return render_template('pages/dog/_assessment.html', model = model)