File Manager

Viewing File: vault_update.rb

# Description: Chef-Vault VaultUpdate class
# Copyright 2014-15, Nordstrom, Inc.

# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at

#     http://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

require_relative "vault_base"
require_relative "vault_admins"
require_relative "vault_clients"

class Chef
  class Knife
    class VaultUpdate < Knife
      include Chef::Knife::VaultBase
      include Chef::Knife::VaultAdmins
      include Chef::Knife::VaultClients

      banner "knife vault update VAULT ITEM VALUES (options)"

      option :search,
        short: "-S SEARCH",
        long: "--search SEARCH",
        description: "Chef SOLR search for clients"

      option :clients,
        short: "-C CLIENTS",
        long: "--clients CLIENTS",
        description: "Chef clients to be added as clients"

      option :admins,
        short: "-A ADMINS",
        long: "--admins ADMINS",
        description: "Chef users to be added as admins"

      option :json,
        short: "-J FILE",
        long: "--json FILE",
        description: "File containing JSON data to encrypt"

      option :file,
        long: "--file FILE",
        description: "File to be added to vault item as file-content"

      option :clean,
        long: "--clean",
        description: "Clean clients before performing search"

      option :keys_mode,
        short: "-K KEYS_MODE",
        long: "--keys-mode KEYS_MODE",
        description: "Mode in which to save vault keys"

      def run
        vault = @name_args[0]
        item = @name_args[1]
        values = @name_args[2]
        search = config[:search]
        json_file = config[:json]
        file = config[:file]
        clean = config[:clean]
        keys_mode = config[:keys_mode]

        set_mode(config[:vault_mode])

        if vault && item && ((values || json_file || file) || (search || clients || admins) || (keys_mode))
          begin
            vault_item = ChefVault::Item.load(vault, item)

            # Keys management first
            if clean
              vault_clients = vault_item.get_clients.clone.sort
              vault_clients.each do |client|
                ui.info "Deleting #{client}"
                vault_item.delete_client(client)
              end
            end

            vault_item.search(search) if search
            vault_item.clients if search
            vault_item.clients(clients) if clients
            vault_item.admins(admins) if admins

            # Save only the keys if no value is provided, otherwise save the item
            if values || json_file || file
              merge_values(values, json_file).each do |key, value|
                vault_item[key] = value
              end

              if file
                vault_item["file-name"] = File.basename(file)
                vault_item["file-content"] = File.open(file, &:read)
              end

              vault_item.save
            else
              vault_item.save_keys
            end

          rescue ChefVault::Exceptions::KeysNotFound,
                 ChefVault::Exceptions::ItemNotFound
            raise ChefVault::Exceptions::ItemNotFound,
              "#{vault}/#{item} does not exist, "\
              "use 'knife vault create' to create."
          end

          if keys_mode
            vault_item.mode(keys_mode)
            vault_item.save_keys
          end
        else
          show_usage
        end
      end
    end
  end
end