#
# Author:: Christine Draper (<christine_draper@thirdwaveinsights.com>)
# Copyright:: Copyright (c) 2014 ThirdWave Insights LLC
# License:: Apache License, Version 2.0
#
# 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 'topology_helper'
require 'chef/knife/cookbook_create'

class Chef
  class Knife
    class TopoCookbookCreate < Chef::Knife

      deps do
        Chef::Knife::CookbookCreate.load_deps
      end

      banner "knife topo cookbook create TOPOLOGY [ TOPOLOGY_FILE ] (options)"
      
      option :data_bag,
      :short => '-D DATA_BAG',
      :long => "--data-bag DATA_BAG",
      :description => "The data bag the topologies are stored in"

      # Make the base cookbook create options available on topo cookbook
      self.options = (Chef::Knife::CookbookCreate.options).merge(self.options)

      def initialize (args)
        super
        @cookbook_create_args  = initialize_cmd_args(args, [ 'cookbook', 'create' ])

        # All called commands need to accept union of options
        Chef::Knife::CookbookCreate.options = options

      end

      def run
        if !@name_args[0]
          show_usage
          ui.fatal("You must specify the name of a topology")
          exit 1
        end
 
        @bag_name = topo_bag_name(config[:data_bag])
        @topo_name = @name_args[0]
        @topo_file = @name_args[1]

        # Get the topology data from either the file or the server
        if @topo_file
          topologies = load_topologies(@topo_file)
          index = topologies.find_index{ |t| t['name'] == @topo_name}
          unless index
            ui.fatal("Topology #{@topo_name} was not found in topology file #{@topo_file}")
            exit(1)
          end
          topo = topologies[index]
        else
          unless topo = load_from_server(@bag_name, @topo_name )
            ui.fatal("Topology #{@bag_name}/#{@topo_name} does not exist on the server - use 'knife topo create' first")
            exit(1)
          end
        end

        # create the topology cookbooks
        attribute_cookbooks = topo['cookbook_attributes']
        cookbook_names = []
        if attribute_cookbooks && attribute_cookbooks.length > 0
          attribute_cookbooks.each do |cookbook_spec|
            cookbook_name = cookbook_spec['cookbook']
            run_create_cookbook(cookbook_name) unless cookbook_names.include?(cookbook_name)
            cookbook_names << cookbook_name
            create_attr_file(@cookbook_path, cookbook_name,
            cookbook_spec['filename'] + ".rb", cookbook_spec)
          end
        else
          ui.info "No cookbook attributes found for topology #{@topo_name}"
        end
      end

      private

      include Chef::Knife::TopologyHelper

      def run_create_cookbook(cookbook_name)
        @cookbook_create_args[2] = cookbook_name
        begin
          command = run_cmd(Chef::Knife::CookbookCreate, @cookbook_create_args)
        rescue Exception => e
          raise if Chef::Config[:verbosity] == 2
          ui.warn "create of cookbook #{cookbook_name} exited with error"
          humanize_exception(e)
        end

        # Store the cookbook path for use later
        @cookbook_path = File.expand_path(Array(command.config[:cookbook_path]).first)
        @copyright = command.config[:cookbook_copyright] || "YOUR_COMPANY_NAME"

      end

      def create_attr_file(dir, cookbook_name, filename, attrs)
        
        ui.info("** Creating attribute file #{filename}")
        
        open(File.join(dir, cookbook_name, "attributes", filename), "w") do |file|
          file.puts <<-EOH
#
# THIS FILE IS GENERATED BY THE KNIFE TOPO PLUGIN - MANUAL CHANGES WILL BE OVERWRITTEN 
#
# Cookbook Name:: #{cookbook_name}
# Attribute File:: #{filename}
#
# Copyright #{Time.now.year}, #{@copyright}
#
          EOH
          
          # Print out attribute line
          def print_attr(file, lhs, value1)
            if value1.is_a?(Hash)
              value1.each do |key, value2|
                print_attr(file, "#{lhs}['#{key}']", value2)
              end                           
            else
              file.write "#{lhs} = \"#{value1}\"\n"
            end
          end
          
          # Print out attributes hashed by priority
          def print_priority_attrs(file, attrs, indent=0)
            %w(default force_default normal override force_override).each do |priority|
              if attrs[priority]
                lhs = ""
                indent.times { |i| lhs += " " }
                lhs += priority
                print_attr(file, lhs, attrs[priority])
              end
            end
          end
          
          # Print out qualified attributes
          def print_qualified_attr(file, qualifier_hash)
            file.puts "if node['topo']['#{qualifier_hash['qualifier']}'] == \"#{qualifier_hash['value']}\""
            print_priority_attrs(file, qualifier_hash, 2)
            file.puts "end"
          end
          
          # Process the attributes not needing qualification
          print_priority_attrs(file, attrs)
          file.puts
          
          # Process attributes that need to be qualified
          if attrs['conditional']
            attrs['conditional'].each do |qualified_attrs|
              file.puts "# Attributes for specific #{qualified_attrs['qualifier']}"
              print_qualified_attr(file, qualified_attrs)
            end
          end        
          
        end
      end
       
    end
  end
end