groove-dl 0.1.0

data/lib/groove-dl/errors.rb

@@ -0,0 +1,8 @@
+# -*- coding: utf-8 -*-
+module GrooveDl
+  # Errors module
+  module Errors
+    class AlreadyDownloaded < StandardError
+    end
+  end
+end

data/lib/groove-dl/version.rb

@@ -0,0 +1,5 @@
+# -*- coding: utf-8 -*-
+# GrooveDl version
+module GrooveDl
+  VERSION = '0.1.0'
+end

data/lib/groove-dl/widgets/download/bar.rb

@@ -0,0 +1,87 @@
+# -*- coding: utf-8 -*-
+module GrooveDl
+  # Widgets components
+  module Widgets
+    # Download section
+    module Download
+      # Download bar
+      class Bar < Gtk::Box
+        attr_reader :type, :query
+
+        ##
+        # Initialize widgets
+        #
+        # @param [Grooveshark::Client] client Grooveshark client
+        # @param [Gtk::Window] window Gtk app
+        #
+        def load(_client, window)
+          set_name('download_bar')
+
+          download_box = Gtk::Box.new(:horizontal, 6)
+
+          add_button = Gtk::Button.new(label: 'Add to queue',
+                                       stock_id: Gtk::Stock::SAVE)
+
+          add_button.signal_connect('released') do
+            l = window.find_by_name('download_list')
+            l.store.clear
+            l.create_model(window.find_by_name('search_list').selection)
+          end
+
+          download_box.pack_start(add_button,
+                                  expand: false,
+                                  fill: true,
+                                  padding: 5)
+
+          directory_chooser =
+            Gtk::FileChooserButton.new('Select directory',
+                                       Gtk::FileChooser::Action::SELECT_FOLDER)
+          directory_chooser.filename = Dir.tmpdir
+          directory_chooser.set_name('directory_chooser')
+          download_box.pack_start(directory_chooser,
+                                  expand: true,
+                                  fill: true,
+                                  padding: 5)
+
+          concurrency_entry = Gtk::Entry.new
+          concurrency_entry.set_name('concurrency_entry')
+          concurrency_entry.text = '5'
+
+          concurrency_label = Gtk::Label.new('Concurrency', true)
+          concurrency_label.mnemonic_widget = concurrency_entry
+
+          download_box.pack_start(concurrency_label,
+                                  expand: false,
+                                  fill: false,
+                                  padding: 5)
+          download_box.pack_start(concurrency_entry,
+                                  expand: false,
+                                  fill: false,
+                                  padding: 5)
+
+          concurrency_entry.signal_connect('changed') do
+            value = concurrency_entry.text.to_i
+            concurrency_entry.text = value.to_s unless value == 0
+          end
+
+          download_button = Gtk::Button.new(label: 'Download',
+                                            stock_id: Gtk::Stock::SAVE)
+
+          download_button.signal_connect('released') do
+            download_button.sensitive = false
+            window.find_by_name('download_list').download
+          end
+
+          download_box.pack_start(download_button,
+                                  expand: false,
+                                  fill: true,
+                                  padding: 5)
+
+          pack_start(download_box,
+                     expand: false,
+                     padding: 10)
+        end
+      end
+    end
+  end
+end

data/lib/groove-dl/widgets/download/book.rb

@@ -0,0 +1,73 @@
+# -*- coding: utf-8 -*-
+module GrooveDl
+  # Widgets components
+  module Widgets
+    # Download section
+    module Download
+      # Download book
+      class Book < Gtk::Notebook
+        attr_reader :window
+
+        QUEUE = 'Queue (%d)'
+        SUCCESS = 'Success (%d)'
+        FAILED = 'Failed (%d)'
+
+        ##
+        # Initialize widgets
+        #
+        # @param [Grooveshark::Client] client Grooveshark client
+        # @param [Gtk::Window] window Gtk app
+        #
+        def load(client, window)
+          @window = window
+          set_name('download_book')
+          set_tab_pos(Gtk::PositionType::TOP)
+
+          # Download list
+          download_list = Widgets::Download::List::Queue.new(:vertical, 6)
+          download_list.set_name('download_list')
+          download_list.load(client, window)
+
+          @download_label = Gtk::Label.new(QUEUE % 0)
+          @download_label.set_name('download_label')
+          append_page(download_list, @download_label)
+
+          # Success list
+          download_success_list =
+            Widgets::Download::List::Success.new(:vertical, 6)
+          download_success_list.set_name('download_success_list')
+          download_success_list.load(client, window)
+
+          @download_success_label = Gtk::Label.new(SUCCESS % 0)
+          @download_success_label.set_name('download_success_label')
+          append_page(download_success_list, @download_success_label)
+
+          # Failed list
+          download_failed_list =
+            Widgets::Download::List::Failed.new(:vertical, 6)
+          download_failed_list.set_name('download_failed_list')
+          download_failed_list.load(client, window)
+
+          @download_failed_label = Gtk::Label.new(FAILED % 0)
+          @download_failed_label.set_name('download_failed_label')
+          append_page(download_failed_list, @download_failed_label)
+        end
+
+        ##
+        # Set label for page in notebook element
+        #
+        # @param [String] type Page type
+        # @param [Integer] nb Number of element in this page
+        #
+        def set_label(type, nb)
+          element = @download_label if type == 'QUEUE'
+          element = @download_success_label if type == 'SUCCESS'
+          element = @download_failed_label if type == 'FAILED'
+
+          return if element.nil?
+          element.set_text(Book.const_get(type.upcase) % nb)
+        end
+      end
+    end
+  end
+end

data/lib/groove-dl/widgets/download/list/failed.rb

@@ -0,0 +1,75 @@
+# -*- coding: utf-8 -*-
+module GrooveDl
+  # Widgets components
+  module Widgets
+    # Download section
+    module Download
+      # List page
+      module List
+        # Failed tree
+        class Failed < Gtk::Box
+          attr_reader :store, :data
+
+          COLUMN_PATH,
+          COLUMN_REASON = *(0..2).to_a
+
+          ##
+          # Initialize widgets
+          #
+          # @param [Grooveshark::Client] client Grooveshark client
+          # @param [Gtk::Window] window Gtk app
+          #
+          def load(_client, _window)
+            @data = {}
+            sw = Gtk::ScrolledWindow.new
+            sw.shadow_type = Gtk::ShadowType::ETCHED_IN
+            sw.set_policy(Gtk::PolicyType::AUTOMATIC,
+                          Gtk::PolicyType::AUTOMATIC)
+            pack_start(sw, expand: true, fill: true, padding: 0)
+
+            @store = Gtk::ListStore.new(String, String)
+            treeview = Gtk::TreeView.new(@store)
+            treeview.rules_hint = true
+
+            sw.add(treeview)
+
+            add_columns(treeview)
+          end
+
+          ##
+          # Append row in list store
+          #
+          # @param [Gtk::TreeIter] i Iterable element
+          # @param [String] reason Why this download have failed
+          #
+          def create_model(i, reason)
+            iter = @store.append
+            path = i[COLUMN_PATH]
+            iter[COLUMN_PATH] = path
+            iter[COLUMN_REASON] = reason
+          end
+
+          ##
+          # Add columns on the treeview element
+          #
+          # @param [Gtk::Treeview] treeview Treeview
+          #
+          def add_columns(treeview)
+            renderer = Gtk::CellRendererText.new
+            column = Gtk::TreeViewColumn.new('Path',
+                                             renderer,
+                                             'text' => COLUMN_PATH)
+            column.fixed_width = 650
+            treeview.append_column(column)
+
+            renderer = Gtk::CellRendererText.new
+            column = Gtk::TreeViewColumn.new('Reason',
+                                             renderer,
+                                             'text' => COLUMN_REASON)
+            treeview.append_column(column)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/groove-dl/widgets/download/list/queue.rb

@@ -0,0 +1,171 @@
+# -*- coding: utf-8 -*-
+module GrooveDl
+  # Widgets components
+  module Widgets
+    # Download section
+    module Download
+      # List page
+      module List
+        # Queue tree
+        class Queue < Gtk::Box
+          attr_reader :store, :data
+          attr_writer :downloader, :client
+          attr_accessor :failed, :success, :queue
+
+          COLUMN_PATH,
+          COLUMN_PGBAR_VALUE,
+          COLUMN_PGBAR_TEXT = *(0..2).to_a
+
+          ##
+          # Initialize widgets
+          #
+          # @param [Grooveshark::Client] client Grooveshark client
+          # @param [Gtk::Window] window Gtk app
+          #
+          def load(client, window)
+            @success = 0
+            @failed = 0
+            @queue = 0
+            @client = client
+            @window = window
+            @data = {}
+            @downloader = GrooveDl::Downloader.new(@client)
+            @downloader.type = 'gui'
+
+            sw = Gtk::ScrolledWindow.new
+            sw.shadow_type = Gtk::ShadowType::ETCHED_IN
+            sw.set_policy(Gtk::PolicyType::AUTOMATIC,
+                          Gtk::PolicyType::AUTOMATIC)
+            pack_start(sw, expand: true, fill: true, padding: 0)
+
+            @store = Gtk::ListStore.new(String, Integer, String)
+            create_model
+            treeview = Gtk::TreeView.new(@store)
+            treeview.rules_hint = true
+
+            sw.add(treeview)
+
+            add_columns(treeview)
+          end
+
+          def create_model(data = {})
+            data.each do |id, element|
+              if element.is_a?(Grooveshark::Song)
+                iter = @store.append
+                iter[COLUMN_PATH] =
+                  @downloader.build_path(@window
+                                           .find_by_name('directory_chooser')
+                                           .filename,
+                                         element)
+                iter[COLUMN_PGBAR_VALUE] = 0
+                iter[COLUMN_PGBAR_TEXT] = nil
+                @data[element.id] = { iter: iter, song: element }
+              else
+                playlist = Grooveshark::Playlist.new(@client,
+                                                     'playlist_id' => id)
+                result = {}
+                playlist.load_songs.each do |song|
+                  result[song.id] = song
+                end
+
+                create_model(result)
+              end
+            end
+
+            return if @data.empty?
+            @queue = @data.count
+            @window.find_by_name('download_book').set_label('QUEUE', @queue)
+          end
+
+          ##
+          # Add columns on the treeview element
+          #
+          # @param [Gtk::Treeview] treeview Treeview
+          #
+          def add_columns(treeview)
+            renderer = Gtk::CellRendererText.new
+            column = Gtk::TreeViewColumn.new('Path',
+                                             renderer,
+                                             'text' => COLUMN_PATH)
+            column.fixed_width = 650
+            treeview.append_column(column)
+
+            renderer = Gtk::CellRendererProgress.new
+            column = Gtk::TreeViewColumn.new('Progress',
+                                             renderer,
+                                             value: COLUMN_PGBAR_VALUE,
+                                             text: COLUMN_PGBAR_TEXT)
+            column.fixed_width = 100
+            treeview.append_column(column)
+          end
+
+          ##
+          # Download files in queue
+          #
+          def download
+            concurrency = @window.find_by_name('concurrency_entry').text.to_i
+            concurrency = 5 if concurrency == 0
+            Thread.abort_on_exception = true
+            Thread.new do
+              nb = 0
+              @data.each do |_id, s|
+                nb += 1
+                Thread.new do
+                  download_file(s)
+                  nb -= 1
+                end
+                sleep(0.5) until nb < concurrency
+              end
+            end
+          end
+
+          ##
+          # Download song
+          #
+          # @param [Grooveshark::Song] song Song to download
+          #
+          def download_file(song)
+            begin
+              @downloader.download(song[:song], song[:iter])
+              notify_success(song)
+            rescue Errors::AlreadyDownloaded => e
+              GrooveDl.configuration.logger.info(e.message)
+              notify_success(song)
+            rescue Grooveshark::GeneralError => e
+              GrooveDl.configuration.logger.error(e)
+              notify_error(song, e)
+            end
+
+            @window.find_by_name('download_book')
+              .set_label('QUEUE', @queue -= 1)
+            @store.remove(song[:iter])
+          end
+
+          ##
+          # Notify success
+          #
+          # @param [Grooveshark::Song] song Song displayed in success page
+          #
+          def notify_success(song)
+            @window.find_by_name('download_success_list')
+              .create_model(song[:iter])
+            @window.find_by_name('download_book')
+              .set_label('SUCCESS', @success += 1)
+          end
+
+          ##
+          # Notify erro
+          #
+          # @param [Grooveshark::Song] song Song displayed in failed page
+          #
+          def notify_error(song, e)
+            @window.find_by_name('download_failed_list')
+              .create_model(song[:iter], e.message)
+            @window.find_by_name('download_book')
+              .set_label('FAILED', @failed += 1)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/groove-dl/widgets/download/list/success.rb

@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*-
+module GrooveDl
+  # Widgets components
+  module Widgets
+    # Download section
+    module Download
+      # List page
+      module List
+        # Success tree
+        class Success < Gtk::Box
+          attr_reader :store, :data
+
+          COLUMN_PATH,
+          COLUMN_SIZE = *(0..1).to_a
+
+          ##
+          # Initialize widgets
+          #
+          # @param [Grooveshark::Client] client Grooveshark client
+          # @param [Gtk::Window] window Gtk app
+          #
+          def load(_client, _window)
+            @data = {}
+            sw = Gtk::ScrolledWindow.new
+            sw.shadow_type = Gtk::ShadowType::ETCHED_IN
+            sw.set_policy(Gtk::PolicyType::AUTOMATIC,
+                          Gtk::PolicyType::AUTOMATIC)
+            pack_start(sw, expand: true, fill: true, padding: 0)
+
+            @store = Gtk::ListStore.new(String, String)
+            treeview = Gtk::TreeView.new(@store)
+            treeview.rules_hint = true
+
+            sw.add(treeview)
+
+            add_columns(treeview)
+          end
+
+          ##
+          # Append row in the list store
+          #
+          # @param [Gtk::TreeIter] i Iterable element
+          #
+          def create_model(i)
+            iter = @store.append
+            path = i[COLUMN_PATH]
+            iter[COLUMN_PATH] = path
+            size = (File.size?(path).to_f / (1024 * 1024)).round(2)
+            iter[COLUMN_SIZE] = "#{size} MB"
+          end
+
+          ##
+          # Add columns on the treeview element
+          #
+          # @param [Gtk::Treeview] treeview Treeview
+          #
+          def add_columns(treeview)
+            renderer = Gtk::CellRendererText.new
+            column = Gtk::TreeViewColumn.new('Path',
+                                             renderer,
+                                             'text' => COLUMN_PATH)
+            column.fixed_width = 650
+            treeview.append_column(column)
+
+            renderer = Gtk::CellRendererText.new
+            column = Gtk::TreeViewColumn.new('Size',
+                                             renderer,
+                                             'text' => COLUMN_SIZE)
+            column.fixed_width = 100
+            treeview.append_column(column)
+          end
+        end
+      end
+    end
+  end
+end