lib/buildbox/agent.rb in buildbox-0.5 vs lib/buildbox/agent.rb in buildbox-0.6.beta1

@@ -4,22 +4,27 @@
   class Agent
     include Celluloid
     include Celluloid::Logger
 
     def initialize(access_token, api = Buildbox::API.new)
-      @api          = api
-      @access_token = access_token
+      @api           = api
+      @access_token  = access_token
+      @uploader_pool = Uploader.pool(:size => 10) # upload 10 things at a time
     end
 
     def process
       return if @current_build
 
       if @current_build = next_build
         @api.update_build(@access_token, @current_build, :agent_accepted => @access_token)
 
-        Monitor.new(@current_build, @access_token, @api).async.monitor
-        Runner.new(@current_build).start
+        montior = Monitor.new(@current_build, @access_token, @api).async.monitor
+        runner  = Runner.start(@current_build)
+
+        @current_build.artifact_paths.each do |path|
+          upload_artifacts_from(runner.build_directory, path)
+        end
       end
 
       @current_build = nil
     end
 
@@ -33,8 +38,25 @@
       nil
     end
 
     def hostname
       `hostname`.chomp
+    end
+
+    def upload_artifacts_from(build_directory, artifact_path)
+      files = Artifact.files_to_upload(build_directory, artifact_path)
+
+      files.each_pair do |relative_path, absolute_path|
+        artifact = @api.create_artifact(@access_token, @current_build,
+                                        :path => relative_path,
+                                        :file_size => File.size(absolute_path))
+
+        @uploader_pool.upload(artifact[:uploader], absolute_path) do |state, response|
+          @api.update_artifact(@access_token, @current_build, artifact[:id], :state => state)
+        end
+      end
+    rescue => e
+      error "There was an error uploading artifacts for path: #{artifact_path} (#{e.class.name}: #{e.message})"
+      e.backtrace[0..3].each { |line| error(line) }
     end
   end
 end