module-tools.py

   1 #!/usr/bin/python -u
   2
   3 import sys, os
   4 import re
   5 import time
   6 import tempfile
   7 from glob import glob
   8 from optparse import OptionParser
   9
  10 # HARDCODED NAME CHANGES
  11 #
  12 # Moving to git we decided to rename some of the repositories. Here is
  13 # a map of name changes applied in git repositories.
  14 RENAMED_SVN_MODULES = {
  15     "PLEWWW": "plewww",
  16     "PLCAPI": "plcapi",
  17     "BootManager": "bootmanager",
  18     "BootCD": "bootcd",
  19     "VserverReference": "vserver-reference",
  20     "BootstrapFS": "bootstrapfs",
  21     "MyPLC": "myplc",
  22     "CoDemux": "codemux",
  23     "NodeManager": "nodemanager",
  24     "NodeUpdate": "nodeupdate"
  25     }
  26
  27 def svn_to_git_name(module):
  28     if RENAMED_SVN_MODULES.has_key(module):
  29         return RENAMED_SVN_MODULES[module]
  30     return module
  31
  32 def git_to_svn_name(module):
  33     for key in RENAMED_SVN_MODULES:
  34         if module == RENAMED_SVN_MODULES[key]:
  35             return key
  36     return module
  37
  38
  39 # e.g. other_choices = [ ('d','iff') , ('g','uess') ] - lowercase
  40 def prompt (question,default=True,other_choices=[],allow_outside=False):
  41     if not isinstance (other_choices,list):
  42         other_choices = [ other_choices ]
  43     chars = [ c for (c,rest) in other_choices ]
  44
  45     choices = []
  46     if 'y' not in chars:
  47         if default is True: choices.append('[y]')
  48         else : choices.append('y')
  49     if 'n' not in chars:
  50         if default is False: choices.append('[n]')
  51         else : choices.append('n')
  52
  53     for (char,choice) in other_choices:
  54         if default == char:
  55             choices.append("["+char+"]"+choice)
  56         else:
  57             choices.append("<"+char+">"+choice)
  58     try:
  59         answer=raw_input(question + " " + "/".join(choices) + " ? ")
  60         if not answer:
  61             return default
  62         answer=answer[0].lower()
  63         if answer == 'y':
  64             if 'y' in chars: return 'y'
  65             else: return True
  66         elif answer == 'n':
  67             if 'n' in chars: return 'n'
  68             else: return False
  69         elif other_choices:
  70             for (char,choice) in other_choices:
  71                 if answer == char:
  72                     return char
  73             if allow_outside:
  74                 return answer
  75         return prompt(question,default,other_choices)
  76     except:
  77         raise
  78
  79 def default_editor():
  80     try:
  81         editor = os.environ['EDITOR']
  82     except:
  83         editor = "emacs"
  84     return editor
  85
  86 ### fold long lines
  87 fold_length=132
  88
  89 def print_fold (line):
  90     while len(line) >= fold_length:
  91         print line[:fold_length],'\\'
  92         line=line[fold_length:]
  93     print line
  94
  95 class Command:
  96     def __init__ (self,command,options):
  97         self.command=command
  98         self.options=options
  99         self.tmp="/tmp/command-%d"%os.getpid()
 100
 101     def run (self):
 102         if self.options.dry_run:
 103             print 'dry_run',self.command
 104             return 0
 105         if self.options.verbose and self.options.mode not in Main.silent_modes:
 106             print '+',self.command
 107             sys.stdout.flush()
 108         return os.system(self.command)
 109
 110     def run_silent (self):
 111         if self.options.dry_run:
 112             print 'dry_run',self.command
 113             return 0
 114         if self.options.verbose:
 115             print '+',self.command,' .. ',
 116             sys.stdout.flush()
 117         retcod=os.system(self.command + " &> " + self.tmp)
 118         if retcod != 0:
 119             print "FAILED ! -- out+err below (command was %s)"%self.command
 120             os.system("cat " + self.tmp)
 121             print "FAILED ! -- end of quoted output"
 122         elif self.options.verbose:
 123             print "OK"
 124         os.unlink(self.tmp)
 125         return retcod
 126
 127     def run_fatal(self):
 128         if self.run_silent() !=0:
 129             raise Exception,"Command %s failed"%self.command
 130
 131     # returns stdout, like bash's $(mycommand)
 132     def output_of (self,with_stderr=False):
 133         if self.options.dry_run:
 134             print 'dry_run',self.command
 135             return 'dry_run output'
 136         tmp="/tmp/status-%d"%os.getpid()
 137         if self.options.debug:
 138             print '+',self.command,' .. ',
 139             sys.stdout.flush()
 140         command=self.command
 141         if with_stderr:
 142             command += " &> "
 143         else:
 144             command += " > "
 145         command += tmp
 146         os.system(command)
 147         result=file(tmp).read()
 148         os.unlink(tmp)
 149         if self.options.debug:
 150             print 'Done',
 151         return result
 152
 153
 154 class SvnRepository:
 155     type = "svn"
 156
 157     def __init__(self, path, options):
 158         self.path = path
 159         self.options = options
 160
 161     def name(self):
 162         return os.path.basename(self.path)
 163
 164     def url(self):
 165         out = Command("svn info %s" % self.path, self.options).output_of()
 166         for line in out.split('\n'):
 167             if line.startswith("URL:"):
 168                 return line.split()[1].strip()
 169
 170     def repo_root(self):
 171         out = Command("svn info %s" % self.path, self.options).output_of()
 172         for line in out.split('\n'):
 173             if line.startswith("Repository Root:"):
 174                 root = line.split()[2].strip()
 175                 return "%s/%s" % (root, self.name())
 176
 177     @classmethod
 178     def checkout(cls, remote, local, options, recursive=False):
 179         if recursive:
 180             svncommand = "svn co %s %s" % (remote, local)
 181         else:
 182             svncommand = "svn co -N %s %s" % (remote, local)
 183         Command("rm -rf %s" % local, options).run_silent()
 184         Command(svncommand, options).run_fatal()
 185
 186         return SvnRepository(local, options)
 187
 188     @classmethod
 189     def remote_exists(cls, remote):
 190         return os.system("svn list %s &> /dev/null" % remote) == 0
 191
 192     def tag_exists(self, tagname):
 193         url = "%s/tags/%s" % (self.repo_root(), tagname)
 194         return SvnRepository.remote_exists(url)
 195
 196     def update(self, subdir="", recursive=True, branch=None):
 197         path = os.path.join(self.path, subdir)
 198         if recursive:
 199             svncommand = "svn up %s" % path
 200         else:
 201             svncommand = "svn up -N %s" % path
 202         Command(svncommand, self.options).run_fatal()
 203
 204     def commit(self, logfile):
 205         # add all new files to the repository
 206         Command("svn status %s | grep '^\?' | sed -e 's/? *//' | sed -e 's/ /\\ /g' | xargs svn add" %
 207                 self.path, self.options).output_of()
 208         Command("svn commit -F %s %s" % (logfile, self.path), self.options).run_fatal()
 209
 210     def to_branch(self, branch):
 211         remote = "%s/branches/%s" % (self.repo_root(), branch)
 212         SvnRepository.checkout(remote, self.path, self.options, recursive=True)
 213
 214     def to_tag(self, tag):
 215         remote = "%s/tags/%s" % (self.repo_root(), branch)
 216         SvnRepository.checkout(remote, self.path, self.options, recursive=True)
 217
 218     def tag(self, tagname, logfile):
 219         tag_url = "%s/tags/%s" % (self.repo_root(), tagname)
 220         self_url = self.url()
 221         Command("svn copy -F %s %s %s" % (logfile, self_url, tag_url), self.options).run_fatal()
 222
 223     def diff(self, f=""):
 224         if f:
 225             f = os.path.join(self.path, f)
 226         else:
 227             f = self.path
 228         return Command("svn diff %s" % f, self.options).output_of(True)
 229
 230     def diff_with_tag(self, tagname):
 231         tag_url = "%s/tags/%s" % (self.repo_root(), tagname)
 232         return Command("svn diff %s %s" % (tag_url, self.url()),
 233                        self.options).output_of(True)
 234
 235     def revert(self, f=""):
 236         if f:
 237             Command("svn revert %s" % os.path.join(self.path, f), self.options).run_fatal()
 238         else:
 239             # revert all
 240             Command("svn revert %s -R" % self.path, self.options).run_fatal()
 241             Command("svn status %s | grep '^\?' | sed -e 's/? *//' | sed -e 's/ /\\ /g' | xargs rm -rf " %
 242                     self.path, self.options).run_silent()
 243
 244     def is_clean(self):
 245         command="svn status %s" % self.path
 246         return len(Command(command,self.options).output_of(True)) == 0
 247
 248     def is_valid(self):
 249         return os.path.exists(os.path.join(self.path, ".svn"))
 250
 251
 252 class GitRepository:
 253     type = "git"
 254
 255     def __init__(self, path, options):
 256         self.path = path
 257         self.options = options
 258
 259     def name(self):
 260         return os.path.basename(self.path)
 261
 262     def url(self):
 263         return self.repo_root()
 264
 265     def gitweb(self):
 266         c = Command("git show | grep commit | awk '{print $2;}'", self.options)
 267         out = self.__run_in_repo(c.output_of).strip()
 268         return "http://git.onelab.eu/?p=%s.git;a=commit;h=%s" % (self.name(), out)
 269
 270     def repo_root(self):
 271         c = Command("git remote show origin", self.options)
 272         out = self.__run_in_repo(c.output_of)
 273         for line in out.split('\n'):
 274             if line.strip().startswith("Fetch URL:"):
 275                 return line.split()[2]
 276
 277     @classmethod
 278     def checkout(cls, remote, local, options, depth=0):
 279         Command("rm -rf %s" % local, options).run_silent()
 280         Command("git clone --depth %d %s %s" % (depth, remote, local), options).run_fatal()
 281         return GitRepository(local, options)
 282
 283     @classmethod
 284     def remote_exists(cls, remote):
 285         return os.system("git --no-pager ls-remote %s &> /dev/null" % remote) == 0
 286
 287     def tag_exists(self, tagname):
 288         command = 'git tag -l | grep "^%s$"' % tagname
 289         c = Command(command, self.options)
 290         out = self.__run_in_repo(c.output_of, with_stderr=True)
 291         return len(out) > 0
 292
 293     def __run_in_repo(self, fun, *args, **kwargs):
 294         cwd = os.getcwd()
 295         os.chdir(self.path)
 296         ret = fun(*args, **kwargs)
 297         os.chdir(cwd)
 298         return ret
 299
 300     def __run_command_in_repo(self, command, ignore_errors=False):
 301         c = Command(command, self.options)
 302         if ignore_errors:
 303             return self.__run_in_repo(c.output_of)
 304         else:
 305             return self.__run_in_repo(c.run_fatal)
 306
 307     def __is_commit_id(self, id):
 308         c = Command("git show %s | grep commit | awk '{print $2;}'" % id, self.options)
 309         ret = self.__run_in_repo(c.output_of, with_stderr=False)
 310         if ret.strip() == id:
 311             return True
 312         return False
 313
 314     def update(self, subdir=None, recursive=None, branch="master"):
 315         if branch == "master":
 316             self.__run_command_in_repo("git checkout %s" % branch)
 317         else:
 318             self.to_branch(branch, remote=True)
 319         self.__run_command_in_repo("git fetch origin --tags")
 320         self.__run_command_in_repo("git fetch origin")
 321         if not self.__is_commit_id(branch):
 322             # we don't need to merge anythign for commit ids.
 323             self.__run_command_in_repo("git merge --ff origin/%s" % branch)
 324
 325     def to_branch(self, branch, remote=True):
 326         self.revert()
 327         if remote:
 328             command = "git branch --track %s origin/%s" % (branch, branch)
 329             c = Command(command, self.options)
 330             self.__run_in_repo(c.output_of, with_stderr=True)
 331         return self.__run_command_in_repo("git checkout %s" % branch)
 332
 333     def to_tag(self, tag):
 334         self.revert()
 335         return self.__run_command_in_repo("git checkout %s" % tag)
 336
 337     def tag(self, tagname, logfile):
 338         self.__run_command_in_repo("git tag %s -F %s" % (tagname, logfile))
 339         self.commit(logfile)
 340
 341     def diff(self, f=""):
 342         c = Command("git diff %s" % f, self.options)
 343         return self.__run_in_repo(c.output_of, with_stderr=True)
 344
 345     def diff_with_tag(self, tagname):
 346         c = Command("git diff %s" % tagname, self.options)
 347         return self.__run_in_repo(c.output_of, with_stderr=True)
 348
 349     def commit(self, logfile, branch="master"):
 350         self.__run_command_in_repo("git add .", ignore_errors=True)
 351         self.__run_command_in_repo("git add -u", ignore_errors=True)
 352         self.__run_command_in_repo("git commit -F  %s" % logfile, ignore_errors=True)
 353         if branch == "master" or self.__is_commit_id(branch):
 354             self.__run_command_in_repo("git push")
 355         else:
 356             self.__run_command_in_repo("git push origin %s:%s" % (branch, branch))
 357         self.__run_command_in_repo("git push --tags")
 358
 359     def revert(self, f=""):
 360         if f:
 361             self.__run_command_in_repo("git checkout %s" % f)
 362         else:
 363             # revert all
 364             self.__run_command_in_repo("git --no-pager reset --hard")
 365             self.__run_command_in_repo("git --no-pager clean -f")
 366
 367     def is_clean(self):
 368         def check_commit():
 369             command="git status"
 370             s="nothing to commit (working directory clean)"
 371             return Command(command, self.options).output_of(True).find(s) >= 0
 372         return self.__run_in_repo(check_commit)
 373
 374     def is_valid(self):
 375         return os.path.exists(os.path.join(self.path, ".git"))
 376
 377
 378 class Repository:
 379     """ Generic repository """
 380     supported_repo_types = [SvnRepository, GitRepository]
 381
 382     def __init__(self, path, options):
 383         self.path = path
 384         self.options = options
 385         for repo in self.supported_repo_types:
 386             self.repo = repo(self.path, self.options)
 387             if self.repo.is_valid():
 388                 break
 389
 390     @classmethod
 391     def has_moved_to_git(cls, module, config):
 392         module = svn_to_git_name(module)
 393         # check if the module is already in Git
 394 #        return SvnRepository.remote_exists("%s/%s/aaaa-has-moved-to-git" % (config['svnpath'], module))
 395         return GitRepository.remote_exists(Module.git_remote_dir(module))
 396
 397
 398     @classmethod
 399     def remote_exists(cls, remote):
 400         for repo in Repository.supported_repo_types:
 401             if repo.remote_exists(remote):
 402                 return True
 403         return False
 404
 405     def __getattr__(self, attr):
 406         return getattr(self.repo, attr)
 407
 408
 409
 410 # support for tagged module is minimal, and is for the Build class only
 411 class Module:
 412
 413     svn_magic_line="--This line, and those below, will be ignored--"
 414     setting_tag_format = "Setting tag %s"
 415
 416     redirectors=[ # ('module_name_varname','name'),
 417                   ('module_version_varname','version'),
 418                   ('module_taglevel_varname','taglevel'), ]
 419
 420     # where to store user's config
 421     config_storage="CONFIG"
 422     #
 423     config={}
 424
 425     import commands
 426     configKeys=[ ('svnpath',"Enter your toplevel svnpath",
 427                   "svn+ssh://%s@svn.planet-lab.org/svn/"%commands.getoutput("id -un")),
 428                  ('gitserver', "Enter your git server's hostname", "git.onelab.eu"),
 429                  ('gituser', "Enter your user name (login name) on git server", commands.getoutput("id -un")),
 430                  ("build", "Enter the name of your build module","build"),
 431                  ('username',"Enter your firstname and lastname for changelogs",""),
 432                  ("email","Enter your email address for changelogs",""),
 433                  ]
 434
 435     @classmethod
 436     def prompt_config_option(cls, key, message, default):
 437         cls.config[key]=raw_input("%s [%s] : "%(message,default)).strip() or default
 438
 439     @classmethod
 440     def prompt_config (cls):
 441         for (key,message,default) in cls.configKeys:
 442             cls.config[key]=""
 443             while not cls.config[key]:
 444                 cls.prompt_config_option(key, message, default)
 445
 446     # for parsing module spec name:branch
 447     matcher_branch_spec=re.compile("\A(?P<name>[\w\.-]+):(?P<branch>[\w\.-]+)\Z")
 448     # special form for tagged module - for Build
 449     matcher_tag_spec=re.compile("\A(?P<name>[\w\.-]+)@(?P<tagname>[\w\.-]+)\Z")
 450     # parsing specfiles
 451     matcher_rpm_define=re.compile("%(define|global)\s+(\S+)\s+(\S*)\s*")
 452
 453     @classmethod
 454     def parse_module_spec(cls, module_spec):
 455         name = branch_or_tagname = module_type = ""
 456
 457         attempt=Module.matcher_branch_spec.match(module_spec)
 458         if attempt:
 459             module_type = "branch"
 460             name=attempt.group('name')
 461             branch_or_tagname=attempt.group('branch')
 462         else:
 463             attempt=Module.matcher_tag_spec.match(module_spec)
 464             if attempt:
 465                 module_type = "tag"
 466                 name=attempt.group('name')
 467                 branch_or_tagname=attempt.group('tagname')
 468             else:
 469                 name=module_spec
 470         return name, branch_or_tagname, module_type
 471
 472
 473     def __init__ (self,module_spec,options):
 474         # parse module spec
 475         self.name, branch_or_tagname, module_type = self.parse_module_spec(module_spec)
 476
 477         if module_type == "branch":
 478             self.branch=branch_or_tagname
 479         elif module_type == "tag":
 480             self.tagname=branch_or_tagname
 481
 482         # when available prefer to use git module name internally
 483         self.name = svn_to_git_name(self.name)
 484
 485         self.options=options
 486         self.module_dir="%s/%s"%(options.workdir,self.name)
 487         self.repository = None
 488         self.build = None
 489
 490     def run (self,command):
 491         return Command(command,self.options).run()
 492     def run_fatal (self,command):
 493         return Command(command,self.options).run_fatal()
 494     def run_prompt (self,message,fun, *args):
 495         fun_msg = "%s(%s)" % (fun.func_name, ",".join(args))
 496         if not self.options.verbose:
 497             while True:
 498                 choice=prompt(message,True,('s','how'))
 499                 if choice is True:
 500                     fun(*args)
 501                     return
 502                 elif choice is False:
 503                     print 'About to run function:', fun_msg
 504         else:
 505             question=message+" - want to run function: " + fun_msg
 506             if prompt(question,True):
 507                 fun(*args)
 508
 509     def friendly_name (self):
 510         if hasattr(self,'branch'):
 511             return "%s:%s"%(self.name,self.branch)
 512         elif hasattr(self,'tagname'):
 513             return "%s@%s"%(self.name,self.tagname)
 514         else:
 515             return self.name
 516
 517     @classmethod
 518     def git_remote_dir (cls, name):
 519         return "%s@%s:/git/%s.git" % (cls.config['gituser'], cls.config['gitserver'], name)
 520
 521     @classmethod
 522     def svn_remote_dir (cls, name):
 523         name = git_to_svn_name(name)
 524         svn = cls.config['svnpath']
 525         if svn.endswith('/'):
 526             return "%s%s" % (svn, name)
 527         return "%s/%s" % (svn, name)
 528
 529     def svn_selected_remote(self):
 530         svn_name = git_to_svn_name(self.name)
 531         remote = self.svn_remote_dir(svn_name)
 532         if hasattr(self,'branch'):
 533             remote = "%s/branches/%s" % (remote, self.branch)
 534         elif hasattr(self,'tagname'):
 535             remote = "%s/tags/%s" % (remote, self.tagname)
 536         else:
 537             remote = "%s/trunk" % remote
 538         return remote
 539
 540     ####################
 541     @classmethod
 542     def init_homedir (cls, options):
 543         if options.verbose and options.mode not in Main.silent_modes:
 544             print 'Checking for', options.workdir
 545         storage="%s/%s"%(options.workdir, cls.config_storage)
 546         # sanity check. Either the topdir exists AND we have a config/storage
 547         # or topdir does not exist and we create it
 548         # to avoid people use their own daily svn repo
 549         if os.path.isdir(options.workdir) and not os.path.isfile(storage):
 550             print """The directory %s exists and has no CONFIG file
 551 If this is your regular working directory, please provide another one as the
 552 module-* commands need a fresh working dir. Make sure that you do not use
 553 that for other purposes than tagging""" % options.workdir
 554             sys.exit(1)
 555
 556         def checkout_build():
 557             print "Checking out build module..."
 558             remote = cls.git_remote_dir(cls.config['build'])
 559             local = os.path.join(options.workdir, cls.config['build'])
 560             GitRepository.checkout(remote, local, options, depth=1)
 561             print "OK"
 562
 563         def store_config():
 564             f=file(storage,"w")
 565             for (key,message,default) in Module.configKeys:
 566                 f.write("%s=%s\n"%(key,Module.config[key]))
 567             f.close()
 568             if options.debug:
 569                 print 'Stored',storage
 570                 Command("cat %s"%storage,options).run()
 571
 572         def read_config():
 573             # read config
 574             f=open(storage)
 575             for line in f.readlines():
 576                 (key,value)=re.compile("^(.+)=(.+)$").match(line).groups()
 577                 Module.config[key]=value
 578             f.close()
 579
 580         if not os.path.isdir (options.workdir):
 581             print "Cannot find",options.workdir,"let's create it"
 582             Command("mkdir -p %s" % options.workdir, options).run_silent()
 583             cls.prompt_config()
 584             checkout_build()
 585             store_config()
 586         else:
 587             read_config()
 588             # check missing config options
 589             old_layout = False
 590             for (key,message,default) in cls.configKeys:
 591                 if not Module.config.has_key(key):
 592                     print "Configuration changed for module-tools"
 593                     cls.prompt_config_option(key, message, default)
 594                     old_layout = True
 595
 596             if old_layout:
 597                 Command("rm -rf %s" % options.workdir, options).run_silent()
 598                 Command("mkdir -p %s" % options.workdir, options).run_silent()
 599                 checkout_build()
 600                 store_config()
 601
 602             build_dir = os.path.join(options.workdir, cls.config['build'])
 603             if not os.path.isdir(build_dir):
 604                 checkout_build()
 605             else:
 606                 build = Repository(build_dir, options)
 607                 if not build.is_clean():
 608                     print "build module needs a revert"
 609                     build.revert()
 610                     print "OK"
 611                 build.update()
 612
 613         if options.verbose and options.mode not in Main.silent_modes:
 614             print '******** Using config'
 615             for (key,message,default) in Module.configKeys:
 616                 print '\t',key,'=',Module.config[key]
 617
 618     def init_module_dir (self):
 619         if self.options.verbose:
 620             print 'Checking for',self.module_dir
 621
 622         if not os.path.isdir (self.module_dir):
 623             if Repository.has_moved_to_git(self.name, Module.config):
 624                 self.repository = GitRepository.checkout(self.git_remote_dir(self.name),
 625                                                          self.module_dir,
 626                                                          self.options)
 627             else:
 628                 remote = self.svn_selected_remote()
 629                 self.repository = SvnRepository.checkout(remote,
 630                                                          self.module_dir,
 631                                                          self.options, recursive=False)
 632
 633         self.repository = Repository(self.module_dir, self.options)
 634         if self.repository.type == "svn":
 635             # check if module has moved to git
 636             if Repository.has_moved_to_git(self.name, Module.config):
 637                 Command("rm -rf %s" % self.module_dir, self.options).run_silent()
 638                 self.init_module_dir()
 639             # check if we have the required branch/tag
 640             if self.repository.url() != self.svn_selected_remote():
 641                 Command("rm -rf %s" % self.module_dir, self.options).run_silent()
 642                 self.init_module_dir()
 643
 644         elif self.repository.type == "git":
 645             if hasattr(self,'branch'):
 646                 print "to branch", self.branch
 647                 self.repository.to_branch(self.branch)
 648             elif hasattr(self,'tagname'):
 649                 self.repository.to_tag(self.tagname)
 650
 651         else:
 652             raise Exception, 'Cannot find %s - check module name'%self.module_dir
 653
 654
 655     def revert_module_dir (self):
 656         if self.options.fast_checks:
 657             if self.options.verbose: print 'Skipping revert of %s' % self.module_dir
 658             return
 659         if self.options.verbose:
 660             print 'Checking whether', self.module_dir, 'needs being reverted'
 661
 662         if not self.repository.is_clean():
 663             self.repository.revert()
 664
 665     def update_module_dir (self):
 666         if self.options.fast_checks:
 667             if self.options.verbose: print 'Skipping update of %s' % self.module_dir
 668             return
 669         if self.options.verbose:
 670             print 'Updating', self.module_dir
 671
 672         if hasattr(self,'branch'):
 673             self.repository.update(branch=self.branch)
 674         elif hasattr(self,'tagname'):
 675             self.repository.update(branch=self.tagname)
 676         else:
 677             self.repository.update()
 678
 679     def main_specname (self):
 680         attempt="%s/%s.spec"%(self.module_dir,self.name)
 681         if os.path.isfile (attempt):
 682             return attempt
 683         pattern1="%s/*.spec"%self.module_dir
 684         level1=glob(pattern1)
 685         if level1:
 686             return level1[0]
 687         pattern2="%s/*/*.spec"%self.module_dir
 688         level2=glob(pattern2)
 689
 690         if level2:
 691             return level2[0]
 692         raise Exception, 'Cannot guess specfile for module %s -- patterns were %s or %s'%(self.name,pattern1,pattern2)
 693
 694     def all_specnames (self):
 695         level1=glob("%s/*.spec" % self.module_dir)
 696         if level1: return level1
 697         level2=glob("%s/*/*.spec" % self.module_dir)
 698         return level2
 699
 700     def parse_spec (self, specfile, varnames):
 701         if self.options.verbose:
 702             print 'Parsing',specfile,
 703             for var in varnames:
 704                 print "[%s]"%var,
 705             print ""
 706         result={}
 707         f=open(specfile)
 708         for line in f.readlines():
 709             attempt=Module.matcher_rpm_define.match(line)
 710             if attempt:
 711                 (define,var,value)=attempt.groups()
 712                 if var in varnames:
 713                     result[var]=value
 714         f.close()
 715         if self.options.debug:
 716             print 'found',len(result),'keys'
 717             for (k,v) in result.iteritems():
 718                 print k,'=',v
 719         return result
 720
 721     # stores in self.module_name_varname the rpm variable to be used for the module's name
 722     # and the list of these names in self.varnames
 723     def spec_dict (self):
 724         specfile=self.main_specname()
 725         redirector_keys = [ varname for (varname,default) in Module.redirectors]
 726         redirect_dict = self.parse_spec(specfile,redirector_keys)
 727         if self.options.debug:
 728             print '1st pass parsing done, redirect_dict=',redirect_dict
 729         varnames=[]
 730         for (varname,default) in Module.redirectors:
 731             if redirect_dict.has_key(varname):
 732                 setattr(self,varname,redirect_dict[varname])
 733                 varnames += [redirect_dict[varname]]
 734             else:
 735                 setattr(self,varname,default)
 736                 varnames += [ default ]
 737         self.varnames = varnames
 738         result = self.parse_spec (specfile,self.varnames)
 739         if self.options.debug:
 740             print '2st pass parsing done, varnames=',varnames,'result=',result
 741         return result
 742
 743     def patch_spec_var (self, patch_dict,define_missing=False):
 744         for specfile in self.all_specnames():
 745             # record the keys that were changed
 746             changed = dict ( [ (x,False) for x in patch_dict.keys() ] )
 747             newspecfile=specfile+".new"
 748             if self.options.verbose:
 749                 print 'Patching',specfile,'for',patch_dict.keys()
 750             spec=open (specfile)
 751             new=open(newspecfile,"w")
 752
 753             for line in spec.readlines():
 754                 attempt=Module.matcher_rpm_define.match(line)
 755                 if attempt:
 756                     (define,var,value)=attempt.groups()
 757                     if var in patch_dict.keys():
 758                         if self.options.debug:
 759                             print 'rewriting %s as %s'%(var,patch_dict[var])
 760                         new.write('%%%s %s %s\n'%(define,var,patch_dict[var]))
 761                         changed[var]=True
 762                         continue
 763                 new.write(line)
 764             if define_missing:
 765                 for (key,was_changed) in changed.iteritems():
 766                     if not was_changed:
 767                         if self.options.debug:
 768                             print 'rewriting missing %s as %s'%(key,patch_dict[key])
 769                         new.write('\n%%define %s %s\n'%(key,patch_dict[key]))
 770             spec.close()
 771             new.close()
 772             os.rename(newspecfile,specfile)
 773
 774     # returns all lines until the magic line
 775     def unignored_lines (self, logfile):
 776         result=[]
 777         white_line_matcher = re.compile("\A\s*\Z")
 778         for logline in file(logfile).readlines():
 779             if logline.strip() == Module.svn_magic_line:
 780                 break
 781             elif white_line_matcher.match(logline):
 782                 continue
 783             else:
 784                 result.append(logline.strip()+'\n')
 785         return result
 786
 787     # creates a copy of the input with only the unignored lines
 788     def stripped_magic_line_filename (self, filein, fileout ,new_tag_name):
 789        f=file(fileout,'w')
 790        f.write(self.setting_tag_format%new_tag_name + '\n')
 791        for line in self.unignored_lines(filein):
 792            f.write(line)
 793        f.close()
 794
 795     def insert_changelog (self, logfile, oldtag, newtag):
 796         for specfile in self.all_specnames():
 797             newspecfile=specfile+".new"
 798             if self.options.verbose:
 799                 print 'Inserting changelog from %s into %s'%(logfile,specfile)
 800             spec=open (specfile)
 801             new=open(newspecfile,"w")
 802             for line in spec.readlines():
 803                 new.write(line)
 804                 if re.compile('%changelog').match(line):
 805                     dateformat="* %a %b %d %Y"
 806                     datepart=time.strftime(dateformat)
 807                     logpart="%s <%s> - %s"%(Module.config['username'],
 808                                                  Module.config['email'],
 809                                                  newtag)
 810                     new.write(datepart+" "+logpart+"\n")
 811                     for logline in self.unignored_lines(logfile):
 812                         new.write("- " + logline)
 813                     new.write("\n")
 814             spec.close()
 815             new.close()
 816             os.rename(newspecfile,specfile)
 817
 818     def show_dict (self, spec_dict):
 819         if self.options.verbose:
 820             for (k,v) in spec_dict.iteritems():
 821                 print k,'=',v
 822
 823     def last_tag (self, spec_dict):
 824         try:
 825             return "%s-%s" % (spec_dict[self.module_version_varname],
 826                               spec_dict[self.module_taglevel_varname])
 827         except KeyError,err:
 828             raise Exception,'Something is wrong with module %s, cannot determine %s - exiting'%(self.name,err)
 829
 830     def tag_name (self, spec_dict, old_svn_name=False):
 831         base_tag_name = self.name
 832         if old_svn_name:
 833             base_tag_name = git_to_svn_name(self.name)
 834         return "%s-%s" % (base_tag_name, self.last_tag(spec_dict))
 835
 836
 837 ##############################
 838     # using fine_grain means replacing only those instances that currently refer to this tag
 839     # otherwise, <module>-{SVNPATH,GITPATH} is replaced unconditionnally
 840     def patch_tags_file (self, tagsfile, oldname, newname,fine_grain=True):
 841         newtagsfile=tagsfile+".new"
 842         tags=open (tagsfile)
 843         new=open(newtagsfile,"w")
 844
 845         matches=0
 846         # fine-grain : replace those lines that refer to oldname
 847         if fine_grain:
 848             if self.options.verbose:
 849                 print 'Replacing %s into %s\n\tin %s .. '%(oldname,newname,tagsfile),
 850             matcher=re.compile("^(.*)%s(.*)"%oldname)
 851             for line in tags.readlines():
 852                 if not matcher.match(line):
 853                     new.write(line)
 854                 else:
 855                     (begin,end)=matcher.match(line).groups()
 856                     new.write(begin+newname+end+"\n")
 857                     matches += 1
 858         # brute-force : change uncommented lines that define <module>-SVNPATH
 859         else:
 860             if self.options.verbose:
 861                 print 'Searching for -SVNPATH or -GITPATH lines referring to /%s/\n\tin %s .. '%(self.name,tagsfile),
 862             pattern="\A\s*%s-(SVNPATH|GITPATH)\s*(=|:=)\s*(?P<url_main>[^\s]+)/%s[^\s]+"\
 863                                           %(self.name,self.name)
 864             matcher_module=re.compile(pattern)
 865             for line in tags.readlines():
 866                 attempt=matcher_module.match(line)
 867                 if attempt:
 868                     if line.find("-GITPATH") >= 0:
 869                         modulepath = "%s-GITPATH"%self.name
 870                         replacement = "%-32s:= %s/%s.git@%s\n"%(modulepath,attempt.group('url_main'),self.name,newname)
 871                     else:
 872                         modulepath = "%s-SVNPATH"%self.name
 873                         replacement = "%-32s:= %s/%s/tags/%s\n"%(modulepath,attempt.group('url_main'),self.name,newname)
 874                     if self.options.verbose:
 875                         print ' ' + modulepath,
 876                     new.write(replacement)
 877                     matches += 1
 878                 else:
 879                     new.write(line)
 880         tags.close()
 881         new.close()
 882         os.rename(newtagsfile,tagsfile)
 883         if self.options.verbose: print "%d changes"%matches
 884         return matches
 885
 886     def check_tag(self, tagname, need_it=False, old_svn_tag_name=None):
 887         if self.options.verbose:
 888             print "Checking %s repository tag: %s - " % (self.repository.type, tagname),
 889
 890         found_tagname = tagname
 891         found = self.repository.tag_exists(tagname)
 892         if not found and old_svn_tag_name:
 893             if self.options.verbose:
 894                 print "KO"
 895                 print "Checking %s repository tag: %s - " % (self.repository.type, old_svn_tag_name),
 896             found = self.repository.tag_exists(old_svn_tag_name)
 897             if found:
 898                 found_tagname = old_svn_tag_name
 899
 900         if (found and need_it) or (not found and not need_it):
 901             if self.options.verbose:
 902                 print "OK",
 903                 if found: print "- found"
 904                 else: print "- not found"
 905         else:
 906             if self.options.verbose:
 907                 print "KO"
 908             if found:
 909                 raise Exception, "tag (%s) is already there" % tagname
 910             else:
 911                 raise Exception, "can not find required tag (%s)" % tagname
 912
 913         return found_tagname
 914
 915
 916 ##############################
 917     def do_tag (self):
 918         self.init_module_dir()
 919         self.revert_module_dir()
 920         self.update_module_dir()
 921         # parse specfile
 922         spec_dict = self.spec_dict()
 923         self.show_dict(spec_dict)
 924
 925         # side effects
 926         old_tag_name = self.tag_name(spec_dict)
 927         old_svn_tag_name = self.tag_name(spec_dict, old_svn_name=True)
 928
 929         if (self.options.new_version):
 930             # new version set on command line
 931             spec_dict[self.module_version_varname] = self.options.new_version
 932             spec_dict[self.module_taglevel_varname] = 0
 933         else:
 934             # increment taglevel
 935             new_taglevel = str ( int (spec_dict[self.module_taglevel_varname]) + 1)
 936             spec_dict[self.module_taglevel_varname] = new_taglevel
 937
 938         new_tag_name = self.tag_name(spec_dict)
 939
 940         # sanity check
 941         old_tag_name = self.check_tag(old_tag_name, need_it=True, old_svn_tag_name=old_svn_tag_name)
 942         new_tag_name = self.check_tag(new_tag_name, need_it=False)
 943
 944         # checking for diffs
 945         diff_output = self.repository.diff_with_tag(old_tag_name)
 946         if len(diff_output) == 0:
 947             if not prompt ("No pending difference in module %s, want to tag anyway"%self.name,False):
 948                 return
 949
 950         # side effect in trunk's specfile
 951         self.patch_spec_var(spec_dict)
 952
 953         # prepare changelog file
 954         # we use the standard subversion magic string (see svn_magic_line)
 955         # so we can provide useful information, such as version numbers and diff
 956         # in the same file
 957         changelog="/tmp/%s-%d.edit"%(self.name,os.getpid())
 958         changelog_svn="/tmp/%s-%d.svn"%(self.name,os.getpid())
 959         setting_tag_line=Module.setting_tag_format%new_tag_name
 960         file(changelog,"w").write("""
 961 %s
 962 %s
 963 Please write a changelog for this new tag in the section above
 964 """%(Module.svn_magic_line,setting_tag_line))
 965
 966         if not self.options.verbose or prompt('Want to see diffs while writing changelog',True):
 967             file(changelog,"a").write('DIFF=========\n' + diff_output)
 968
 969         if self.options.debug:
 970             prompt('Proceed ?')
 971
 972         # edit it
 973         self.run("%s %s"%(self.options.editor,changelog))
 974         # strip magic line in second file - looks like svn has changed its magic line with 1.6
 975         # so we do the job ourselves
 976         self.stripped_magic_line_filename(changelog,changelog_svn,new_tag_name)
 977         # insert changelog in spec
 978         if self.options.changelog:
 979             self.insert_changelog (changelog,old_tag_name,new_tag_name)
 980
 981         ## update build
 982         build_path = os.path.join(self.options.workdir,
 983                                   Module.config['build'])
 984         build = Repository(build_path, self.options)
 985         if self.options.build_branch:
 986             build.to_branch(self.options.build_branch)
 987         if not build.is_clean():
 988             build.revert()
 989
 990         tagsfiles=glob(build.path+"/*-tags*.mk")
 991         tagsdict=dict( [ (x,'todo') for x in tagsfiles ] )
 992         default_answer = 'y'
 993         tagsfiles.sort()
 994         while True:
 995             for tagsfile in tagsfiles:
 996                 status=tagsdict[tagsfile]
 997                 basename=os.path.basename(tagsfile)
 998                 print ".................... Dealing with %s"%basename
 999                 while tagsdict[tagsfile] == 'todo' :
1000                     choice = prompt ("insert %s in %s    "%(new_tag_name,basename),default_answer,
1001                                      [ ('y','es'), ('n', 'ext'), ('f','orce'),
1002                                        ('d','iff'), ('r','evert'), ('c', 'at'), ('h','elp') ] ,
1003                                      allow_outside=True)
1004                     if choice == 'y':
1005                         self.patch_tags_file(tagsfile,old_tag_name,new_tag_name,fine_grain=True)
1006                     elif choice == 'n':
1007                         print 'Done with %s'%os.path.basename(tagsfile)
1008                         tagsdict[tagsfile]='done'
1009                     elif choice == 'f':
1010                         self.patch_tags_file(tagsfile,old_tag_name,new_tag_name,fine_grain=False)
1011                     elif choice == 'd':
1012                         print build.diff(f=os.path.basename(tagsfile))
1013                     elif choice == 'r':
1014                         build.revert(f=tagsfile)
1015                     elif choice == 'c':
1016                         self.run("cat %s"%tagsfile)
1017                     else:
1018                         name=self.name
1019                         print """y: change %(name)s-{SVNPATH,GITPATH} only if it currently refers to %(old_tag_name)s
1020 f: unconditionnally change any line that assigns %(name)s-SVNPATH to using %(new_tag_name)s
1021 d: show current diff for this tag file
1022 r: revert that tag file
1023 c: cat the current tag file
1024 n: move to next file"""%locals()
1025
1026             if prompt("Want to review changes on tags files",False):
1027                 tagsdict = dict ( [ (x, 'todo') for x in tagsfiles ] )
1028                 default_answer='d'
1029             else:
1030                 break
1031
1032         def diff_all_changes():
1033             print build.diff()
1034             print self.repository.diff()
1035
1036         def commit_all_changes(log):
1037             if hasattr(self,'branch'):
1038                 self.repository.commit(log, branch=self.branch)
1039             else:
1040                 self.repository.commit(log)
1041             build.commit(log)
1042
1043         self.run_prompt("Review module and build", diff_all_changes)
1044         self.run_prompt("Commit module and build", commit_all_changes, changelog_svn)
1045         self.run_prompt("Create tag", self.repository.tag, new_tag_name, changelog_svn)
1046
1047         if self.options.debug:
1048             print 'Preserving',changelog,'and stripped',changelog_svn
1049         else:
1050             os.unlink(changelog)
1051             os.unlink(changelog_svn)
1052
1053
1054 ##############################
1055     def do_version (self):
1056         self.init_module_dir()
1057         self.revert_module_dir()
1058         self.update_module_dir()
1059         spec_dict = self.spec_dict()
1060         if self.options.www:
1061             self.html_store_title('Version for module %s (%s)' % (self.friendly_name(),
1062                                                                   self.last_tag(spec_dict)))
1063         for varname in self.varnames:
1064             if not spec_dict.has_key(varname):
1065                 self.html_print ('Could not find %%define for %s'%varname)
1066                 return
1067             else:
1068                 self.html_print ("%-16s %s"%(varname,spec_dict[varname]))
1069         self.html_print ("%-16s %s"%('url',self.repository.url()))
1070         if self.options.verbose:
1071             self.html_print ("%-16s %s"%('main specfile:',self.main_specname()))
1072             self.html_print ("%-16s %s"%('specfiles:',self.all_specnames()))
1073         self.html_print_end()
1074
1075
1076 ##############################
1077     def do_diff (self):
1078         self.init_module_dir()
1079         self.revert_module_dir()
1080         self.update_module_dir()
1081         spec_dict = self.spec_dict()
1082         self.show_dict(spec_dict)
1083
1084         # side effects
1085         tag_name = self.tag_name(spec_dict)
1086         old_svn_tag_name = self.tag_name(spec_dict, old_svn_name=True)
1087
1088         # sanity check
1089         tag_name = self.check_tag(tag_name, need_it=True, old_svn_tag_name=old_svn_tag_name)
1090
1091         if self.options.verbose:
1092             print 'Getting diff'
1093         diff_output = self.repository.diff_with_tag(tag_name)
1094
1095         if self.options.list:
1096             if diff_output:
1097                 print self.name
1098         else:
1099             thename=self.friendly_name()
1100             do_print=False
1101             if self.options.www and diff_output:
1102                 self.html_store_title("Diffs in module %s (%s) : %d chars"%(\
1103                         thename,self.last_tag(spec_dict),len(diff_output)))
1104
1105                 self.html_store_raw ('<p> &lt; (left) %s </p>' % tag_name)
1106                 self.html_store_raw ('<p> &gt; (right) %s </p>' % thename)
1107                 self.html_store_pre (diff_output)
1108             elif not self.options.www:
1109                 print 'x'*30,'module',thename
1110                 print 'x'*20,'<',tag_name
1111                 print 'x'*20,'>',thename
1112                 print diff_output
1113
1114 ##############################
1115     # store and restitute html fragments
1116     @staticmethod
1117     def html_href (url,text): return '<a href="%s">%s</a>'%(url,text)
1118
1119     @staticmethod
1120     def html_anchor (url,text): return '<a name="%s">%s</a>'%(url,text)
1121
1122     @staticmethod
1123     def html_quote (text):
1124         return text.replace('&','&#38;').replace('<','&lt;').replace('>','&gt;')
1125
1126     # only the fake error module has multiple titles
1127     def html_store_title (self, title):
1128         if not hasattr(self,'titles'): self.titles=[]
1129         self.titles.append(title)
1130
1131     def html_store_raw (self, html):
1132         if not hasattr(self,'body'): self.body=''
1133         self.body += html
1134
1135     def html_store_pre (self, text):
1136         if not hasattr(self,'body'): self.body=''
1137         self.body += '<pre>' + self.html_quote(text) + '</pre>'
1138
1139     def html_print (self, txt):
1140         if not self.options.www:
1141             print txt
1142         else:
1143             if not hasattr(self,'in_list') or not self.in_list:
1144                 self.html_store_raw('<ul>')
1145                 self.in_list=True
1146             self.html_store_raw('<li>'+txt+'</li>')
1147
1148     def html_print_end (self):
1149         if self.options.www:
1150             self.html_store_raw ('</ul>')
1151
1152     @staticmethod
1153     def html_dump_header(title):
1154         nowdate=time.strftime("%Y-%m-%d")
1155         nowtime=time.strftime("%H:%M (%Z)")
1156         print """<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
1157 <html xmlns="http://www.w3.org/1999/xhtml" lang="en" xml:lang="en">
1158 <head>
1159 <title> %s </title>
1160 <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
1161 <style type="text/css">
1162 body { font-family:georgia, serif; }
1163 h1 {font-size: large; }
1164 p.title {font-size: x-large; }
1165 span.error {text-weight:bold; color: red; }
1166 </style>
1167 </head>
1168 <body>
1169 <p class='title'> %s - status on %s at %s</p>
1170 <ul>
1171 """%(title,title,nowdate,nowtime)
1172
1173     @staticmethod
1174     def html_dump_middle():
1175         print "</ul>"
1176
1177     @staticmethod
1178     def html_dump_footer():
1179         print "</body></html"
1180
1181     def html_dump_toc(self):
1182         if hasattr(self,'titles'):
1183             for title in self.titles:
1184                 print '<li>',self.html_href ('#'+self.friendly_name(),title),'</li>'
1185
1186     def html_dump_body(self):
1187         if hasattr(self,'titles'):
1188             for title in self.titles:
1189                 print '<hr /><h1>',self.html_anchor(self.friendly_name(),title),'</h1>'
1190         if hasattr(self,'body'):
1191             print self.body
1192             print '<p class="top">',self.html_href('#','Back to top'),'</p>'
1193
1194
1195
1196 class Build(Module):
1197
1198     def __get_modules(self, tagfile):
1199         self.init_module_dir()
1200         modules = {}
1201
1202         tagfile = os.path.join(self.module_dir, tagfile)
1203         for line in open(tagfile):
1204             try:
1205                 name, url = line.split(':=')
1206                 name, git_or_svn_path = name.rsplit('-', 1)
1207                 name = svn_to_git_name(name.strip())
1208                 modules[name] = (git_or_svn_path.strip(), url.strip())
1209             except:
1210                 pass
1211         return modules
1212
1213     def get_modules(self, tagfile):
1214         modules = self.__get_modules(tagfile)
1215         for module in modules:
1216             module_type = tag_or_branch = ""
1217
1218             path_type, url = modules[module]
1219             if path_type == "GITPATH":
1220                 module_spec = os.path.split(url)[-1].replace(".git","")
1221                 name, tag_or_branch, module_type = self.parse_module_spec(module_spec)
1222             else:
1223                 tag_or_branch = os.path.split(url)[-1].strip()
1224                 if url.find('/tags/') >= 0:
1225                     module_type = "tag"
1226                 elif url.find('/branches/') >= 0:
1227                     module_type = "branch"
1228
1229             modules[module] = {"module_type" : module_type,
1230                                "path_type": path_type,
1231                                "tag_or_branch": tag_or_branch,
1232                                "url":url}
1233         return modules
1234
1235
1236
1237 def modules_diff(first, second):
1238     diff = {}
1239
1240     for module in first:
1241         if first[module]['tag_or_branch'] != second[module]['tag_or_branch']:
1242             diff[module] = (first[module]['tag_or_branch'], second[module]['tag_or_branch'])
1243
1244     first_set = set(first.keys())
1245     second_set = set(second.keys())
1246
1247     new_modules = list(second_set - first_set)
1248     removed_modules = list(first_set - second_set)
1249
1250     return diff, new_modules, removed_modules
1251
1252 def release_changelog(options, buildtag_old, buildtag_new):
1253
1254     tagfile = options.distrotags[0]
1255     if not tagfile:
1256         print "ERROR: provide a tagfile name (eg. onelab, onelab-k27, planetlab)"
1257         return
1258     tagfile = "%s-tags.mk" % tagfile
1259
1260     print '= build tag %s to %s =' % (buildtag_old, buildtag_new)
1261     print '== distro %s (%s to %s) ==' % (tagfile, buildtag_old, buildtag_new)
1262
1263     build = Build("build@%s" % buildtag_old, options)
1264     build.init_module_dir()
1265     first = build.get_modules(tagfile)
1266
1267     print '* from', buildtag_old, build.repository.gitweb()
1268
1269     build = Build("build@%s" % buildtag_new, options)
1270     build.init_module_dir()
1271     second = build.get_modules(tagfile)
1272
1273     print '* to', buildtag_new, build.repository.gitweb()
1274
1275     diff, new_modules, removed_modules = modules_diff(first, second)
1276
1277
1278     def get_module(name, tag):
1279         print name, tag
1280         if not tag or  tag == "trunk":
1281             return Module("%s" % (module), options)
1282         else:
1283             return Module("%s@%s" % (module, tag), options)
1284
1285
1286     for module in diff:
1287         print '=== %s - %s to %s : package %s ===' % (tagfile, buildtag_old, buildtag_new, module)
1288
1289         first, second = diff[module]
1290         m = get_module(module, first)
1291         os.system('rm -rf %s' % m.module_dir) # cleanup module dir
1292         m.init_module_dir()
1293
1294         if m.repository.type == "svn":
1295             print '* from', first, m.repository.url()
1296         else:
1297             print '* from', first, m.repository.gitweb()
1298
1299         specfile = m.main_specname()
1300         (tmpfd, tmpfile) = tempfile.mkstemp()
1301         os.system("cp -f /%s %s" % (specfile, tmpfile))
1302
1303         m = get_module(module, second)
1304         m.init_module_dir()
1305         specfile = m.main_specname()
1306
1307         if m.repository.type == "svn":
1308             print '* to', second, m.repository.url()
1309         else:
1310             print '* to', second, m.repository.gitweb()
1311
1312         print '{{{'
1313         os.system("diff -u %s %s" % (tmpfile, specfile))
1314         print '}}}'
1315
1316         os.unlink(tmpfile)
1317
1318     for module in new_modules:
1319         print '=== %s : new package in build %s ===' % (tagfile, module)
1320
1321     for module in removed_modules:
1322         print '=== %s : removed package from build %s ===' % (tagfile, module)
1323
1324
1325 ##############################
1326 class Main:
1327
1328     module_usage="""Usage: %prog [options] module_desc [ .. module_desc ]
1329
1330 module-tools : a set of tools to manage subversion tags and specfile
1331   requires the specfile to either
1332   * define *version* and *taglevel*
1333   OR alternatively
1334   * define redirection variables module_version_varname / module_taglevel_varname
1335 Trunk:
1336   by default, the trunk of modules is taken into account
1337   in this case, just mention the module name as <module_desc>
1338 Branches:
1339   if you wish to work on a branch rather than on the trunk,
1340   you can use something like e.g. Mom:2.1 as <module_desc>
1341 """
1342     release_usage="""Usage: %prog [options] tag1 .. tagn
1343   Extract release notes from the changes in specfiles between several build tags, latest first
1344   Examples:
1345       release-changelog 4.2-rc25 4.2-rc24 4.2-rc23 4.2-rc22
1346   You can refer to a (build) branch by prepending a colon, like in
1347       release-changelog :4.2 4.2-rc25
1348   You can refer to the build trunk by just mentioning 'trunk', e.g.
1349       release-changelog -t coblitz-tags.mk coblitz-2.01-rc6 trunk
1350 """
1351     common_usage="""More help:
1352   see http://svn.planet-lab.org/wiki/ModuleTools"""
1353
1354     modes={
1355         'list' : "displays a list of available tags or branches",
1356         'version' : "check latest specfile and print out details",
1357         'diff' : "show difference between module (trunk or branch) and latest tag",
1358         'tag'  : """increment taglevel in specfile, insert changelog in specfile,
1359                 create new tag and and monitor its adoption in build/*-tags*.mk""",
1360         'branch' : """create a branch for this module, from the latest tag on the trunk,
1361                   and change trunk's version number to reflect the new branch name;
1362                   you can specify the new branch name by using module:branch""",
1363         'sync' : """create a tag from the module
1364                 this is a last resort option, mostly for repairs""",
1365         'changelog' : """extract changelog between build tags
1366                 expected arguments are a list of tags""",
1367         }
1368
1369     silent_modes = ['list']
1370     release_modes = ['changelog']
1371
1372     @staticmethod
1373     def optparse_list (option, opt, value, parser):
1374         try:
1375             setattr(parser.values,option.dest,getattr(parser.values,option.dest)+value.split())
1376         except:
1377             setattr(parser.values,option.dest,value.split())
1378
1379     def run(self):
1380
1381         mode=None
1382         for function in Main.modes.keys():
1383             if sys.argv[0].find(function) >= 0:
1384                 mode = function
1385                 break
1386         if not mode:
1387             print "Unsupported command",sys.argv[0]
1388             print "Supported commands:" + " ".join(Main.modes.keys())
1389             sys.exit(1)
1390
1391         if mode not in Main.release_modes:
1392             usage = Main.module_usage
1393             usage += Main.common_usage
1394             usage += "\nmodule-%s : %s"%(mode,Main.modes[mode])
1395         else:
1396             usage = Main.release_usage
1397             usage += Main.common_usage
1398
1399         parser=OptionParser(usage=usage)
1400
1401         if mode == "tag" or mode == 'branch':
1402             parser.add_option("-s","--set-version",action="store",dest="new_version",default=None,
1403                               help="set new version and reset taglevel to 0")
1404         if mode == "tag" :
1405             parser.add_option("-c","--no-changelog", action="store_false", dest="changelog", default=True,
1406                               help="do not update changelog section in specfile when tagging")
1407             parser.add_option("-b","--build-branch", action="store", dest="build_branch", default=None,
1408                               help="specify a build branch; used for locating the *tags*.mk files where adoption is to take place")
1409         if mode == "tag" or mode == "sync" :
1410             parser.add_option("-e","--editor", action="store", dest="editor", default=default_editor(),
1411                               help="specify editor")
1412
1413         if mode in ["diff","version"] :
1414             parser.add_option("-W","--www", action="store", dest="www", default=False,
1415                               help="export diff in html format, e.g. -W trunk")
1416
1417         if mode == "diff" :
1418             parser.add_option("-l","--list", action="store_true", dest="list", default=False,
1419                               help="just list modules that exhibit differences")
1420
1421         default_modules_list=os.path.dirname(sys.argv[0])+"/modules.list"
1422         parser.add_option("-a","--all",action="store_true",dest="all_modules",default=False,
1423                           help="run on all modules as found in %s"%default_modules_list)
1424         parser.add_option("-f","--file",action="store",dest="modules_list",default=None,
1425                           help="run on all modules found in specified file")
1426         parser.add_option("-n","--dry-run",action="store_true",dest="dry_run",default=False,
1427                           help="dry run - shell commands are only displayed")
1428         parser.add_option("-t","--distrotags",action="callback",callback=Main.optparse_list, dest="distrotags",
1429                           default=[], nargs=1,type="string",
1430                           help="""specify distro-tags files, e.g. onelab-tags-4.2.mk
1431 -- can be set multiple times, or use quotes""")
1432
1433         parser.add_option("-w","--workdir", action="store", dest="workdir",
1434                           default="%s/%s"%(os.getenv("HOME"),"modules"),
1435                           help="""name for dedicated working dir - defaults to ~/modules
1436 ** THIS MUST NOT ** be your usual working directory""")
1437         parser.add_option("-F","--fast-checks",action="store_true",dest="fast_checks",default=False,
1438                           help="skip safety checks, such as svn updates -- use with care")
1439
1440         # default verbosity depending on function - temp
1441         verbose_modes= ['tag', 'sync', 'branch']
1442
1443         if mode not in verbose_modes:
1444             parser.add_option("-v","--verbose", action="store_true", dest="verbose", default=False,
1445                               help="run in verbose mode")
1446         else:
1447             parser.add_option("-q","--quiet", action="store_false", dest="verbose", default=True,
1448                               help="run in quiet (non-verbose) mode")
1449         (options, args) = parser.parse_args()
1450         options.mode=mode
1451         if not hasattr(options,'dry_run'):
1452             options.dry_run=False
1453         if not hasattr(options,'www'):
1454             options.www=False
1455         options.debug=False
1456
1457         ########## module-*
1458         if len(args) == 0:
1459             if options.all_modules:
1460                 options.modules_list=default_modules_list
1461             if options.modules_list:
1462                 args=Command("grep -v '#' %s"%options.modules_list,options).output_of().split()
1463             else:
1464                 parser.print_help()
1465                 sys.exit(1)
1466         Module.init_homedir(options)
1467
1468
1469         if mode not in Main.release_modes:
1470             modules=[ Module(modname,options) for modname in args ]
1471             # hack: create a dummy Module to store errors/warnings
1472             error_module = Module('__errors__',options)
1473
1474             for module in modules:
1475                 if len(args)>1 and mode not in Main.silent_modes:
1476                     if not options.www:
1477                         print '========================================',module.friendly_name()
1478                 # call the method called do_<mode>
1479                 method=Module.__dict__["do_%s"%mode]
1480                 try:
1481                     method(module)
1482                 except Exception,e:
1483                     if options.www:
1484                         title='<span class="error"> Skipping module %s - failure: %s </span>'%\
1485                             (module.friendly_name(), str(e))
1486                         error_module.html_store_title(title)
1487                     else:
1488                         import traceback
1489                         traceback.print_exc()
1490                         print 'Skipping module %s: '%modname,e
1491
1492             if options.www:
1493                 if mode == "diff":
1494                     modetitle="Changes to tag in %s"%options.www
1495                 elif mode == "version":
1496                     modetitle="Latest tags in %s"%options.www
1497                 modules.append(error_module)
1498                 error_module.html_dump_header(modetitle)
1499                 for module in modules:
1500                     module.html_dump_toc()
1501                 Module.html_dump_middle()
1502                 for module in modules:
1503                     module.html_dump_body()
1504                 Module.html_dump_footer()
1505         else:
1506             release_changelog(options, *args)
1507
1508
1509 ####################
1510 if __name__ == "__main__" :
1511     try:
1512         Main().run()
1513     except KeyboardInterrupt:
1514         print '\nBye'