class Ivar::CheckAllManager

def disable
  @mutex.synchronize do
    if @trace_point
      @trace_point.disable
      @trace_point = nil
    end
  end
end

def enable(project_root, &block)
  disable if @trace_point
  root_pathname = Pathname.new(project_root)
  @mutex.synchronize do
    # :end means "end of module or class definition" in TracePoint
    @trace_point = TracePoint.new(:end) do |tp|
      next unless tp.path
      file_path = Pathname.new(File.expand_path(tp.path))
      if file_path.to_s.start_with?(root_pathname.to_s)
        klass = tp.self
        next if klass.included_modules.include?(Ivar::Checked)
        klass.include(Ivar::Checked)
      end
    end
    @trace_point.enable
  end
  if block
    begin
      yield
    ensure
      disable
    end
  end
  nil
end

def enabled?
  @mutex.synchronize { !@trace_point.nil? && @trace_point.enabled? }
end

def initialize
  @trace_point = nil
  @mutex = Mutex.new
end

def trace_point
  @mutex.synchronize { @trace_point }
end