#!/usr/bin/env python

import os, os.path as op, re, sys

# line starts with two chars one of which is not a space (and both are not

# newlines obviously) and ends with one or more newlines followed by two spaces

# on a next line (indented text)

CODEBLOCK = re.compile(r'()\n(([^ \n][^\n]|[^\n][^ \n])[^\n]*)\n+  ')

INDEX = '''

Mercurial tests

===============

.. toctree::

   :maxdepth: 1

'''

def rstify(orig, name):

    header = '%s\n%s\n\n' % (name, '=' * len(name))

    content = header + orig

    content = CODEBLOCK.sub(r'\n\1\n\n::\n\n  ', content)

    return content

def main(base):

    if os.path.isdir(base):

        one_dir(base)

    else:

        one_file(base)

def one_dir(base):

    index = INDEX

    #doc = lambda x: op.join(op.dirname(__file__), 'docs', x)

    for fn in sorted(os.listdir(base)):

        if not fn.endswith('.t'):

            continue

        name = os.path.splitext(fn)[0]

        content = one_file(op.join(base, fn))

        target = op.join(base, name + '.rst')

        #with file(doc(name + '.rst'), 'w') as f:

        with file(target, 'w') as f:

            f.write(content)

        index += '\n   ' + name

    #with file(doc('index.rst'), 'w') as f:

    #    f.write(index)

def one_file(path):

    name = os.path.basename(path)[:-2]

    return rstify(file(path).read(), name)

if __name__ == '__main__':

    if len(sys.argv) != 2:

        print 'Please supply a path to tests dir as parameter'

        sys.exit()

    main(sys.argv[1])