# File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 94
94:     def api_url(path)
95:       "#{@plugin_config['ssl'] ? 'https' : 'http'}://#{CGI.escape(@plugin_config['username'])}:#{@plugin_config['password']}@#{@plugin_config['endpoint']}#{path}"
96:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 94
94:     def api_url(path)
95:       "#{@plugin_config['ssl'] ? 'https' : 'http'}://#{CGI.escape(@plugin_config['username'])}:#{@plugin_config['password']}@#{@plugin_config['endpoint']}#{path}"
96:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 130
130:     def compress(data)
131:       @log.trace "Compressing #{data.size / 1024} kB chunk of data..."
132: 
133:       io = StringIO.new
134: 
135:       writer = Zlib::GzipWriter.new(io, Zlib::DEFAULT_COMPRESSION, Zlib::FINISH)
136:       writer.write(data)
137:       writer.close
138: 
139:       @log.trace "Data compressed to #{io.size / 1024} kB."
140: 
141:       io.string
142:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 130
130:     def compress(data)
131:       @log.trace "Compressing #{data.size / 1024} kB chunk of data..."
132: 
133:       io = StringIO.new
134: 
135:       writer = Zlib::GzipWriter.new(io, Zlib::DEFAULT_COMPRESSION, Zlib::FINISH)
136:       writer.write(data)
137:       writer.close
138: 
139:       @log.trace "Data compressed to #{io.size / 1024} kB."
140: 
141:       io.string
142:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 70
70:     def create_remote_disk
71:       size = disk_size
72: 
73:       @log.info "Creating new #{size} GB disk..."
74: 
75: 
76:       body = hash_to_request(
77:           'size' => size * 1024 *1024 * 1024,
78:           'name' => @plugin_config['drive_name']
79:       )
80: 
81:       begin
82:         ret = response_to_hash(RestClient.post(api_url('/drives/create'), body))
83: 
84: 
85:         @log.info "Disk created with UUID: #{ret['drive']}."
86:       rescue => e
87:         @log.error e.info
88:         raise PluginError, "An error occured while creating the drive, #{e.message}. See logs for more info."
89:       end
90: 
91:       ret['drive']
92:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 70
70:     def create_remote_disk
71:       size = disk_size
72: 
73:       @log.info "Creating new #{size} GB disk..."
74: 
75: 
76:       body = hash_to_request(
77:           'size' => size * 1024 *1024 * 1024,
78:           'name' => @plugin_config['drive_name']
79:       )
80: 
81:       begin
82:         ret = response_to_hash(RestClient.post(api_url('/drives/create'), body))
83: 
84: 
85:         @log.info "Disk created with UUID: #{ret['drive']}."
86:       rescue => e
87:         @log.error e.info
88:         raise PluginError, "An error occured while creating the drive, #{e.message}. See logs for more info."
89:       end
90: 
91:       ret['drive']
92:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 180
180:     def create_server
181:       @log.info "Creating new server..."
182: 
183:       memory = ((is_cloudsigma? and @appliance_config.hardware.memory < 512) ? 512 : @appliance_config.hardware.memory)
184: 
185:       body = hash_to_request(
186:           'name' => "#{@appliance_config.name}-#{@appliance_config.version}.#{@appliance_config.release}",
187:           'cpu' => @appliance_config.hardware.cpus * 1000, # MHz
188:           'smp' => 'auto',
189:           'mem' => memory,
190:           'persistent' => 'true', # hack
191:           'ide:0:0' => @plugin_config['drive_uuid'],
192:           'boot' => 'ide:0:0',
193:           'nic:0:model' => 'e1000',
194:           'nic:0:dhcp' => 'auto',
195:           'vnc:ip' => 'auto',
196:           'vnc:password' => (0...8).map { (('a'..'z').to_a + ('A'..'Z').to_a)[rand(52)] }.join # 8 character VNC password
197:       )
198: 
199:       begin
200:         path = is_cloudsigma? ? '/servers/create' : '/servers/create/stopped'
201:         ret = response_to_hash(RestClient.post(api_url(path), body))
202: 
203:         @log.info "Server was registered with '#{ret['name']}' name as '#{ret['server']}' UUID. Use web UI or API tools to start your server."
204:       rescue => e
205:         @log.error e.info
206:         raise PluginError, "An error occured while creating the server, #{e.message}. See logs for more info."
207:       end
208:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 180
180:     def create_server
181:       @log.info "Creating new server..."
182: 
183:       memory = ((is_cloudsigma? and @appliance_config.hardware.memory < 512) ? 512 : @appliance_config.hardware.memory)
184: 
185:       body = hash_to_request(
186:           'name' => "#{@appliance_config.name}-#{@appliance_config.version}.#{@appliance_config.release}",
187:           'cpu' => @appliance_config.hardware.cpus * 1000, # MHz
188:           'smp' => 'auto',
189:           'mem' => memory,
190:           'persistent' => 'true', # hack
191:           'ide:0:0' => @plugin_config['drive_uuid'],
192:           'boot' => 'ide:0:0',
193:           'nic:0:model' => 'e1000',
194:           'nic:0:dhcp' => 'auto',
195:           'vnc:ip' => 'auto',
196:           'vnc:password' => (0...8).map { (('a'..'z').to_a + ('A'..'Z').to_a)[rand(52)] }.join # 8 character VNC password
197:       )
198: 
199:       begin
200:         path = is_cloudsigma? ? '/servers/create' : '/servers/create/stopped'
201:         ret = response_to_hash(RestClient.post(api_url(path), body))
202: 
203:         @log.info "Server was registered with '#{ret['name']}' name as '#{ret['server']}' UUID. Use web UI or API tools to start your server."
204:       rescue => e
205:         @log.error e.info
206:         raise PluginError, "An error occured while creating the server, #{e.message}. See logs for more info."
207:       end
208:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 44
44:     def disk_size
45:       size = 0
46:       @appliance_config.hardware.partitions.each_value { |partition| size += partition['size'] }
47:       size
48:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 44
44:     def disk_size
45:       size = 0
46:       @appliance_config.hardware.partitions.each_value { |partition| size += partition['size'] }
47:       size
48:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 39
39:     def execute
40:       upload
41:       create_server
42:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 39
39:     def execute
40:       upload
41:       create_server
42:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 50
50:     def hash_to_request(h)
51:       body = ""
52: 
53:       h.each do |k, v|
54:         body << "#{k} #{v.to_s}\n"
55:       end
56: 
57:       body
58:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 50
50:     def hash_to_request(h)
51:       body = ""
52: 
53:       h.each do |k, v|
54:         body << "#{k} #{v.to_s}\n"
55:       end
56: 
57:       body
58:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 175
175:     def is_cloudsigma?
176:       !@plugin_config['endpoint'].match(/cloudsigma\.com$/).nil?
177:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 175
175:     def is_cloudsigma?
176:       !@plugin_config['endpoint'].match(/cloudsigma\.com$/).nil?
177:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 60
60:     def response_to_hash(r)
61:       h = {}
62: 
63:       r.each_line do |l|
64:         h[$1] = $2 if l =~ /(\w+) (.*)/
65:       end
66: 
67:       h
68:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 60
60:     def response_to_hash(r)
61:       h = {}
62: 
63:       r.each_line do |l|
64:         h[$1] = $2 if l =~ /(\w+) (.*)/
65:       end
66: 
67:       h
68:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 98
 98:     def upload
 99:       @log.info "Uploading appliance..."
100: 
101:       # Create the disk with specific size or use already existing
102:       @plugin_config['drive_uuid'] = create_remote_disk unless @plugin_config['drive_uuid']
103: 
104:       upload_chunks
105: 
106:       @log.info "Appliance uploaded."
107:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 98
 98:     def upload
 99:       @log.info "Uploading appliance..."
100: 
101:       # Create the disk with specific size or use already existing
102:       @plugin_config['drive_uuid'] = create_remote_disk unless @plugin_config['drive_uuid']
103: 
104:       upload_chunks
105: 
106:       @log.info "Appliance uploaded."
107:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 144
144:     def upload_chunk(data, part)
145:       try = 1
146: 
147:       url = api_url("/drives/#{@plugin_config['drive_uuid']}/write/#{@step * part}")
148: 
149:       begin
150:         @log.info "Uploading part #{part+1}..."
151: 
152:         headers = {:content_type => "application/octet-stream"}
153:         headers['Content-Encoding'] = 'gzip' unless is_cloudsigma?
154: 
155:         RestClient.post url,
156:                         data,
157:                         headers
158: 
159:         @log.info "Part #{part+1} uploaded."
160:       rescue => e
161:         @log.warn "An error occured while uploading #{part} chunk, #{e.message}"
162:         try += 1
163: 
164:         unless try > @plugin_config['retry']
165:           # Let's sleep for specified amount of time
166:           sleep @plugin_config['wait']
167:           retry
168:         else
169:           @log.error e.info
170:           raise PluginError, "Couldn't upload appliance, #{e.message}."
171:         end
172:       end
173:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 144
144:     def upload_chunk(data, part)
145:       try = 1
146: 
147:       url = api_url("/drives/#{@plugin_config['drive_uuid']}/write/#{@step * part}")
148: 
149:       begin
150:         @log.info "Uploading part #{part+1}..."
151: 
152:         headers = {:content_type => "application/octet-stream"}
153:         headers['Content-Encoding'] = 'gzip' unless is_cloudsigma?
154: 
155:         RestClient.post url,
156:                         data,
157:                         headers
158: 
159:         @log.info "Part #{part+1} uploaded."
160:       rescue => e
161:         @log.warn "An error occured while uploading #{part} chunk, #{e.message}"
162:         try += 1
163: 
164:         unless try > @plugin_config['retry']
165:           # Let's sleep for specified amount of time
166:           sleep @plugin_config['wait']
167:           retry
168:         else
169:           @log.error e.info
170:           raise PluginError, "Couldn't upload appliance, #{e.message}."
171:         end
172:       end
173:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 109
109:     def upload_chunks
110:       @step = @plugin_config['chunk'] * 1024 * 1024 # in bytes
111:       part = @plugin_config['start_part']
112: 
113:       @log.info "Uploading disk in #{disk_size * 1024 / @plugin_config['chunk']} parts."
114: 
115:       File.open(@previous_deliverables.disk, 'rb') do |f|
116:         while !f.eof?
117:           f.seek(part * @step, File::SEEK_SET)
118: 
119:           data = f.read(@step)
120:           data = compress(data) unless is_cloudsigma?
121:           upload_chunk(data, part)
122: 
123:           part += 1
124:         end
125:       end
126: 
127:       @log.info "Appliance #{@appliance_config.name} uploaded to drive with UUID #{@plugin_config['drive_uuid']}."
128:     end

     # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 109
109:     def upload_chunks
110:       @step = @plugin_config['chunk'] * 1024 * 1024 # in bytes
111:       part = @plugin_config['start_part']
112: 
113:       @log.info "Uploading disk in #{disk_size * 1024 / @plugin_config['chunk']} parts."
114: 
115:       File.open(@previous_deliverables.disk, 'rb') do |f|
116:         while !f.eof?
117:           f.seek(part * @step, File::SEEK_SET)
118: 
119:           data = f.read(@step)
120:           data = compress(data) unless is_cloudsigma?
121:           upload_chunk(data, part)
122: 
123:           part += 1
124:         end
125:       end
126: 
127:       @log.info "Appliance #{@appliance_config.name} uploaded to drive with UUID #{@plugin_config['drive_uuid']}."
128:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 27
27:     def validate
28:       set_default_config_value('chunk', 64) # chunk size in MB
29:       set_default_config_value('start_part', 0) # part number to start uploading
30:       set_default_config_value('wait', 5) # wait time before retrying upload
31:       set_default_config_value('retry', 3) # number of retries
32:       set_default_config_value('ssl', false) # use SSL?
33:       set_default_config_value('drive_name', @appliance_config.name)
34: 
35:       validate_plugin_config(['endpoint', 'username', 'password'], 'http://boxgrinder.org/tutorials/boxgrinder-build-plugins/#ElasticHosts_Delivery_Plugin')
36:       raise PluginValidationError, "You can use ElasticHosts plugin with base appliances (appliances created with operating system plugins) only, see http://boxgrinder.org/tutorials/boxgrinder-build-plugins/#ElasticHosts_Delivery_Plugin." unless @previous_plugin_info[:type] == :os
37:     end

    # File lib/boxgrinder-build/plugins/delivery/elastichosts/elastichosts-plugin.rb, line 27
27:     def validate
28:       set_default_config_value('chunk', 64) # chunk size in MB
29:       set_default_config_value('start_part', 0) # part number to start uploading
30:       set_default_config_value('wait', 5) # wait time before retrying upload
31:       set_default_config_value('retry', 3) # number of retries
32:       set_default_config_value('ssl', false) # use SSL?
33:       set_default_config_value('drive_name', @appliance_config.name)
34: 
35:       validate_plugin_config(['endpoint', 'username', 'password'], 'http://boxgrinder.org/tutorials/boxgrinder-build-plugins/#ElasticHosts_Delivery_Plugin')
36:       raise PluginValidationError, "You can use ElasticHosts plugin with base appliances (appliances created with operating system plugins) only, see http://boxgrinder.org/tutorials/boxgrinder-build-plugins/#ElasticHosts_Delivery_Plugin." unless @previous_plugin_info[:type] == :os
37:     end