#-*- coding: utf-8 -*-

"""
 (c) 2014 - Copyright Red Hat Inc

 Authors:
   Pierre-Yves Chibon <pingou@pingoured.fr>

"""

import flask
import os
from math import ceil

import pygit2
from sqlalchemy.exc import SQLAlchemyError
from pygments import highlight
from pygments.lexers import guess_lexer
from pygments.lexers.text import DiffLexer
from pygments.formatters import HtmlFormatter


import progit.doc_utils
import progit.lib
import progit.forms
from progit import APP, SESSION, LOG, cla_required


def __get_tree(repo_obj, tree, filepath, startswith=False):
    ''' Retrieve the entry corresponding to the provided filename in a
    given tree.
    '''
    filename = filepath[0]
    if isinstance(tree, pygit2.Blob):
        return (tree, None)
    cnt = 0
    for el in tree:
        cnt += 1
        ok = False
        if cnt == len(tree) and el.name.startswith(filename):
            ok = True
        if el.name == filename:
            ok = True
        if ok:
            if len(filepath) == 1:
                return (el, tree)
            else:
                return __get_tree(
                    repo_obj, repo_obj[el.oid], filepath[1:])


def __get_tree_and_content(repo_obj, commit, path, startswith):
    ''' Return the tree and the content of the specified file. '''

    (blob_or_tree, tree_obj) = __get_tree(
        repo_obj, commit.tree, path, startswith=startswith)

    if not repo_obj[blob_or_tree.oid]:
        flask.abort(404, 'File not found')


    blob_or_tree_obj = repo_obj[blob_or_tree.oid]
    blob = repo_obj[blob_or_tree.oid]
    if isinstance(blob, pygit2.Blob):  # Returned a file
        name, ext = os.path.splitext(blob_or_tree.name)
        content = progit.doc_utils.convert_readme(blob_or_tree_obj.data, ext)
    else:  # Returned a tree
        path.append('index')
        return __get_tree_and_content(repo_obj, commit, path, startswith)

    tree = sorted(tree_obj, key=lambda x: x.filemode)
    return (tree, content)


def do_view_wiki(repo, username=None, branchname=None, filename=None):
    """ List all issues associated to a repo
    """
    repo = progit.lib.get_project(SESSION, repo, user=username)

    if not repo:
        flask.abort(404, 'Project not found')

    if not repo.project_wiki:
        flask.abort(404, 'No documentation found for this project')

    reponame = os.path.join(APP.config['WIKI_FOLDER'], repo.path)
    repo_obj = pygit2.Repository(reponame)

    if branchname in repo_obj.listall_branches():
        branch = repo_obj.lookup_branch(branchname)
        commit = branch.get_object()
    else:
        if not repo_obj.is_empty:
            commit = repo_obj[repo_obj.head.target]
        else:
            commit = None
        branchname = 'master'

    content = None
    tree = None
    startswith = False
    if not filename:
        path = ['index']
        startswith = True
    else:
        path = filename.split('/')

    if commit:
        (tree, content ) = __get_tree_and_content(
            repo_obj, commit, path, startswith)

    return flask.render_template(
        'wiki.html',
        select='docs',
        repo_obj=repo_obj,
        repo=repo,
        username=username,
        branchname=branchname,
        filename=filename,
        tree=tree,
        content=content,
    )


## URLs


@APP.route('/<repo>/docs')
@APP.route('/<repo>/docs/<path:filename>')
@APP.route('/<repo>/docs/<branchname>')
@APP.route('/<repo>/docs/<branchname>/<path:filename>')

@APP.route('/fork/<username>/<repo>/docs')
@APP.route('/fork/<username>/<repo>/docs/<path:filename>')
@APP.route('/fork/<username>/<repo>/docs/<branchname>')
@APP.route('/fork/<username>/<repo>/docs/<branchname>/<path:filename>')
def view_wiki(repo, username=None, branchname=None, filename=None):
    """ Display the documentation
    """
    status = flask.request.args.get('status', None)
    return do_view_wiki(
        repo, username=username, branchname=branchname, filename=filename)