#!/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-storage - Manages OCCI storage resource

Usage:
    occi-storage <COMMAND> [OPTIONS] [PARAMETERS]

Commands:

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

list
       lists available storage resources

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

delete <storage id>
       deletes the storage resource idenfitied by <storage 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'
require 'pp'
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],
            ['--debug',      '-D',GetoptLong::NO_ARGUMENT],
            ['--timeout',    '-T',GetoptLong::REQUIRED_ARGUMENT],
            ['--multipart',  '-M',GetoptLong::NO_ARGUMENT]
        )

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

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
            when '--multipart'
                curb = false
        end
    end
rescue Exception => e
    exit(-1)
end

if !ARGV[0]
    puts "#{cmd_name}: [COMMAND] not present"
    puts "#{cmd_name}: Execute #{cmd_name} -h for help."
    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

case ARGV[0].downcase
    when 'create'
        image_xml = ARGV[1]

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

        rc = occi_client.post_image(image_xml, curb)

    when 'list'
        rc = occi_client.get_images

    when 'show'
        image_id = ARGV[1]

        if !image_id
            puts "#{cmd_name} show: missing storage id parameter"
            exit(-1)
        end

        rc = occi_client.get_image(image_id)

    when 'delete'
        image_id = ARGV[1]

        if !image_id
            puts "#{cmd_name} show: missing storage id parameter"
            exit(-1)
        end

        rc = occi_client.delete_image(image_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
