#!/usr/bin/env ruby
# -------------------------------------------------------------------------- #
# Copyright 2002-2010, OpenNebula Project Leads (OpenNebula.org)             #
#                                                                            #
# 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.                                             #
#--------------------------------------------------------------------------- #

ONE_LOCATION=ENV["ONE_LOCATION"]

if !ONE_LOCATION
    RUBY_LIB_LOCATION="/usr/lib/one/ruby"
else
    RUBY_LIB_LOCATION=ONE_LOCATION+"/lib/ruby"
    TEMPLATES_LOCATION=ONE_LOCATION+"/etc/occi_templates"
    CONF_LOCATION=ONE_LOCATION+"/etc"
end

$: << RUBY_LIB_LOCATION
$: << RUBY_LIB_LOCATION+"/cloud"

COMMANDS_HELP=<<-EOT
occi-compute - Manages compute resources

Usage:
    occi-compute <COMMAND> [OPTIONS] [ARGUMENTS]

Commands:

* create <occi xml file>
       creates a new compute resource described by the provided 
       <occi xml file>

* list
       lists available compute resources

* show <compute id>
       retrieves the OCCI XML representation of the compute resource
       identified by <compute id>

* update <occi xml file>
       updates the representation of the compute resource represented by the
       provided <occi xml file>

* delete <compute id>
       deletes the compute resource idenfitied by <compute id>


Options:

--help, -h
    Show help

--username <id>, -U <id>
    The username of the user

--password <key>, -P <key>
    The password of the user

--url <url>, -R <url>
    Set url as the web service url to use

--timeout <seconds>, -T <seconds>
    Sets a timeout for the http connection

--debug, -D
    Enables verbosity

EOT

require 'occi/OCCIClient'
require 'getoptlong'
require "rexml/document"

include CloudCLI

opts = GetoptLong.new(
            ['--help',         '-h',GetoptLong::NO_ARGUMENT],
            ['--version',      '-v',GetoptLong::NO_ARGUMENT],
            ['--username',     '-U',GetoptLong::REQUIRED_ARGUMENT],
            ['--password',     '-P',GetoptLong::REQUIRED_ARGUMENT],
            ['--url',          '-R',GetoptLong::REQUIRED_ARGUMENT],
            ['--timeout',      '-T',GetoptLong::REQUIRED_ARGUMENT],
            ['--debug',        '-D',GetoptLong::NO_ARGUMENT]
        )

url      = nil
username = nil
password = nil
auth     = nil
timeout  = nil
debug    = false

begin
    opts.each do |opt, arg|
        case opt
            when '--help'
                puts COMMANDS_HELP
                return
            when '--version'
                puts CloudCLI.version_text
                exit 0
            when '--username'
                username = arg
            when '--password'
                password = arg
            when '--url'
                url = arg
            when '--timeout'
                timeout = arg
            when '--debug'
                debug = true
        end
    end
rescue Exception => e
    exit(-1)
end 

begin
    occi_client = OCCIClient::Client.new(url,username,password, timeout, debug)
rescue Exception => e
    puts "#{cmd_name}: #{e.message}"
    exit(-1)
end

if !ARGV[0]
    puts "#{cmd_name}: [COMMAND] not present"
    puts "#{cmd_name}: Execute #{cmd_name} -h for help."
    exit(-1)
end   

case ARGV[0].downcase
    when 'list'
        rc = occi_client.get_vms

    when 'create'
        vm_xml = ARGV[1]

        if !vm_xml || !File.exists?(vm_xml)
            puts "#{cmd_name} create: missing OCCI-XML or file not found"
            exit(-1)
        end

        rc = occi_client.post_vms(vm_xml)
        
    when 'show'
        vm_id = ARGV[1]

        if !vm_id 
            puts "#{cmd_name} show: missing VM-ID parameter"
            exit(-1)
        end

        rc = occi_client.get_vm(vm_id)
        
    when 'update'
        vm_xml = ARGV[1]

        if !vm_xml || !File.exists?(vm_xml)
            puts "#{cmd_name} update: missing OCCI-XML or file not found"
            exit -1
        end

        rc = occi_client.put_vm(vm_xml)
        
    when 'delete'
        vm_id = ARGV[1]

        if !vm_id 
            puts "#{cmd_name} delete: missing VM-ID parameter"
            exit -1
        end

        rc = occi_client.delete_vm(vm_id)
        
    else
        puts "Command #{ARGV[0]} not valid."
        exit(-1)
end

if CloudClient::is_error?(rc)
    puts rc.to_s()
else 
    begin
        doc = REXML::Document.new(rc)
    rescue REXML::ParseException => e
        puts e.message 
        exit(-1)
    end
    
    xml = doc.root
    
    if xml.nil?
        puts rc
        exit(-1)
    end
    
    str = ""
    REXML::Formatters::Pretty.new(4).write(xml,str)
    puts "\n" + str + "\n " 
end

