]> git.donarmstrong.com Git - dsa-puppet.git/blobdiff - 3rdparty/modules/neutron/lib/puppet/provider/neutron_subnet/neutron.rb
try again, with puppetforge modules, correctly included now
[dsa-puppet.git] / 3rdparty / modules / neutron / lib / puppet / provider / neutron_subnet / neutron.rb
diff --git a/3rdparty/modules/neutron/lib/puppet/provider/neutron_subnet/neutron.rb b/3rdparty/modules/neutron/lib/puppet/provider/neutron_subnet/neutron.rb
new file mode 100644 (file)
index 0000000..0da2180
--- /dev/null
@@ -0,0 +1,219 @@
+require File.join(File.dirname(__FILE__), '..','..','..',
+                  'puppet/provider/neutron')
+
+Puppet::Type.type(:neutron_subnet).provide(
+  :neutron,
+  :parent => Puppet::Provider::Neutron
+) do
+  desc <<-EOT
+    Neutron provider to manage neutron_subnet type.
+
+    Assumes that the neutron service is configured on the same host.
+  EOT
+
+  commands :neutron => 'neutron'
+
+  mk_resource_methods
+
+  def self.neutron_type
+    'subnet'
+  end
+
+  def self.instances
+    list_neutron_resources(neutron_type).collect do |id|
+      attrs = get_neutron_resource_attrs(neutron_type, id)
+      new(
+        :ensure                    => :present,
+        :name                      => attrs['name'],
+        :id                        => attrs['id'],
+        :cidr                      => attrs['cidr'],
+        :ip_version                => attrs['ip_version'],
+        :gateway_ip                => parse_gateway_ip(attrs['gateway_ip']),
+        :allocation_pools          => parse_allocation_pool(attrs['allocation_pools']),
+        :host_routes               => parse_host_routes(attrs['host_routes']),
+        :dns_nameservers           => parse_dns_nameservers(attrs['dns_nameservers']),
+        :enable_dhcp               => attrs['enable_dhcp'],
+        :network_id                => attrs['network_id'],
+        :tenant_id                 => attrs['tenant_id']
+      )
+    end
+  end
+
+  def self.prefetch(resources)
+    subnets = instances
+    resources.keys.each do |name|
+      if provider = subnets.find{ |subnet| subnet.name == name }
+        resources[name].provider = provider
+      end
+    end
+  end
+
+  def self.parse_gateway_ip(value)
+    return '' if value.nil?
+    return value
+  end
+
+  def self.parse_allocation_pool(values)
+    allocation_pools = []
+    return [] if values.empty?
+    for value in Array(values)
+      matchdata = /\{\s*"start"\s*:\s*"(.*)"\s*,\s*"end"\s*:\s*"(.*)"\s*\}/.match(value.gsub(/\\"/,'"'))
+      start_ip = matchdata[1]
+      end_ip = matchdata[2]
+      allocation_pools << "start=#{start_ip},end=#{end_ip}"
+    end
+    return allocation_pools
+  end
+
+  def self.parse_host_routes(values)
+    host_routes = []
+    return [] if values.empty?
+    for value in Array(values)
+      matchdata = /\{\s*"destination"\s*:\s*"(.*)"\s*,\s*"nexthop"\s*:\s*"(.*)"\s*\}/.match(value.gsub(/\\"/,'"'))
+      destination = matchdata[1]
+      nexthop = matchdata[2]
+      host_routes << "destination=#{destination},nexthop=#{nexthop}"
+    end
+    return host_routes
+  end
+
+  def self.parse_dns_nameservers(values)
+    # just enforce that this is actually an array
+    return Array(values)
+  end
+
+  def exists?
+    @property_hash[:ensure] == :present
+  end
+
+  def create
+    opts = ["--name=#{@resource[:name]}"]
+
+    if @resource[:ip_version]
+      opts << "--ip-version=#{@resource[:ip_version]}"
+    end
+
+    if @resource[:gateway_ip]
+      if @resource[:gateway_ip] == ''
+        opts << '--no-gateway'
+      else
+        opts << "--gateway-ip=#{@resource[:gateway_ip]}"
+      end
+    end
+
+    if @resource[:enable_dhcp] == 'False'
+      opts << "--disable-dhcp"
+    else
+      opts << "--enable-dhcp"
+    end
+
+    if @resource[:allocation_pools]
+      Array(@resource[:allocation_pools]).each do |allocation_pool|
+        opts << "--allocation-pool=#{allocation_pool}"
+      end
+    end
+
+    if @resource[:dns_nameservers]
+      Array(@resource[:dns_nameservers]).each do |nameserver|
+        opts << "--dns-nameserver=#{nameserver}"
+      end
+    end
+
+    if @resource[:host_routes]
+      Array(@resource[:host_routes]).each do |host_route|
+        opts << "--host-route=#{host_route}"
+      end
+    end
+
+    if @resource[:tenant_name]
+      tenant_id = self.class.get_tenant_id(model.catalog,
+                                           @resource[:tenant_name])
+      opts << "--tenant_id=#{tenant_id}"
+    elsif @resource[:tenant_id]
+      opts << "--tenant_id=#{@resource[:tenant_id]}"
+    end
+
+    if @resource[:network_name]
+      opts << resource[:network_name]
+    elsif @resource[:network_id]
+      opts << resource[:network_id]
+    end
+
+    results = auth_neutron('subnet-create', '--format=shell',
+                           opts, resource[:cidr])
+
+    if results =~ /Created a new subnet:/
+      attrs = self.class.parse_creation_output(results)
+      @property_hash = {
+        :ensure                    => :present,
+        :name                      => resource[:name],
+        :id                        => attrs['id'],
+        :cidr                      => attrs['cidr'],
+        :ip_version                => attrs['ip_version'],
+        :gateway_ip                => self.class.parse_gateway_ip(attrs['gateway_ip']),
+        :allocation_pools          => self.class.parse_allocation_pool(attrs['allocation_pools']),
+        :host_routes               => self.class.parse_host_routes(attrs['host_routes']),
+        :dns_nameservers           => self.class.parse_dns_nameservers(attrs['dns_nameservers']),
+        :enable_dhcp               => attrs['enable_dhcp'],
+        :network_id                => attrs['network_id'],
+        :tenant_id                 => attrs['tenant_id'],
+      }
+    else
+      fail("did not get expected message on subnet creation, got #{results}")
+    end
+  end
+
+  def destroy
+    auth_neutron('subnet-delete', name)
+    @property_hash[:ensure] = :absent
+  end
+
+  def gateway_ip=(value)
+    if value == ''
+      auth_neutron('subnet-update', '--no-gateway', name)
+    else
+      auth_neutron('subnet-update', "--gateway-ip=#{value}", name)
+    end
+  end
+
+  def enable_dhcp=(value)
+    if value == 'False'
+      auth_neutron('subnet-update', "--disable-dhcp", name)
+    else
+      auth_neutron('subnet-update', "--enable-dhcp", name)
+    end
+  end
+
+  def dns_nameservers=(values)
+    unless values.empty?
+      opts = ["#{name}", "--dns-nameservers", "list=true"]
+      for value in values
+        opts << value
+      end
+      auth_neutron('subnet-update', opts)
+    end
+  end
+
+  def host_routes=(values)
+    unless values.empty?
+      opts = ["#{name}", "--host-routes", "type=dict", "list=true"]
+      for value in values
+        opts << value
+      end
+      auth_neutron('subnet-update', opts)
+    end
+  end
+
+  [
+   :cidr,
+   :ip_version,
+   :network_id,
+   :allocation_pools,
+   :tenant_id,
+  ].each do |attr|
+     define_method(attr.to_s + "=") do |value|
+       fail("Property #{attr.to_s} does not support being updated")
+     end
+  end
+
+end