class Solargraph::Arc::Devise

Public Class Methods

instance() click to toggle source
# File lib/solargraph/arc/devise.rb, line 4
def self.instance
  @instance ||= self.new
end
new() click to toggle source
# File lib/solargraph/arc/devise.rb, line 8
def initialize
  @seen_devise_closures = []
end

Public Instance Methods

process(source_map, ns) click to toggle source
# File lib/solargraph/arc/devise.rb, line 12
def process(source_map, ns)
  if source_map.filename.include?("app/models")
    process_model(source_map, ns)
  elsif source_map.filename.end_with?("app/controllers/application_controller.rb")
    process_controller(source_map, ns)
  else
    []
  end
end

Private Instance Methods

process_controller(source_map, ns) click to toggle source
# File lib/solargraph/arc/devise.rb, line 49
def process_controller(source_map, ns)
  pins = [
    Util.build_module_include(
      ns,
      "Devise::Controllers::Helpers",
      Util.dummy_location(ns.filename)
    )
  ]

  mapping_pins = @seen_devise_closures.map do |model_ns|
    ast = Walker.normalize_ast(source_map.source)
    mapping = model_ns.name.underscore

    [
      Util.build_public_method(
        ns,
        "authenticate_#{mapping}!",
        location: Util.build_location(ast, ns.filename)
      ),
      Util.build_public_method(
        ns,
        "#{mapping}_signed_in?",
        types: ["true", "false"],
        location: Util.build_location(ast, ns.filename)
      ),
      Util.build_public_method(
        ns,
        "current_#{mapping}",
        types: [model_ns.name, "nil"],
        location: Util.build_location(ast, ns.filename)
      ),
      Util.build_public_method(
        ns,
        "#{mapping}_session",
        location: Util.build_location(ast, ns.filename)
      )
    ]
  end.flatten

  pins += mapping_pins
  Solargraph.logger.debug("[ARC][Devise] added #{pins.map(&:name)} to #{ns.path}") if pins.any?
  pins
end
process_model(source_map, ns) click to toggle source
# File lib/solargraph/arc/devise.rb, line 24
def process_model(source_map, ns)
  walker = Walker.from_source(source_map.source)
  pins   = []

  walker.on :send, [nil, :devise] do |ast|
    @seen_devise_closures << ns

    modules = ast.children[2..-1]
      .map {|c| c.children.first }
      .select {|s| s.is_a?(Symbol) }

    modules.each do |mod|
      pins << Util.build_module_include(
        ns,
        "Devise::Models::#{mod.to_s.capitalize}",
        Util.build_location(ast, ns.filename)
      )
    end
  end

  walker.walk
  Solargraph.logger.debug("[ARC][Devise] added #{pins.map(&:name)} to #{ns.path}") if pins.any?
  pins
end