nehm 1.6.1 → 2.0.1

data/lib/nehm/http_client.rb

@@ -0,0 +1,66 @@
+require 'json'
+require 'net/http'
+
+module Nehm
+
+  class HTTPClient
+
+    ##
+    # SoundCloud API client ID
+
+    CLIENT_ID = '11a37feb6ccc034d5975f3f803928a32'
+
+    ##
+    # Exception classes
+
+    class Status404 < StandardError; end
+    class ConnectionError < StandardError; end
+
+    def get(api_version, uri_string)
+      uri = form_uri(api_version, uri_string)
+      get_hash(uri)
+    end
+
+    def resolve(url)
+      response = get(1, "/resolve?url=#{url}")
+
+      errors = response['errors']
+      if errors
+        if errors[0]['error_message'] =~ /404/
+          raise Status404
+        else
+          raise ConnectionError # HACK
+        end
+      end
+
+      if response['status'] =~ /302/
+        get_hash(response['location'])
+      end
+    end
+
+    private
+
+    def form_uri(api_version, uri_string)
+      uri =
+        case api_version
+        when 1
+          'https://api.soundcloud.com'
+        when 2
+          'https://api-v2.soundcloud.com'
+        end
+      uri += uri_string
+      uri += "&client_id=#{CLIENT_ID}" if api_version == 1
+
+      uri
+    end
+
+    def get_hash(uri)
+      uri = URI.parse(URI.escape(uri))
+      http = Net::HTTP.new(uri.host, uri.port)
+      http.use_ssl = true
+      res = http.get(uri.request_uri)
+      JSON.parse(res.body)
+    end
+
+  end
+end

data/lib/nehm/menu.rb

@@ -0,0 +1,96 @@
+module Nehm
+  module UI
+    class Menu
+
+      attr_writer :msg_bar, :header
+
+      def initialize
+        @choices = {}
+        @inc_index = 1
+        @items = []
+
+        yield self
+        select
+      end
+
+      def choice(index, desc, &block)
+        # Visual index - index that you see in menu
+        # Select index - index than can be selected
+        # For example, if you use ':added' index
+        # In menu you see 'A', but you can select it by number
+        # You receive a warning though
+
+        visual_index = select_index = index
+
+        if index == :inc
+          visual_index = select_index = @inc_index.to_s
+          @inc_index += 1
+        end
+
+        if index == :added
+          visual_index = 'A'.green
+          select_index = @inc_index.to_s
+          @inc_index += 1
+        end
+
+        @choices[select_index] = block
+        @items << "#{visual_index} #{desc}"
+      end
+
+      def show_header
+        UI.say @header
+        UI.newline
+      end
+
+      ##
+      # Message bar used to show messages of
+      # last successful or not operations
+      # Shown before menu
+
+      def show_msg_bar
+        UI.say 'Message:'
+        UI.say "  #{@msg_bar}"
+        UI.newline
+        @msg_bar.clear
+      end
+
+      def newline
+        @items << "\n"
+      end
+
+      def select
+        show_header unless @header.to_s.empty?
+        show_msg_bar unless @msg_bar.to_s.empty?
+
+        # Add exit option
+        newline
+        choice('e', 'Exit'.red) { UI.term }
+
+        # Output items
+        @items.each do |item|
+          UI.say item
+        end
+
+        UI.newline
+
+        selected = UI.ask('Enter option'.yellow.freeze)
+        call_selected_block(selected)
+
+        UI.newline
+      end
+
+      def call_selected_block(selected)
+        loop do
+          if @choices.keys.include? selected
+            block = @choices[selected]
+            block.call
+            break
+          else
+            selected = UI.ask "You must choose one of [#{@choices.keys.join(', ')}]"
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/nehm/option_parser.rb

@@ -3,8 +3,6 @@ module Nehm
   ##
   # OptionParser parses options and add hash with options to specified command
 
-  # TODO: make abbrebiated options
-
   class OptionParser
 
     def initialize(args, command)

data/lib/nehm/os.rb

@@ -6,8 +6,8 @@ module Nehm
   # OS module returns information about OS on this computer
 
   module OS
-    def self.linux?
-      RbConfig::CONFIG['host_os'] =~ /linux/ ? true : false
+    def self.mac?
+      RbConfig::CONFIG['host_os'] =~ /darwin|mac os/ ? true : false
     end
   end
 end

data/lib/nehm/path_manager.rb

@@ -1,13 +1,10 @@
 module Nehm
 
   ##
-  # Path manager works with download pathes
+  # Path manager works with download paths
 
   module PathManager
 
-    ##
-    # Returns default download path (contains in ~/.nehmconfig)
-
     def self.default_dl_path
       Cfg[:dl_path]
     end
@@ -16,13 +13,10 @@ module Nehm
     # Checks path for validation and returns it if valid
 
     def self.get_path(path)
-      # If path begins with ~
-      path = tilde_to_home(path) if tilde_at_top?(path)
-
       # Check path for existence
       UI.term 'Invalid download path! Please enter correct path' unless Dir.exist?(path)
 
-      path
+      File.expand_path(path)
     end
 
     def self.set_dl_path
@@ -31,21 +25,18 @@ module Nehm
         default_path = File.join(ENV['HOME'], '/Music')
 
         if Dir.exist?(default_path)
-          ask_sentence << " (press enter to set it to #{default_path.magenta})"
+          ask_sentence << " (press Enter to set it to #{default_path.magenta})"
         else
           default_path = nil
         end
 
-        path = HighLine.new.ask(ask_sentence + ':')
+        path = UI.ask(ask_sentence + ':')
 
         # If user press enter, set path to default
         path = default_path if path == '' && default_path
 
-        # If tilde at top of the line of path
-        path = PathManager.tilde_to_home(path) if PathManager.tilde_at_top?(path)
-
         if Dir.exist?(path)
-          Cfg[:dl_path] = path
+          Cfg[:dl_path] = File.expand_path(path)
           UI.say "#{'Download directory set up to'.green} #{path.magenta}"
           break
         else
@@ -54,15 +45,5 @@ module Nehm
       end
     end
 
-    module_function
-
-    def tilde_at_top?(path)
-      path[0] == '~'
-    end
-
-    def tilde_to_home(path)
-      File.join(ENV['HOME'], path[1..-1])
-    end
-
   end
 end

data/lib/nehm/playlist.rb

@@ -11,9 +11,9 @@ module Nehm
       @name = name.chomp
     end
 
-    def add_track(track_path)
+    def add_track(track)
       UI.say 'Adding to iTunes'
-      AppleScript.add_track_to_playlist(track_path, @name)
+      AppleScript.add_track_to_playlist(track.file_path, @name)
     end
 
     def to_s

data/lib/nehm/playlist_manager.rb

@@ -1,3 +1,4 @@
+require 'nehm/applescript'
 require 'nehm/playlist'
 
 module Nehm
@@ -7,11 +8,8 @@ module Nehm
 
   module PlaylistManager
 
-    ##
-    # Returns default iTunes playlist (contains in ~/.nehmconfig)
-
     def self.default_playlist
-      default_user_playlist || music_master_library unless OS.linux?
+      default_user_playlist || music_master_library if OS.mac?
     end
 
     ##
@@ -27,7 +25,7 @@ module Nehm
 
     def self.set_playlist
       loop do
-        playlist = HighLine.new.ask('Enter name of default iTunes playlist to that you want add tracks (press Enter to set it to default iTunes Music library):')
+        playlist = UI.ask('Enter name of default iTunes playlist to that you want add tracks (press Enter to set it to default iTunes Music library):')
 
         # If entered nothing, unset iTunes playlist
         if playlist == ''

data/lib/nehm/track.rb

@@ -42,7 +42,7 @@ module Nehm
 
       separators = [' - ', ' ~ ']
       separators.each do |sep|
-        return title.split(sep) if title.include? sep
+        return title.split(sep, 2) if title.include? sep
       end
 
       [@hash['user']['username'], title]
@@ -53,7 +53,9 @@ module Nehm
     end
 
     def url
-      "#{@hash['stream_url']}?client_id=#{Client::CLIENT_ID}"
+      # API V2 track hash has no 'stream_url' but 'uri'
+      dl_url = @hash['uri'] ? "#{@hash['uri']}/stream" : @hash['stream_url']
+      "#{dl_url}?client_id=#{HTTPClient::CLIENT_ID}"
     end
 
     def year

data/lib/nehm/track_manager.rb

@@ -0,0 +1,135 @@
+require 'taglib'
+
+require 'nehm/track'
+
+module Nehm
+
+  class TrackManager
+
+    def initialize(options)
+      setup_environment(options)
+    end
+
+    def process_tracks(tracks)
+      tracks.reverse_each do |track|
+        UI.newline
+        dl(track)
+        tag(track)
+        track.artwork.suicide
+        @playlist.add_track(track) if @playlist
+        UI.newline
+      end
+      UI.success 'Done!'
+    end
+
+    def likes(limit, offset)
+      likes = Client.tracks(limit, offset, :likes, @uid)
+      return nil if likes.empty?
+
+      filter(likes)
+      convert(likes)
+    end
+
+    def posts(limit, offset)
+      posts = Client.tracks(limit, offset, :posts, @uid)
+      return nil if posts.empty?
+
+      posts.reject! { |hash| hash['type'] == 'playlist' }
+      posts.map! { |hash| hash['track'] }
+      filter(posts)
+      convert(posts)
+    end
+
+    def track_from_url(url)
+      track = [Client.track(url)]
+      return nil if track.empty?
+
+      filter(track)
+      convert(track)
+    end
+
+    def search(query, limit, offset)
+      found = Client.search(query, limit, offset)
+      return nil if found.empty?
+
+      filter(found)
+      convert(found)
+    end
+
+    private
+
+    def dl(track)
+      # Downloading track
+      UI.say 'Downloading ' + track.full_name
+      `curl -# -o \"#{track.file_path}\" -L #{track.url}`
+
+      # Downloading artwork
+      UI.say 'Downloading artwork'
+      artwork = track.artwork
+      `curl -# -o \"#{artwork.file_path}\" -L #{artwork.url}`
+    end
+
+    def tag(track)
+      UI.say 'Setting tags'
+      TagLib::MPEG::File.open(track.file_path) do |file|
+        tag = file.id3v2_tag
+        tag.artist = track.artist
+        tag.title = track.title
+        tag.year = track.year
+
+        # Adding artwork
+        apic = TagLib::ID3v2::AttachedPictureFrame.new
+        apic.mime_type = 'image/jpeg'
+        apic.description = 'Cover'
+        apic.type = TagLib::ID3v2::AttachedPictureFrame::FrontCover
+        apic.picture = File.open(track.artwork.file_path, 'rb') { |f| f.read }
+        tag.add_frame(apic)
+
+        file.save
+      end
+    end
+
+    def setup_environment(options)
+      # Setting up user id
+      permalink = options[:from]
+      @uid = permalink ? UserManager.get_uid(permalink) : UserManager.default_uid
+      unless @uid
+        UI.error "You didn't logged in"
+        UI.say "Login from #{'nehm configure'.yellow} or use #{'[from PERMALINK]'.yellow} option"
+        UI.term
+      end
+
+      # Setting up download path
+      temp_path = options[:to]
+      dl_path = temp_path ? PathManager.get_path(temp_path) : PathManager.default_dl_path
+      if dl_path
+        ENV['dl_path'] = dl_path
+      else
+        UI.error "You don't set up download path!"
+        UI.say "Set it up from #{'nehm configure'.yellow} or use #{'[to PATH_TO_DIRECTORY]'.yellow} option"
+        UI.term
+      end
+
+      # Setting up iTunes playlist
+      @playlist = nil
+      if !options[:dl] && OS.mac?
+        playlist_name = options[:pl]
+        @playlist = playlist_name ? PlaylistManager.get_playlist(playlist_name) : PlaylistManager.default_playlist
+      end
+    end
+
+    def filter(tracks)
+      # Removing unstreamable tracks
+      first_length = tracks.length
+      tracks.select! { |hash| hash['streamable'] }
+      diff = first_length - tracks.length
+
+      UI.warning "Was skipped #{diff} undownloadable track(s)" if diff > 0
+    end
+
+    def convert(tracks)
+      tracks.map! { |hash| Track.new(hash) }
+    end
+
+  end
+end