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


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

COMMANDS_HELP=<<-EOT
econe-run-instances

Runs an instance of a particular image

Usage:
    econe-run-instances [OPTIONS] IMAGE-ID

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

    --type <type>, -t <type>
        OpenNebula template in which is based this instance
        
    --user-data, -d
        Specifies Base64-encoded MIME user data to be made 
        available to the instance
        
    --headers, -H
        Display column headers
        
IMAGE-ID: The image identification as returned by
the econe-upload command

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],
            ['--type',       '-t',GetoptLong::REQUIRED_ARGUMENT],
            ['--user-data',  '-d',GetoptLong::REQUIRED_ARGUMENT],
            ['--headers',    '-H',GetoptLong::NO_ARGUMENT]
       )

headers = false
url     = nil
access  = nil
secret  = nil
auth    = nil
type    = nil
user_data = nil

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 '--type'
                type = arg
            when '--headers'
                headers = true
            when '--user-data'
                user_data = arg
        end
    end
rescue Exception => e
    exit -1
end 

image_id = ARGV.shift

if !image_id
    puts "#{cmd_name}: missing IMAGE-ID parameter"
    exit -1
end

if !type
    type = "m1.small"
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.run_instances(image_id,type, user_data)

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

images = rc['instancesSet']['item']

fmt = "%-12s  %-13s   %-14s %s"  

if headers
    puts fmt % ["Owner", "ImageId", "InstanceId", "InstanceType"]
    puts "------------------------------------------------------------------------------" 
end

images.each { |img|
        puts fmt % [rc['ownerId'],img['imageId'],img['instanceId'],img['instanceType']]
    }

exit 0
