kentaroi-okayu 0.0.1

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,6 @@
+*.sw?
+.DS_Store
+coverage
+rdoc
+pkg
+test/id_pass.txt

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Kentaro Imai
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,18 @@
+= okayu
+
+Description goes here.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but
+   bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2009 Kentaro Imai. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,63 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "okayu"
+    gem.summary = %Q{A comment viewer for http://live.nicovideo.jp/}
+    gem.description = %Q{A comment viewer for http://live.nicovideo.jp/}
+    gem.email = "kentaroi@gmail.com"
+    gem.homepage = "http://github.com/kentaroi/okayu"
+    gem.authors = ["Kentaro Imai"]
+    gem.add_dependency "mechanize"
+    gem.add_dependency "nokogiri"
+    gem.add_dependency "facter"
+    gem.add_dependency "sqlite3-ruby"
+    gem.add_dependency "activerecord", ">=2.2.0"
+    gem.add_dependency "wxruby"
+    gem.add_development_dependency "thoughtbot-shoulda"
+    gem.add_development_dependency "redgreen"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/*_test.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/*_test.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  if File.exist?('VERSION')
+    version = File.read('VERSION')
+  else
+    version = ""
+  end
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "okayu #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.0.1

data/bin/okayu

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'okayu.rb'
+
+okayu = Okayu::App.new
+okayu.main_loop

data/lib/app.rb

@@ -0,0 +1,25 @@
+module Okayu  
+  class App < Wx::App
+    def initialize
+      @teardowned = false
+      super()
+    end
+
+
+    def on_init
+      UserInfo.load
+      Okayu::connect_db
+
+      frame = AppFrame.new("おかゆ")
+      frame.show
+    end
+
+    def on_exit
+      unless @teardowned
+        UserInfo.save
+        @teardowned = true
+        puts "teardown!!"
+      end
+    end
+  end
+end

data/lib/app_dir.rb

@@ -0,0 +1,29 @@
+require 'rubygems'
+require 'facter'
+
+module Okayu
+  class AppDir
+    if Facter.value(:kernel) == 'Linux'
+      if File.exist?(@@app_dir = ENV['HOME'] + '/.okayu')
+      else
+        Dir.mkdir(@@app_dir)
+      end
+    elsif Facter.value(:kernel) == 'windows'
+      if Facter.value(:operatingsystemrelease).slice(/^\d*/).to_i < 6 # Before Vista
+        if File.exist?(@@app_dir = ENV['HOME'] + '/Local Settings/Application Data/okayu')
+        else
+          Dir.mkdir(@@app_dir)
+        end
+      else # From Vista
+        if File.exist?(@@app_dir = ENV['HOME'] + '/AppData/Local/okayu')
+        else
+          Dir.mkdir(@@app_dir)
+        end
+      end
+    end
+
+    def self.to_s
+      @@app_dir
+    end
+  end
+end

data/lib/client.rb

@@ -0,0 +1,210 @@
+require 'singleton'
+require 'rubygems'
+require 'mechanize'
+require 'thread'
+require 'player_status.rb'
+require 'thread_info.rb'
+require 'ruby-debug'
+
+class WWW::Mechanize::CookieJar
+  def load_cookie cookie
+    @jar = cookie
+  end
+end
+
+
+module Okayu
+
+  class Channel
+    attr_accessor :id, :player_status, :thread, :queue
+    def initialize
+      @queue = Queue.new
+    end
+  end
+
+  class Client
+    TOP_URL = 'http://www.nicovideo.jp/'
+    LOGIN_URL = 'https://secure.nicovideo.jp/secure/login?site=niconico'
+    PLAYER_STATUS_URL = 'http://live.nicovideo.jp/api/getplayerstatus?v='
+    @@channels = []
+    @@connected = false
+
+    class << self
+      def channels
+        @@channels
+      end
+      def connected?
+        @@connected
+      end
+
+      def agent
+        @@agent
+      end
+
+      def connect auth={}
+        @@agent = WWW::Mechanize.new
+        begin
+          if auth[:mail] && auth[:password]
+            @@login_page = @@agent.post(LOGIN_URL, 'mail' => auth[:mail], 'password' => auth[:password])
+          elsif auth[:cookie]
+            @@agent.cookie_jar.load_cookiestxt(auth[:cookie])
+            @@login_page = @@agent.get(TOP_URL)
+          end
+
+          if @@login_page
+            @@connected = true unless @@login_page.forms.first.name == 'login'
+          else
+            @@connected = false
+          end
+        rescue
+          @@connected = false
+        end
+      end
+
+      def live_id_from_url url
+        if /lv\d+/ =~ url
+          $&
+        else
+          nil
+        end
+      end
+
+      def open live_id, res_from=-200
+        if live_id
+          channel = Channel.new
+          channel.id = live_id
+          channel.player_status = get_player_status(live_id)
+          if channel.player_status.status
+            @@channels << channel
+            channel.thread = Thread.new(@@channels.index(channel), channel.player_status, res_from) do |index, ps, res_from|
+              begin
+                s = TCPSocket.open(ps.addr, ps.port)
+                s.print live_query_msg(ps.thread, res_from)
+                thread_info = ThreadInfo.new s.gets("\0")
+                while true
+                  @@channels[index].queue.enq s.gets("\0").slice(0...-1)
+                end
+              ensure
+                s.close
+              end
+            end
+            self.new channel
+          end
+        end
+      end
+
+      def send_comment live_id, comment, mail
+        if live_id
+          player_status = get_player_status(live_id)
+          if player_status.status
+            s = TCPSocket.open(player_status.addr, player_status.port)
+            s.print live_query_msg(player_status.thread, -0)
+            thread_info = ThreadInfo.new s.gets("\0")
+            postkey = get_postkey(thread_info)
+            s.print(send_comment_msg(comment, mail, player_status, thread_info, postkey))
+            s.close
+          end
+        end
+      end
+
+      def get_player_status live_id
+          PlayerStatus.new(@@agent.get(PLAYER_STATUS_URL + live_id).body)
+      end
+
+      def live_query_msg thread, res_from
+        %Q[<thread thread="#{thread}" res_from="#{res_from}" version="20061206" />\0]
+      end
+
+      def postkey_url thread, last_res
+        block_no = last_res.to_i / 100
+        %Q[http://live.nicovideo.jp/api/getpostkey?thread=#{thread}&block_no=#{block_no}]
+      end
+      
+      def get_postkey thread_info
+        postkey_page = @@agent.get(postkey_url(thread_info.thread, thread_info.last_res))
+        if /postkey=(.+)/ =~ postkey_page.body
+          $1
+        else
+          nil
+        end
+      end
+
+      def send_comment_msg comment, mail, player_status, thread_info, postkey
+        vpos = ((Time.new - Time.at(player_status.open_time.to_i)) * 100).to_i
+        %Q[<chat thread="#{thread_info.thread}" ticket="#{thread_info.ticket}" vpos="#{vpos}" postkey="#{postkey}" mail="#{mail}" user_id="#{player_status.user_id}" premium="#{player_status.is_premium}">#{comment}</chat>\0]
+      end
+
+      def close channel
+        index = @@channels.index(channel)
+        @@channels[index].thread.kill
+        @@channels[index] = nil
+      end
+
+      def close_all
+        @@channels.each do |channel|
+          channel.thread.kill
+        end
+      end
+    end
+
+    # instance methods
+    #
+    attr_reader :channel
+    def initialize channel
+      @channel = channel
+    end
+
+    def close
+      self.class.close @channel
+    end
+
+    def comments
+      comments = []
+      until @channel.queue.empty?
+        comments << @channel.queue.deq
+      end
+      comments.map{|comment| comment_hash_from_xml(comment)}
+    end
+
+    def comment_hash_from_xml xml
+      hash = Hash.new
+      reader = Nokogiri::XML::Reader(xml)
+      reader.read
+      if reader.name == 'chat'
+        hash[:commented_at] = Time.at(reader.attribute('date').to_i)
+        hash[:mail]         = reader.attribute('mail') || ''
+        hash[:number]       = reader.attribute('no')
+        hash[:user_id]      = reader.attribute('user_id')
+        case reader.attribute('premium')
+        when 1
+          hash[:premium]    = true
+          hash[:owner]      = false
+          hash[:command]    = false
+        when 2
+          hash[:premium]    = true
+          hash[:owner]      = true
+          hash[:command]    = true
+        when 3
+          hash[:premium]    = true
+          hash[:owner]      = true
+          hash[:command]    = false
+        else
+          hash[:premium]    = false
+          hash[:owner]      = false
+          hash[:command]    = false
+        end
+        reader.read
+        hash[:message]      = reader.value
+      end
+      hash
+    end
+
+    def community
+      @channel.player_status.default_community
+    end
+
+    def base_time
+      Time.at(@channel.player_status.base_time.to_i)
+    end
+  end
+end

data/lib/controllers/listeners_controller.rb

@@ -0,0 +1,18 @@
+module Okayu
+  class ListenersController
+    def self.set_color_index params
+      begin
+        listener = Listener.find params[:id]
+        listener.color_index = params[:color_index]
+        listener.save!
+      rescue
+        return false
+      end
+      true
+    end
+    
+    def self.unset_color_index params
+      self.set_color_index params
+    end
+  end
+end

data/lib/controllers/live_controller.rb

@@ -0,0 +1,82 @@
+module Okayu
+  class LiveController
+    def initialize live_id
+      @live_id = live_id
+      if @client = Client.open(live_id)
+        @live_now = true
+        if @live = Live.find(:first, :conditions => {:live_id => @live_id})
+          @community = @live.community
+        else
+          @live = Live.new
+          @live.live_id = @live_id
+          @live.base_time = @client.base_time
+          unless @community = Community.find(:first, :conditions => {:community_id => @client.community})
+            @community = Community.new
+            @community.community_id = @client.community
+            @community.save
+          end
+          @live.community = @community
+          @live.save
+        end
+      else
+        @live_now = false
+        if @live = Live.find(:first, :conditions => {:live_id => @live_id})
+          @community = @live.community
+        else
+          @community = nil
+        end
+      end
+    end
+ 
+    def valid?
+      !!@live
+    end
+
+    def live_now?
+      @live_now
+    end
+
+
+    def base_time
+      @live.base_time
+    end
+
+    def comments
+      @live.comments
+    end
+
+    def new_comments
+      if @live_now
+        @client.comments.map{|comment_hash|
+          if Comment.find :first, :conditions => {:live_id => @live_id, :number => comment_hash[:number]}
+            nil
+          else
+            comment = Comment.new(comment_hash)
+            comment.community = @community
+            comment.live = @live
+            if comment_hash[:user_id]
+              if visitor = Visitor.find(:first, :conditions => {:user_id => comment_hash[:user_id]})
+                comment.visitor = visitor
+                if visitor.listener
+                  comment.listener = visitor.listener
+                end
+              else
+                comment.create_visitor(:user_id => comment_hash[:user_id])
+              end
+            end
+            comment.save
+            comment
+          end
+        }.compact
+      else
+        []
+      end
+    end
+
+    def close
+      if @live_now && @client
+        @client.close
+      end
+    end
+  end
+end

data/lib/controllers/visitors_controller.rb

@@ -0,0 +1,19 @@
+module Okayu
+  class VisitorsController
+    def self.set_color_index params
+      begin
+        visitor = Visitor.find params[:id]
+        visitor.color_index = params[:color_index]
+        visitor.save!
+      rescue
+        return false
+      end
+      true
+    end
+
+    def self.unset_color_index params
+      self.set_color_index params
+    end
+  end
+end
+

data/lib/controllers.rb

@@ -0,0 +1,3 @@
+require 'controllers/live_controller.rb'
+require 'controllers/listeners_controller.rb'
+require 'controllers/visitors_controller.rb'

data/lib/cookie_thief.rb

@@ -0,0 +1,172 @@
+require 'rubygems'
+require 'sqlite3'
+
+module Okayu
+  class CookieThief
+    class << self
+      def get args
+        cookiestxt_from_array(send_query(sql_from_hash(args)))
+      end
+      
+      def send_query sql
+        db = SQLite3::Database.new(cookies_filename)
+        cookie_array = db.execute(sql).first
+        db.close
+        cookie_array
+      end
+      
+      def cookiestxt_from_array cookie_array
+        if cookie_array
+          cookie_array[1] = 'TRUE'
+          cookie_array[3] = cookie_array[3] != '0' ? 'TRUE' : 'FALSE'
+          cookie_array.join("\t") + "\n"
+        end
+      end
+    end
+  end
+
+  class IECookieThief < Okayu::CookieThief
+    class << self
+      def get args
+        cookies = []
+        if args[:host]
+          cookie_filenames(args[:host]).map do |cookie_filename|
+            cookies = cookies_from_cookie_file(cookie_filename)
+          end
+        end
+
+        if args[:name]
+          cookies = select_by_name(cookies, args[:name])
+        end
+        cookies.join("\n")
+      end
+
+      def select_by_name cookies, name
+        cookies.select{|cookie|cookie.split("\t")[5] == name}
+      end
+
+      def cookies_from_cookie_file filename
+        cookies = []
+        File.open filename, 'r' do |f|
+          cookies = cookies_from_cookies_string(f.read)
+        end
+        cookies
+      end
+
+      def cookies_from_cookies_string str
+        str.split(/^\*\n/m).map{|s|cookie_from_cookie_string(s)}
+      end
+
+      def cookie_from_cookie_string str
+        a = str.split(/\n/m)
+        name = a[0]
+        value = a[1]
+        host = "." + a[2].sub(%r{/.*$}, '')
+        path = "/" + a[2].sub(%r{^.*/}, '')
+        expiry = ((10**-7) * (a[4].to_i * (2**32) + a[3].to_i) - 11644473600).to_i # 11644473600 is midnight January 1, 1970
+        "#{host}\tTRUE\t#{path}\tTRUE\t#{expiry}\t#{name}\t#{value}" 
+      end
+
+      def cookie_filenames host
+        cookie_filename = []
+        case Facter.value(:kernel)
+        when 'windows'
+          cookie_filenames =  Dir.glob("#{ENV['HOME']}/AppData/Roaming/Microsoft/Windows/Cookies/Low/*").select{|path| path =~ /#{host_main_str(host)}/}
+        end
+        cookie_filenames
+      end
+      def host_main_str host
+        host_main_str = host.sub(/^\./, '').sub(/\..*$/, '')
+      end
+    end
+  end
+
+  class FirefoxCookieThief < Okayu::CookieThief
+    class << self
+      def sql_from_hash args
+        sql = "select host, isHttpOnly, path, isSecure, expiry, name, value from moz_cookies where "
+        sql << args.map{|key, value|"#{key.to_s} = '#{value}'"}.join(' and ')
+        sql << ';'
+      end
+
+      def cookies_filename
+        cookies_filename = nil
+        case Facter.value(:kernel)
+        when 'Linux'
+          if dir = Dir.glob("#{ENV['HOME']}/.mozilla/firefox/*").select{|path| path =~ /.*\.default$/}.first
+            if File.exist?("#{dir}/cookies.sqlite")
+              cookies_filename = "#{dir}/cookies.sqlite"
+            end
+          end
+        when 'windows'
+          if Facter.value(:operatingsystemrelease).to_i < 6 # before Vista
+            unless dir = Dir.glob("#{ENV['HOME']}/Application Data/Mozilla/Firefox/Profiles/*").select{|path| path =~ /.*\.default$/}.first
+              dir = Dir.glob("#{ENV['HOME']}/../Administrator/Application Data/Mozilla/Firefox/Profiles/*").select{|path| path =~ /.*\.default$/}.first
+            end
+            if dir && File.exist?("#{dir}/cookies.sqlite")
+                cookies_filename = "#{dir}/cookies.sqlite"
+            end
+          else # from Vista
+            if dir = Dir.glob("#{ENV['HOME']}/AppData/Roaming/Mozilla/Firefox/Profiles/*").select{|path| path =~ /.*\.default$/}.first
+              if File.exist?("#{dir}/cookies.sqlite")
+                cookies_filename = "#{dir}/cookies.sqlite"
+              end
+            end
+          end
+        end
+        cookies_filename
+      end
+    end
+  end
+
+  class ChromeCookieThief < Okayu::CookieThief
+    class << self
+      def cookies_filename
+        cookies_filename = nil
+        case Facter.value(:kernel)
+        when 'windows'
+          if Facter.value(:operatingsystemrelease).to_i < 6 # before Vista
+            if File.exist?(filename = "#{ENV['HOME']}/LocalApplication Data/Google/Chrome/User Data/Default/Cookies")
+              cookies_filename = filename
+            elsif File.exist?(filename = "#{ENV['HOME']}/../Administrator/LocalApplication Data/Google/Chrome/User Data/Default/Cookies")
+              cookies_filename = filename
+            end
+          else # from Vista
+            if File.exist?(filename = "#{ENV['HOME']}/AppData/Local/Google/Chrome/User Data/Default/Cookies")
+              cookies_filename = filename
+            end
+          end
+        end
+        cookies_filename
+      end
+      
+      def sql_from_hash args
+        args = chrome_from_firefox(args)
+        sql = "select host_key, httponly, path, secure, expires_utc, name, value from cookies where "
+        sql << args.map{|key, value|"#{key.to_s} = '#{value}'"}.join(' and ')
+        sql << ';'
+      end
+
+      def chrome_from_firefox args
+        chrome_args = {}
+        args.each do |key, value|
+          case key
+          when :host
+            chrome_args[:host_key] = value
+          when :isHttpOnly
+            chrome_args[:httponly] = value
+          when :isSecure
+            chrome_args[:secure] = value
+          when :expiry
+            chrome_args[:expires_utc] = value
+          else
+            chrome_args[key] = value
+          end
+        end
+        chrome_args
+      end
+
+
+    end
+  end
+end