add basic working route resource and provider

Author: btm

chef/lib/chef/platform.rb

@@ -60,6 +60,7 @@ class Platform
         :user => Chef::Provider::User::Useradd,
         :group => Chef::Provider::Group::Groupadd,
         :http_request => Chef::Provider::HttpRequest,
+        :route => Chef::Provider::Route
       }
     }
 

chef/lib/chef/provider/route.rb

@@ -0,0 +1,97 @@
+#
+# Author:: Bryan McLellan (btm@loftninjas.org)
+# Copyright:: Copyright (c) 2009 Bryan McLellan
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/log'
+require 'chef/mixin/command'
+require 'chef/provider'
+
+class Chef
+  class Provider
+    class Route < Chef::Provider
+      include Chef::Mixin::Command
+
+      def load_current_resource
+        @current_resource = Chef::Resource::Route.new(@new_resource.name)
+
+        Chef::Log.debug("Checking routes for #{@current_resource.target}")
+        status = popen4("route -n") do |pid, stdin, stdout, stderr|
+          stdin.close
+          stdout.each do |line|
+            case line
+            # Destination     Gateway         Genmask         Flags Metric Ref    Use Iface
+            when /^#{@new_resource.target}\s+([\d.]+)\s+([\d.]+)\s+(.+)\s+(\d+)\s+(.+)\s+(.+)\s+(\w+)$/
+              @current_resource.target(@new_resource.target)
+              @current_resource.gateway($1)
+              @current_resource.netmask($2)
+              @current_resource.metric($4.to_i)
+              @current_resource.device($7)
+              Chef::Log.debug("Found route ip:#{@current_resource.target} gw:#{@current_resource.gateway} nm:#{@current_resource.netmask} " +
+                  "metric:#{@current_resource.metric} dev:#{@current_resource.device}")
+            end
+          end
+        end
+
+        unless status.exitstatus == 0
+          raise Chef::Exception::Package, "route failed - #{status.inspect}!"
+        end
+
+        @current_resource
+      end
+
+      def action_add
+        # check to see if load_current_resource found the route
+        unless @current_resource.gateway
+          if @new_resource.route_type == :net
+            command = "route add -net #{@new_resource.target}"
+          else
+            command = "route add #{@new_resource.target}"
+          end
+          command << " netmask #{@new_resource.netmask}" if @new_resource.netmask
+          command << " gw #{@new_resource.gateway}" if @new_resource.gateway
+          command << " metric #{@new_resource.metric}" if @new_resource.metric
+          command << " dev #{@new_resource.device}" if @new_resource.device
+  
+          run_command(
+            :command => command
+          )
+          @new_resource.updated = true
+        else
+          Chef::Log.debug("Route #{@current_resource.target} already exists")
+        end
+      end
+
+      def action_delete
+        # check to see if load_current_resource found the route
+        if @current_resource.gateway 
+          command = "route del #{@new_resource.target}"
+          command << " netmask #{@new_resource.netmask}" if @new_resource.netmask
+          command << " gw #{@new_resource.gateway}" if @new_resource.gateway
+          command << " metric #{@new_resource.metric}" if @new_resource.metric
+          command << " dev #{@new_resource.device}" if @new_resource.device
+  
+          run_command(
+            :command => command
+          )
+          @new_resource.updated = true
+        else
+          Chef::Log.debug("Route #{@current_resource.target} does not exist")
+        end
+      end
+    end
+  end
+end

chef/lib/chef/resource/route.rb

@@ -0,0 +1,90 @@
+#
+# Author:: Bryan McLellan (btm@loftninjas.org)
+# Copyright:: Copyright (c) 2009 Bryan McLellan
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/resource'
+
+class Chef
+  class Resource
+    class Route < Chef::Resource
+
+      def initialize(name, collection=nil, node=nil)
+        super(name, collection, node)
+        @resource_name = :route
+        @target = name
+        @action = :add
+        @allowed_actions.push(:add, :delete)
+        @netmask = nil
+        @gateway = nil
+        @metric = nil
+        @device = nil 
+        @route_type = :host
+      end
+
+      def target(arg=nil)
+        set_or_return(
+          :target,
+          arg,
+          :kind_of => String
+        )
+      end
+
+      def netmask(arg=nil)
+        set_or_return(
+          :netmask,
+          arg,
+          :kind_of => String
+        )
+      end
+
+      def gateway(arg=nil)
+        set_or_return(
+          :gateway,
+          arg,
+          :kind_of => String
+        )
+      end
+
+      def metric(arg=nil)
+        set_or_return(
+          :metric,
+          arg,
+          :kind_of => Integer
+        )
+      end
+
+      def device(arg=nil)
+        set_or_return(
+          :device,
+          arg,
+          :kind_of => String
+        )
+      end
+
+      def route_type(arg=nil)
+        real_arg = arg.kind_of?(String) ? arg.to_sym : arg
+        set_or_return(
+          :route_type,
+          real_arg,
+          :equal_to => [ :host, :net]
+        )
+      end
+    end
+  end
+end
+
+