Merge pull request #9 from Thermatix/master

allow start method to accept blocks which it can execute and animate …

lib/tty/spinner.rb

@@ -117,16 +117,26 @@ def on(name, &block)
     #
     #
     # @api public
-    def start
+    # @param [String] stop_message
+    #   If you provide a block, this is the stop message given when the block finishes
+    # @yield Provide a block to be executed and have the spinner automatically animate for you
+    def start(stop_message=nil,&block)
       @started_at = Time.now
       sleep_time = 1.0 / @interval
-
-      @thread = Thread.new do
+      @thread =
+      Thread.new do
         while @started_at do
           spin
           sleep(sleep_time)
         end
       end
+      if block_given?
+        @thread = Thread.new(&block)
+        until(@thread.status == false)  do
+          sleep(sleep_time)
+        end
+        stop(stop_message)
+      end
     end
 
     # Duration of the spinning animation

spec/unit/start_spec.rb

@@ -12,4 +12,15 @@
 
     expect(spinner).to have_received(:spin).at_least(5).times
   end
+
+  it "accepts block and stop message and will wait and spin until provided block finishes executing" do
+    spinner = TTY::Spinner.new(output: output, interval: 100)
+    allow(spinner).to receive(:spin)
+    spinner.start "done" do
+      100000000.times do |n|
+        n * n
+      end
+    end
+    expect(spinner).to have_received(:spin).at_least(5).times
+  end
 end