rmd 0.0.1 → 0.1.0

data/lib/rmd/base/playlist.rb

@@ -0,0 +1,52 @@
+require 'mechanize'
+
+module RMD
+  module Base
+    class Playlist
+      attr_reader :errors, :link, :songs
+
+      def initialize(link)
+        @link = link
+        @errors = []
+        @songs = []
+      end
+
+      def success?
+        !!songs && songs.count > 0
+      end
+
+      def fetch; end
+
+      private
+
+      def agent
+        agent ||= Mechanize.new
+      end
+
+      def page
+        @page ||= agent.get(link)
+      end
+
+      def song_elements
+        @song_element ||= page.search(song_css)
+      end
+
+      def calculate_progress
+        if song_elements.count > 0
+          progress_bar = ProgressBar.create(
+            starting_at: 0,
+            total: song_elements.count,
+            format: '%c / %C Songs %B %p%%'
+          )
+
+          song_elements.each do |element|
+            yield element.attr('href')
+            progress_bar.increment
+          end
+        else
+          @errors << 'Can not get song lists from this playlist page.'
+        end
+      end
+    end
+  end
+end

data/lib/rmd/base/song.rb

@@ -0,0 +1,17 @@
+module RMD
+  module Base
+    class Song
+      attr_reader :errors, :link, :data_link
+
+      def initialize(link)
+        @link = link
+      end
+
+      def fetch; end
+
+      def success?
+        !!data_link && data_link != ''
+      end
+    end
+  end
+end

data/lib/rmd/downloader.rb

@@ -44,8 +44,17 @@ module RMD
     private
 
     def file_name
+      @file_name ||= uncached_file_name
+    end
+
+    def uncached_file_name
       uri = URI.parse(link)
-      File.basename(uri.path)
+      name = CGI::parse(uri.query.to_s)['filename'].first
+      if name
+        name
+      else
+        File.basename(uri.path)
+      end
     end
   end
 end

data/lib/rmd/factory.rb

@@ -1,6 +1,8 @@
 require 'rmd/song_playlist_adapter'
 require 'rmd/nct/song'
 require 'rmd/nct/playlist'
+require 'rmd/zing/song'
+require 'rmd/zing/playlist'
 
 module RMD
   class Factory
@@ -19,6 +21,15 @@ module RMD
         when /playlist/
           RMD::NCT::Playlist.new(link)
         end
+      when /mp3\.zing\.vn/
+        case link
+        when /bai-hat/
+          RMD::SongPlaylistAdapter.new(RMD::Zing::Song.new(link))
+        when /album|playlist/
+          RMD::Zing::Playlist.new(link)
+        end
+      else
+        raise 'Your url is not valid. Please check again.'
       end
     end
 

data/lib/rmd/nct/getter/key_from_page.rb

@@ -17,7 +17,7 @@ module RMD
 
         def uncached_key
           page.search('script').each do |element|
-            if match_data = /NCTNowPlaying.intFlashPlayer\("flashPlayer", "song", "(.+)"\)\;/.match(element.text)
+            if match_data = regex.match(element.text)
               return match_data.to_a.last
             end
           end
@@ -44,6 +44,10 @@ module RMD
         def element
           @element ||= response.at_xpath('.//tracklist//location')
         end
+
+        def regex
+          /NCTNowPlaying.intFlashPlayer\("flashPlayer", "song", "(.+)"\)\;/
+        end
       end
     end
   end

data/lib/rmd/nct/playlist.rb

@@ -1,63 +1,26 @@
-require 'mechanize'
+require 'rmd/base/playlist'
 
 module RMD
   module NCT
-    class Playlist
-      attr_reader :errors, :link, :songs
-
-      def initialize(link)
-        @link = link
-      end
+    class Playlist < RMD::Base::Playlist
 
       def fetch
-        if song_elements.count > 0
-          @errors = []
-          progress_bar = ProgressBar.create(
-            starting_at: 0,
-            total: song_elements.count,
-            format: '%c / %C Songs %B %p%%'
-          )
-
-          @songs = song_elements.inject([]) do |result, element|
-            link = element.attr('href')
-            song = RMD::NCT::Song.new(link)
-            song.fetch
-            if song.success?
-              result << song.data_link
-            else
-              @errors << song.errors
-            end
-
-            progress_bar.increment
-            result
+        calculate_progress do |link|
+          song = RMD::NCT::Song.new(link)
+          song.fetch
+          if song.success?
+            @songs << song.data_link
+          else
+            @errors << song.errors
           end
-        else
-          @songs = []
-          @errors = ['Can not get song lists from this playlist page.']
         end
       end
 
-      def success?
-        !!songs && songs.count > 0
-      end
-
       private
 
       def song_css
         '.item_content .name_song'
       end
-
-      def song_elements
-        @song_element ||= page.search(song_css)
-      end
-
-      def page
-        @page ||= agent.get(link)
-      end
-
-      def agent
-        agent ||= Mechanize.new
-      end
     end
   end
 end

data/lib/rmd/nct/song.rb

@@ -1,14 +1,10 @@
+require 'rmd/base/song'
 require 'rmd/nct/getter/key_from_page'
 require 'rmd/nct/getter/key_from_url'
 
 module RMD
   module NCT
-    class Song
-      attr_reader :errors, :link, :data_link
-
-      def initialize(link)
-        @link = link
-      end
+    class Song < RMD::Base::Song
 
       def fetch
         getters.each do |getter|
@@ -19,10 +15,6 @@ module RMD
         end
       end
 
-      def success?
-        !!data_link
-      end
-
       private
 
       def getters

data/lib/rmd/version.rb

@@ -1,3 +1,3 @@
 module Rmd
-  VERSION = "0.0.1"
+  VERSION = "0.1.0"
 end

data/lib/rmd/zing/playlist.rb

@@ -0,0 +1,21 @@
+require 'rmd/base/playlist'
+require 'rmd/zing/utils/correct_url'
+
+module RMD
+  module Zing
+    class Playlist < RMD::Base::Playlist
+
+      def fetch
+        calculate_progress do |link|
+          @songs << RMD::Zing::Utils::CorrectUrl.correct(link)
+        end
+      end
+
+      private
+
+      def song_css
+        '._btnDownload'
+      end
+    end
+  end
+end

data/lib/rmd/zing/song.rb

@@ -0,0 +1,45 @@
+require 'mechanize'
+require 'rmd/zing/utils/correct_url'
+
+module RMD
+  module Zing
+    class Song < RMD::Base::Song
+
+      def fetch
+        if new_link
+          @data_link = RMD::Zing::Utils::CorrectUrl.correct(new_link)
+        else
+          @errors = 'Can not get song from this link!'
+        end
+      end
+
+      private
+
+      def agent
+        @agent ||= Mechanize.new
+      end
+
+      def page
+        @page ||= agent.get(link)
+      end
+
+      def regex
+        /http\:\/\/mp3.zing.vn\/download\/song\/\S+\/\w+/
+      end
+
+      def new_link
+        @new_link ||= uncached_new_link
+      end
+
+      def uncached_new_link
+        page.search('.detail-function script').each do |element|
+          if match_data = regex.match(element.text)
+            return match_data.to_s
+          end
+        end
+
+        nil
+      end
+    end
+  end
+end

data/lib/rmd/zing/utils/correct_url.rb

@@ -0,0 +1,43 @@
+module RMD
+  module Zing
+    module Utils
+      class CorrectUrl
+        attr_reader :url
+
+        def initialize(url)
+          @url = url
+        end
+
+        def correct
+          if redirect_location
+            URI.escape(redirect_location)
+          else
+            url
+          end
+        end
+
+        def self.correct(url)
+          new(url).correct
+        end
+
+        private
+
+        def uri
+          @uri ||= URI.parse(url)
+        end
+
+        def http_client
+          @http_client ||= Net::HTTP.start(uri.host)
+        end
+
+        def header_response
+          @header_response ||= http_client.head(uri.path)
+        end
+
+        def redirect_location
+          header_response['location']
+        end
+      end
+    end
+  end
+end

data/rmd.gemspec

@@ -24,6 +24,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'colorize', '~> 0.7'
 
   spec.add_development_dependency "bundler", "~> 1.6"
+  spec.add_development_dependency "rake", "~> 10.1"
   spec.add_development_dependency "rspec", '~> 3.1'
   spec.add_development_dependency "webmock", '~> 1.20'
   spec.add_development_dependency "vcr", '~> 2.9'

data/spec/acceptants/main_spec.rb

@@ -0,0 +1,37 @@
+require 'spec_helper'
+
+describe RMD::Main do
+  describe '#download' do
+    let(:run_method) { RMD::Processor.process(link) }
+    let(:file_path) { File.expand_path("../../../#{file_name}", __FILE__) }
+
+    context 'when downloads song from nhaccuatui' do
+      let(:link) { 'http://www.nhaccuatui.com/bai-hat/honjitsu-mijukumono-juken-no-kamisama-ost-tv-version-tokio.tFBwsJKDGQX9.html' }
+      let(:file_name) { 'HonjitsuMijukumonoJukenNoKamisamaOSTTVVersion-TOKIO-3164566.mp3' }
+      let(:scenario) { 'nhaccuatui song'}
+      it_behaves_like 'download'
+    end
+
+    context 'when downloads playlist from nhaccuatui' do
+      let(:link) { 'http://www.nhaccuatui.com/playlist/dev-playlist-dang-cap-nhat.jn0g1fg4Z6UO.html' }
+      let(:file_name) { 'BlazeTvSize-KajiuraYuki_35ax5_hq.mp3' }
+      let(:file_path) { File.expand_path("../../../#{file_name}", __FILE__) }
+      let(:scenario) { 'nhaccuatui playlist' }
+      it_behaves_like 'download'
+    end
+
+    context 'when downloads song from zing mp3' do
+      let(:link) { 'http://mp3.zing.vn/bai-hat/Bird-TV-Size-Yuya-Matsushita/ZWZCO98B.html' }
+      let(:file_name) { 'Bird TV Size - Yuya Matsushita.mp3' }
+      let(:scenario) { 'zing song' }
+      it_behaves_like 'download'
+    end
+
+    context 'when downloads playlist from zing mp3' do
+      let(:link) { 'http://mp3.zing.vn/playlist/dev-playlist-zid-sincepast/IO0E698Z.html' }
+      let(:file_name) { 'sharp TV Size - Negoto.mp3' }
+      let(:scenario) { 'zing playlist' }
+      it_behaves_like 'download'
+    end
+  end
+end

data/spec/rmd/downloader_spec.rb

@@ -14,20 +14,26 @@ describe RMD::Downloader do
 
   describe '#download' do
     let(:downloader) { described_class.new(link) }
+    let(:run_method) { downloader.download }
+    let(:scenario) { 'song download' }
     let(:link) { 'http://db.gamefaqs.com/portable/3ds/file/fire_emblem_awakening_shop.txt' }
     let(:file_name) { 'fire_emblem_awakening_shop.txt' }
-    let(:content_length) { body.length }
     let(:file_path) { File.expand_path("../../../#{file_name}", __FILE__) }
-    let(:result) { File.read(file_path) }
+    it_behaves_like 'download'
+  end
 
-    it 'downloads' do
-      VCR.use_cassette("song download") do
-        expect {
-          downloader.download
-        }.to output("\e[0;32;49m#{file_name}\e[0m\n").to_stdout
-        expect(File.exists?(file_path)).to eq true
-      end
-      File.delete(file_path)
+  describe '#file_name' do
+    let(:downloader) { described_class.new(link) }
+    subject { downloader.send(:file_name) }
+
+    context 'when link does not have file name' do
+      let(:link) { 'www.example.com/playlist/abc.mp3' }
+      it { is_expected.to eq 'abc.mp3' }
+    end
+
+    context 'when link have file name' do
+      let(:link) { 'www.example.com/playlist/abc.mp3?filename=abc%20xyz.mp3' }
+      it { is_expected.to eq 'abc xyz.mp3' }
     end
   end
 end