git-remote-hg

   1 #!/usr/bin/env python
   2
   3 import sys
   4
   5
   6 def get_repo(alias, url):
   7     try:
   8         from mercurial import hg, ui
   9     except ImportError:
  10         sys.stderr.write("Mercurial python libraries not installed!\n")
  11         sys.exit(1)
  12
  13     ui = ui.ui()
  14     source, revs, checkout = hg.parseurl(ui.expandpath(url), [])
  15     repo = hg.repository(ui, source)
  16     repo.alias = alias
  17     if not revs:
  18         revs = ['tip']
  19     repo.revs = revs
  20     return repo
  21
  22
  23 def do_capabilities(repo, args):
  24     """
  25     """
  26
  27     sys.stderr.write("Capabilities are AWESUM.\n")
  28
  29     print # end capabilities
  30
  31
  32 def do_list(repo, args):
  33     """
  34     """
  35
  36     from mercurial.node import hex
  37     for ref in repo.revs:
  38         sys.stderr.write("? %s (%s)\n" % (ref, hex(repo.lookup(ref))))
  39         print '? %s' % (ref)
  40
  41     print # end list
  42
  43
  44 COMMANDS = {
  45     'capabilities': do_capabilities,
  46     'list': do_list,
  47 }
  48
  49
  50 def main(args):
  51     if len(args) < 3:
  52         sys.stderr.write("Missing url and stuff.\n")
  53         sys.exit(1)
  54
  55     alias = args[1]
  56     url = args[2]
  57     if url.startswith('hg+'):
  58         url = url[3:]
  59     elif url.startswith('hg::'):
  60         url = url[4:]
  61
  62     repo = get_repo(alias, url)
  63
  64     sys.stderr.write("How can I help u todayz on %s?\n" % args[1:])
  65     line = sys.stdin.readline()
  66
  67     while (line):
  68         cmdline = line.strip().split()
  69
  70         if not cmdline:
  71             break  # Blank line means we're about to quit
  72
  73         cmd = cmdline.pop(0)
  74         sys.stderr.write("Got command '%s' with args '%s'\n" % (cmd, ' '.join(cmdline)))
  75
  76         if cmd not in COMMANDS:
  77             sys.stderr.write("Dunno how to handle that, SUCKAH\n")
  78             sys.exit(1)
  79
  80         func = COMMANDS[cmd]
  81         func(repo, cmdline)
  82         sys.stdout.flush()
  83
  84         line = sys.stdin.readline()
  85
  86
  87 if __name__ == '__main__':
  88     sys.exit(main(sys.argv))