summaryrefslogtreecommitdiffstats
path: root/lib/puppet/util/settings/file_setting.rb
blob: 7ddc4697c367738c157dc9fadcc5c33955859c2a (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
require 'puppet/util/settings/setting'

# A file.
class Puppet::Util::Settings::FileSetting < Puppet::Util::Settings::Setting
    AllowedOwners = %w{root service}
    AllowedGroups = %w{service}

    class SettingError < StandardError; end

    attr_accessor :mode, :create

    # Should we create files, rather than just directories?
    def create_files?
        create
    end

    def group=(value)
        unless AllowedGroups.include?(value)
            raise SettingError, "Invalid group %s on setting %s. Valid groups are %s." % [value, name, AllowedGroups.join(', ')]
        end
        @group = value
    end

    def group
        return unless defined?(@group) && @group
        @settings[:group]
    end

    def owner=(value)
        unless AllowedOwners.include?(value)
            raise SettingError, "Invalid owner %s on setting %s. Valid owners are %s." % [value, name, AllowedOwners.join(', ')]
        end
        @owner = value
    end

    def owner
        return unless defined?(@owner) && @owner
        return "root" if @owner == "root"
        @settings[:user]
    end

    # Set the type appropriately.  Yep, a hack.  This supports either naming
    # the variable 'dir', or adding a slash at the end.
    def munge(value)
        # If it's not a fully qualified path...
        if value.is_a?(String) and value !~ /^\$/ and value !~ /^\// and value != 'false'
            # Make it one
            value = File.join(Dir.getwd, value)
        end
        if value.to_s =~ /\/$/
            @type = :directory
            return value.sub(/\/$/, '')
        end
        return value
    end

    # Return the appropriate type.
    def type
        value = @settings.value(self.name)
        if @name.to_s =~ /dir/
            return :directory
        elsif value.to_s =~ /\/$/
            return :directory
        elsif value.is_a? String
            return :file
        else
            return nil
        end
    end

    # Turn our setting thing into a Puppet::Resource instance.
    def to_resource
        return nil unless type = self.type

        path = self.value

        return nil unless path.is_a?(String)

        # Make sure the paths are fully qualified.
        path = File.join(Dir.getwd, path) unless path =~ /^\//

        return nil unless type == :directory or create_files? or File.exist?(path)
        return nil if path =~ /^\/dev/

        resource = Puppet::Resource.new(:file, path)
        resource[:mode] = self.mode if self.mode

        if Puppet.features.root?
            resource[:owner] = self.owner if self.owner
            resource[:group] = self.group if self.group
        end

        resource[:ensure] = type
        resource[:loglevel] = :debug
        resource[:backup] = false

        resource.tag(self.section, self.name, "settings")

        resource
    end

    # Make sure any provided variables look up to something.
    def validate(value)
        return true unless value.is_a? String
        value.scan(/\$(\w+)/) { |name|
            name = $1
            unless @settings.include?(name)
                raise ArgumentError,
                    "Settings parameter '%s' is undefined" %
                    name
            end
        }
    end
end