Sha256: fc0fe1b34f8c20a0ab915f74e441aee853cb867f84be0cec1d71d45b2d95a347
Contents?: true
Size: 1.25 KB
Versions: 73
Compression:
Stored size: 1.25 KB
Contents
module Brakeman
class FileTypeDetector < BaseProcessor
def initialize
super(nil)
reset
end
def detect_type(file)
reset
process(file.ast)
if @file_type.nil?
@file_type = guess_from_path(file.path.relative)
end
@file_type || :libs
end
MODEL_CLASSES = [
:'ActiveRecord::Base',
:ApplicationRecord
]
def process_class exp
name = class_name(exp.class_name)
parent = class_name(exp.parent_name)
if name.match(/Controller$/)
@file_type = :controllers
return exp
elsif MODEL_CLASSES.include? parent
@file_type = :models
return exp
end
super
end
def guess_from_path path
case
when path.include?('app/models')
:models
when path.include?('app/controllers')
:controllers
when path.include?('config/initializers')
:initializers
when path.include?('lib/')
:libs
when path.match?(%r{config/environments/(?!production\.rb)$})
:skip
when path.match?(%r{environments/production\.rb$})
:skip
when path.match?(%r{application\.rb$})
:skip
end
end
private
def reset
@file_type = nil
end
end
end
Version data entries
73 entries across 73 versions & 3 rubygems