1# Copyright (C) 2011 Canonical Ltd
2#
3# This program is free software; you can redistribute it and/or modify
4# it under the terms of the GNU General Public License as published by
5# the Free Software Foundation; either version 2 of the License, or
6# (at your option) any later version.
7#
8# This program is distributed in the hope that it will be useful,
9# but WITHOUT ANY WARRANTY; without even the implied warranty of
10# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11# GNU General Public License for more details.
12#
13# You should have received a copy of the GNU General Public License
14# along with this program; if not, write to the Free Software
15# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
17from ... import (
18    controldir,
19    commands,
20    errors,
21    )
22
23
24class cmd_file_refs(commands.Command):
25    __doc__ = \
26        """Find the inventories that reference a particular version of a text."""
27
28    hidden = True
29    takes_args = ['file_id', 'rev_id']
30    takes_options = ['directory']
31
32    def run(self, file_id, rev_id, directory=u'.'):
33        file_id = file_id.encode()
34        rev_id = rev_id.encode()
35        bd, relpath = controldir.ControlDir.open_containing(directory)
36        repo = bd.find_repository()
37        self.add_cleanup(repo.lock_read().unlock)
38        inv_vf = repo.inventories
39        all_invs = [k[-1] for k in inv_vf.keys()]
40        # print len(all_invs)
41        for inv in repo.iter_inventories(all_invs, 'unordered'):
42            try:
43                entry = inv.get_entry(file_id)
44            except errors.NoSuchId:
45                # This file doesn't even appear in this inv.
46                continue
47            if entry.revision == rev_id:
48                self.outf.write(inv.revision_id + b'\n')
49