import uuid

from models import *
from app import app, git_command, get_permission_level, get_visibility, db, bcrypt
import os
import shutil
import config
import flask
import git
import subprocess
from flask_httpauth import HTTPBasicAuth
import zlib
import re
import datetime

auth = HTTPBasicAuth(realm=config.AUTH_REALM)

auth_required = flask.Response("Unauthorized Access", 401, {"WWW-Authenticate": 'Basic realm="Login Required"'})


@auth.verify_password
def verify_password(username, password):
    user = User.query.filter_by(username=username).first()

    if user and bcrypt.check_password_hash(user.password_hashed, password):
        flask.g.user = username
        return True

    return False


@app.route("/<username>/<repository>/git-upload-pack", methods=["POST"])
@app.route("/git/<username>/<repository>/git-upload-pack", methods=["POST"])
@auth.login_required(optional=True)
def git_upload_pack(username, repository):
    if auth.current_user() is None and not get_visibility(username, repository):
        return auth_required
    if not (get_visibility(username, repository) or get_permission_level(flask.g.user, username,
                                                                         repository) is not None):
        flask.abort(403)

    server_repo_location = os.path.join(config.REPOS_PATH, username, repository, ".git")
    text = git_command(server_repo_location, flask.request.data, "upload-pack", "--stateless-rpc", ".")

    return flask.Response(text, content_type="application/x-git-upload-pack-result")


@app.route("/<username>/<repository>/git-receive-pack", methods=["POST"])
@app.route("/git/<username>/<repository>/git-receive-pack", methods=["POST"])
@auth.login_required
def git_receive_pack(username, repository):
    if not get_permission_level(flask.g.user, username, repository):
        flask.abort(403)

    server_repo_location = os.path.join(config.REPOS_PATH, username, repository, ".git")
    text = git_command(server_repo_location, flask.request.data, "receive-pack", "--stateless-rpc", ".")

    sha = flask.request.data.split(b" ", 2)[1].decode()
    info = git_command(server_repo_location, None, "show", "-s", "--format='%H%n%at%n%cn <%ce>%n%B'", sha).decode()

    if re.match("^[0-9a-fA-F]{40}$", info[:40]):
        print(info.split("\n", 3))
        sha, time, identity, body = info.split("\n", 3)
        login = flask.g.user

        if not Commit.query.filter_by(identifier=f"/{username}/{repository}/{sha}").first():
            user = User.query.filter_by(username=login).first()
            repo = Repo.query.filter_by(route=f"/{username}/{repository}").first()

            commit = Commit(sha, user, repo, time, body, identity)

            db.session.add(commit)
            db.session.commit()

    return flask.Response(text, content_type="application/x-git-receive-pack-result")


@app.route("/<username>/<repository>/info/refs", methods=["GET", "POST"])
@app.route("/git/<username>/<repository>/info/refs", methods=["GET", "POST"])
@auth.login_required(optional=True)
def git_info_refs(username, repository):
    server_repo_location = os.path.join(config.REPOS_PATH, username, repository, ".git")

    repo = git.Repo(server_repo_location)
    repo_data = Repo.query.filter_by(route=f"/{username}/{repository}").first()
    if not repo_data.default_branch:
        if repo.heads:
            repo_data.default_branch = repo.heads[0].name
            repo.git.checkout("-f", repo_data.default_branch)

    if auth.current_user() is None and (
            not get_visibility(username, repository) or flask.request.args.get("service") == "git-receive-pack"):
        return auth_required
    try:
        if not (get_visibility(username, repository) or get_permission_level(flask.g.user, username,
                                                                             repository) is not None):
            flask.abort(403)
    except AttributeError:
        return auth_required

    service = flask.request.args.get("service")

    if service.startswith("git"):
        service = service[4:]
    else:
        flask.abort(403)

    if service == "receive-pack":
        try:
            if not get_permission_level(flask.g.user, username, repository):
                flask.abort(403)
        except AttributeError:
            return auth_required

    service_line = f"# service=git-{service}\n"
    service_line = (f"{len(service_line) + 4:04x}" + service_line).encode()

    if service == "upload-pack":
        text = service_line + b"0000" + git_command(server_repo_location, None, "upload-pack", "--stateless-rpc",
                                                    "--advertise-refs", "--http-backend-info-refs", ".")
    elif service == "receive-pack":
        refs = git_command(server_repo_location, None, "receive-pack", "--http-backend-info-refs", ".")
        text = service_line + b"0000" + refs
    else:
        flask.abort(403)

    response = flask.Response(text, content_type=f"application/x-git-{service}-advertisement")
    response.headers["Cache-Control"] = "no-cache"

    return response