class Solargraph::Arc::Relation

Public Class Methods

instance() click to toggle source
# File lib/solargraph/arc/relation.rb, line 4
def self.instance
  @instance ||= self.new
end

Public Instance Methods

plural_association(ns, ast) click to toggle source

TODO: handle custom class for relation

# File lib/solargraph/arc/relation.rb, line 36
def plural_association(ns, ast)
  relation_name = ast.children[2].children.first

  Util.build_public_method(
    ns,
    relation_name.to_s,
    types: ["ActiveRecord::Associations::CollectionProxy<#{relation_name.to_s.singularize.camelize}>"],
    location: Util.build_location(ast, ns.filename)
  )
end
process(source_map, ns) click to toggle source
# File lib/solargraph/arc/relation.rb, line 8
def process(source_map, ns)
  return [] unless source_map.filename.include?("app/models")

  walker = Walker.from_source(source_map.source)
  pins   = []

  walker.on :send, [nil, :belongs_to] do |ast|
    pins << singular_association(ns, ast)
  end

  walker.on :send, [nil, :has_one] do |ast|
    pins << singular_association(ns, ast)
  end

  walker.on :send, [nil, :has_many] do |ast|
    pins << plural_association(ns, ast)
  end

  walker.on :send, [nil, :has_and_belongs_to_many] do |ast|
    pins << plural_association(ns, ast)
  end

  walker.walk
  Solargraph.logger.debug("[ARC][Relation] added #{pins.map(&:name)} to #{ns.path}") if pins.any?
  pins
end
singular_association(ns, ast) click to toggle source

TODO: handle custom class for relation

# File lib/solargraph/arc/relation.rb, line 48
def singular_association(ns, ast)
  relation_name = ast.children[2].children.first

  Util.build_public_method(
    ns,
    relation_name.to_s,
    types: [relation_name.to_s.camelize],
    location: Util.build_location(ast, ns.filename)
  )
end