plugins/svn.py

   1
   2 from yap import YapPlugin, YapError
   3 from yap.util import get_output, takes_options, run_command, short_help
   4 import os
   5
   6 class SvnPlugin(YapPlugin):
   7     def __init__(self, yap):
   8         self.yap = yap
   9
  10     def _clone_svn(self, url, directory=None, **flags):
  11         url = url.rstrip('/')
  12         if directory is None:
  13             directory = url.rsplit('/')[-1]
  14             directory = directory.replace('.git', '')
  15
  16         try:
  17             os.mkdir(directory)
  18         except OSError:
  19             raise YapError("Directory exists: %s" % directory)
  20         os.chdir(directory)
  21         self.yap.cmd_init()
  22         os.system("git svn init %s" % url)
  23         if '-r' in flags:
  24             os.system("git svn fetch -r %s:" % flags['-r'])
  25         else:
  26             os.system("git svn fetch")
  27         self.yap.cmd_repo("svn", url)
  28         os.system("git config yap.svn.enabled 1")
  29
  30     def _enabled(self):
  31         enabled = get_output("git config yap.svn.enabled")
  32         return bool(enabled)
  33
  34     def _get_svn_ref(self):
  35         fetch = get_output("git config svn-remote.svn.fetch")
  36         if not fetch:
  37             raise YapError("No svn remote configured")
  38         ref = fetch[0].split(':')[1]
  39         return ref
  40
  41     # Ensure users don't accidentally kill our "svn" repo
  42     def pre_repo(self, *args, **flags):
  43         if not self._enabled():
  44             return
  45         if '-d' in flags and args and args[0] == "svn":
  46             raise YapError("Refusing to delete special svn repository")
  47
  48     @takes_options("r:")
  49     def cmd_clone(self, *args, **flags):
  50         if args and not run_command("svn info %s" % args[0]):
  51             self._clone_svn(*args, **flags)
  52         else:
  53             self.yap._call_base("cmd_commit", *args, **flags)
  54
  55     def cmd_fetch(self, *args, **flags):
  56         if self._enabled():
  57             if args and args[0] == 'svn':
  58                 os.system("git svn fetch")
  59                 return
  60         self.yap._call_base("cmd_fetch", *args, **flags)
  61
  62     def cmd_push(self, *args, **flags):
  63         if self._enabled():
  64             if args and args[0] == 'svn':
  65                 print "Verifying branch is up-to-date"
  66                 self.yap.cmd_fetch("svn")
  67                 ref = self._get_svn_ref()
  68                 rev = get_output("git rev-parse %s" % ref)
  69                 assert rev
  70                 base = get_output("git merge-base HEAD %s" % rev[0])
  71                 if base[0] != rev[0]:
  72                     raise YapError("Branch not up-to-date.  Update first.")
  73                 self.yap._confirm_push(current, "trunk", "svn")
  74                 os.system("git svn dcommit")
  75                 return
  76         self.yap._call_base("cmd_push", *args, **flags)
  77
  78     @short_help("change options for the svn plugin")
  79     def cmd_svn(self, subcmd):
  80         "<enable>"
  81
  82         if subcmd not in ["enable"]:
  83             raise TypeError
  84
  85         if "svn" not in [x[0] for x in self._list_remotes()]:
  86             raise YapError("The svn plugin requires a remote named 'svn'")
  87
  88         os.system("git config yap.svn.enabled 1")