Sha256: ad372b429b927f01f4f1243be47b4a44fe2606955f992c75ef47b3ba420824fd

Contents?: true

Size: 1.46 KB

Versions: 6

Compression:

Stored size: 1.46 KB

Contents

=begin
    Copyright 2010-2015 Tasos Laskos <tasos.laskos@arachni-scanner.com>

    This file is part of the Arachni Framework project and is subject to
    redistribution and commercial restrictions. Please see the Arachni Framework
    web site for more information on licensing and terms of use.
=end

module Arachni
module Component

# Includes some useful methods for the components.
#
# @author Tasos "Zapotek" Laskos <tasos.laskos@arachni-scanner.com>
module Utilities
    include Arachni::Utilities

    # @param    [String]    filename
    #   Filename, without the path.
    # @param    [Block]     block
    #   The block to be passed each line as it's read.
    def read_file( filename, &block )
        component_path = block_given? ?
            block.source_location.first : caller_path(1)

        # The name of the component that called us.
        component_name = File.basename( component_path, '.rb' )

        # The path to the component's data file directory.
        path  = File.expand_path( File.dirname( component_path ) ) +
            "/#{component_name}/"

        File.open( "#{path}/#{filename}" ) do |file|
            if block_given?
                # I really hope that ruby frees each line as soon as possible
                # otherwise this provides no advantage
                file.each { |line| yield line.strip }
            else
                file.read.lines.map { |l| l.strip }
            end
        end
    end

    extend self

end

end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version Path
arachni-1.3.2 lib/arachni/component/utilities.rb
arachni-1.3.1 lib/arachni/component/utilities.rb
arachni-1.3 lib/arachni/component/utilities.rb
arachni-1.2.1 lib/arachni/component/utilities.rb
arachni-1.2 lib/arachni/component/utilities.rb
arachni-1.1 lib/arachni/component/utilities.rb