class Chef::Knife::NodeEnvironmentSet

def run
  if @name_args.size < 2
    ui.fatal "You must specify a node name and an environment."
    show_usage
    exit 1
  else
    @node_name = @name_args[0]
    @environment = @name_args[1]
  end
  node = Chef::Node.load(@node_name)
  node.chef_environment = @environment
  node.save
  config[:environment] = @environment
  output(format_for_display(node))
end