#!/usr/bin/env python

"""
setup a repository

Parameters:
- host URL
- location on host disk
"""

import os
import subprocess
import sys
import urlparse

from subprocess import check_call as call
from optparse import OptionParser


# global variables
HOST='k0s.org'
HGRC="""[paths]
default = http://%(host)s/%(repo)s/%(name)s
default-push = ssh://%(host)s/%(repo)s/%(name)s
"""

def call(command, *args, **kw):

    code = subprocess.call(command, *args, **kw)
    if isinstance(command, basestring):
        cmdstr = command
    else:
        cmdstr = ' '.join(command)
    print cmdstr
    if code:
        raise SystemExit("Command `%s` exited with code %d" % (cmdstr, code))

def main(args=sys.argv[1:]):

    # parse command line arguments
    parser = OptionParser('%prog [options] location')
    parser.add_option('-m', '--message', dest='message',
                      default='initial commit',
                      help='commit message [Default: %default]')
    parser.add_option('-u', '--remote-url', dest='remote_url',
                      default="http://k0s.org/hg/",
                      help="URL of host hg repository collection [Default: %default]")
    parser.add_option('-p', '--remote-path', dest='remote_path',
                      help="remote path of hg repository links; otherwise taken from --remote-url, if specified")
    parser.add_option
    options, args = parser.parse_args(args)
    if len(args) != 1:
        parser.print_usage()
        parser.exit()
    # TODO: sanity check for remote_url, remote_path

    # setup repository
    directory = args[0]
    name = os.path.basename(directory) # XXX unnecessary?
    call(['hg', 'init', directory])
    call(['hg', 'add', '-R', directory])
    call(['hg', 'commit', '-m', options.message, '-R', directory])

    # setup remote, if specified
    if options.remote_url:

        # parse remote URL
        host, netloc, path, query, anchor = urlparse.urlsplit(options.remote_url)
        if options.remote_path:
            remote_host = host
            if ':' in remote_path:
                remote_host, remote_path = remote_path.split(':', 1)
        else:
            remote_path = path

        # setup remote repository
        call(['ssh', host, "mkdir -p cd %s/%s && hg init && hg add && hg ci -m '%s'" % (repo, name, options.message)])

        # write local .hgrc file
        # TODO: use ConfigParser
        with file(os.path.join(directory, '.hg', 'hgrc'), 'w') as f:
            print >> f, HGRC % { 'host': host, 'repo': repo, 'name': name}

        # push changes
        call(['hg', 'push', '-R', directory])

if __name__ == '__main__':
  main()