import os.path

import subprocess

import sys

import reindent

def status(message, modal=False, info=None):

    """Decorator to output status info to stdout."""

    def decorated_fxn(fxn):

        def call_fxn(*args, **kwargs):

            sys.stdout.write(message + ' ... ')

            sys.stdout.flush()

            result = fxn(*args, **kwargs)

            if not modal and not info:

                print "done"

            elif info:

                print info(result)

            else:

                if result:

                    print "yes"

                else:

                    print "NO"

            return result

        return call_fxn

    return decorated_fxn

@status("Getting the list of files that have been added/changed",

            info=lambda x: "%s files" % len(x))

def changed_files():

    """Run ``svn status`` and return a set of files that have been

    changed/added."""

    cmd = 'svn status --quiet --non-interactive --ignore-externals'

    svn_st = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE)

    svn_st.wait()

    output = [line.strip() for line in svn_st.stdout.readlines()]

    files = set()

    for line in output:

        if not line[0] in ('A', 'M'):

            continue

        line_parts = line.split()

        path = line_parts[-1]

        if os.path.isfile(path):

            files.add(path)

    return files

@status("Fixing whitespace", info=lambda x: "%s files" % x)

def normalize_whitespace(file_paths):

    """Make sure that the whitespace for .py files have been normalized."""

    reindent.makebackup = False  # No need to create backups.

    result = map(reindent.check, (x for x in file_paths if x.endswith('.py')))

    return sum(result)

@status("Docs modified", modal=True)

def docs_modified(file_paths):

    """Report if any files in the Docs directory."""

    for path in file_paths:

        if path.startswith("Doc"):

            return True

    return False

@status("Misc/ACKS updated", modal=True)

def credit_given(file_paths):

    """Check if Misc/ACKS has been changed."""

    return True if 'Misc/ACKS' in file_paths else False

@status("Misc/NEWS updated", modal=True)

def reported_news(file_paths):

    """Check if Misc/NEWS has been changed."""

    return True if 'Misc/NEWS' in file_paths else False

def main():

    file_paths = changed_files()

    # PEP 7/8 verification.

    normalize_whitespace(file_paths)

    # Docs updated.

    docs_modified(file_paths)

    # Misc/ACKS changed.

    credit_given(file_paths)

    # Misc/NEWS changed.

    reported_news(file_paths)

    # Test suite run and passed.

    print

    print "Did you run the test suite?"

if __name__ == '__main__':

    main()