lib/netbox_client_ruby/api/ipam/prefix.rb in netbox-client-ruby-0.0.4 vs lib/netbox_client_ruby/api/ipam/prefix.rb in netbox-client-ruby-0.1.0

@@ -6,44 +6,46 @@
 require 'netbox_client_ruby/api/ipam/vlan_group'
 require 'netbox_client_ruby/api/tenancy/tenant'
 require 'ipaddress'
 
 module NetboxClientRuby
-  class Prefix
-    include NetboxClientRuby::Entity
+  module IPAM
+    class Prefix
+      include Entity
 
-    id id: :id
-    deletable true
-    path 'ipam/prefixes/:id.json'
-    creation_path 'ipam/prefixes/'
-    object_fields(
-      site: proc { |raw_data| NetboxClientRuby::Site.new raw_data['id'] },
-      vrf: proc { |raw_data| NetboxClientRuby::Vrf.new raw_data['id'] },
-      tenant: proc { |raw_data| NetboxClientRuby::Tenant.new raw_data['id'] },
-      vlan: proc { |raw_data| NetboxClientRuby::Vlan.new raw_data['id'] },
-      status: proc { |raw_data| NetboxClientRuby::PrefixStatus.new raw_data['value'] },
-      role: proc { |raw_data| NetboxClientRuby::Role.new raw_data['id'] },
-      prefix: proc { |raw_data| IPAddress.parse raw_data }
-    )
-    readonly_fields :display_name
-  end
+      id id: :id
+      deletable true
+      path 'ipam/prefixes/:id.json'
+      creation_path 'ipam/prefixes/'
+      object_fields(
+        site: proc { |raw_data| DCIM::Site.new raw_data['id'] },
+        vrf: proc { |raw_data| Vrf.new raw_data['id'] },
+        tenant: proc { |raw_data| Tenancy::Tenant.new raw_data['id'] },
+        vlan: proc { |raw_data| Vlan.new raw_data['id'] },
+        status: proc { |raw_data| PrefixStatus.new raw_data['value'] },
+        role: proc { |raw_data| Role.new raw_data['id'] },
+        prefix: proc { |raw_data| IPAddress.parse raw_data }
+      )
+      readonly_fields :display_name
+    end
 
-  class PrefixStatus
-    attr_reader :value, :label
+    class PrefixStatus
+      attr_reader :value, :label
 
-    def initialize(status_value)
-      @value = status_value
-      @label = case status_value
-               when 0 then
-                 'Container'.freeze
-               when 1 then
-                 'Active'.freeze
-               when 2 then
-                 'Reserved'.freeze
-               when 3 then
-                 'Deprecated'.freeze
-               else
-                 'UNDEFINED'.freeze
-               end
+      def initialize(status_value)
+        @value = status_value
+        @label = case status_value
+                 when 0 then
+                   'Container'.freeze
+                 when 1 then
+                   'Active'.freeze
+                 when 2 then
+                   'Reserved'.freeze
+                 when 3 then
+                   'Deprecated'.freeze
+                 else
+                   'UNDEFINED'.freeze
+                 end
+      end
     end
   end
 end