#
# Copyright (C) 2008 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


import os
import sys
import subprocess
import tempfile
from signal import SIGTERM
from error import GitError
import portable
from repo_trace import REPO_TRACE, IsTrace, Trace

GIT = 'git'
MIN_GIT_VERSION = (1, 5, 4)
GIT_DIR = 'GIT_DIR'

LAST_GITDIR = None
LAST_CWD = None

_ssh_proxy_path = None
_ssh_sock_path = None
_ssh_clients = []


def ssh_sock(create=True):
    global _ssh_sock_path
    if _ssh_sock_path is None:
        if not create:
            return None
        tmp_dir = '/tmp'
        if not os.path.exists(tmp_dir):
            tmp_dir = tempfile.gettempdir()
        _ssh_sock_path = os.path.join(
            tempfile.mkdtemp('', 'ssh-', tmp_dir),
            'master-%r@%h:%p')
    return _ssh_sock_path


def _ssh_proxy():
    global _ssh_proxy_path
    if _ssh_proxy_path is None:
        _ssh_proxy_path = os.path.join(
            os.path.dirname(__file__),
            'git_ssh')
    return _ssh_proxy_path


def _add_ssh_client(p):
    _ssh_clients.append(p)


def _remove_ssh_client(p):
    try:
        _ssh_clients.remove(p)
    except ValueError:
        pass


def terminate_ssh_clients():
    global _ssh_clients
    for p in _ssh_clients:
        try:
            os.kill(p.pid, SIGTERM)
            p.wait()
        except OSError:
            pass
    _ssh_clients = []


_git_version = None


class _GitCall(object):
    def version(self):
        p = GitCommand(None, ['--version'], capture_stdout=True)
        if p.Wait() == 0:
            return portable.stream2str(p.stdout)
        return None

    def version_tuple(self):
        global _git_version

        if _git_version is None:
            ver_str = git.version()
            if ver_str.startswith('git version '):
                _git_version = tuple(
                    map(int,
                        ver_str[len('git version '):].strip().split('-')[0].split('.')[0:3]
                    ))
            else:
                print('fatal: "%s" unsupported' % ver_str, file=sys.stderr)
                sys.exit(1)
        return _git_version

    def __getattr__(self, name):
        name = name.replace('_', '-')

        def fun(*cmdv):
            command = [name]
            command.extend(cmdv)
            return GitCommand(None, command).Wait() == 0

        return fun


git = _GitCall()


def git_require(min_version, fail=False):
    git_version = git.version_tuple()
    if min_version <= git_version:
        return True
    if fail:
        need = '.'.join(map(str, min_version))
        print('fatal: git %s or later required' % need, file=sys.stderr)
        sys.exit(1)
    return False


def _setenv(env, name, value):
    env[name] = value.encode(encoding='UTF-8')


class GitCommand(object):
    def __init__(self,
                 project,
                 cmdv,
                 bare=False,
                 provide_stdin=False,
                 capture_stdout=False,
                 capture_stderr=False,
                 disable_editor=False,
                 ssh_proxy=False,
                 cwd=None,
                 gitdir=None):
        env = os.environ.copy()

        for key in [REPO_TRACE,
                    GIT_DIR,
                    'GIT_ALTERNATE_OBJECT_DIRECTORIES',
                    'GIT_OBJECT_DIRECTORY',
                    'GIT_WORK_TREE',
                    'GIT_GRAFT_FILE',
                    'GIT_INDEX_FILE']:
            if key in env:
                del env[key]

        if disable_editor:
            _setenv(env, 'GIT_EDITOR', ':')
        if ssh_proxy:
            _setenv(env, 'REPO_SSH_SOCK', ssh_sock())
            _setenv(env, 'GIT_SSH', _ssh_proxy())
        if 'http_proxy' in env and 'darwin' == sys.platform:
            s = "'http.proxy=%s'" % (env['http_proxy'],)
            p = env.get('GIT_CONFIG_PARAMETERS')
            if p is not None:
                s = p + ' ' + s
            _setenv(env, 'GIT_CONFIG_PARAMETERS', s)

        if project:
            if not cwd:
                cwd = project.worktree
            if not gitdir:
                gitdir = project.gitdir

        command = [GIT]
        if bare:
            if gitdir:
                env[GIT_DIR] = gitdir
            cwd = None
        command.extend(cmdv)

        if provide_stdin:
            stdin = subprocess.PIPE
        else:
            stdin = None

        if capture_stdout:
            stdout = subprocess.PIPE
        else:
            stdout = None

        if capture_stderr:
            stderr = subprocess.PIPE
        else:
            stderr = None

        if IsTrace():
            global LAST_CWD
            global LAST_GITDIR

            dbg = ''

            if cwd and LAST_CWD != cwd:
                if LAST_GITDIR or LAST_CWD:
                    dbg += '\n'
                dbg += ': cd %s\n' % cwd
                LAST_CWD = cwd

            if GIT_DIR in env and LAST_GITDIR != env[GIT_DIR]:
                if LAST_GITDIR or LAST_CWD:
                    dbg += '\n'
                dbg += ': export GIT_DIR=%s\n' % env[GIT_DIR]
                LAST_GITDIR = env[GIT_DIR]

            dbg += ': '
            dbg += ' '.join(command)
            if stdin == subprocess.PIPE:
                dbg += ' 0<|'
            if stdout == subprocess.PIPE:
                dbg += ' 1>|'
            if stderr == subprocess.PIPE:
                dbg += ' 2>|'
            Trace('%s', dbg)

        try:
            p = subprocess.Popen(command,
                                 cwd=cwd,
                                 env=env,
                                 stdin=stdin,
                                 stdout=stdout,
                                 stderr=stderr)
        except Exception as e:
            raise GitError('%s: %s' % (command[1], e))

        if ssh_proxy:
            _add_ssh_client(p)

        portable.SUBPROCESSES.append(p)
        self.process = p
        self.stdin = p.stdin

    def Wait(self):
        try:
            p = self.process
            (self.stdout, self.stderr) = p.communicate()
            rc = p.returncode
            portable.SUBPROCESSES.remove(p)
        finally:
            _remove_ssh_client(p)
        return rc