vnehm 1.1

data/lib/vnehm/commands/help_command.rb

@@ -0,0 +1,85 @@
+module Vnehm
+  class HelpCommand < Command
+
+    SPACES_BTWN_NAME_AND_DESC = 3
+
+    def execute
+      command_name = options[:args].pop
+      if command_name.nil?
+        UI.say HELP
+        UI.term
+      end
+
+      @cmd = CommandManager.find_command(command_name)
+
+      show_usage
+      show_summary
+      show_arguments unless @cmd.arguments.empty?
+      show_options unless @cmd.options.empty?
+    end
+
+    def arguments
+      { 'КОМАНДА' => 'имя команды (может быть сокращено), справку которой нужно показать' }
+    end
+
+    def program_name
+      'vnehm help'
+    end
+
+    def summary
+      'Показ справки для определенной команды'
+    end
+
+    def usage
+      "#{program_name} КОМАНДА"
+    end
+
+    private
+
+    def find_longest_name(names)
+      names.inject do |longest, word|
+        word.length > longest.length ? word : longest
+      end
+    end
+
+    def show_usage
+      UI.say "#{'Использование:'.yellow} #{@cmd.usage}"
+    end
+
+    def show_summary
+      UI.newline
+      UI.say "#{'Описание:'.yellow}"
+      UI.say "  #{@cmd.summary}"
+    end
+
+    def show_arguments
+      UI.newline
+      UI.say "#{'Аргументы:'.yellow}"
+      show_info(@cmd.arguments)
+    end
+
+    def show_options
+      UI.newline
+      UI.say "#{'Опции:'.yellow}"
+      show_info(@cmd.options_descs)
+    end
+
+    def show_info(hash)
+      @longest ||= nil
+
+      unless @longest
+        names = []
+        names += @cmd.arguments.keys unless @cmd.arguments.empty?
+        names += @cmd.options_descs.keys unless @cmd.options_descs.empty?
+        @longest ||= find_longest_name(names).length
+      end
+
+      hash.each do |name, desc|
+        need_spaces = @longest - name.length
+
+        UI.say "  #{name.green}#{' ' * (need_spaces + SPACES_BTWN_NAME_AND_DESC)}#{desc}"
+      end
+    end
+
+  end
+end

data/lib/vnehm/commands/list_command.rb

@@ -0,0 +1,50 @@
+require 'vnehm/tracks_view_command'
+
+module Vnehm
+
+  ##
+  # This command gets likes/posts from user's account,
+  # Prints as menu, and downloads selected tracks
+
+  class ListCommand < TracksViewCommand
+
+    def initialize
+      super
+
+      add_option(:to, 'to ПУТЬ',
+                 'Скачивать аудиозапись(и) в ПУТЬ')
+
+      add_option(:pl, 'pl ПЛЕЙЛИСТ',
+                 "Добавлять аудиозапись(и) в плейлист iTunes'a c именем" \
+                                                                  ' ПЛЕЙЛИСТ')
+
+      add_option(:limit, 'limit ЧИСЛО',
+                 'Показывать ЧИСЛО аудиозаписей на каждой страницe')
+
+      add_option(:offset, 'offset ЧИСЛО',
+                 'Показывать с ЧИСЛО+1 аудиозаписи')
+
+      add_option(:dl, 'dl yes',
+                 'Не добавлять аудиозапись(и) в iTunes. Просто скачать их')
+    end
+
+    def program_name
+      'vnehm list'
+    end
+
+    def summary
+      'Вывод списка ваших аудиозаписей из VK и загрузка выбранных треков'
+    end
+
+    def usage
+      "#{program_name} [ОПЦИИ]"
+    end
+
+    protected
+
+    def get_tracks
+      @track_manager.tracks(@limit, @offset)
+    end
+
+  end
+end

data/lib/vnehm/commands/search_command.rb

@@ -0,0 +1,67 @@
+require 'vnehm/tracks_view_command'
+
+module Vnehm
+
+  class SearchCommand < TracksViewCommand
+
+    def initialize
+      super
+
+      add_option(:"-t", '-t ПУТЬ',
+                 'Скачивать аудиозапись(и) в ПУТЬ')
+
+      add_option(:"-pl", '-pl ПЛЕЙЛИСТ',
+                 "Добавлять аудиозапись(и) в плейлист iTunes'a c именем" \
+                                                                  ' ПЛЕЙЛИСТ')
+
+      add_option(:"-lim", '-lim ЧИСЛО',
+                 'Показывать ЧИСЛО аудиозаписей на каждой страницe')
+
+      add_option(:"-dl", '-dl yes',
+                 'Не добавлять аудиозаписи в iTunes. Просто скачать их')
+
+    end
+
+    def execute
+      # Convert dash-options to normal options
+      options_to_convert = { :"-t"   => :to,
+                             :"-pl"  => :pl,
+                             :"-lim" => :limit,
+                             :"-dl"  => :dl }
+
+      options_to_convert.each do |k,v|
+        value = @options[k]
+        @options.delete(k)
+        @options[v] = value unless value.nil?
+      end
+
+      @query = @options[:args].join(' ')
+      super
+    end
+
+    def arguments
+      { 'ЗАПРОС' => 'Искать аудиозаписи по ЗАПРОСу' }
+    end
+
+    def program_name
+      'vnehm search'
+    end
+
+    def summary
+      'Поиск, вывод и загрузка определенных аудиозаписей по запросу'
+    end
+
+    def usage
+      "#{program_name} ЗАПРОС [ОПЦИИ]"
+    end
+
+    protected
+
+    def get_tracks
+      UI.term 'Вы должны ввести запрос' if @query.empty?
+
+      @track_manager.search(@query, @limit, @offset)
+    end
+
+  end
+end

data/lib/vnehm/commands/version_command.rb

@@ -0,0 +1,23 @@
+require 'vnehm/version'
+
+module Vnehm
+  class VersionCommand < Command
+
+    def execute
+      UI.say VERSION
+    end
+
+    def program_name
+      'vnehm version'
+    end
+
+    def summary
+      'Вывод версии приложения'
+    end
+
+    def usage
+      program_name
+    end
+
+  end
+end

data/lib/vnehm/menu.rb

@@ -0,0 +1,78 @@
+module Vnehm
+  module UI
+    class Menu
+
+      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 header=(value)
+        @items.unshift(value)
+      end
+
+      def newline
+        @items << "\n"
+      end
+
+      def select
+        # Add exit option
+        newline
+        choice('e', 'Выйти'.red) { UI.term }
+
+        # Output items
+        @items.each do |item|
+          UI.say item
+        end
+
+        UI.newline
+
+        selected = UI.ask('Введите опцию:'.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 unless block.nil?
+            break
+          else
+            selected = UI.ask "Вы должны выбрать одно из [#{@choices.keys.join(', ')}]"
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/vnehm/option_parser.rb

@@ -0,0 +1,29 @@
+module Vnehm
+
+  ##
+  # OptionParser parses options and add hash with options to specified command
+
+  class OptionParser
+
+    def initialize(args, command)
+      @args = args
+      @command = command
+    end
+
+    def parse
+      options = @command.options.keys.map(&:to_s)
+      options.each do |option|
+        if @args.include? option
+          index = @args.index(option)
+          value = @args[index + 1]
+          @args.delete_at(index + 1)
+          @args.delete_at(index)
+
+          @command.options[option.to_sym] = value
+        end
+      end
+      @command.options[:args] = @args
+    end
+
+  end
+end

data/lib/vnehm/os.rb

@@ -0,0 +1,13 @@
+require 'rbconfig'
+
+module Vnehm
+
+  ##
+  # OS module returns information about OS on this computer
+
+  module OS
+    def self.mac?
+      RbConfig::CONFIG['host_os'] =~ /darwin|mac os/ ? true : false
+    end
+  end
+end

data/lib/vnehm/path_manager.rb

@@ -0,0 +1,49 @@
+module Vnehm
+
+  ##
+  # Path manager works with download paths
+
+  module PathManager
+
+    def self.default_dl_path
+      Cfg[:dl_path]
+    end
+
+    ##
+    # Checks path for validation and returns it if valid
+
+    def self.get_path(path)
+      # Check path for existence
+      UI.term 'Такой папки не существует! Пожалуйста, введите корректный путь' unless Dir.exist?(path)
+
+      File.expand_path(path)
+    end
+
+    def self.set_dl_path
+      loop do
+        ask_sentence = 'Введите путь в желаемую директорию скачиваемых аудиозаписей'
+        default_path = File.join(ENV['HOME'], '/Music')
+
+        if Dir.exist?(default_path)
+          ask_sentence << " (нажмите Enter, чтобы установить #{default_path.magenta} в качестве этой директории)"
+        else
+          default_path = nil
+        end
+
+        path = UI.ask(ask_sentence + ':')
+
+        # If user press enter, set path to default
+        path = default_path if path == '' && default_path
+
+        if Dir.exist?(path)
+          Cfg[:dl_path] = File.expand_path(path)
+          UI.say "#{'Директория загружаемых аудиотреков установлена по пути'.green} #{path.magenta}"
+          break
+        else
+          UI.error 'Такой папки не существует! Пожалуйста, введите корректный путь'
+        end
+      end
+    end
+
+  end
+end

data/lib/vnehm/playlist.rb

@@ -0,0 +1,24 @@
+module Vnehm
+
+  ##
+  # iTunes playlist primitive
+
+  class Playlist
+
+    attr_reader :name
+
+    def initialize(name)
+      @name = name.chomp
+    end
+
+    def add_track(track)
+      UI.say 'Adding to iTunes'
+      AppleScript.add_track_to_playlist(track.file_path, @name)
+    end
+
+    def to_s
+      @name
+    end
+
+  end
+end

data/lib/vnehm/playlist_manager.rb

@@ -0,0 +1,66 @@
+require 'vnehm/applescript'
+require 'vnehm/playlist'
+
+module Vnehm
+
+  ##
+  # Playlist manager works with iTunes playlists
+
+  module PlaylistManager
+
+    def self.default_playlist
+      default_user_playlist || music_master_library if OS.mac?
+    end
+
+    ##
+    # Checks path for existence and returns it if exists
+
+    def self.get_playlist(playlist_name)
+      if AppleScript.list_of_playlists.include? playlist_name
+        Playlist.new(playlist_name)
+      else
+        UI.term 'Такого плейлиста не существует. Введите корректное название'
+      end
+    end
+
+    def self.set_playlist
+      loop do
+        playlist = UI.ask('Введите имя плейлиста iTunes, в который вы ' \
+                          'хотите добавлять треки (нажмите Enter, чтобы ' \
+                          'не добавлять аудиозаписи в плейлист, ' \
+                          'а в музыкальную медиатеку iTunes)')
+
+        # If entered nothing, unset iTunes playlist
+        if playlist == ''
+          Cfg[:playlist] = nil
+          UI.success 'Треки будут добавляться в музыкальную медиатеку iTunes'
+          break
+        end
+
+        if AppleScript.list_of_playlists.include? playlist
+          Cfg[:playlist] = playlist
+          UI.say 'Аудиозаписи будут добавляться в плейлист '.green +
+                                                              playlist.magenta
+          break
+        else
+          UI.error 'Такого плейлиста не существует. Введите корректное название'
+        end
+      end
+    end
+
+
+    module_function
+
+    def default_user_playlist
+      Playlist.new(Cfg[:playlist]) unless Cfg[:playlist].nil?
+    end
+
+    ##
+    # Music master library is main iTunes music library
+
+    def music_master_library
+      Playlist.new(AppleScript.music_master_library)
+    end
+
+  end
+end

data/lib/vnehm/token_manager.rb

@@ -0,0 +1,33 @@
+require 'cgi'
+
+module Vnehm
+
+  module TokenManager
+
+    def self.token
+      Cfg['token'] if Cfg.exist?
+    end
+
+    def self.authorize
+      auth_url = Client.authorization_url
+      UI.say "1. Переходите по этой ссылке с помощью вашего браузера: #{auth_url.magenta}"
+      UI.say "2. Жмите #{'Разрешить'.green}"
+      UI.say '3. Скопируйте ссылку из адресной строки вашего браузера сюда:'
+      uri = UI.ask
+
+      uri = URI(uri)
+      hash = CGI::parse(uri.fragment)
+
+      if hash['access_token']
+        Cfg['token'] = hash['access_token'].first
+      else
+        UI.term 'Вы ввели неверную ссылку! ' \
+                'Пожалуйста, авторизируйтесь ещё раз и перепроверьте введеные данные'
+      end
+
+      UI.success 'Вы успешно авторизованы!'
+    end
+
+  end
+
+end

data/lib/vnehm/track.rb

@@ -0,0 +1,51 @@
+module Vnehm
+
+  # Primitive for SoundCloud track
+
+  class Track
+
+    attr_reader :hash
+
+    def initialize(hash)
+      @hash = hash
+    end
+
+    def artist
+      CGI::unescapeHTML(@hash.artist)
+    end
+
+    def duration
+      seconds = @hash['duration']
+
+      time = Time.at(seconds)
+      time -= time.utc_offset
+
+      time.hour > 0 ? time.strftime("%H:%M:%S") : time.strftime("%M:%S")
+    end
+
+    def file_name
+      "#{full_name.tr(',./\\\'$%"', '')}.mp3"
+    end
+
+    def file_path
+      File.join(ENV['dl_path'], file_name)
+    end
+
+    def full_name
+      "#{artist} - #{title}"
+    end
+
+    def id
+      @hash.aid
+    end
+
+    def title
+      CGI::unescapeHTML(@hash.title)
+    end
+
+    def url
+      @hash.url
+    end
+
+  end
+end

data/lib/vnehm/track_manager.rb

@@ -0,0 +1,86 @@
+require 'taglib'
+
+require 'vnehm/track'
+
+module Vnehm
+
+  class TrackManager
+
+    def initialize(options)
+      setup_environment(options)
+    end
+
+    def process_tracks(tracks)
+      tracks.reverse_each do |track|
+        UI.newline
+        dl(track)
+        tag(track)
+        @playlist.add_track(track) if @playlist
+        UI.newline
+      end
+      UI.success 'Done!'
+    end
+
+    def tracks(limit, offset)
+      UI.term "Неверное значение опции 'limit'\nОно должно быть больше 0" if limit <= 0
+      UI.term "Неверное значении опции 'offset'\nОно должно быть больше или равно 0" if offset < 0
+
+      tracks = Client.tracks(limit, offset)
+      return nil if tracks.empty?
+
+      tracks.map! { |hash| Track.new(hash) }
+    end
+
+    def search(query, limit, offset)
+      UI.term "Неверное значение опции 'limit'\nОно должно быть больше 0" if limit <= 0
+      UI.term "Неверное значении опции 'offset'\nОно должно быть больше или равно 0" if offset < 0
+
+      found = Client.search(query, limit, offset)
+      return nil if found.empty?
+
+      found.shift # Deleting first item - 'count'
+
+      found.map! { |hash| Track.new(hash) }
+    end
+
+    private
+
+    def dl(track)
+      # Downloading track
+      UI.say 'Загрузка ' + track.full_name
+      `curl -# -o \"#{track.file_path}\" -L \"#{track.url}\"`
+    end
+
+    def tag(track)
+      TagLib::MPEG::File.open(track.file_path) do |file|
+        tag = file.id3v2_tag
+        tag.artist = track.artist
+        tag.title = track.title
+
+        file.save
+      end
+    end
+
+    def setup_environment(options)
+      # 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 "Вы не указали путь, куда должны скачиваться треки!"
+        UI.say "Укажите его через команду  #{'vnehm configure'.yellow} " \
+        " или используйте опцию #{'to ПУТЬ'.yellow}"
+        UI.term
+      end
+
+      # Setting up iTunes playlist
+      @playlist = nil
+      if options[:dl] != 'yes' && OS.mac?
+        playlist_name = options[:pl]
+        @playlist = playlist_name ? PlaylistManager.get_playlist(playlist_name) : PlaylistManager.default_playlist
+      end
+    end
+
+  end
+end