summaryrefslogtreecommitdiff
path: root/lib/chef/provider/group/groupmod.rb
blob: c8b6458db02d9bac35205c4c9d458676c47db412 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
#
# Author:: Dan Crosta (<dcrosta@late.am>)
# Copyright:: Copyright (c) 2012 Opscode, Inc.
# 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/mixin/shell_out'

class Chef
  class Provider
    class Group
      class Groupmod < Chef::Provider::Group

        include Chef::Mixin::ShellOut

        def load_current_resource
          super
          [ "group", "user" ].each do |binary|
            raise Chef::Exceptions::Group, "Could not find binary /usr/sbin/#{binary} for #{@new_resource}" unless ::File.exists?("/usr/sbin/#{binary}")
          end
        end

        # Create the group
        def create_group
          command = "group add"
          command << set_options
          shell_out!(command)

          add_group_members(@new_resource.members)
        end

        # Manage the group when it already exists
        def manage_group
          if @new_resource.append
            members_to_be_added = [ ]
            if @new_resource.excluded_members && !@new_resource.excluded_members.empty?
              # First find out if any member needs to be removed
              members_to_be_removed = [ ]
              @new_resource.excluded_members.each do |member|
                members_to_be_removed << member if @current_resource.members.include?(member)
              end

              unless members_to_be_removed.empty?
                # We are using a magic trick to remove the groups.
                reset_group_membership

                # Capture the members we need to add in
                # members_to_be_added to be added later on.
                @current_resource.members.each do |member|
                  members_to_be_added << member unless members_to_be_removed.include?(member)
                end
              end
            end

            if @new_resource.members && !@new_resource.members.empty?
              @new_resource.members.each do |member|
                members_to_be_added << member if !@current_resource.members.include?(member)
              end
            end

            Chef::Log.debug("#{@new_resource} not changing group members, the group has no members to add") if members_to_be_added.empty?

            add_group_members(members_to_be_added)
          else
            # We are resetting the members of a group so use the same trick
            reset_group_membership
            Chef::Log.debug("#{@new_resource} setting group members to: none") if @new_resource.members.empty?
            add_group_members(@new_resource.members)
          end
        end

        # Remove the group
        def remove_group
          shell_out!("group del #{@new_resource.group_name}")
        end

        # Adds a list of usernames to the group using `user mod`
        def add_group_members(members)
          Chef::Log.debug("#{@new_resource} adding members #{members.join(', ')}") if !members.empty?
          members.each do |user|
            shell_out!("user mod -G #{@new_resource.group_name} #{user}")
          end
        end

        # This is tricky, but works: rename the existing group to
        # "<name>_bak", create a new group with the same GID and
        # "<name>", then set correct members on that group
        def reset_group_membership
          rename = "group mod -n #{@new_resource.group_name}_bak #{@new_resource.group_name}"
          shell_out!(rename)

          create = "group add"
          create << set_options(:overwrite_gid => true)
          shell_out!(create)

          remove = "group del #{@new_resource.group_name}_bak"
          shell_out!(remove)
        end

        # Little bit of magic as per Adam's useradd provider to pull and assign the command line flags
        #
        # ==== Returns
        # <string>:: A string containing the option and then the quoted value
        def set_options(overwrite_gid=false)
          opts = ""
          if overwrite_gid || @new_resource.gid && (@current_resource.gid != @new_resource.gid)
            opts << " -g '#{@new_resource.gid}'"
          end
          if overwrite_gid
            opts << " -o"
          end
          opts << " #{@new_resource.group_name}"
          opts
        end
      end
    end
  end
end