summaryrefslogtreecommitdiffstats
path: root/lib/puppet/file_serving/metadata.rb
blob: 1cc3fa355e5b78adc0fd4311ad0681f1997804f3 (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
#
#  Created by Luke Kanies on 2007-10-16.
#  Copyright (c) 2007. All rights reserved.

require 'puppet'
require 'puppet/indirector'
require 'puppet/file_serving'
require 'puppet/file_serving/base'
require 'puppet/util/checksums'
require 'puppet/file_serving/indirection_hooks'

# A class that handles retrieving file metadata.
class Puppet::FileServing::Metadata < Puppet::FileServing::Base

    include Puppet::Util::Checksums

    extend Puppet::Indirector
    indirects :file_metadata, :extend => Puppet::FileServing::IndirectionHooks

    attr_reader :path, :owner, :group, :mode, :checksum_type, :checksum, :ftype, :destination

    PARAM_ORDER = [:mode, :ftype, :owner, :group]

    def attributes_with_tabs
        desc = []
        PARAM_ORDER.each { |check|
            check = :ftype if check == :type
            desc << send(check)
        }

        case ftype
        when "file", "directory": desc << checksum
        when "link": desc << @destination
        else
            raise ArgumentError, "Cannot manage files of type %s" % ftype
        end

        return desc.join("\t")
    end

    def checksum_type=(type)
        raise(ArgumentError, "Unsupported checksum type %s" % type) unless respond_to?("%s_file" % type)

        @checksum_type = type
    end

    # Retrieve the attributes for this file, relative to a base directory.
    # Note that File.stat raises Errno::ENOENT if the file is absent and this
    # method does not catch that exception.
    def collect
        real_path = full_path()
        stat = stat()
        @owner = stat.uid
        @group = stat.gid
        @ftype = stat.ftype


        # We have to mask the mode, yay.
        @mode = stat.mode & 007777

        case stat.ftype
        when "file":
            @checksum = ("{%s}" % @checksum_type) + send("%s_file" % @checksum_type, real_path)
        when "directory": # Always just timestamp the directory.
            sumtype = @checksum_type.to_s =~ /time/ ? @checksum_type : "ctime"
            @checksum = ("{%s}" % sumtype) + send("%s_file" % sumtype, path).to_s
        when "link":
            @destination = File.readlink(real_path)
        else
            raise ArgumentError, "Cannot manage files of type %s" % stat.ftype
        end
    end

    def initialize(*args)
        @checksum_type = "md5"
        super
    end
end