Class: Simp::Rake::Build::Unpack

Inherits:
Rake::TaskLib
  • Object
show all
Includes:
Constants
Defined in:
lib/simp/rake/build/unpack.rb

Instance Method Summary collapse

Methods included from Constants

#distro_build_dir, #init_member_vars

Constructor Details

#initialize(base_dir) ⇒ Unpack 



11
12
13
14
15
 
# File 'lib/simp/rake/build/unpack.rb', line 11

def initialize( base_dir )
  init_member_vars( base_dir )

  define_tasks
end

Instance Method Details

#define_tasks ⇒ Object 



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
 
# File 'lib/simp/rake/build/unpack.rb', line 17

def define_tasks
  #!/usr/bin/rake -T

=begin
  desc "Unpack an ISO. Unpacks either a RHEL or CentOS ISO into
  <targetdir>/<RHEL|CentOS><version>-<arch>.
   * :iso_path - Full path to the ISO image to unpack.
   * :merge - If true, then automatically merge any existing
     directories. Defaults to prompting.
   * :targetdir - The parent directory for the to-be-created directory
     containing the unpacked ISO. Defaults to the current directory.
   * :isoinfo - The isoinfo executable to use to extract stuff from the ISO.
     Defaults to 'isoinfo'.
   * :version - optional override for the <version> number (e.g., '7.0' instead of '7')
   * :exclude_repos - Don't unpack repos or packages when true

  "
=end

  task :unpack,[:iso_path, :merge, :targetdir, :isoinfo, :version, :exclude_repos] do |t,args|
    args.with_defaults(
      :iso_path      => '',
      :isoinfo       => 'isoinfo',
      :targetdir     => Dir.pwd,
      :merge         => false,
      :version       => false,
      :exclude_repos => false,
    )

    iso_path      = args.iso_path
    iso_info      = which(args.isoinfo)
    targetdir     = args.targetdir
    merge         = args.merge
    version       = args.version
    exclude_repos = args.exclude_repos

    # Checking for valid arguments
    File.exist?(args.iso_path) or
      fail "Error: You must provide the full path and filename of the ISO image."

    %x{file --keep-going '#{iso_path}'}.split(":")[1..-1].to_s =~ /ISO/ or
      fail "Error: The file provided is not a valid ISO."

    pieces = File.basename(iso_path,'.iso').split('-')

    # Mappings of ISO name to target directory name.
    # This is a hash of hashes to provide room for growth.
    dvd_map = {
      # RHEL structure as provided from RHN:
      #   rhel-server-<version>-<arch>-<whatever>
      'rhel' => {
        'baseos'  => 'RedHat',
        'version' => version || pieces[2],
        'arch'    => pieces[3]
      },
      # CentOS structure as provided from the CentOS website:
      #   CentOS-<version>-<arch>-<whatever>
      'CentOS' => {
        'baseos'  => 'CentOS',
        'version' => version || pieces[1],
        'arch'    => pieces[2]
      }
    }

    # Determine the target directory
    map = dvd_map[pieces[0]]
    map.nil? and fail "Error: Could not find a mapping for '#{iso_path}'."
    out_dir = "#{File.expand_path(targetdir)}/#{map['baseos']}#{map['version']}-#{map['arch']}"

    # Attempt a merge
    if File.exist?(out_dir) and merge.to_s.strip == 'false'
      puts "Directory '#{out_dir}' already exists! Would you like to merge? [Yn]?"
      unless $stdin.gets.strip.match(/^(y.*|$)/i)
        puts "Skipping #{iso_path}"
        next
      end
    end

    puts "Target dir: #{out_dir}"
    mkdir_p(out_dir)

    # Unpack the ISO
    iso_toc = %x{#{iso_info} -Rf -i #{iso_path}}.split("\n")
    iso_toc.each do |iso_entry|
      iso_toc.delete(File.dirname(iso_entry))
    end

    progress = ProgressBar.create(:title => 'Unpacking', :total => iso_toc.size)

    iso_toc.each do |iso_entry|
      target = "#{out_dir}#{iso_entry}"
      unless File.exist?(target)
        if exclude_repos && target =~ %r{\.rpm$|repodata|repomd.xml}
          puts "  [exclude_repos] SKIPPING repo/package file: #{target}"
          next
        end
        FileUtils.mkdir_p(File.dirname(target))
        system("#{iso_info} -R -x #{iso_entry} -i #{iso_path} > #{target}")
      end
      if progress
        progress.increment
      else
        print "#"
      end
    end

    # The repodata directory needs to exist (but not be copied) for el7 distros
    FileUtils.mkdir("#{out_dir}/repodata") if version.start_with?('7')
  end
end