niconico 1.3.0 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 36a98bd1ac8a923065e75fb31ce853304d878dd6
-  data.tar.gz: 6117581998277bd370b01516ab7f4f33c5f07c6c
+  metadata.gz: df82c4e26df5f6e3fd88b09775a014a65600fb3b
+  data.tar.gz: dccb4490003953bcbc1bac19b533f775e506cdde
 SHA512:
-  metadata.gz: f597570ac2329a7cd8da0bb401e735aa1242e2da6e35eb4d80d369ada23e1ca1deebe9cc7541afb6b9be4088b62b6aebb006c90251ad4adb910b5d669482ee0d
-  data.tar.gz: e8b48d6985978f6f00d3fd3d7965c36661b289824d7346066a16beb34cf1787683e1baaf8f89a4a65a036037365cc424162520e9532b9dd93a4e9ff1067971e8
+  metadata.gz: c56db982b4c804be9d889360b752396bb321bd2e88c874e709dc5d9bb2853348b330de2421d4aca67888e1fa19688953be8353945ed1ce2b890984c0a7f422d8
+  data.tar.gz: 5bea438a9fd9264d97328918663b1287bc5164d320ca2a0237bfc448a6c6915861cf875d4e4f6cbc81dd83d9eca5dbf970ff08f251cc8aa688f77d5542a06de4

data/lib/niconico.rb

@@ -1,12 +1,12 @@
 # -*- coding: utf-8 -*-
 
-require 'rubygems'
 require 'mechanize'
 require 'cgi'
 require 'niconico/version'
 
 class Niconico
   URL = {
+    top: 'http://www.nicovideo.jp/',
     login: 'https://secure.nicovideo.jp/secure/login?site=niconico',
     watch: 'http://www.nicovideo.jp/watch/',
     getflv: 'http://flapi.nicovideo.jp/api/getflv'
@@ -16,34 +16,76 @@ class Niconico
 
   attr_reader :agent, :logined
 
-  def initialize(mail, pass)
-    @mail = mail
-    @pass = pass
+  def initialize(*args)
+    case args.size
+    when 2
+      @mail, @pass = args
+    when 1
+      @token = args.first
+    else
+      raise ArgumentError, "wrong number of arguments (#{args.size} for 1..2)"
+    end
 
     @logined = false
 
     @agent = Mechanize.new.tap do |agent|
       agent.user_agent = "Niconico.gem (#{Niconico::VERSION}, https://github.com/sorah/niconico)"
       agent.ssl_version = 'SSLv3'
+
+      agent.cookie_jar.add(
+        HTTP::Cookie.new(
+          domain: '.nicovideo.jp', path: '/',
+          name: 'lang', value: 'ja-jp',
+        )
+      )
     end
   end
 
   def login(force=false)
     return false if !force && @logined
 
+    if @token
+      login_with_token
+    elsif @mail && @pass
+      login_with_email
+    else
+      raise 'huh? (may be bug)'
+    end
+  end
+
+  def inspect
+    "#<Niconico: #{@mail || '(token)'}, #{@logined ? "" : "not "}logined>"
+  end
+
+  class LoginError < StandardError; end
+
+  private
+
+  def login_with_email
     page = @agent.post(URL[:login], 'mail' => @mail, 'password' => @pass)
 
     raise LoginError, "Failed to login (x-niconico-authflag is 0)" if page.header["x-niconico-authflag"] == '0'
     @logined = true
   end
 
-  def inspect
-    "#<Niconico: #{@mail} (#{@logined ? "" : "not "}logined)>"
+  def login_with_token
+    @agent.cookie_jar.add(
+      HTTP::Cookie.new(
+        domain: '.nicovideo.jp', path: '/',
+        name: 'user_session', value: @token
+      )
+    )
+
+    page = @agent.get(URL[:top])
+    raise LoginError, "Failed to login (x-niconico-authflag is 0)" if page.header["x-niconico-authflag"] == '0'
+
+    @logined = true
   end
-  class LoginError < StandardError; end
+
 end
 
-require_relative './niconico/video'
-require_relative './niconico/mylist'
-require_relative './niconico/ranking'
-require_relative './niconico/channel'
+require 'niconico/video'
+require 'niconico/mylist'
+require 'niconico/ranking'
+require 'niconico/channel'
+require 'niconico/live'

data/lib/niconico/channel.rb

@@ -2,7 +2,7 @@
 require 'json'
 require 'open-uri'
 require 'nokogiri'
-require_relative './video'
+require 'niconico/video'
 
 class Niconico
   def channel_videos(ch)

data/lib/niconico/deferrable.rb

@@ -0,0 +1,40 @@
+class Niconico
+  module Deferrable
+    module ClassMethods
+      def deferrable(*keys)
+        keys.each do |key|
+          binding.eval(<<-EOM, __FILE__, __LINE__.succ)
+            define_method(:#{key}) do
+              get() unless fetched?
+              @#{key}
+            end
+          EOM
+        end
+        self.deferred_methods.push *keys
+      end
+
+      def deferred_methods
+        @deferred_methods ||= []
+      end
+    end
+
+    def self.included(klass)
+      klass.extend ClassMethods
+    end
+
+    def fetched?; @fetched; end
+
+    def get
+      @fetched = true
+    end
+
+    private
+
+    def preload_deffered_values(vars={})
+      vars.each do |k,v|
+        next unless self.class.deferred_methods.include?(k)
+        instance_variable_set "@#{k}", v
+      end
+    end
+  end
+end

data/lib/niconico/live.rb

@@ -0,0 +1,204 @@
+# coding: utf-8
+require 'niconico/live/api'
+
+class Niconico
+  def live(live_id)
+    Live.new(self, live_id)
+  end
+
+  class Live
+    class ReservationOutdated < Exception; end
+    class ReservationNotAccepted < Exception; end
+    class TicketRetrievingFailed < Exception; end
+    class AcceptingReservationFailed < Exception; end
+
+    class << self
+      def public_key
+        @public_key ||= begin
+          if ENV["NICONICO_LIVE_PUBLIC_KEY"]
+            File.read(File.expand_path(ENV["NICONICO_LIVE_PUBLIC_KEY"]))
+          else
+            nil
+          end
+        end
+      end
+
+      def public_key=(other)
+        @public_key = other
+      end
+    end
+
+    def initialize(parent, live_id)
+      @parent = parent
+      @agent = parent.agent
+      @id = @live_id = live_id
+      @client = Niconico::Live::API.new(@agent)
+
+      get()
+    end
+
+    attr_reader :id, :live, :ticket
+    attr_writer :public_key
+
+    def public_key
+      @public_key || self.class.public_key
+    end
+
+    def fetched?
+      !!@fetched
+    end
+
+    def get(force=false)
+      return self if @fetched && !force
+      @live = @client.get(@live_id)
+      @fetched = true
+      self
+    end
+
+    def seat(force=false)
+      return @seat if @seat && !force
+      raise ReservationNotAccepted if reserved? && !reservation_accepted?
+
+      @seat = @client.get_player_status(self.id, self.public_key)
+
+      raise TicketRetrievingFailed, @seat[:error] if @seat[:error]
+
+      @seat
+    end
+
+    def accept_reservation
+      return self if reservation_accepted?
+      raise ReservationOutdated if reservation_outdated?
+
+      result = @client.accept_watching_reservation(self.id)
+      raise AcceptingReservationFailed unless result
+
+      get(:reload)
+
+      self
+    end
+
+    def inspect
+      "#<Niconico::Live: #{id}, #{title}>"
+    end
+
+    def title
+      get.live[:title]
+    end
+
+    def description
+      get.live[:description]
+    end
+
+    def opens_at
+      get.live[:opens_at]
+    end
+
+    def starts_at
+      get.live[:starts_at]
+    end
+
+    def status
+      get.live[:status]
+    end
+
+    def scheduled?
+      status == :scheduled
+    end
+
+    def on_air?
+      status == :on_air
+    end
+
+    def closed?
+      status == :closed
+    end
+
+    def reserved?
+      get.live.key? :reservation
+    end
+
+    def reservation_unaccepted?
+      reserved? && live[:reservation][:status] == :reserved
+    end
+
+    def reservation_accepted?
+      reserved? && live[:reservation][:status] == :accepted
+    end
+
+    def reservation_outdated?
+      reserved? && live[:reservation][:status] == :outdated
+    end
+
+    def reservation_available?
+      reservation_unaccepted? || reservation_accepted?
+    end
+
+    def reservation_expires_at
+      reserved? ? live[:reservation][:expires_at] : nil
+    end
+
+    def channel
+      get.live[:channel]
+    end
+
+    def premium?
+      !!seat[:premium?]
+    end
+
+    def rtmp_url
+      seat[:rtmp][:url]
+    end
+
+    def ticket
+      seat[:rtmp][:ticket]
+    end
+
+    def quesheet
+      seat[:quesheet]
+    end
+
+    def rtmpdump_commands(file_base)
+      file_base = File.expand_path(file_base)
+
+      publishes = quesheet.select{ |_| /^\/publish / =~ _[:body] }.map do |publish|
+        publish[:body].split(/ /).tap(&:shift)
+      end
+
+      plays = quesheet.select{ |_| /^\/play / =~ _[:body] }
+ 
+      plays.map.with_index do |play, i|
+        cases = play[:body].sub(/^case:/,'').split(/ /)[1].split(/,/)
+        publish_id = nil
+
+        publish_id   = cases.find { |_| _.start_with?('premium:') } if premium?
+        publish_id ||= cases.find { |_| _.start_with?('default:') }
+        publish_id ||= cases[0]
+
+        publish_id = publish_id.split(/:/).last
+
+        contents = publishes.select{ |_| _[0] == publish_id }
+
+        contents.map.with_index do |content, j|
+          content = content[1]
+          rtmp = "#{self.rtmp_url}/mp4:#{content}"
+
+          seq = 0
+          begin
+            file = "#{file_base}.#{i}.#{j}.#{seq}.flv"
+            seq += 1
+          end while File.exist?(file)
+
+          ['rtmpdump',
+           '-V',
+           '-o', file,
+           '-r', rtmp,
+           '-C', "S:#{ticket}",
+           '--playpath', "mp4:#{content}",
+           '--app', URI.parse(self.rtmp_url).path.sub(/^\//,'')
+          ]
+        end
+      end.flatten(1)
+    end
+  end
+end

data/lib/niconico/live/api.rb

@@ -0,0 +1,212 @@
+# coding: utf-8
+require 'time'
+require 'openssl'
+
+class Niconico
+  class Live
+    class API
+      class NoPublicKeyProvided < Exception; end
+
+      URL_GETPLAYERSTATUS = 'http://ow.live.nicovideo.jp/api/getplayerstatus'.freeze
+      URL_WATCHINGRESERVATION_LIST = 'http://live.nicovideo.jp/api/watchingreservation?mode=list'
+
+      def initialize(agent)
+        @agent = agent
+      end
+
+      attr_reader :agent
+
+      def get(id)
+        id = normalize_id(id)
+
+        page = agent.get("http://live.nicovideo.jp/gate/#{id}")
+
+        comment_area = page.at("#comment_area#{id}").inner_text
+
+        result = {
+          title: page.at('h2 span').inner_text,
+          id: id,
+          description: page.at('.stream_description .text_area').inner_html,
+        }
+
+        kaijo = page.search('.kaijo strong').map(&:inner_text)
+        result[:opens_at] = Time.parse("#{kaijo[0]} #{kaijo[1]} +0900")
+        result[:starts_at] = Time.parse("#{kaijo[0]} #{kaijo[2]} +0900")
+
+        result[:status] = :scheduled if comment_area.include?('開場まで、あと')
+        result[:status] = :on_air if comment_area.include?('現在放送中')
+        close_message = comment_area.match(/この番組は(.+?)に終了いたしました。/)
+        if close_message
+          result[:status] = :closed
+          result[:closed_at] = Time.parse("#{close_message[1]} +0900")
+        end
+
+        reservation_valid_until_message = comment_area.match(/使用期限: (.+?)まで/)
+        if reservation_valid_until_message 
+          result[:reservation] = {}
+          result[:reservation][:expires_at] = Time.parse("#{reservation_valid_until_message[1]} +0900")
+
+          if comment_area.include?('視聴権を使用し、タイムシフト視聴を行いますか？')
+            result[:reservation][:status] = :reserved
+            result[:reservation][:available] = true
+          elsif comment_area.include?('本番組は、タイムシフト視聴を行う事が可能です。')
+            result[:reservation][:status] = :accepted
+            result[:reservation][:available] = true
+          elsif comment_area.include?('タイムシフト視聴をこれ以上行う事は出来ません。') || comment_area.include?('視聴権の利用期限が過ぎています。')
+            result[:reservation][:status] = :outdated
+          end
+        end
+
+        channel = page.at('div.chan')
+        if channel
+          result[:channel] = {
+            name: channel.at('.shosai a').inner_text,
+            id: channel.at('.shosai a')['href'].split('/').last,
+            link: channel.at('.shosai a')['href'],
+          }
+        end
+
+        result
+      end
+
+      def heartbeat
+        raise NotImplementedError
+      end
+
+      def get_player_status(id, public_key = nil)
+        id = normalize_id(id)
+        page = agent.get("http://ow.live.nicovideo.jp/api/getplayerstatus?locale=GLOBAL&lang=ja%2Djp&v=#{id}&seat%5Flocale=JP")
+        if page.body[0] == 'c' # encrypted
+          page = Nokogiri::XML(decrypt_encrypted_player_status(page.body, public_key))
+        end
+
+        status = page.at('getplayerstatus')
+
+        if status['status'] == 'fail'
+          error = page.at('error code').inner_text
+
+          case error
+          when 'notlogin'
+            return {error: :not_logged_in}
+          when 'comingsoon'
+            return {error: :not_yet_started}
+          when 'closed'
+            return {error: :closed}
+          when 'require_accept_print_timeshift_ticket'
+            return {error: :reservation_not_accepted}
+          when 'timeshift_ticket_expire'
+            return {error: :reservation_expired}
+          when 'noauth'
+            return {error: :archive_closed}
+          when 'notfound'
+            return {error: :not_found}
+          else
+            return {error: error}
+          end
+        end
+
+        result = {}
+
+        # Strings
+        %w(id title description provider_type owner_name
+           bourbon_url full_video kickout_video).each do |key|
+          item = status.at(key)
+          result[key.to_sym] = item.inner_text if item
+        end
+
+        # Integers
+        %w(watch_count comment_count owner_id watch_count comment_count).each do |key|
+          item = status.at(key)
+          result[key.to_sym] = item.inner_text.to_i if item
+        end
+
+        # Flags
+        %w(is_premium is_reserved is_owner international is_rerun_stream is_archiveplayserver
+           archive allow_netduetto
+           is_nonarchive_timeshift_enabled is_timeshift_reserved).each do |key|
+          item = status.at(key)
+          result[key.sub(/^is_/,'').concat('?').to_sym] = item.inner_text == '1' if item
+        end
+
+        # Datetimes
+        %w(base_time open_time start_time end_time).each do |key|
+          item = status.at(key)
+          result[key.to_sym] = Time.at(item.inner_text.to_i) if item
+        end
+
+        rtmp = status.at('rtmp')
+        result[:rtmp] = {
+          url: rtmp.at('url').inner_text,
+          ticket: rtmp.at('ticket').inner_text,
+        }
+
+        ms = status.at('ms')
+        result[:ms] = {
+          address: ms.at('addr').inner_text,
+          port:    ms.at('port').inner_text.to_i,
+          thread:  ms.at('thread').inner_text,
+        }
+
+        quesheet = status.search('quesheet que')
+        result[:quesheet] = quesheet.map do |que|
+          {vpos: que['vpos'].to_i, mail: que['mail'], name: que['name'], body: que.inner_text}
+        end
+
+        result
+      end
+
+      def watching_reservations
+        page = agent.get(URL_WATCHINGRESERVATION_LIST)
+        page.search('vid').map(&:inner_text).map{ |_| normalize_id(_) }
+      end
+
+      def accept_watching_reservation(id_)
+        id = normalize_id(id_, with_lv: false)
+        page = agent.get("http://live.nicovideo.jp/api/watchingreservation?mode=confirm_watch_my&vid=#{id}&next_url&analytic")
+        token = page.at('#reserve img')['onclick'].scan(/'(.+?)'/)[1][0]
+
+        page = agent.post("http://live.nicovideo.jp/api/watchingreservation",
+                          accept: 'true', mode: 'use', vid: id, token: token)
+
+        page.at('nicolive_video_response')['status'] == 'ok'
+      end
+
+      def decrypt_encrypted_player_status(body, public_key)
+        unless public_key
+          raise NoPublicKeyProvided,
+            'You should provide proper public key to decrypt ' \
+            'encrypted player status'
+        end
+
+        lines = body.lines
+        pubkey = OpenSSL::PKey::RSA.new(public_key)
+
+        encrypted_shared_key = lines[1].unpack('m*')[0]
+        shared_key_raw = pubkey.public_decrypt(encrypted_shared_key)
+        shared_key = shared_key_raw.unpack('L>*')[0].to_s
+
+        cipher = OpenSSL::Cipher.new('bf-ecb').decrypt
+        cipher.padding = 0
+        cipher.key_len = shared_key.size
+        cipher.key = shared_key
+
+        encrypted_body = lines[2].unpack('m*')[0]
+
+        body = cipher.update(encrypted_body) + cipher.final
+        body.force_encoding('utf-8')
+      end
+
+      private
+
+      def normalize_id(id, with_lv: true)
+        id = id.to_s
+
+        if with_lv
+          id.start_with?('lv') ? id : "lv#{id}"
+        else
+          id.start_with?('lv') ? id[2..-1] : id
+        end
+      end
+    end
+  end
+end

data/lib/niconico/version.rb

@@ -1,3 +1,3 @@
 class Niconico
-  VERSION = "1.3.0"
+  VERSION = "1.4.0"
 end

data/lib/niconico/video.rb

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 require 'json'
+require 'niconico/deferrable'
 
 class Niconico
   def video(video_id)
@@ -8,35 +9,29 @@ class Niconico
   end
 
   class Video
-    DEFERRABLES = [:id, :title, :url, :video_url, :type, :tags, :mylist_comment, :description, :description_raw]
-    DEFERRABLES_VAR = DEFERRABLES.map{|k| :"@#{k}" }
+    include Niconico::Deferrable
 
-    DEFERRABLES.zip(DEFERRABLES_VAR).each do |(k,i)|
-      define_method(k) do
-        instance_variable_get(i) || (get && instance_variable_get(i))
-      end
-    end
+    deferrable :id, :title,
+      :description, :description_raw,
+      :url, :video_url, :type,
+      :tags, :mylist_comment
 
     def initialize(parent, video_id, defer=nil)
       @parent = parent
       @agent = parent.agent
       @fetched = false
       @thread_id = @id = video_id
+      @page = nil
       @url = "#{Niconico::URL[:watch]}#{@id}"
 
       if defer
-        defer.each do |k,v|
-          next unless DEFERRABLES.include?(k)
-          instance_variable_set :"@#{k}", v
-        end
-        @page = nil
+        preload_deffered_values(defer)
       else
-        @page = get()
+        get()
       end
     end
 
     def economy?; @eco; end
-    def fetched?; @fetched; end
 
     def get(options = {})
       begin

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: niconico
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 1.4.0
 platform: ruby
 authors:
 - Shota Fukumori (sora_h)
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-04-07 00:00:00.000000000 Z
+date: 2014-05-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mechanize
@@ -52,6 +52,9 @@ files:
 - Rakefile
 - lib/niconico.rb
 - lib/niconico/channel.rb
+- lib/niconico/deferrable.rb
+- lib/niconico/live.rb
+- lib/niconico/live/api.rb
 - lib/niconico/mylist.rb
 - lib/niconico/ranking.rb
 - lib/niconico/version.rb
@@ -76,9 +79,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.0
+rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: wrapper of Mechanize, optimized for nicovideo.
 test_files: []
-has_rdoc: