summaryrefslogtreecommitdiffstats
path: root/lib/puppet/provider/package/pkgdmg.rb
blob: 4e9e50802c63368abfb051879bb4788fde0f7a3b (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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
#
# pkgdmg.rb
#
# Install Installer.app packages wrapped up inside a DMG image file.
#
# Copyright (C) 2007 Jeff McCune Jeff McCune <jeff@northstarlabs.net>
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation (version 2 of the License)
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston MA  02110-1301 USA
#
# Motivation: DMG files provide a true HFS file system
# and are easier to manage and .pkg bundles.
#
# Note: the 'apple' Provider checks for the package name
# in /L/Receipts.  Since we install multiple pkg's from a single
# source, we treat the source .pkg.dmg file as the package name.
# As a result, we store installed .pkg.dmg file names
# in /var/db/.puppet_pkgdmg_installed_<name>

require 'puppet/provider/package'
require 'facter/util/plist'

Puppet::Type.type(:package).provide :pkgdmg, :parent => Puppet::Provider::Package do
    desc "Package management based on Apple's Installer.app and DiskUtility.app.  This package works by checking the contents of a DMG image for Apple pkg or mpkg files. Any number of pkg or mpkg files may exist in the root directory of the DMG file system. Sub directories are not checked for packages.  See `the wiki docs </trac/puppet/wiki/DmgPackages>` for more detail."

    confine :operatingsystem => :darwin
    defaultfor :operatingsystem => :darwin
    commands :installer => "/usr/sbin/installer"
    commands :hdiutil => "/usr/bin/hdiutil"
    commands :curl => "/usr/bin/curl"

    # JJM We store a cookie for each installed .pkg.dmg in /var/db
    def self.instance_by_name
        Dir.entries("/var/db").find_all { |f|
            f =~ /^\.puppet_pkgdmg_installed_/
        }.collect do |f|
            name = f.sub(/^\.puppet_pkgdmg_installed_/, '')
            yield name if block_given?
            name
        end
    end

    def self.instances
        instance_by_name.collect do |name|

            new(

                :name => name,
                :provider => :pkgdmg,

                :ensure => :installed
            )
        end
    end

    def self.installpkg(source, name, orig_source)
        installer "-pkg", source, "-target", "/"
        # Non-zero exit status will throw an exception.
        File.open("/var/db/.puppet_pkgdmg_installed_#{name}", "w") do |t|
            t.print "name: '#{name}'\n"
            t.print "source: '#{orig_source}'\n"
        end
    end

    def self.installpkgdmg(source, name)
        unless source =~ /\.dmg$/i || source =~ /\.pkg$/i
            raise Puppet::Error.new("Mac OS X PKG DMG's must specificy a source string ending in .dmg or flat .pkg file")
        end
        require 'open-uri'
        cached_source = source
        if %r{\A[A-Za-z][A-Za-z0-9+\-\.]*://} =~ cached_source
            cached_source = "/tmp/#{name}"
            begin
                curl "-o", cached_source, "-C", "-", "-k", "-s", "--url", source
                Puppet.debug "Success: curl transfered [#{name}]"
            rescue Puppet::ExecutionFailure
                Puppet.debug "curl did not transfer [#{name}].  Falling back to slower open-uri transfer methods."
                cached_source = source
            end
        end

        begin
            if source =~ /\.dmg$/i
                File.open(cached_source) do |dmg|
                    xml_str = hdiutil "mount", "-plist", "-nobrowse", "-readonly", "-noidme", "-mountrandom", "/tmp", dmg.path
                    hdiutil_info = Plist::parse_xml(xml_str)
                    unless hdiutil_info.has_key?("system-entities")
                        raise Puppet::Error.new("No disk entities returned by mount at %s" % dmg.path)
                    end
                    mounts = hdiutil_info["system-entities"].collect { |entity|
                        entity["mount-point"]
                    }.compact
                    begin
                        mounts.each do |mountpoint|
                            Dir.entries(mountpoint).select { |f|
                                f =~ /\.m{0,1}pkg$/i
                            }.each do |pkg|
                                installpkg("#{mountpoint}/#{pkg}", name, source)
                            end
                        end
                    ensure
                        mounts.each do |mountpoint|
                            hdiutil "eject", mountpoint
                        end
                    end
                end
            elsif source =~ /\.pkg$/i
                installpkg(cached_source, name, source)
            else
                raise Puppet::Error.new("Mac OS X PKG DMG's must specificy a source string ending in .dmg or flat .pkg file")
            end
        ensure
            # JJM Remove the file if open-uri didn't already do so.
            File.unlink(cached_source) if File.exist?(cached_source)
        end
    end

    def query
        if FileTest.exists?("/var/db/.puppet_pkgdmg_installed_#{@resource[:name]}")
            Puppet.debug "/var/db/.puppet_pkgdmg_installed_#{@resource[:name]} found"
            return {:name => @resource[:name], :ensure => :present}
        else
            return nil
        end
    end

    def install
        source = nil
        unless source = @resource[:source]
            raise Puppet::Error.new("Mac OS X PKG DMG's must specify a package source.")
        end
        unless name = @resource[:name]
            raise Puppet::Error.new("Mac OS X PKG DMG's must specify a package name.")
        end
        self.class.installpkgdmg(source,name)
    end
end