Extract DownloadProgressBar into its own class

Author: frosas

Diff for: lib/spotify_to_mp3/app.rb

@@ -1,6 +1,6 @@
 require 'fileutils'
 require 'spotify_to_mp3/app/stream_queries'
-require 'ruby-progressbar'
+require 'spotify_to_mp3/app/download_progress_bar'
 
 module SpotifyToMp3
   class App
@@ -48,15 +48,12 @@ def run
             @grooveshark.download(
               track: track,
               on_response: Proc.new { |response|
-                win_half = $stdout.winsize[1] / 2
-                title = "[#{i.next}/#{tracks_to_download.length}] #{track}"
-                cut_title = title[0..win_half.pred].ljust win_half
-                cut_title = cut_title.gsub(/.{3}$/, '...') if title.length > win_half                
-                progress_bar = ProgressBar.create(
-                  :title => cut_title,
-                  :total => response['content-length'].to_i,
-                  :format => "%t %p%% [%B] %E"
-                )
+                progress_bar = DownloadProgressBar.new(
+                  track: track,
+                  track_number: i.next,
+                  track_size: response['content-length'].to_i,
+                  total_tracks: tracks_to_download.length
+                )                
               },
               on_body_chunk: Proc.new { |chunk|
                 progress_bar.progress += chunk.length

Diff for: lib/spotify_to_mp3/app/download_progress_bar.rb

@@ -0,0 +1,34 @@
+require 'ruby-progressbar'
+require 'forwardable'
+
+module SpotifyToMp3
+  class App
+    class DownloadProgressBar
+      extend Forwardable
+      
+      def_delegators :@progress_bar, :progress, :progress=, :finish
+      
+      def initialize(options)
+        track = options.fetch(:track)
+        track_number = options.fetch(:track_number)
+        track_size = options.fetch(:track_size)
+        total_tracks = options.fetch(:total_tracks)
+        
+        @progress_bar = ProgressBar.create(
+          title: cut_title("[#{track_number}/#{total_tracks}] #{track}"),
+          total: track_size,
+          format: "%t %p%% [%B] %E"
+        )
+      end
+      
+      private
+      
+      def cut_title(title)
+        win_half = $stdout.winsize[1] / 2
+        cut_title = title[0..win_half.pred].ljust win_half
+        cut_title.gsub(/.{3}$/, '...') if title.length > win_half
+        cut_title
+      end
+    end
+  end
+end