opennebula-tools 3.2.1-2 / usr / bin / occi-storage

#!/usr/bin/env ruby
# -------------------------------------------------------------------------- #
# Copyright 2002-2012, 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>


* update <occi xml file>
      updates the representation of the storage resource represented by the
      provided <occi xml file>
      
* 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'

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 'update'
        image_xml = ARGV[1]

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

        rc = occi_client.put_image(image_xml)

    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()
    exit(-1)
else 
    str, code = print_xml(rc)
    puts str
    exit(code)
end