Change to extract character genertion.

piotrmurach · Jul 13, 2016 · b2198fd · b2198fd
1 parent 87b204c
commit b2198fd
Showing 1 changed file with 16 additions and 9 deletions.
diff --git a/lib/tty/spinner.rb b/lib/tty/spinner.rb
@@ -211,15 +211,7 @@ def stop(stop_message = '')
       end
       return clear_line if @clear
 
-      char = if success?
-               @success_mark
-             elsif error?
-               @error_mark
-             else
-               @frames[@current - 1]
-             end
-      data = message.gsub(MATCHER, char)
-
+      data = message.gsub(MATCHER, next_char)
       if !stop_message.empty?
         data << ' ' + stop_message
       end
@@ -234,6 +226,21 @@ def stop(stop_message = '')
       kill
     end
 
+    # Retrieve next character
+    #
+    # @return [String]
+    #
+    # @api private
+    def next_char
+      if success?
+        @success_mark
+      elsif error?
+        @error_mark
+      else
+        @frames[@current - 1]
+      end
+    end
+
     # Finish spinning and set state to :success
     #
     # @api public