#!/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-network - Manages virtual networks

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

Commands:

create <occi xml file>
       creates a new virtual network described by the provided 
       <occi xml file>

list
       lists available virtual networks

show <network id>
       retrieves the OCCI XML representation of the virtual network
       identified by <network id>

delete <network id>
       deletes the virtual network idenfitied by <network 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

--multipart, -M
    Use 'multipart-post' library instead of Curb/Curl

EOT

require 'occi/OCCIClient'
require 'CloudClient'
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 'create'
        network_xml = ARGV[1]

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

        rc = occi_client.post_network(network_xml)

    when 'list'
        rc = occi_client.get_networks

    when 'show'
        network_id = ARGV[1]

        if !network_id 
            puts "#{cmd_name} show: missing NETWORK-ID or file not found"
            exit(-1)
        end
        
        rc = occi_client.get_network(network_id)

    when 'delete'
        network_id = ARGV[1]

        if !network_id 
            puts "#{cmd_name} delete: missing NETWORK-ID parameter"
            exit(-1)
        end

        rc = occi_client.delete_network(network_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
