# import sqlite3
import os
import argparse
import json
import getpass
import datetime
import logging
import sys

import action

class Project:
    name = ''
    status = 'active'


def handle_project(args, last_action, conn):
    logging.info('> handle project')
    logging.debug('args: ' + str(args))

    if not args:
        list_project(last_action.project_id, conn)
        sys.exit(0)

    if args[0].isdigit():
        view_project()
        sys.exit(0)

    # Use sys.exit when no add/update in Database

    if args[0] == '-c':
        if len(args) == 1:
            print('required project name')
            sys.exit(1)

        project = Project()
        project.name = arg_string(args[1], 'project name')
        create_project(project, conn)
    elif args[0] == '-e':
        print('Update project')
    elif args[0] == '-d':
        if len(args) > 1:
            delete_project(arg_number(args[1]), conn)
        else:
            delete_project(last_action.project_id, conn)

def arg_string(arg, required = ''):
    if required and arg.startswith('-'):
        print('required ', required)
        sys.exit('1')
    else:
        return arg.strip()
def arg_number(arg, required = ''):
    if not arg.isdigit():
        print('Invalid value') # TODO
        sys.exit('1')

    return int(arg)

def create_project(project, conn):
    # TODO Project is same name is forbidden
    logging.info('>> Create project')

    query = """
    INSERT INTO project (username, name, status, created_at)
    VALUES (?, ?, ?, ?);
    """

    cursor = conn.cursor()
    cursor.execute(query, (getpass.getuser(), project.name, project.status, datetime.datetime.now(),))
    project_id = cursor.lastrowid
    action_message = '{} (status: {})'.format(project.name, project.status)
    action.create_action(cursor, project_id=project_id, message=action_message)
    logging.debug(action_message)

    print('created project {}: {} (status: {})'.format(project_id, project.name, project.status))

def edit_project(args, project_id, conn):
    logging.info('>> Edit project')

    update_args = {}
    if args.status:
        update_args['status'] = args.status
    if args.edit_name:
        update_args['name'] = args.edit_name

    query = 'UPDATE project SET {} WHERE id = ?'
    query = query.format(', '.join("%s = '%s'" % (k, v) for k, v in update_args.items()))
    logging.debug('update project query: ' + query)

    cursor = conn.cursor()
    if update_args:
        logging.debug('Do a project update')
        cursor.execute(query, (project_id,))

        log_args = ', '.join("%s: '%s'" % (k, v) for k, v in update_args.items())
        print('updated project {}: ({})'.format(project_id, log_args))
    else:
        print('updated project {}: set active project')

    action.create_action(cursor, project_id, message = 'update ' + str(update_args))

def delete_project(project_id, conn):
    logging.info('>> Remove project')

    query = """
    DELETE FROM project
    WHERE id = ?;
    """

    cursor = conn.cursor()
    cursor.execute(query, (project_id,))
    if cursor.rowcount != 1:
        logging.error('DELETE FAILED')
    print('deleted project {}: {}'.format(project_id, 'project_name'))

def list_project(active_project_id, conn):
    logging.info('>> No arguments')
    query = """
    SELECT id, username, name, status, created_at,
        (SELECT count(*) FROM task WHERE task.project_id = project.id)
    FROM project;
    """

    cursor = conn.cursor()
    cursor.execute(query)
    for row in cursor.fetchall():
        logging.debug('Project row: {}'.format(row))
        project_id, username, name, status, date, nb_task = row
        print('{:1} {:2d}: ({:8}) | {} | {} ({} tasks)'.format('*' if active_project_id == project_id else '',
            project_id, username, status, name, nb_task))

def view_project():
    print("TODO View Project")
    #TODO