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

require 'uri'
require 'puppet/file_serving'

# This module is used to pick the appropriate terminus
# in file-serving indirections.  This is necessary because
# the terminus varies based on the URI asked for.
module Puppet::FileServing::IndirectionHooks
    PROTOCOL_MAP = {"puppet" => :rest, "file" => :file, "puppetmounts" => :file_server}

    # Pick an appropriate terminus based on the protocol.
    def select_terminus(full_uri, options = {})
        # Short-circuit to :file if it's a fully-qualified path.
        return PROTOCOL_MAP["file"] if full_uri =~ /^#{::File::SEPARATOR}/
        begin
            uri = URI.parse(URI.escape(full_uri))
        rescue => detail
            raise ArgumentError, "Could not understand URI %s: %s" % [full_uri, detail.to_s]
        end

        terminus = PROTOCOL_MAP[uri.scheme] || raise(ArgumentError, "URI protocol '%s' is not supported for file serving" % uri.scheme)

        # This provides a convenient mechanism for people to write configurations work
        # well in both a networked and local setting.
        if uri.host.nil? and uri.scheme == "puppet" and Puppet.settings[:name] == "puppet"
            terminus = :file_server
        end

        if terminus == :file_server and uri.path =~ %r{^/([^/]+)\b}
            modname = $1
            if modname == "modules"
                terminus = :modules
            elsif terminus(:modules).find_module(modname, options[:node])
                Puppet.warning "DEPRECATION NOTICE: Found file '%s' in module without using the 'modules' mount; please prefix path with '/modules'" % uri.path
                terminus = :modules
            end
        end

        return terminus
    end
end