Sha256: 1caa4e2283cf44b25ee7fc992664252c87e58e2e521fce4fe6c2c968a76ff2cc
Contents?: true
Size: 1.67 KB
Versions: 8
Compression:
Stored size: 1.67 KB
Contents
# frozen_string_literal: true
module Orthoses
module ActiveRecord
class BelongsTo
def initialize(loader)
@loader = loader
end
def call
@loader.call.tap do |store|
::ActiveRecord::Base.descendants.each do |base|
next if base.abstract_class?
base_name = Orthoses::Utils.module_name(base) || next
lines = base.reflect_on_all_associations(:belongs_to).flat_map do |ref|
# FIXME: Can I get list of class for polymorphic?
type = ref.polymorphic? ? 'untyped' : ref.klass.to_s
opt = "#{type}?"
[
"def #{ref.name}: () -> #{opt}",
"def #{ref.name}=: (#{opt}) -> #{opt}",
"def reload_#{ref.name}: () -> #{opt}",
].tap do |ary|
if !ref.polymorphic?
ary << "def build_#{ref.name}: (?untyped attributes) ?{ (#{type}) -> void } -> #{type}"
ary << "def create_#{ref.name}: (?untyped attributes) ?{ (#{type}) -> void } -> #{type}"
ary << "def create_#{ref.name}!: (?untyped attributes) ?{ (#{type}) -> void } -> #{type}"
end
end
end
generated_association_methods = "#{base_name}::GeneratedAssociationMethods"
store[generated_association_methods].tap do |content|
content.header = "module #{generated_association_methods}"
content.concat(lines)
end
store[base_name].tap do |content|
store[base_name] << "include #{generated_association_methods}"
end
end
end
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems