glark 1.9.0 → 1.10.0

data/lib/glark/app/rcfile.rb

@@ -0,0 +1,49 @@
+#!/usr/bin/ruby -w
+# -*- ruby -*-
+
+module Glark
+  class RCFile
+    COMMENT_RE = Regexp.new '\s*#.*'
+    NAME_VALUE_RE = Regexp.new '\s*[=:]\s*'
+    
+    def initialize file
+      @values = Array.new
+
+      pn = file.kind_of?(Pathname) ? file : Pathname.new(file)
+      
+      return unless pn.exist?
+
+      pn.each_line do |line|
+        read_line line
+      end
+    end
+
+    def read_line line
+      line.sub! COMMENT_RE, ''
+      line.chomp!
+      return if line.empty?
+      
+      name, value = line.split NAME_VALUE_RE
+      return unless name && value
+
+      add name, value
+    end
+
+    def names
+      @values.collect { |x| x[0] }
+    end
+
+    def values name
+      ary = @values.assoc name
+      ary && ary[1 .. -1]
+    end
+
+    def add name, value
+      if ary = @values.assoc(name)
+        ary << value
+      else
+        @values << [ name, value ]
+      end
+    end
+  end
+end

data/lib/glark/app/runner.rb

@@ -0,0 +1,140 @@
+#!/usr/bin/ruby -w
+#!ruby -w
+# vim: set filetype=ruby : set sw=2
+
+require 'rubygems'
+require 'riel/log/loggable'
+require 'glark/app/options'
+require 'glark/io/file/binary_file'
+require 'glark/io/file/gz_file'
+require 'glark/io/file/tar_file'
+require 'glark/io/file/tar_gz_file'
+require 'glark/io/file/zip_file'
+
+$stdout.sync = true             # unbuffer
+$stderr.sync = true             # unbuffer
+
+module Glark
+  # The main processor.
+  class Runner
+    include Loggable
+
+    GZ_RE = Regexp.new '\.gz$'
+    TAR_GZ_RE = Regexp.new '\.(?:tgz|tar\.gz)$'
+    TAR_RE = Regexp.new '\.tar$'
+    ZIP_RE = Regexp.new '\.(?:zip|jar)$'
+    
+    attr_reader :exit_status
+    
+    def initialize opts, files
+      @opts = opts
+      @expr = opts.match_spec.expr
+      @searched_files = Array.new          # files searched, so we don't cycle through links
+
+      @exclude_matching = @opts.input_spec.exclude_matching
+
+      @range = @opts.input_spec.range
+      @output_opts = @opts.output_options
+      @invert_match = @output_opts.invert_match
+
+      # 0 == matches, 1 == no matches, 2 == error
+      @exit_status = @invert_match ? 0 : 1
+
+      @output_type_cls = @output_opts.output_type_cls
+      
+      @opts.fileset.each do |type, file|
+        search type, file
+      end
+    end
+
+    def search_file file, output_type_cls = @output_type_cls
+      output_type = output_type_cls.new file, @output_opts
+      update_status file.search @expr, output_type
+    end
+
+    def update_status matched
+      if matched
+        @exit_status = @invert_match ? 1 : 0
+      end
+    end
+
+    def search_text fname, io
+      file = Glark::File.new fname, io, @range
+      search_file file
+    end
+
+    def search_binary fname
+      file = Glark::BinaryFile.new fname
+      update_status file.search_as_binary @expr, @output_opts
+    end
+
+    def search_read_archive_file fname, cls
+      @output_opts.show_file_names = true
+      file = cls.new fname, @range
+      update_status file.search @expr, @output_type_cls, @output_opts
+    end
+
+    def search_read fname
+      fstr = fname.to_s
+      
+      case
+      when TAR_GZ_RE.match(fstr)
+        search_read_archive_file fname, Glark::TarGzFile
+      when GZ_RE.match(fstr)
+        search_file Glark::GzFile.new(fname, @range)
+      when TAR_RE.match(fstr)
+        search_read_archive_file fname, Glark::TarFile
+      when ZIP_RE.match(fstr)
+        search_read_archive_file fname, Glark::ZipFile
+      else
+        write "file '#{fname}' does not have a handled extension"
+        return
+      end
+    end
+
+    def search_list fname
+      fstr = fname.to_s
+
+      cls = case
+            when TAR_RE.match(fstr)
+              Glark::TarFile
+            when ZIP_RE.match(fstr)
+              Glark::ZipFile
+            when TAR_GZ_RE.match(fstr)
+              Glark::TarGzFile
+            else
+              write "file '#{fname}' does not have a handled extension"
+              return
+            end
+
+      file = cls.new fname, @range
+
+      update_status file.search_list(@expr, @output_type_cls, @output_opts)
+    end
+    
+    def search type, name
+      if @exclude_matching
+        expr = @opts.match_spec.expr
+        return if expr.respond_to?(:re) && expr.re.match(name.to_s)
+      end
+      
+      if name == "-" 
+        write "reading standard input..."
+        search_text name, $stdin
+      else
+        case type
+        when :binary
+          search_binary name 
+        when :text
+          search_text name, ::File.new(name)
+        when :read
+          search_read name 
+        when :list
+          search_list name 
+        else
+          raise "type unknown: file: #{name}; type: #{type}"
+        end
+      end
+    end
+  end
+end

data/lib/glark/app/spec.rb

@@ -0,0 +1,19 @@
+#!/usr/bin/ruby -w
+# -*- ruby -*-
+
+module Glark
+  class AppSpec
+    attr_reader :local_config_files
+    
+    attr_reader :input_spec
+    attr_reader :match_spec
+    attr_reader :output_spec
+    
+    def initialize input_spec, match_spec, output_spec
+      @input_spec = input_spec
+      @match_spec = match_spec
+      @output_spec = output_spec
+      @local_config_files = false      # use local .glarkrc files    
+    end
+  end
+end

data/lib/glark/input/filter/criteria_opts.rb

@@ -0,0 +1,41 @@
+#!/usr/bin/ruby -w
+#!ruby -w
+# vim: set filetype=ruby : set sw=2
+
+require 'glark/util/io/filter/criteria'
+
+module Glark
+  class CriteriaOpts < Criteria
+    def initialize
+      super
+      @options = opt_classes.collect { |optcls| optcls.new self }
+    end
+
+    def opt_classes
+      Array.new
+    end
+
+    def config_fields
+      fields = {
+      }
+    end
+
+    def dump_fields
+      config_fields
+    end
+
+    def update_fields rcfields
+      rcfields.each do |name, values|
+        @options.each do |opt|
+          opt.match_rc name, values
+        end
+      end
+    end
+
+    def add_as_options optdata
+      @options.each do |opt|
+        opt.add_to_option_data optdata
+      end
+    end
+  end
+end

data/lib/glark/input/filter/dir_criteria_opts.rb

@@ -0,0 +1,37 @@
+#!/usr/bin/ruby -w
+#!ruby -w
+# vim: set filetype=ruby : set sw=2
+
+require 'glark/input/filter/criteria_opts'
+require 'glark/input/filter/options'
+require 'glark/util/optutil'
+
+module Glark
+  class DirCriteriaOpts < CriteriaOpts
+    include OptionUtil
+
+    attr_accessor :skip_all
+
+    def initialize skip_all
+      super()
+      @skip_all = skip_all
+      
+      add :name, :negative, BaseNameFilter.new('.svn')
+      add :name, :negative, BaseNameFilter.new('.git')
+    end
+
+    def skipped? pn, depth
+      return true if @skip_all || !depth.nonzero?
+      super pn
+    end
+
+    def opt_classes
+      [
+       MatchDirNameOption,
+       SkipDirNameOption,
+       MatchDirPathOption,
+       SkipDirPathOption,
+      ]
+    end
+  end
+end

data/lib/glark/input/filter/file_criteria_opts.rb

@@ -0,0 +1,33 @@
+#!/usr/bin/ruby -w
+#!ruby -w
+# vim: set filetype=ruby : set sw=2
+
+require 'glark/input/filter/criteria_opts'
+require 'glark/input/filter/filter'
+require 'glark/input/filter/options'
+require 'glark/util/optutil'
+
+module Glark
+  class FileCriteriaOpts < CriteriaOpts
+    include OptionUtil
+
+    def opt_classes
+      [
+       SizeLimitOption,
+       MatchNameOption,
+       SkipNameOption,
+       MatchPathOption,
+       SkipPathOption,
+       MatchExtOption,
+       SkipExtOption,
+      ]
+    end
+
+    def config_fields
+      maxsize = (filter = find_by_class(:size, :negative, SizeLimitFilter)) && filter.max_size
+      fields = {
+        "size-limit" => maxsize
+      }
+    end
+  end
+end

data/lib/glark/input/filter/filter.rb

@@ -0,0 +1,59 @@
+#!/usr/bin/ruby -w
+#!ruby -w
+# vim: set filetype=ruby : set sw=2
+
+require 'glark/util/io/filter/filter'
+
+module Glark
+  class PatternFilter < Filter
+    include Loggable
+
+    attr_reader :pattern
+
+    def initialize pattern
+      @pattern = pattern
+    end
+
+    def pattern_match? str
+      if @pattern.kind_of? String
+        @pattern == str
+      else
+        @pattern.match str
+      end
+    end
+
+    def to_s
+      @pattern.to_s
+    end
+  end
+
+  class BaseNameFilter < PatternFilter
+    def match? pn
+      pattern_match? pn.basename.to_s
+    end
+  end
+
+  class FullNameFilter < PatternFilter
+    def match? pn
+      pattern_match? pn.to_s
+    end
+  end
+
+  class SizeLimitFilter < Glark::Filter
+    attr_reader :max_size
+    
+    def initialize maxsize
+      @max_size = maxsize
+    end
+
+    def match? pn
+      pn.size > @max_size
+    end
+  end
+
+  class ExtFilter < PatternFilter
+    def match? pn
+      pattern_match? pn.extname.to_s[1 .. -1]
+    end
+  end
+end

data/lib/glark/input/filter/options.rb

@@ -0,0 +1,171 @@
+#!/usr/bin/ruby -w
+#!ruby -w
+# vim: set filetype=ruby : set sw=2
+
+require 'glark/input/filter/filter'
+require 'glark/util/option'
+
+module Glark
+  class SizeLimitOption < Option
+    def argtype
+      :integer
+    end
+
+    def posneg
+      :negative
+    end
+
+    def field
+      :size
+    end
+
+    def set val
+      @optee.add field, posneg, SizeLimitFilter.new(val.to_i)
+    end
+
+    def rcfield
+      'size-limit'
+    end
+  end
+
+  class RegexpOption < Glark::Option
+    def set val
+      @optee.add field, posneg, cls.new(Regexp.create val)
+    end
+
+    def argtype
+      :string
+    end
+  end
+
+  module MatchOption
+    def rcfield
+      'match-' + field.to_s
+    end
+
+    def posneg
+      :positive
+    end
+  end
+
+  module SkipOption
+    def tags
+      %w{ not skip }.collect { |x| '--' + x + '-' + field.to_s }
+    end
+
+    def rcfield
+      'skip-' + field.to_s
+    end
+
+    def posneg
+      :negative
+    end
+  end
+
+  class ExtOption < RegexpOption
+    def cls
+      ExtFilter
+    end
+
+    def field
+      :ext
+    end
+  end
+
+  class MatchExtOption < ExtOption
+    include MatchOption
+  end
+
+  class SkipExtOption < ExtOption
+    include SkipOption
+  end
+
+  class NameOption < RegexpOption
+    def cls
+      BaseNameFilter
+    end
+
+    def field
+      :name
+    end
+  end
+
+  class MatchNameOption < NameOption
+    include MatchOption
+
+    def tags
+      %w{ --basename --name --with-basename --with-name } + super
+    end
+  end
+
+  class SkipNameOption < NameOption
+    include SkipOption
+
+    def tags
+      %w{ --without-basename --without-name } + super
+    end
+  end
+
+  class DirNameOption < RegexpOption
+    def cls
+      BaseNameFilter
+    end
+
+    def field
+      :dirname
+    end
+  end
+
+  class MatchDirNameOption < DirNameOption
+    include MatchOption
+  end
+
+  class SkipDirNameOption < DirNameOption
+    include SkipOption
+  end
+
+  class PathOption < RegexpOption
+    def cls
+      FullNameFilter
+    end
+
+    def field
+      :path
+    end
+  end
+
+  class MatchPathOption < PathOption
+    include MatchOption
+
+    def tags
+      %w{ --fullname --path --with-fullname --with-path } + super
+    end
+  end
+
+  class SkipPathOption < PathOption
+    include SkipOption
+
+    def tags
+      %w{ --without-fullname --without-path } + super
+    end
+  end
+
+
+  class DirPathOption < RegexpOption
+    def cls
+      FullNameFilter
+    end
+
+    def field
+      :dirpath
+    end
+  end
+
+  class MatchDirPathOption < DirPathOption
+    include MatchOption
+  end
+
+  class SkipDirPathOption < DirPathOption
+    include SkipOption
+  end
+end