from distutils.command.sdist import sdist as _sdist
from distutils.util import convert_path
import os,re

entities = [
    ("&lt;","<"), ("&gt;", ">"), ("&quot;", '"'), ("&apos;", "'"),
    ("&amp;", "&")
]
def unescape(data):
    for old,new in entities:
        data = data.replace(old,new)
    return data

patterns = [
    (convert_path('CVS/Entries'), re.compile(r"^\w?/([^/]+)/", re.M), None),
    (convert_path('.svn/entries'), re.compile(r'name="([^"]+)"'), unescape),
]

def joinpath(prefix,suffix):
    if not prefix:
        return suffix
    return os.path.join(prefix,suffix)



















def walk_revctrl(dirname='', memo=None):
    """Find all files under revision control"""
    if memo is None:
        memo = {}
    if dirname in memo:
        # Don't rescan a scanned directory
        return
    for path, pattern, postproc in patterns:
        path = joinpath(dirname,path)
        if os.path.isfile(path):
            f = open(path,'rU')
            data = f.read()
            f.close()
            for match in pattern.finditer(data):
                path = match.group(1)
                if postproc:
                    path = postproc(path)
                path = joinpath(dirname,path)
                if os.path.isfile(path):
                    yield path
                elif os.path.isdir(path):
                    for item in walk_revctrl(path, memo):
                        yield item

class sdist(_sdist):
    """Smart sdist that finds anything supported by revision control"""
    def finalize_options(self):
        _sdist.finalize_options(self)
        if not os.path.isfile(self.template):
            self.force_manifest = 1     # always regen if no template

    def add_defaults(self):
        _sdist.add_defaults(self)
        self.filelist.extend(walk_revctrl())