import click
import os

from zope import component
from ceo_common.interfaces import IConfig

from ..utils import http_post, http_get, http_delete
from .utils import handle_sync_response, check_file_path, check_if_in_development


def psql_cli_response(file, username, password):
    cfg_srv = component.getUtility(IConfig)
    psql_host = cfg_srv.get('postgresql_host')
    info = f"""PostgreSQL Database Information for {username}

    Your new PostgreSQL database was created. To connect, use the following options:

    Database: {username}
    Username: {username}
    Password: {password}
    Host: {psql_host}

    On {psql_host} to connect using the PostgreSQL command-line client use

      psql -d {username} -U {username} -W

    From other CSC machines you can connect using

      psql -d {username} -h {psql_host} -U {username} -W

    """
    with click.open_file(file, "w") as f:
        f.write(info)
    os.chown(file, username, username)
    os.chmod(file, 0o640)

    click.echo(f"""PostgreSQL database created

    Connection Information:

    Database: {username}
    Username: {username}
    Password: {password}
    Host: {psql_host}

    Settings and more info has been written to {file}""")


@click.group(short_help='Perform operations on PostgreSQL')
def postgresql():
    pass


@postgresql.command(short_help='Create a PostgreSQL database for a user')
@click.argument('username')
def create(username):
    resp = http_get(f'/api/members/{username}')
    result = handle_sync_response(resp)
    info_file_path = os.path.join(result['home_directory'], "ceo-psql-info")

    check_file_path(info_file_path)

    resp = http_post(f'/api/db/postgresql/{username}')
    result = handle_sync_response(resp)
    password = result['password']

    psql_cli_response(info_file_path, username, password)


@postgresql.command(short_help='Reset the password of a PostgreSQL user')
@click.argument('username')
def pwreset(username):
    resp = http_get(f'/api/members/{username}')
    result = handle_sync_response(resp)
    info_file_path = os.path.join(result['home_directory'], "ceo-psql-info")

    check_file_path(info_file_path)

    resp = http_post(f'/api/db/postgresql/{username}/pwreset')
    result = handle_sync_response(resp)
    password = result['password']

    psql_cli_response(info_file_path, username, password)


@postgresql.command(short_help="Delete the database of a PostgreSQL user")
@click.argument('username')
def delete(username):
    check_if_in_development()
    click.confirm("Are you sure?", abort=True)
    resp = http_delete(f'/api/db/postgresql/{username}')
    handle_sync_response(resp)