Adds support to use "assign where" in usergroups

Icinga · Dec 14, 2015 · cd2155e · cd2155e
1 parent 6e19cd0
commit cd2155e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/libraries/resource_usergroup.rb b/libraries/resource_usergroup.rb
@@ -15,6 +15,14 @@ def initialize(name, run_context = nil)
         @name = name
       end
 
+      def assign_where(arg = nil)
+        set_or_return(
+          :assign_where, arg,
+          :kind_of => Array,
+          :default => nil
+        )
+      end
+
       def display_name(arg = nil)
         set_or_return(
           :display_name, arg,
@@ -51,7 +59,7 @@ def resource_properties(arg = nil)
         set_or_return(
           :resource_properties, arg,
           :kind_of => Array,
-          :default => %w(display_name groups zone)
+          :default => %w(assign_where display_name groups zone)
         )
       end
     end

diff --git a/templates/default/object.usergroup.conf.erb b/templates/default/object.usergroup.conf.erb
@@ -15,6 +15,11 @@ object UserGroup <%= object.inspect -%> {
   <% if options['groups'] && !options['groups'].empty? -%>
   groups = <%= options['groups'].sort.uniq.inspect %>
   <% end -%>
+  <% if options['assign_where'] -%>
+  <%- options['assign_where'].each do |a|-%>
+  assign where <%= a %>
+  <% end -%>
+  <% end -%>
 }
 
 <% end -%>