Class: Console::Resolver

Inherits:
Object
  • Object
show all
Defined in:
lib/console/resolver.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Resolver 



76
77
78
79
80
 
# File 'lib/console/resolver.rb', line 76

def initialize
  @names = {}
  
  @trace_point = TracePoint.new(:class, &self.method(:resolve))
end

Class Method Details

.default_resolver(logger, env = ENV) ⇒ Object

You can change the log level for different classes using CONSOLE_<LEVEL> env vars.

e.g. ‘CONSOLE_WARN=Acorn,Banana CONSOLE_DEBUG=Cat` will set the log level for the classes Acorn and Banana to `warn` and Cat to `debug`. This overrides the default log level.

You can enable all log levels for a given class by using ‘CONSOLE_ON=MyClass`. Similarly you can disable all logging using `CONSOLE_OFF=MyClass`.



36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
 
# File 'lib/console/resolver.rb', line 36

def self.default_resolver(logger, env = ENV)
  # Find all CONSOLE_<LEVEL> variables from environment:
  levels = logger.class::LEVELS
    .map{|label, level| [level, env["CONSOLE_#{label.upcase}"]&.split(',')]}
    .to_h
    .compact
  
  off_klasses = env['CONSOLE_OFF']&.split(',')
  on_klasses = env['CONSOLE_ON']&.split(',')
  
  resolver = nil
  
  # If we have any levels, then create a class resolver, and each time a class is resolved, set the log level for that class to the specified level:
  if on_klasses&.any?
    resolver ||= Resolver.new
    
    resolver.bind(on_klasses) do |klass|
      logger.enable(klass, logger.class::MINIMUM_LEVEL - 1)
    end
  end
  
  if off_klasses&.any?
    resolver ||= Resolver.new
    
    resolver.bind(off_klasses) do |klass|
      logger.disable(klass)
    end
  end
  
  levels.each do |level, names|
    resolver ||= Resolver.new
    
    resolver.bind(names) do |klass|
      logger.enable(klass, level)
    end
  end
  
  return resolver
end

Instance Method Details

#bind(names, &block) ⇒ Object 



82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
 
# File 'lib/console/resolver.rb', line 82

def bind(names, &block)
  names.each do |name|
    if klass = Object.const_get(name) rescue nil
      yield klass
    else
      @names[name] = block
    end
  end
  
  if @names.any?
    @trace_point.enable
  else
    @trace_point.disable
  end
end

#resolve(trace_point) ⇒ Object 



102
103
104
105
106
107
108
109
110
 
# File 'lib/console/resolver.rb', line 102

def resolve(trace_point)
  if block = @names.delete(trace_point.self.to_s)
    block.call(trace_point.self)
  end
  
  if @names.empty?
    @trace_point.disable
  end
end

#waiting? ⇒ Boolean 



98
99
100
 
# File 'lib/console/resolver.rb', line 98

def waiting?
  @trace_point.enabled?
end