Dry API modules loading

Author: n-rodriguez

lib/netbox_client_ruby.rb

@@ -68,6 +68,16 @@ module NetboxClientRuby
     setting :ssl_options, default: { verify: true }
   end
 
+  def self.load_collection(collection, method_name, class_name)
+    collection.define_method(method_name) { class_name.new }
+    collection.__send__(:module_function, method_name)
+  end
+
+  def self.load_entity(entity, method_name, class_name)
+    entity.define_method(method_name) { |id| class_name.new id }
+    entity.__send__(:module_function, method_name)
+  end
+
   def self.circuits
     NetboxClientRuby::Circuits
   end

lib/netbox_client_ruby/api/circuits.rb

@@ -8,8 +8,7 @@ module Circuits
       circuit_types: CircuitTypes,
       circuit_terminations: CircuitTerminations,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { class_name.new }
-      module_function(method_name)
+      NetboxClientRuby.load_collection(self, method_name, class_name)
     end
 
     {
@@ -18,8 +17,7 @@ module Circuits
       circuit_type: CircuitType,
       circuit_termination: CircuitTermination,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { |id| class_name.new id }
-      module_function(method_name)
+      NetboxClientRuby.load_entity(self, method_name, class_name)
     end
   end
 end

lib/netbox_client_ruby/api/dcim.rb

@@ -26,8 +26,7 @@ module DCIM
       sites: Sites,
       virtual_chassis_list: VirtualChassisList,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { class_name.new }
-      module_function(method_name)
+      NetboxClientRuby.load_collection(self, method_name, class_name)
     end
 
     {
@@ -54,8 +53,7 @@ module DCIM
       site: Site,
       virtual_chassis: VirtualChassis,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { |id| class_name.new id }
-      module_function(method_name)
+      NetboxClientRuby.load_entity(self, method_name, class_name)
     end
   end
 end

lib/netbox_client_ruby/api/extras.rb

@@ -7,17 +7,15 @@ module Extras
       journal_entries: JournalEntries,
       tags: Tags,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { class_name.new }
-      module_function(method_name)
+      NetboxClientRuby.load_collection(self, method_name, class_name)
     end
 
     {
       config_context: ConfigContext,
       journal_entry: JournalEntry,
       tag: Tag,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { |id| class_name.new id }
-      module_function(method_name)
+      NetboxClientRuby.load_entity(self, method_name, class_name)
     end
   end
 end

lib/netbox_client_ruby/api/ipam.rb

@@ -14,8 +14,7 @@ module IPAM
       vlan_groups: VlanGroups,
       vrfs: Vrfs,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { class_name.new }
-      module_function(method_name)
+      NetboxClientRuby.load_collection(self, method_name, class_name)
     end
 
     {
@@ -30,8 +29,7 @@ module IPAM
       vlan_group: VlanGroup,
       vrf: Vrf,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { |id| class_name.new id }
-      module_function(method_name)
+      NetboxClientRuby.load_entity(self, method_name, class_name)
     end
   end
 end

lib/netbox_client_ruby/api/secrets.rb

@@ -8,16 +8,14 @@ module Secrets
       generate_rsa_key_pair: RSAKeyPair,
       get_session_key: SessionKey,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { class_name.new }
-      module_function(method_name)
+      NetboxClientRuby.load_collection(self, method_name, class_name)
     end
 
     {
       secret_role: SecretRole,
       secret: Secret,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { |id| class_name.new id }
-      module_function(method_name)
+      NetboxClientRuby.load_entity(self, method_name, class_name)
     end
 
     def session_key=(session_key)

lib/netbox_client_ruby/api/tenancy.rb

@@ -8,8 +8,7 @@ module Tenancy
       contacts: Contacts,
       contact_groups: ContactGroups,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { class_name.new }
-      module_function(method_name)
+      NetboxClientRuby.load_collection(self, method_name, class_name)
     end
 
     {
@@ -18,8 +17,7 @@ module Tenancy
       contact: Contact,
       contact_group: ContactGroup,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { |id| class_name.new id }
-      module_function(method_name)
+      NetboxClientRuby.load_entity(self, method_name, class_name)
     end
   end
 end

lib/netbox_client_ruby/api/virtualization.rb

@@ -9,8 +9,7 @@ module Virtualization
       virtual_machines: VirtualMachines,
       interfaces: Interfaces,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { class_name.new }
-      module_function(method_name)
+      NetboxClientRuby.load_collection(self, method_name, class_name)
     end
 
     {
@@ -20,8 +19,7 @@ module Virtualization
       virtual_machine: VirtualMachine,
       interface: Interface,
     }.each_pair do |method_name, class_name|
-      define_method(method_name) { |id| class_name.new id }
-      module_function(method_name)
+      NetboxClientRuby.load_entity(self, method_name, class_name)
     end
   end
 end