from datetime import timedelta, datetime

from functools import wraps

import requests

from urllib.parse import urlparse, parse_qs

import json

import re

from docusign_esign import ApiClient, AccountsApi

from flask import session, flash, url_for, redirect, render_template, current_app

from jinja2 import environment

from .ds_client import DSClient

from ..consts import minimum_buffer_min, API_TYPE

from ..error_handlers import process_error

def ds_logout_internal():

# remove the keys and their values from the session

session.pop("ds_access_token", None)

session.pop("ds_refresh_token", None)

session.pop("ds_user_email", None)

session.pop("ds_user_name", None)

session.pop("ds_expiration", None)

session.pop("ds_account_id", None)

session.pop("ds_account_name", None)

session.pop("ds_base_path", None)

session.pop("envelope_id", None)

session.pop("eg", None)

session.pop("envelope_documents", None)

session.pop("template_id", None)

session.pop("auth_type", None)

session.pop("api", None)

DSClient.destroy()

def create_api_client(base_path, access_token):

"""Create api client and construct API headers"""

api_client = ApiClient()

api_client.host = base_path

api_client.set_default_header(header_name="Authorization", header_value=f"Bearer {access_token}")

return api_client

def ds_token_ok(buffer_min=60):

"""

:param buffer_min: buffer time needed in minutes

:return: true iff the user has an access token that will be good for another buffer min

"""

ok = "ds_access_token" in session and "ds_expiration" in session

ok = ok and (session["ds_expiration"] - timedelta(minutes=buffer_min)) > datetime.utcnow()

return ok

def get_manifest(manifest_url):

try:

manifest = requests.get(manifest_url).json()

return manifest

except Exception as e:

current_app.logger.info(f"Could not load code examples manifest. Manifest URL: {manifest_url} with error {str(e)}")

raise Exception(f"Could not load code examples manifest. Manifest URL: {manifest_url} with error {str(e)}")

def get_example_by_number(manifest, number, apiName):

for api in manifest["APIs"]:

if api["Name"] == apiName:

for group in api["Groups"]:

for example in group["Examples"]:

if example["ExampleNumber"] == number:

return example

def authenticate(eg, api):

def decorator(func):

@wraps(func)

def wrapper(*args, **kwargs):

if not (session.get("api") == api or (api == API_TYPE["ESIGNATURE"] and not session.get("api"))):

session["api"] = api

session["eg"] = url_for(eg + ".get_view")

if api == API_TYPE["MONITOR"]:

session["auth_type"] = "jwt"

return redirect(url_for("ds.ds_login"))

else:

return redirect(url_for("ds.ds_must_authenticate"))

if ds_token_ok(minimum_buffer_min):

return func(*args, **kwargs)

else:

# We could store the parameters of the requested operation

# so it could be restarted automatically.

# But since it should be rare to have a token issue here,

# we"ll make the user re-enter the form data after

# authentication.

session["eg"] = url_for(eg + ".get_view")

if session.get("auth_type"):

flash("Token has been updated")

return redirect(url_for("ds.ds_login"))

else:

return redirect(url_for("ds.ds_must_authenticate"))

return wrapper

return decorator

def authenticate_agent(eg):

def decorator(func):

@wraps(func)

def wrapper(*args, **kwargs):

session["eg"] = url_for(eg + ".list_envelopes")

if ds_token_ok(minimum_buffer_min):

return func(*args, **kwargs)

else:

return redirect(url_for("ds.ds_must_authenticate"))

return wrapper

return decorator

def ensure_manifest(manifest_url):

def decorator(func):

@wraps(func)

def wrapper(*args, **kwargs):

manifest = get_manifest(manifest_url=manifest_url)

session["manifest"] = manifest

return func(*args, **kwargs)

return wrapper

return decorator

def to_json(value):

return json.dumps(value)

environment.DEFAULT_FILTERS['to_json'] = to_json

def is_cfr(accessToken, accountId, basePath):

api_client = create_api_client(basePath, accessToken)

accounts_api = AccountsApi(api_client)

account_details = accounts_api.get_account_information(accountId)

return account_details.status21_cfr_part11

def get_user_info(access_token, base_path, oauth_host_name):

api_client = create_api_client(base_path, access_token)

api_client.set_oauth_host_name(oauth_host_name)

return api_client.get_user_info(access_token)

def get_parameter_value_from_url(url, param_name):

parsed_url = urlparse(url)

query_params = parse_qs(parsed_url.query)

# Access the parameter value (returns a list)

param_value_list = query_params.get(param_name, [])

# If the parameter exists, return the first value; otherwise, return None

return param_value_list[0] if param_value_list else None

def replace_template_id(file_path, template_id):

with open(file_path, 'r') as file:

content = file.read()

content = re.sub('template-id', template_id, content)

with open(file_path, 'w') as file:

file.write(content)