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

try:
    #from gevent.monkey import patch_all
    #patch_all(subprocess=False, aggressive=False)
    from gevent.monkey import get_original
    from gevent.server import StreamServer
    import select
    select.poll = get_original('select', 'poll')
    import subprocess
    subprocess.Popen = get_original('subprocess', 'Popen')
except ImportError:
    print 'You need install gevent manually! System shutdown.'

from binascii import hexlify
from maria import Maria
from maria.config import config
from vilya.libs.permdir import get_repo_root
from vilya.models.sshkey import SSHKey
from vilya.models.project import CodeDoubanProject as Project
from vilya.models.gist import Gist

config.project_root = get_repo_root()
config.log_file = None
config.host_key_path = './host.key'

app = Maria(config=config)


@app.get_environ
def get_environ_handler(user, path):
    return {'CODE_REMOTE_USER': user}


@app.has_permission
def has_permission_handler(username, path, perm):
    if not username or not path:
        return False

    if path.endswith('.git'):
        path = path[:-4]

    # gist
    if path.startswith('gist/'):
        gist_id = path.rpartition("/")[-1]
        if username == Gist.get(gist_id).owner_id:
            return  True
        return False

    # project
    project = Project.get_by_name(path)
    if not project:
        return False
    if perm == 'read':
        return True
    if not project.can_push:
        return False
    if project.has_push_perm(username):
        return True
    return False


@app.get_user
def get_user_handler(username, key):
    # ssh username
    if username != 'git':
        return None

    # ssh key
    if not key:
        return None
    fingerprint = hexlify(key.get_fingerprint())
    fingerprint = ":".join([fingerprint[i:2+i] for i in range(0, len(fingerprint), 2)])
    sshkey = SSHKey.get_by_fingerprint(fingerprint)
    if not sshkey:
        return 'guest'
    return sshkey.user_id


if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser(description='code ssh daemon.')
    parser.add_argument('--host', default='0.0.0.0', type=str)
    parser.add_argument('--port', default=2200, type=int)
    args = parser.parse_args()

    server = StreamServer((args.host, args.port), app)
    server.serve_forever()