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 import itertools
  20
  21 from cvs2svn_lib.symbol import Trunk
  22 from cvs2svn_lib.cvs_item import CVSRevisionDelete
  23 from cvs2svn_lib.cvs_item import CVSSymbol
  24 from cvs2svn_lib.revision_manager import RevisionCollector
  25 from cvs2svn_lib.key_generator import KeyGenerator
  26
  27
  28 class GitRevisionCollector(RevisionCollector):
  29   """Output file revisions to git-fast-import."""
  30
  31   def __init__(self, blob_filename, revision_reader):
  32     self.blob_filename = blob_filename
  33     self.revision_reader = revision_reader
  34
  35   def register_artifacts(self, which_pass):
  36     self.revision_reader.register_artifacts(which_pass)
  37
  38   def start(self):
  39     self.revision_reader.start()
  40     self.dump_file = open(self.blob_filename, 'wb')
  41     self._mark_generator = KeyGenerator()
  42
  43   def _process_revision(self, cvs_rev):
  44     """Write the revision fulltext to a blob if it is not dead."""
  45
  46     if isinstance(cvs_rev, CVSRevisionDelete):
  47       # There is no need to record a delete revision, and its token
  48       # will never be needed:
  49       return
  50
  51     # FIXME: We have to decide what to do about keyword substitution
  52     # and eol_style here:
  53     fulltext = self.revision_reader.get_content(
  54         cvs_rev, suppress_keyword_substitution=False,
  55         )
  56
  57     mark = self._mark_generator.gen_id()
  58     self.dump_file.write('blob\n')
  59     self.dump_file.write('mark :%d\n' % (mark,))
  60     self.dump_file.write('data %d\n' % (len(fulltext),))
  61     self.dump_file.write(fulltext)
  62     self.dump_file.write('\n')
  63     cvs_rev.revision_reader_token = mark
  64
  65   def _process_symbol(self, cvs_symbol, cvs_file_items):
  66     """Record the original source of CVS_SYMBOL.
  67
  68     Determine the original revision source of CVS_SYMBOL, and store it
  69     as the symbol's revision_reader_token."""
  70
  71     cvs_source = cvs_symbol.get_cvs_revision_source(cvs_file_items)
  72     cvs_symbol.revision_reader_token = cvs_source.revision_reader_token
  73
  74   def process_file(self, cvs_file_items):
  75     for lod_items in cvs_file_items.iter_lods():
  76       for cvs_rev in lod_items.cvs_revisions:
  77         self._process_revision(cvs_rev)
  78
  79     # Now that all CVSRevisions' revision_reader_tokens are set,
  80     # iterate through symbols and set their tokens to those of their
  81     # original source revisions:
  82     for lod_items in cvs_file_items.iter_lods():
  83       if lod_items.cvs_branch is not None:
  84         self._process_symbol(lod_items.cvs_branch, cvs_file_items)
  85       for cvs_tag in lod_items.cvs_tags:
  86         self._process_symbol(cvs_tag, cvs_file_items)
  87
  88   def finish(self):
  89     self.revision_reader.finish()
  90     self.dump_file.close()
  91
  92