1 # (Be in -*- python -*- mode.)
3 # ====================================================================
4 # Copyright (c) 2007-2009 CollabNet. All rights reserved.
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.
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 # ====================================================================
17 """Write file contents to a stream of git-fast-import blobs."""
19 from cvs2svn_lib
.cvs_item
import CVSRevisionDelete
20 from cvs2svn_lib
.revision_manager
import RevisionCollector
21 from cvs2svn_lib
.key_generator
import KeyGenerator
24 class GitRevisionCollector(RevisionCollector
):
25 """Output file revisions to git-fast-import."""
27 def __init__(self
, blob_filename
, revision_reader
):
28 self
.blob_filename
= blob_filename
29 self
.revision_reader
= revision_reader
31 def register_artifacts(self
, which_pass
):
32 self
.revision_reader
.register_artifacts(which_pass
)
35 self
.revision_reader
.start()
36 self
.dump_file
= open(self
.blob_filename
, 'wb')
37 self
._mark
_generator
= KeyGenerator()
39 def _process_revision(self
, cvs_rev
):
40 """Write the revision fulltext to a blob if it is not dead."""
42 if isinstance(cvs_rev
, CVSRevisionDelete
):
43 # There is no need to record a delete revision, and its token
44 # will never be needed:
47 # FIXME: We have to decide what to do about keyword substitution
49 fulltext
= self
.revision_reader
.get_content(cvs_rev
)
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
59 def _process_symbol(self
, cvs_symbol
, cvs_file_items
):
60 """Record the original source of CVS_SYMBOL.
62 Determine the original revision source of CVS_SYMBOL, and store it
63 as the symbol's revision_reader_token."""
65 cvs_source
= cvs_symbol
.get_cvs_revision_source(cvs_file_items
)
66 cvs_symbol
.revision_reader_token
= cvs_source
.revision_reader_token
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
)
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
)
83 self
.revision_reader
.finish()
84 self
.dump_file
.close()