nicovideo 0.0.1 → 0.0.2

data/lib/nicovideo.rb

@@ -3,6 +3,7 @@ $:.unshift File.dirname(__FILE__)
 require 'nicovideo/mechanize-ext'
 require 'nicovideo/base'
 require 'nicovideo/video'
+require 'nicovideo/videopage'
 require 'nicovideo/comments'
 #require 'nicovideo/tags'
 #require 'nicovideo/ichiba'

data/lib/nicovideo/base.rb

@@ -1,6 +1,3 @@
-require 'cgi'
-require 'kconv'
-
 module Nicovideo
 
   class NicovideoArgError      < StandardError ; end
@@ -8,8 +5,6 @@ module Nicovideo
   class NicovideoNotFoundError < StandardError ; end
 
   class Base
-    NV_DEBUG_LEVEL = 0
-    TITLE_NICONICO = 'ニコニコ動画\(RC\)‐'.toutf8
 
     def initialize mail=nil, password=nil
       @mail = mail
@@ -18,9 +13,7 @@ module Nicovideo
       @agent = WWW::Mechanize.new()
       
       # for parameters current video
-      @video_id = nil
-      @params   = nil
-      @page     = nil
+      @vp = nil
       self
     end
     
@@ -31,8 +24,8 @@ module Nicovideo
         account = {'mail' => @mail, 'password' => @password }
         
         begin
-          @page = @agent.post('https://secure.nicovideo.jp/secure/login?site=niconico', account)
-          puts_debug @page.title
+          @agent.post('https://secure.nicovideo.jp/secure/login?site=niconico', account)
+          # TODO: check login success or failure
           @logged_in = true
         rescue
           @logged_in = false
@@ -43,96 +36,44 @@ module Nicovideo
     end
 
     def watch(video_id)
-      @video_id = video_id
+      @vp = get_videopage(video_id)
       if block_given?
-        yield self
+        yield @vp
       end
     end
 
-    def logged_in?()      @logged_in end
-    def comments(num=500) self.get_comments(@video_id, num) end
-    def flv()             self.get_flv @video_id end
-    def title()           self.get_title @video_id end
-    def tags()            self.get_tags @video_id end
+    def logged_in?() @logged_in end
 
-    def get_tags video_id
-      page = get_page(video_id)
-      div = page.parser.search("div#video_tags")
-      tags = div.to_html.scan(/<a href=\"tag\/[\w\%]+?\">(.+?)<\/a>/ou).inject([]) {|arr, v|
-        puts_debug v[0]
-        arr << v[0]
-      }
-      tags
+    def get_tags(video_id)
+      get_videopage(video_id).tags
     end
     
-    def get_title video_id
-      page = get_page(video_id)
-      title = page.title
-      puts_info title.toutf8.sub(/^#{TITLE_NICONICO}/ou, '')
-      title.toutf8.sub(/^#{TITLE_NICONICO}/ou, '')
+    def get_title(video_id)
+      get_videopage(video_id).title
     end
     
-    def get_video video_id=nil
-      puts_info 'getting video : id = ' + video_id
-      self.get_flv video_id
+    def get_video(video_id)
+      self.get_flv(video_id)
     end
     
-    def get_flv video_id=nil
-      puts_info 'getting flv : id = ' + video_id if video_id
-      begin
-        params = get_params video_id
-        
-        video_url = CGI.unescape(params['url'])
-        video_flv = @agent.get_file(video_url)
-        Video.new(video_id, video_flv)
-      end
+    def get_flv(video_id)
+      get_videopage(video_id).flv
     end
     
     def get_comments video_id, num=500
-      puts_info 'getting comment xml : id = ' + video_id
-      begin
-        params = get_params video_id
-        raise unless params['ms']
-        
-        thread_id = params['thread_id']
-        body = %!<thread res_from="-#{num}" version="20061206" thread="#{thread_id}" />!
-        post_url = CGI.unescape(params['ms'])
-        comment_xml = @agent.post_data(post_url, body).body
-        puts_debug comment_xml
-        Comments.new(video_id, comment_xml)
-      end
+      get_videopage(video_id).comments(num)
     end
 
     private
-    def get_params video_id=nil
+    def get_videopage(video_id)
       self.login unless @logged_in
-      
-      begin
-        if !((video_id == @video_id) && @params)
-          puts_info 'getting params : id = ' + video_id
-          @video_id = video_id
-          page = get_page video_id
-          content = @agent.get_file('http://www.nicovideo.jp/api/getflv?v=' + video_id)
-          puts_debug content
-          @params = content.scan(/([^&]+)=([^&]*)/).inject({}){|h, v| h[v[0]] = v[1]; h}
-        else
-          puts_info 'params are already in cache : id = ' + video_id
-        end
-        @params
-      rescue
-        raise NicovideoNotFoundError
-      end
-    end
 
-    def get_page video_id
-      puts_info 'getting HTML page : id = ' + video_id
-      @agent.get('http://www.nicovideo.jp/watch/' + video_id)
+      if @vp.nil? || video_id != @vp.video_id
+        @vp = VideoPage.new(@agent, video_id)
+      end
+      @vp
     end
 
-    def puts_error str ; puts str if (NV_DEBUG_LEVEL >= 1) ; end
-    def puts_info  str ; puts str if (NV_DEBUG_LEVEL >= 2) ; end
-    def puts_debug str ; puts str if (NV_DEBUG_LEVEL >= 3) ; end
-
   end
 
   def Nicovideo.new(mail, password)

data/lib/nicovideo/version.rb

@@ -2,7 +2,7 @@ module Nicovideo #:nodoc:
   module VERSION #:nodoc:
     MAJOR = 0
     MINOR = 0
-    TINY  = 1
+    TINY  = 2
 
     STRING = [MAJOR, MINOR, TINY].join('.')
   end

data/lib/nicovideo/videopage.rb

@@ -0,0 +1,94 @@
+require 'kconv'
+require 'cgi'
+
+module Nicovideo
+  class VideoPage
+    NV_DEBUG_LEVEL = 0
+    TITLE_NICONICO = 'ニコニコ動画\(RC\)‐'.toutf8
+
+    def initialize agent, video_id=nil
+      @agent    = agent
+      @video_id = video_id
+      @params   = nil
+      @page     = nil
+    end
+
+    attr_reader :video_id
+
+    def comments(num=500)
+      puts_info 'getting comment xml : id = ' + @video_id
+      begin
+        @params = @params || get_params
+        ms = @params['ms']
+        raise unless ms
+        
+        thread_id = @params['thread_id']
+        body = %!<thread res_from="-#{num}" version="20061206" thread="#{thread_id}" />!
+        post_url = CGI.unescape(ms)
+        comment_xml = @agent.post_data(post_url, body).body
+        puts_debug comment_xml
+        Comments.new(@video_id, comment_xml)
+      end
+    end
+
+    def flv() return video() end
+
+    def video()
+      begin
+        @params = @params || get_params
+        video_url = CGI.unescape(@params['url'])
+        video_flv = @agent.get_file(video_url)
+        Video.new(@video_id, video_flv)
+      end
+    end
+
+    def tags()
+      @page = @page || get_page
+      div = @page.parser.search("div#video_tags")
+      tags = div.to_html.scan(/<a href=\"tag\/[\w\%]+?\">(.+?)<\/a>/ou).inject([]) {|arr, v|
+        puts_debug v[0]
+        arr << v[0]
+      }
+      tags
+    end
+
+    def title()
+      @page = @page || get_page
+      title = @page.title
+      title.toutf8.sub(/^#{TITLE_NICONICO}/ou, '')
+    end
+
+    private
+    def get_params
+      begin
+        unless @params
+          puts_info 'getting params : id = ' + @video_id
+          @page = @page || get_page
+          content = @agent.get_file('http://www.nicovideo.jp/api/getflv?v=' + @video_id)
+          puts_debug content
+          @params = content.scan(/([^&]+)=([^&]*)/).inject({}){|h, v| h[v[0]] = v[1]; h}
+        else
+          puts_info 'params have already gotten : id = ' + @video_id
+        end
+        @params
+      rescue
+        raise NicovideoNotFoundError
+      end
+    end
+
+    def get_page
+      if @video_id
+        puts_info 'getting HTML page : id = ' + @video_id
+        begin
+          @agent.get('http://www.nicovideo.jp/watch/' + @video_id)
+        rescue
+          raise NicovideoNotFoundError
+        end
+      end
+    end
+
+    def puts_error str ; puts str if (NV_DEBUG_LEVEL >= 1) ; end
+    def puts_info  str ; puts str if (NV_DEBUG_LEVEL >= 2) ; end
+    def puts_debug str ; puts str if (NV_DEBUG_LEVEL >= 3) ; end
+  end
+end

data/sample/nv_download2.rb

@@ -0,0 +1,33 @@
+require 'rubygems'
+require 'nicovideo'
+require 'yaml'
+
+video_ids = ARGV
+
+# set account
+account = YAML.load_file(ENV['HOME'] + '/.nicovideo/account.yml')
+mail = account['mail']
+password = account['password']
+
+# you must login to Nicovideo
+nv = Nicovideo.new(mail, password).login
+
+# get videos and comments
+video_ids.each {|video_id|
+
+  # the another way of nv_download
+  puts nv.get_title(video_id)
+  puts nv.get_tags(video_id).join(' ')
+  puts 'getting comments xml'
+  File.open("#{video_id}.xml", "wb") {|f|
+    f.write nv.get_comments(video_id, 100).to_xml
+  }
+  puts 'getting flv file'
+  File.open("#{video_id}.flv", "wb") {|f|
+    f.write nv.get_flv(video_id)
+  }
+
+  sleep 1
+}
+
+

metadata

@@ -3,8 +3,8 @@ rubygems_version: 0.9.4
 specification_version: 1
 name: nicovideo
 version: !ruby/object:Gem::Version 
-  version: 0.0.1
-date: 2008-01-06 00:00:00 +09:00
+  version: 0.0.2
+date: 2008-01-07 00:00:00 +09:00
 summary: utils for nicovideo
 require_paths: 
 - lib
@@ -37,11 +37,13 @@ files:
 - lib/nicovideo/version.rb
 - lib/nicovideo/mechanize-ext.rb
 - lib/nicovideo/base.rb
+- lib/nicovideo/videopage.rb
 - lib/nicovideo/video.rb
 - lib/nicovideo/comments.rb
 - test/test_helper.rb
 - test/test_nicovideo.rb
 - sample/nv_download.rb
+- sample/nv_download2.rb
 test_files: 
 - test/test_nicovideo.rb
 - test/test_helper.rb