module SimpleCov
  class SourceFile
    class Line
      def reinitialize_line(coverage)
        raise ArgumentError, "Only Integer and nil accepted for coverage" unless coverage.is_a?(Integer) || coverage.nil?
        @coverage    = coverage
        @skipped     = false
      end
    end
  end
end

SimpleCov.start 'rails' do
  add_filter do |source_file|
    if source_file.filename.include?("api/v1")
      swagger_lines = []
      source_file.lines.each do |line|
        if line.src.include? "param :" or line.src.include? "swagger_controller" or line.src.include? "swagger_api" or line.src.include? "ApplicationController::add_common_params(api)" or line.src.include? "summary" or line.src.include? "notes" or line.src.include? "response"
          swagger_lines << line
        end
      end

      swagger_lines.each do |line|
        line.reinitialize_line(nil)
      end
    end
  end
end