Class: KnifeSharp::SharpCookbookAlign

Inherits:
Chef::Knife
  • Object
show all
Includes:
Common
Defined in:
lib/chef/knife/sharp-cookbook-align.rb

Instance Method Summary collapse

Methods included from Common

included

Instance Method Details

#bump_cookbooks(env_name, cookbook_list) ⇒ Object 



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
 
# File 'lib/chef/knife/sharp-cookbook-align.rb', line 92

def bump_cookbooks(env_name, cookbook_list)
  unless cookbook_list.empty?
    env = Chef::Environment.load(env_name)
    cbs = Array.new
    backup_data = Hash.new
    backup_data["environment"] = env.name
    backup_data["cookbook_versions"] = Hash.new
    cookbook_list.each do |cb_name|
      cb = cookbook_loader[cb_name]
      if sharp_config["rollback"] && sharp_config["rollback"]["enabled"] == true
        backup_data["cookbook_versions"][cb_name] = env.cookbook_versions[cb_name]
      end
      # Force "= a.b.c" in cookbook version, as chef11 will not accept "a.b.c"
      env.cookbook_versions[cb_name] = "= #{cb.version}"
      cbs << cb
    end

    ui.msg "* Uploading cookbook(s) #{cookbook_list.join(", ")}"
    cookbook_uploader(cbs).upload_cookbooks

    if env.save
      cbs.each do |cb|
        ui.msg "* Bumping #{cb.name} to #{cb.version} for environment #{env.name}"
        log_action("bumping #{cb.name} to #{cb.version} for environment #{env.name}")
      end
    end

    if sharp_config["rollback"] && sharp_config["rollback"]["enabled"] == true
      identifier = Time.now.to_i
      Dir.mkdir(sharp_config["rollback"]["destination"]) unless File.exists?(sharp_config["rollback"]["destination"])
      fp = open(File.join(sharp_config["rollback"]["destination"], "#{identifier}.json"), "w")
      fp.write(JSON.pretty_generate(backup_data))
      fp.close()
    end
  end
end

#check_cookbooks(env) ⇒ Object

Cookbook methods ###



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
 
# File 'lib/chef/knife/sharp-cookbook-align.rb', line 23

def check_cookbooks(env)
  to_update = Array.new

  unless File.exists?(cookbook_path)
    ui.warn "Bad cookbook path, skipping cookbook sync."
    return to_update
  end

  ui.msg(ui.color("== Cookbooks ==", :bold))

  updated_versions = Hash.new
  local_versions = local_cookbook_versions
  remote_versions = remote_cookbook_versions(env)

  if local_versions.empty?
    ui.warn "No local cookbooks found, is the cookbook path correct ? (#{cookbook_path})"
    return to_update
  end

  # get local-only cookbooks
  (local_versions.keys - remote_versions.keys).each do |cb|
    updated_versions[cb] = local_versions[cb]
    ui.msg "* #{cb} is local only (version #{local_versions[cb]})"
  end

  # get cookbooks not up-to-date
  (remote_versions.keys & local_versions.keys).each do |cb|
    if Chef::VersionConstraint.new("> #{remote_versions[cb]}").include?(local_versions[cb])
      updated_versions[cb] = local_versions[cb]
      ui.msg "* #{cb} is not up-to-date (local: #{local_versions[cb]}/remote: #{remote_versions[cb]})"
    elsif Chef::VersionConstraint.new("> #{local_versions[cb]}").include?(remote_versions[cb]) and config[:force_align]
      updated_versions[cb] = local_versions[cb]
      ui.msg "* #{cb} is to be downgraded (local: #{local_versions[cb]}/remote: #{remote_versions[cb]})"
    end
  end

  if sharp_config[chef_server] and sharp_config[chef_server].has_key?("ignore_cookbooks")
    (updated_versions.keys & sharp_config[chef_server]["ignore_cookbooks"]).each do |cb|
      updated_versions.delete(cb)
      ui.msg "* Skipping #{cb} cookbook (ignore list)"
    end
  end

  if !updated_versions.empty?
    all = false
    updated_versions.each_pair do |cb,version|
      answer = ui.ask_question("> Update #{cb} cookbook to #{version} on server ? Y/N/(A)ll/(Q)uit ", :default => "N").upcase unless all

      if answer == "A"
        all = true
      elsif answer == "Q"
        ui.msg "* Skipping next cookbooks alignment."
        break
      end

      if all or answer == "Y"
        to_update << cb
      else
        ui.msg "* Skipping #{cb} cookbook"
      end
    end
  else
    ui.msg "* Environment #{env} is up-to-date."
  end

  to_update
end

#cookbook_loader ⇒ Object 



129
130
131
 
# File 'lib/chef/knife/sharp-cookbook-align.rb', line 129

def cookbook_loader
  @cookbook_loader ||= Chef::CookbookLoader.new(Chef::Config.cookbook_path)
end

#cookbook_uploader(cookbooks) ⇒ Object 



133
134
135
136
137
138
139
 
# File 'lib/chef/knife/sharp-cookbook-align.rb', line 133

def cookbook_uploader(cookbooks)
  if Gem::Version.new(Chef::VERSION).release >= Gem::Version.new('12.0.0')
    uploader = Chef::CookbookUploader.new(cookbooks)
  else
    uploader = Chef::CookbookUploader.new(cookbooks, Chef::Config.cookbook_path)
  end
end

#local_cookbook_versions ⇒ Object 



141
142
143
 
# File 'lib/chef/knife/sharp-cookbook-align.rb', line 141

def local_cookbook_versions
  Hash[Dir.glob("#{cookbook_path}/*").select {|cb| File.directory?(cb)}.map {|cb| [File.basename(cb), cookbook_loader[File.basename(cb)].version] }]
end

#remote_cookbook_versions(env) ⇒ Object 



145
146
147
 
# File 'lib/chef/knife/sharp-cookbook-align.rb', line 145

def remote_cookbook_versions(env)
  Chef::Environment.load(env).cookbook_versions.each_value {|v| v.gsub!("= ", "")}
end

#run ⇒ Object 



17
18
19
 
# File 'lib/chef/knife/sharp-cookbook-align.rb', line 17

def run
  ui.error "Not implemented yet. Use ```knife sharp align -C```"
end