class Solargraph::Arc::Storage

Public Class Methods

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

Public Instance Methods

process(source_map, ns) click to toggle source
# File lib/solargraph/arc/storage.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, :has_one_attached] do |ast|
    name = ast.children[2].children.first

    pins << Util.build_public_method(
      ns,
      name.to_s,
      types: ["ActiveStorage::Attached::One"],
      location: Util.build_location(ast, ns.filename)
    )
  end

  walker.on :send, [nil, :has_many_attached] do |ast|
    name = ast.children[2].children.first

    pins << Util.build_public_method(
      ns,
      name.to_s,
      types: ["ActiveStorage::Attached::Many"],
      location: Util.build_location(ast, ns.filename)
    )
  end

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