145 lines
4.5 KiB
Ruby
145 lines
4.5 KiB
Ruby
require "set"
|
|
require "tempfile"
|
|
|
|
require "hbc/container/base"
|
|
|
|
module Hbc
|
|
class Container
|
|
class Dmg < Base
|
|
def self.me?(criteria)
|
|
!criteria.command.run("/usr/bin/hdiutil",
|
|
# realpath is a failsafe against unusual filenames
|
|
args: ["imageinfo", Pathname.new(criteria.path).realpath],
|
|
print_stderr: false).stdout.empty?
|
|
end
|
|
|
|
def extract
|
|
mount do |mounts|
|
|
begin
|
|
raise CaskError, "No mounts found in '#{@path}'; perhaps it is a bad DMG?" if mounts.empty?
|
|
mounts.each(&method(:extract_mount))
|
|
ensure
|
|
mounts.each(&method(:eject))
|
|
end
|
|
end
|
|
end
|
|
|
|
def mount
|
|
# realpath is a failsafe against unusual filenames
|
|
path = Pathname.new(@path).realpath
|
|
|
|
Dir.mktmpdir do |unpack_dir|
|
|
cdr_path = Pathname.new(unpack_dir).join("#{path.basename(".dmg")}.cdr")
|
|
|
|
without_eula = @command.run("/usr/bin/hdiutil",
|
|
args: ["attach", "-plist", "-nobrowse", "-readonly", "-noidme", "-mountrandom", unpack_dir, path],
|
|
input: "qn\n",
|
|
print_stderr: false)
|
|
|
|
# If mounting without agreeing to EULA succeeded, there is none.
|
|
plist = if without_eula.success?
|
|
without_eula.plist
|
|
else
|
|
@command.run!("/usr/bin/hdiutil", args: ["convert", "-quiet", "-format", "UDTO", "-o", cdr_path, path])
|
|
|
|
with_eula = @command.run!("/usr/bin/hdiutil",
|
|
args: ["attach", "-plist", "-nobrowse", "-readonly", "-noidme", "-mountrandom", unpack_dir, cdr_path])
|
|
|
|
if verbose? && !(eula_text = without_eula.stdout).empty?
|
|
ohai "Software License Agreement for '#{path}':"
|
|
puts eula_text
|
|
end
|
|
|
|
with_eula.plist
|
|
end
|
|
|
|
yield mounts_from_plist(plist)
|
|
end
|
|
end
|
|
|
|
def eject(mount)
|
|
# realpath is a failsafe against unusual filenames
|
|
mountpath = Pathname.new(mount).realpath
|
|
return unless mountpath.exist?
|
|
|
|
begin
|
|
tries ||= 3
|
|
if tries > 1
|
|
@command.run("/usr/sbin/diskutil",
|
|
args: ["eject", mountpath],
|
|
print_stderr: false)
|
|
else
|
|
@command.run("/usr/sbin/diskutil",
|
|
args: ["unmount", "force", mountpath],
|
|
print_stderr: false)
|
|
end
|
|
raise CaskError, "Failed to eject #{mountpath}" if mountpath.exist?
|
|
rescue CaskError => e
|
|
raise e if (tries -= 1).zero?
|
|
sleep 1
|
|
retry
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def extract_mount(mount)
|
|
Tempfile.open(["", ".bom"]) do |bomfile|
|
|
bomfile.close
|
|
|
|
Tempfile.open(["", ".list"]) do |filelist|
|
|
filelist.write(bom_filelist_from_path(mount))
|
|
filelist.close
|
|
|
|
@command.run!("/usr/bin/mkbom", args: ["-s", "-i", filelist.path, "--", bomfile.path])
|
|
@command.run!("/usr/bin/ditto", args: ["--bom", bomfile.path, "--", mount, @cask.staged_path])
|
|
end
|
|
end
|
|
end
|
|
|
|
def bom_filelist_from_path(mount)
|
|
Dir.chdir(mount) do
|
|
Dir.glob("**/*", File::FNM_DOTMATCH).map do |path|
|
|
next if skip_path?(Pathname(path))
|
|
(path == ".") ? path : path.prepend("./")
|
|
end.compact.join("\n").concat("\n")
|
|
end
|
|
end
|
|
|
|
def skip_path?(path)
|
|
dmg_metadata?(path) || system_dir_symlink?(path)
|
|
end
|
|
|
|
# unnecessary DMG metadata
|
|
DMG_METADATA_FILES = Set.new %w[
|
|
.background
|
|
.com.apple.timemachine.donotpresent
|
|
.com.apple.timemachine.supported
|
|
.DocumentRevisions-V100
|
|
.DS_Store
|
|
.fseventsd
|
|
.MobileBackups
|
|
.Spotlight-V100
|
|
.TemporaryItems
|
|
.Trashes
|
|
.VolumeIcon.icns
|
|
].freeze
|
|
|
|
def dmg_metadata?(path)
|
|
relative_root = path.sub(%r{/.*}, "")
|
|
DMG_METADATA_FILES.include?(relative_root.basename.to_s)
|
|
end
|
|
|
|
def system_dir_symlink?(path)
|
|
# symlinks to system directories (commonly to /Applications)
|
|
path.symlink? && MacOS.system_dir?(path.readlink)
|
|
end
|
|
|
|
def mounts_from_plist(plist)
|
|
return [] unless plist.respond_to?(:fetch)
|
|
plist.fetch("system-entities", []).map { |e| e["mount-point"] }.compact
|
|
end
|
|
end
|
|
end
|
|
end
|