Implement chat API key generation, and create a tenant_overview

eveai_app/views/user_views.py

@@ -1,16 +1,17 @@
 # from . import user_bp
 import uuid
 from datetime import datetime as dt, timezone as tz
-from flask import request, redirect, url_for, flash, render_template, Blueprint, session, current_app
+from flask import request, redirect, url_for, flash, render_template, Blueprint, session, current_app, jsonify
 from flask_security import hash_password, roles_required, roles_accepted, current_user
 from sqlalchemy.exc import SQLAlchemyError
 import ast
 
 from common.models.user import User, Tenant, Role, TenantDomain
-from common.extensions import db
+from common.extensions import db, kms_client
 from .user_forms import TenantForm, CreateUserForm, EditUserForm, TenantDomainForm
 from common.utils.database import Database
 from common.utils.view_assistants import prepare_table_for_macro
+from common.utils.key_encryption import generate_api_key
 
 user_bp = Blueprint('user_bp', __name__, url_prefix='/user')
 
@@ -324,6 +325,45 @@ def edit_tenant_domain(tenant_domain_id):
     return render_template('user/edit_tenant_domain.html', form=form, tenant_domain_id=tenant_domain_id)
 
 
+@user_bp.route('/check_chat_api_key', methods=['POST'])
+@roles_accepted('Super User', 'Tenant Admin')
+def check_chat_api_key():
+    tenant_id = session['tenant']['id']
+    tenant = Tenant.query.get_or_404(tenant_id)
+
+    if tenant.encrypted_chat_api_key:
+        return jsonify({'api_key_exists': True})
+    return jsonify({'api_key_exists': False})
+
+
+@user_bp.route('/generate_chat_api_key', methods=['POST'])
+@roles_accepted('Super User', 'Tenant Admin')
+def generate_chat_api_key():
+    tenant = Tenant.query.get_or_404(session['tenant']['id'])
+
+    new_api_key = generate_api_key(prefix="EveAI-CHAT")
+    tenant.encrypted_chat_api_key = kms_client.encrypt_api_key(new_api_key)
+    update_logging_information(tenant, dt.now(tz.utc))
+
+    try:
+        db.session.add(tenant)
+        db.session.commit()
+    except SQLAlchemyError as e:
+        db.session.rollback()
+        current_app.logger.error(f'Unable to store api key for tenant {tenant.id}. Error: {str(e)}')
+
+    return jsonify({'new_api_key': 'API key generated successfully.', 'api_key': new_api_key}), 200
+
+
+@user_bp.route('/tenant_overview', methods=['GET'])
+@roles_accepted('Super User', 'Tenant Admin')
+def tenant_overview():
+    tenant_id = session['tenant']['id']
+    tenant = Tenant.query.get_or_404(tenant_id)
+    form = TenantForm(obj=tenant)
+    return render_template('user/tenant_overview.html', form=form)
+
+
 def set_logging_information(obj, timestamp):
     obj.created_at = timestamp
     obj.updated_at = timestamp
@@ -332,5 +372,5 @@ def set_logging_information(obj, timestamp):
 
 
 def update_logging_information(obj, timestamp):
-    obj.created_by = current_user.id
+    obj.updated_at = timestamp
     obj.updated_by = current_user.id