#!/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"
end

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

COMMANDS_HELP=<<-EOT
econe-upload

Uploads an image for use with an OpenNebula Cloud. This image should
be later register with econe-register using the returned ImageId

Usage:
    econe-upload [OPTIONS] IMAGE-PATH

Options:

    --help, -h
        Show help

    --access-key <id>, -K <id>
        The username of the user

    --secret-key <key>, -S <key>
        The password of the user

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

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

IMAGE-PATH: Path to the image to upload

EOT

require 'econe/EC2QueryClient'
require 'CloudClient'
require 'getoptlong'

include CloudCLI

opts = GetoptLong.new(
            ['--help',       '-h',GetoptLong::NO_ARGUMENT],
            ['--version',    '-v',GetoptLong::NO_ARGUMENT],
            ['--access-key', '-K',GetoptLong::REQUIRED_ARGUMENT],
            ['--secret-key', '-S',GetoptLong::REQUIRED_ARGUMENT],
            ['--url',        '-U',GetoptLong::REQUIRED_ARGUMENT],
            ['--multipart',  '-M',GetoptLong::NO_ARGUMENT]
        )

url     = nil
access  = nil
secret  = nil
auth    = nil
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 '--access-key'
                access = arg
            when '--secret-key'
                secret = arg
            when '--url'
                url = arg
            when '--multipart'
                curb = false
        end
    end
rescue Exception => e
    exit -1
end 

image = ARGV.shift

if !image || !File.exists?(image)
    puts "#{cmd_name}: missing IMAGE-PATH parameter or file not found"
    exit -1
end

auth = "#{access}:#{secret}" if secret && access

begin
    ec2_client = EC2QueryClient::Client.new(auth,url)
rescue Exception => e
    puts "#{cmd_name}: #{e.message}"
    exit -1
end

rc = ec2_client.upload_image(image, curb)

if CloudClient::is_error?(rc)
    puts "#{cmd_name}: #{rc.message}"
    exit -1
end

puts "Success: ImageId #{rc['imageId']}"

exit 0

