/brz/remove-bazaar

# Copyright (C) 2010 Parth Malwankar <parth.malwankar@gmail.com>

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

"""bzr grep"""

import os

from bzrlib.commands import Command, register_command, display_command

from bzrlib.option import (

    Option,

    )

from bzrlib.lazy_import import lazy_import

lazy_import(globals(), """

import bzrlib

from bzrlib import (

    errors,

    bzrdir,

    )

""")

version_info = (0, 1)

class cmd_grep(Command):

    """Print lines matching PATTERN for specified files.

    """

    takes_args = ['pattern', 'path*']

    takes_options = [

        'verbose',

        Option('line-number', short_name='n',

               help='prefix each line of output with 1-based line number.'),

        Option('null', short_name='z',

               help='Write an ascii NUL (\\0) separator '

               'between files rather than a newline.'),

        Option('ignore-case', short_name='i',

               help='ignore case distinctions while matching.'),

        Option('recursive', short_name='R',

               help='Recurse into subdirectories.'),

        Option('from-root',

               help='Search for pattern starting from the root of the branch.'),

        ]

    @display_command

    def run(self, verbose=False, line_number=False, null=False,

            ignore_case=False, recursive=False, from_root=False,

            path_list=None, pattern=None):

        if path_list == None:

            path_list = ['.']

        else:

            if from_root:

                raise errors.BzrCommandError('cannot specify both --from-root and PATH.')

        print 'pattern:', pattern

        print 'path_list:', path_list

        print 'ignore-case:', ignore_case

        print 'line-number:', line_number

        print 'null:', null

        print 'recursive:', recursive

        print 'from-root:', from_root

        print '=' * 20

        tree, branch, relpath = bzrdir.BzrDir.open_containing_tree_or_branch('.')

        tree.lock_read()

        self.add_cleanup(tree.unlock)

        for path in path_list:

            for fp, fc, fkind, fid, entry in tree.list_files(include_root=False,

                from_dir=relpath, recursive=recursive):

                print 'fp:', fp

                print 'fc:', fc

                print 'fkind:', fkind

                print 'fid:', fid

                print 'entry:', entry

                if fc == 'V' and fkind == 'file':

                    self.file_grep(fp)

                print '~' * 30

    def file_grep(self, path):

        print path + ': grepping ...'

register_command(cmd_grep)