cvs2svn_lib/git_revision_collector.py

   1 # (Be in -*- python -*- mode.)
   2 #
   3 # ====================================================================
   4 # Copyright (c) 2007-2009 CollabNet.  All rights reserved.
   5 #
   6 # This software is licensed as described in the file COPYING, which
   7 # you should have received as part of this distribution.  The terms
   8 # are also available at http://subversion.tigris.org/license-1.html.
   9 # If newer versions of this license are posted there, you may use a
  10 # newer version instead, at your option.
  11 #
  12 # This software consists of voluntary contributions made by many
  13 # individuals.  For exact contribution history, see the revision
  14 # history and logs, available at http://cvs2svn.tigris.org/.
  15 # ====================================================================
  16
  17 """Write file contents to a stream of git-fast-import blobs."""
  18
  19 from cvs2svn_lib.cvs_item import CVSRevisionDelete
  20 from cvs2svn_lib.revision_manager import RevisionCollector
  21 from cvs2svn_lib.key_generator import KeyGenerator
  22
  23
  24 class GitRevisionCollector(RevisionCollector):
  25   """Output file revisions to git-fast-import."""
  26
  27   def __init__(self, blob_filename, revision_reader):
  28     self.blob_filename = blob_filename
  29     self.revision_reader = revision_reader
  30
  31   def register_artifacts(self, which_pass):
  32     self.revision_reader.register_artifacts(which_pass)
  33
  34   def start(self):
  35     self.revision_reader.start()
  36     self.dump_file = open(self.blob_filename, 'wb')
  37     self._mark_generator = KeyGenerator()
  38
  39   def _process_revision(self, cvs_rev):
  40     """Write the revision fulltext to a blob if it is not dead."""
  41
  42     if isinstance(cvs_rev, CVSRevisionDelete):
  43       # There is no need to record a delete revision, and its token
  44       # will never be needed:
  45       return
  46
  47     # FIXME: We have to decide what to do about keyword substitution
  48     # and eol_style here:
  49     fulltext = self.revision_reader.get_content(cvs_rev)
  50
  51     mark = self._mark_generator.gen_id()
  52     self.dump_file.write('blob\n')
  53     self.dump_file.write('mark :%d\n' % (mark,))
  54     self.dump_file.write('data %d\n' % (len(fulltext),))
  55     self.dump_file.write(fulltext)
  56     self.dump_file.write('\n')
  57     cvs_rev.revision_reader_token = mark
  58
  59   def _process_symbol(self, cvs_symbol, cvs_file_items):
  60     """Record the original source of CVS_SYMBOL.
  61
  62     Determine the original revision source of CVS_SYMBOL, and store it
  63     as the symbol's revision_reader_token."""
  64
  65     cvs_source = cvs_symbol.get_cvs_revision_source(cvs_file_items)
  66     cvs_symbol.revision_reader_token = cvs_source.revision_reader_token
  67
  68   def process_file(self, cvs_file_items):
  69     for lod_items in cvs_file_items.iter_lods():
  70       for cvs_rev in lod_items.cvs_revisions:
  71         self._process_revision(cvs_rev)
  72
  73     # Now that all CVSRevisions' revision_reader_tokens are set,
  74     # iterate through symbols and set their tokens to those of their
  75     # original source revisions:
  76     for lod_items in cvs_file_items.iter_lods():
  77       if lod_items.cvs_branch is not None:
  78         self._process_symbol(lod_items.cvs_branch, cvs_file_items)
  79       for cvs_tag in lod_items.cvs_tags:
  80         self._process_symbol(cvs_tag, cvs_file_items)
  81
  82   def finish(self):
  83     self.revision_reader.finish()
  84     self.dump_file.close()
  85
  86