crowbar/crowbar-core

#
# Copyright 2011-2013, Dell
# Copyright 2013-2014, SUSE LINUX Products GmbH
#
# 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.
#

class ChefObject
  class_attribute :chef_type

  NOT_SET = "not_set"

  def self.query_chef
    begin
      return Chef::Search::Query.new
    rescue Errno::ECONNREFUSED => e
      raise Crowbar::Error::ChefOffline.new
    rescue StandardError => e
      return Chef::Node.new
    end
  end

  def self.fetch_nodes_from_cdb
    # get all nodes directly from CouchDB, skip Chef API and Solr search
    nodes = Chef::Node.cdb_list(true)
    return [nodes, 0, nodes.count]
  rescue Errno::ECONNREFUSED => e
    raise Crowbar::Error::ChefOffline
  rescue StandardError
    return [[], 0, 0]
  end

  def self.fetch_roles_from_cdb
    # get all roles directly from CouchDB, skip Chef API and Solr search
    roles = Chef::Role.cdb_list(true)
    return [roles, 0, roles.count]
  rescue Errno::ECONNREFUSED
    raise Crowbar::Error::ChefOffline
  rescue StandardError
    return [[], 0, 0]
  end

  def self.chef_escape(str)
    str.gsub("-:") { |c| '\\' + c }
  end

  # FIXME: the second argument was added so that the Proposal model
  # can be exported in a same way as the ProposalObject. When the replacement
  # is completed, remove it and implement the export in the Proposal.
  # Also check the logging barclamp that it did not break.
  def self.export(obj, name = nil)
    name ||= obj.respond_to?(:name) ? obj.name : "unknown"
    file   = Rails.root.join("db", "#{obj.chef_type}_#{name}.json")
    File.open(file, "w") { |f| f.write(obj.to_json) }
  end

  def export(name = nil)
    self.class.export(self, name)
  end
end