# vim: tabstop=4 shiftwidth=4 softtabstop=4

# Copyright 2011 OpenStack LLC.
# All Rights Reserved.
#
#    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.

"""
Utilities with minimum-depends for use in setup.py
"""

import os
import re
import subprocess


def parse_mailmap(mailmap='.mailmap'):
    mapping = {}
    if os.path.exists(mailmap):
        fp = open(mailmap, 'r')
        for l in fp:
            l = l.strip()
            if not l.startswith('#') and ' ' in l:
                canonical_email, alias = l.split(' ')
                mapping[alias] = canonical_email
    return mapping


def str_dict_replace(s, mapping):
    for s1, s2 in mapping.iteritems():
        s = s.replace(s1, s2)
    return s


# Get requirements from the first file that exists
def get_reqs_from_files(requirements_files):
    reqs_in = []
    for requirements_file in requirements_files:
        if os.path.exists(requirements_file):
            return open(requirements_file, 'r').read().split('\n')
    return []


def parse_requirements(requirements_files=['requirements.txt',
                                           'tools/pip-requires']):
    requirements = []
    for line in get_reqs_from_files(requirements_files):
        if re.match(r'\s*-e\s+', line):
            requirements.append(re.sub(r'\s*-e\s+.*#egg=(.*)$', r'\1',
                                line))
        elif re.match(r'\s*-f\s+', line):
            pass
        else:
            requirements.append(line)

    return requirements


def parse_dependency_links(requirements_files=['requirements.txt',
                                               'tools/pip-requires']):
    dependency_links = []
    for line in get_reqs_from_files(requirements_files):
        if re.match(r'(\s*#)|(\s*$)', line):
            continue
        if re.match(r'\s*-[ef]\s+', line):
            dependency_links.append(re.sub(r'\s*-[ef]\s+', '', line))
    return dependency_links


def write_requirements():
    venv = os.environ.get('VIRTUAL_ENV', None)
    if venv is not None:
        with open("requirements.txt", "w") as req_file:
            output = subprocess.Popen(["pip", "-E", venv, "freeze", "-l"],
                                      stdout=subprocess.PIPE)
            requirements = output.communicate()[0].strip()
            req_file.write(requirements)

@@ -2837,8 +2837,940 @@
 ements)%0A
+%0A%0Adef _run_shell_command(cmd):%0A    output = subprocess.Popen(%5B%22/bin/sh%22, %22-c%22, cmd%5D,%0A                              stdout=subprocess.PIPE)%0A    return output.communicate()%5B0%5D.strip()%0A%0A%0Adef write_vcsversion(location):%0A    %22%22%22 Produce a vcsversion dict that mimics the old one produced by bzr%0A    %22%22%22%0A    if os.path.isdir('.git'):%0A        branch_nick_cmd = 'git branch %7C grep -Ei %22%5C* (.*)%22 %7C cut -f2 -d%22 %22'%0A        branch_nick = _run_shell_command(branch_nick_cmd)%0A        revid_cmd = %22git rev-parse HEAD%22%0A        revid = _run_shell_command(revid_cmd).split()%5B0%5D%0A        revno_cmd = %22git log --oneline %7C wc -l%22%0A        revno = _run_shell_command(revno_cmd)%0A        with open(location, 'w') as version_file:%0A            version_file.write(%22%22%22%0A# This file is automatically generated by setup.py, So don't edit it. :)%0Aversion_info = %7B%0A    'branch_nick': '%25s',%0A    'revision_id': '%25s',%0A    'revno': %25s%0A%7D%0A%22%22%22 %25 (branch_nick, revid, revno))%0A

Add git vcsversion method.