require 'chef/knife/joyent_base'

class Chef
  class Knife
    class JoyentServerList < Knife

      include Knife::JoyentBase

      banner "knife joyent server list <options>"

      def run
        columns = 11
        price_column_width = 11

        servers = [
          ui.color('ID', :bold),
          ui.color('Name', :bold),
          ui.color('State', :bold),
          ui.color('Type', :bold),
          ui.color('Image', :bold),
          ui.color('Flavor', :bold),
          ui.color('IPs', :bold),
          ui.color('      RAM', :bold),
          ui.color('    Disk', :bold),
          ui.color('Price $/Month'),
          ui.color('Tags', :bold)
        ]

        total_cost = 0

        self.connection.servers.sort do |a, b|
          (a.name || '') <=> (b.name || '')
        end.each do |s|
          servers << s.id.to_s
          servers << s.name

          servers << case s.state
                       when 'running'
                         ui.color(s.state, :green)
                       when 'stopping', 'provisioning'
                         ui.color(s.state, :yellow)
                       when 'stopped'
                         ui.color(s.state, :red)
                       else
                         ui.color('unknown', :red)
                     end

          flavor = s.respond_to?(:attributes) ? s.attributes["package"] : 'unknown'

          servers << s.type
          servers << s.dataset
          servers << flavor
          servers << s.ips.join(",")
          servers << "#{sprintf "%6.2f", s.memory/1024.0} GB"
          servers << "#{sprintf "%5.0f", s.disk/1024} GB"
          servers << pricing.monthly_formatted_price_for_flavor(flavor, price_column_width)

          total_cost += (pricing[flavor] || 0)

          if (s.tags rescue nil)
            servers << (show_tags? ? s.tags.map { |k, v| "#{k}:#{v}" }.join(' ') : "")
          else
            servers << "No Tags"
          end
        end

        (columns - 3).times{servers << ""}
        servers << "   Total"
        servers << pricing.formatted_price_for_value(total_cost * pricing.class::HOURS_PER_MONTH,
                                           price_column_width)

        puts ui.list(servers, :uneven_columns_across, columns)
      rescue => e
        output_error(e)
      end

      def show_tags?
        !ENV['SHOW_TAGS'].nil?
      end
    end
  end
end