kentaroi-okayu 0.0.1

data/lib/views/live_panel.rb

@@ -0,0 +1,335 @@
+module Okayu
+  COLORIZE = 300
+  NAME     = 301
+
+  class LiveListCtrl < Wx::ListCtrl
+    COLUMNS = { :no => 0,
+                :message => 1,
+                :mail    => 2,
+                :time    => 3,
+                :name    => 4,
+                :id      => 5,
+                :premium => 6,
+                :ngword  => 7 }
+    attr_accessor :base_time
+
+    def initialize(parent)
+      super(parent, :id => Wx::ID_ANY, :style => Wx::LC_REPORT)
+      insert_column COLUMNS[:no],      "No", Wx::LIST_FORMAT_LEFT, -1
+      insert_column COLUMNS[:message], "Message", Wx::LIST_FORMAT_LEFT, -1
+      insert_column COLUMNS[:mail],    "Command", Wx::LIST_FORMAT_LEFT, -1
+      insert_column COLUMNS[:time],    "Time", Wx::LIST_FORMAT_LEFT, -1
+      insert_column COLUMNS[:name],    "Name", Wx::LIST_FORMAT_LEFT, -1
+      insert_column COLUMNS[:id],      "Id", Wx::LIST_FORMAT_LEFT, -1
+      insert_column COLUMNS[:premium], "Premium", Wx::LIST_FORMAT_LEFT, -1
+      insert_column COLUMNS[:ngword],  "NG word", Wx::LIST_FORMAT_LEFT, -1
+      load_column_width
+
+      evt_list_col_end_drag(Wx::ID_ANY){|event| on_column_resized(event)}
+
+      evt_list_item_right_click(get_id) {|event| on_list_item_right_click(event)}
+      evt_list_item_activated(Wx::ID_ANY) {|event| on_activated(event)}
+      evt_key_down(){|event| get_parent.on_key_down(event)}
+
+      evt_menu(COLORIZE){show_color_dialog}
+      evt_menu(NAME){show_name_dialog}
+    end
+    
+    def load_column_width
+      return unless UserInfo.column_width      
+      set_column_width COLUMNS[:no], UserInfo.column_width[:no]
+      set_column_width COLUMNS[:message], UserInfo.column_width[:message]
+      set_column_width COLUMNS[:mail], UserInfo.column_width[:mail]
+      set_column_width COLUMNS[:time], UserInfo.column_width[:time]
+      set_column_width COLUMNS[:name], UserInfo.column_width[:name]
+      set_column_width COLUMNS[:id], UserInfo.column_width[:id]
+      set_column_width COLUMNS[:premium], UserInfo.column_width[:premium]
+      set_column_width COLUMNS[:ngword], UserInfo.column_width[:ngword]
+    end 
+  
+    def insert_comment comment
+      item = Wx::ListItem.new
+      item.set_data({:id => comment.id,
+                     :visitor_id => comment.visitor_id,
+                     :listener_id => comment.listener_id})
+      item.set_id 0
+      item.set_column COLUMNS[:no]
+      item.set_text comment.number.to_s
+      # item.set_text_colour Wx::GREEN
+      insert_item(item)
+    
+      item.set_column COLUMNS[:message]
+      item.set_text comment.message
+      set_item(item)
+
+      item.set_column COLUMNS[:mail]
+      item.set_text comment.mail
+      set_item(item)
+
+      item.set_column COLUMNS[:time]
+      item.set_text Time.at(comment.commented_at - @base_time).strftime("%M:%S")
+      set_item(item)
+
+      if comment.listener
+        item.set_column COLUMNS[:name]
+        item.set_text comment.listener.name
+        set_item(item)
+      end
+
+      item.set_column COLUMNS[:id]
+      item.set_text comment.user_id || comment.visitor.user_id
+      set_item(item)
+
+      item.set_column COLUMNS[:premium]
+      if comment.owner
+        item.set_text '主'
+      elsif comment.premium
+        item.set_text 'P'
+      elsif comment.command
+        item.set_text 'System'
+      else
+        item.set_text ''
+      end
+      set_item(item)
+
+      if comment.listener
+        if color_index = comment.listener.color_index
+          set_color_to_an_item(0, COLORS[color_index])
+        end
+      elsif comment.visitor
+        if color_index = comment.visitor.color_index
+          set_color_to_an_item(0, COLORS[color_index])
+        end
+      end
+
+    end
+
+    def on_column_resized event
+      column = event.get_column
+      if UserInfo[:column_width]
+        UserInfo[:column_width][COLUMNS.index(column)] = get_column_width(column)
+      else
+        UserInfo[:column_width] = {
+          :no      => get_column_width(COLUMNS[:no]),
+          :message => get_column_width(COLUMNS[:message]),
+          :mail    => get_column_width(COLUMNS[:mail]),
+          :time    => get_column_width(COLUMNS[:time]),
+          :name    => get_column_width(COLUMNS[:name]),
+          :id      => get_column_width(COLUMNS[:id]),
+          :premium => get_column_width(COLUMNS[:premium]),
+          :ngword  => get_column_width(COLUMNS[:ngword])
+        }
+      end
+    end
+
+    def on_list_item_right_click event
+      menu = Wx::Menu.new
+      menu.append(COLORIZE, "色をつける(&C)", "")
+      menu.append(NAME, "名前をつける(&M)", "")
+      popup_menu(menu, event.get_point)
+    end
+
+    def show_color_dialog
+      dialog = ColorDialog.new(self)
+      color_index = dialog.show_modal
+      item_index = get_selections.first
+      if color_index >= 0 && color_index < COLORS.length
+        on_color_set item_index, color_index
+      end
+    end
+
+    def show_name_dialog
+      dialog = Wx::TextEntryDialog.new(self, "名前を入力してください")
+      dialog.show_modal
+      item_index = get_selections.first
+
+      item = get_item(item_index)
+      item.set_text_colour(COLORS[WHITE][:color])
+      item.set_column COLUMNS[:name]
+      item.set_text(dialog.get_value)
+      set_item(item)
+    end
+
+    def is_colored? index
+      get_item_background_colour(index) != Wx::NULL_COLOUR &&
+      get_item_background_colour(index) != COLORS[WHITE][:color]
+    end
+    
+    def on_activated event
+      item_index = event.get_index
+      if is_colored?(item_index)
+        on_color_unset item_index
+      else
+        on_color_set item_index
+      end
+    end
+
+    def on_color_set index, color_index=nil
+      color_index = rand(COLORS.length) unless color_index
+      data = get_item_data(index)
+      return unless data
+
+      if data[:listener_id]
+        set_color({:listener_id => data[:listener_id]}, COLORS[color_index])
+        unless ListenersController.set_color_index :id => data[:listener_id],
+                                                   :color_index => color_index
+          Wx::message_box('色設定の保存に失敗しました')
+        end 
+      else
+        set_color({:visitor_id => data[:visitor_id]}, COLORS[color_index])
+        unless VisitorsController.set_color_index :id => data[:visitor_id],
+                                                  :color_index => color_index
+          Wx::message_box('色設定の保存に失敗しました')
+        end
+      end
+    end
+
+    def on_color_unset index
+      data = get_item_data(index)
+      return unless data
+
+      if data[:listener_id]
+        set_color({:listener_id => data[:listener_id]}, COLORS[WHITE])
+        unless ListenersController.unset_color_index :id => data[:listener_id]
+          Wx::message_box('色設定の消去に失敗しました')
+        end 
+      else
+        set_color({:visitor_id => data[:visitor_id]}, COLORS[WHITE])
+        unless VisitorsController.unset_color_index :id => data[:visitor_id]
+          Wx::message_box('色設定の消去に失敗しました')
+        end
+      end
+    end
+
+    def set_color target, color
+      if target[:listener_id]
+        self.each do |index|
+          if get_item_data(index)[:listener_id] == target[:listener_id]
+            set_color_to_an_item(index, color)
+          end
+        end
+      elsif target[:visitor_id]
+        self.each do |index|
+          if get_item_data(index)[:visitor_id] == target[:visitor_id]
+            set_color_to_an_item(index, color)
+          end
+        end
+      end
+    end
+
+    def set_color_to_an_item index, color
+      set_item_background_colour(index, color[:color])
+      if color[:white_text]
+        get_item(index).set_text_colour(COLORS[WHITE][:color])
+      end
+    end
+
+    def on_name_set index, name
+    end
+    def on_name_unset index, name
+    end
+    def set_name target, name
+    end
+    def set_name_to_an_item index, name
+    end
+  end
+
+  class LivePanel < Wx::Panel
+    attr_reader :title
+    def initialize parent
+      super
+      @title = 'live'
+      @ready = false
+
+      @live_url = Wx::TextCtrl.new(self, 
+                                   :id => Wx::ID_ANY,
+                                   :style => Wx::TE_PROCESS_ENTER)
+      @list = LiveListCtrl.new(self)
+
+      sizer = Wx::BoxSizer.new(Wx::VERTICAL)
+      sizer.add(@live_url, 0, Wx::ALL|Wx::EXPAND, 10)
+      sizer.add(@list, 1, Wx::ALL|Wx::EXPAND, 10)
+      set_sizer(sizer)
+      sizer.set_size_hints(self)
+
+      @live_url.set_focus
+
+      #@live_url.evt_key_down(){|event| on_key_down(event)}
+      evt_key_down(){|event| on_key_down(event)}
+
+
+      @live_url.evt_text_enter(@live_url.get_id) {|event| on_live_url_enter(event)}
+    end
+
+    def set_index index
+      @index = index
+    end
+
+    def on_new_url
+      @live_url.set_focus
+      @live_url.set_selection(-1, -1)
+    end
+
+    def on_live_url_enter event
+      close
+
+      if live_id = live_id_from_url(@live_url.get_value)
+        @ready = false
+        @live_controller = LiveController.new(live_id)
+        if @live_controller.valid?
+          @list.base_time = @live_controller.base_time
+          get_parent.set_page_text(@index, live_id)
+          
+          @list.delete_all_items
+          @live_controller.comments.each do |comment|
+            @list.insert_comment(comment)
+          end
+          
+          @ready = true
+        end
+      end
+    end
+    
+    def live_id_from_url url
+      if /^\d+$/ =~ url
+        "lv#{url}"
+      elsif /lv\d+/ =~ url
+        $&
+      else
+        nil
+      end
+    end
+
+    def on_key_down event
+      case event.get_key_code
+      when 0x16E # Ctrl+PageUp
+        if event.control_down
+          get_parent.advance_selection false
+        end
+      when 0x16F # Ctrl+PageDown
+        if event.control_down
+          get_parent.advance_selection true
+        end
+      end
+    end
+
+    def close
+      if @live_controller
+        @live_controller.close
+      end
+    end
+
+
+    def on_update
+      if @ready && @live_controller.live_now?
+        comments = @live_controller.new_comments
+        comments.each do |comment|
+          @list.insert_comment(comment)
+        end
+      end
+    end
+
+  end
+end
+
+

data/lib/views/login_dialog.rb

@@ -0,0 +1,80 @@
+module Okayu
+  IE_BTN      = 250
+  FIREFOX_BTN = 251
+  CHROME_BTN  = 252
+
+  ID_PASS_BTN = 255
+  class LoginDialog < Wx::Dialog
+    def initialize parent
+      super(parent, Wx::ID_ANY, "ログイン")
+      @ie_btn      = Wx::RadioButton.new(self, :id => IE_BTN,      :label => 'Internet Explorer')
+      @firefox_btn = Wx::RadioButton.new(self, :id => FIREFOX_BTN, :label => 'Firefox / Iceweasel')
+      @chrome_btn  = Wx::RadioButton.new(self, :id => CHROME_BTN,  :label => 'Google Chrome')
+      @id_pass_btn = Wx::RadioButton.new(self, :id => ID_PASS_BTN, :label => 'Id && Password')
+      @mail = Wx::TextCtrl.new self,
+                               :id => Wx::ID_ANY,
+                               :value => UserInfo.mail || '',
+                               :style => Wx::TE_PROCESS_ENTER
+      @password = Wx::TextCtrl.new self,
+                                   :id => Wx::ID_ANY,
+                                   :value => UserInfo.password || '',
+                                   :style => Wx::TE_PROCESS_ENTER | Wx::TE_PASSWORD
+      @ok_btn = Wx::Button.new(self, :id => Wx::ID_OK, :label => 'OK')
+      sizer = Wx::BoxSizer.new Wx::VERTICAL
+      sizer.add(@ie_btn,      0, Wx::ALL | Wx::EXPAND, 10)
+      sizer.add(@firefox_btn, 0, Wx::ALL | Wx::EXPAND, 10)
+      sizer.add(@chrome_btn,  0, Wx::ALL | Wx::EXPAND, 10)
+      sizer.add(@id_pass_btn, 0, Wx::ALL | Wx::EXPAND, 10)
+      sizer.add(@mail,        0, Wx::ALL | Wx::EXPAND, 10)
+      sizer.add(@password,    0, Wx::ALL | Wx::EXPAND, 10)
+      sizer.add(@ok_btn,      0, Wx::ALL | Wx::EXPAND, 10)
+      set_sizer(sizer)
+      sizer.set_size_hints(self)
+      sizer.fit(self)
+
+      case UserInfo.login
+      when :ie
+        @ie_btn.set_value true
+      when :firefox
+        @firefox_btn.set_value true
+      when :chrome
+        @chrome_btn.set_value true
+      when :id_pass
+        @id_pass_btn.set_value true
+      end
+      @ok_btn.set_focus
+
+      @mail.evt_text_enter(@mail.get_id){ on_enter }
+      @password.evt_text_enter(@password.get_id){ on_enter}
+
+      @ok_btn.evt_button(Wx::ID_OK){|event| on_ok(event)}
+    end
+
+    def on_enter
+      UserInfo.login = :id_pass
+      UserInfo.mail = @mail.get_value
+      UserInfo.password = @password.get_value
+      Client.connect :mail => @mail.get_value, :password => @password.get_value
+      end_modal 1
+    end
+
+    def on_ok event
+      if @ie_btn.get_value
+        UserInfo.login = :ie
+        Client.connect :cookie => IECookieThief.get(:host => '.nicovideo.jp', :name => 'user_session')
+        end_modal 1
+      elsif @firefox_btn.get_value
+        UserInfo.login = :firefox
+        Client.connect :cookie => FirefoxCookieThief.get(:host => '.nicovideo.jp', :name => 'user_session')
+        end_modal 1
+      elsif @firefox_btn.get_value
+        UserInfo.login = :chrome
+        Client.connect :cookie => ChromeCookieThief.get(:host => '.nicovideo.jp', :name => 'user_session')
+        end_modal 1
+      else
+        on_enter
+      end
+    end
+  end
+end
+

data/lib/views/main_panel.rb

@@ -0,0 +1,50 @@
+module Okayu
+  class MainPanel < Wx::Panel
+    attr_reader :timer
+    def initialize(frame, x, y, w, h)
+      super(frame, -1, Wx::Point.new(x, y), Wx::Size.new(w, h))
+      @notebook = Wx::Notebook.new self,
+                                   :id =>ID_NOTEBOOK,
+                                   :pos => Wx::Point.new(x, y),
+                                   :size => Wx::Size.new(w, h)
+      # panel = LivePanel.new(@notebook)
+      # @notebook.add_page(panel, panel.title, TRUE)
+      create_tab
+
+      @timer = Wx::Timer.new(self, Wx::ID_ANY)
+      @timer.start
+      evt_timer(@timer.get_id) {|event| on_update}
+
+      evt_size{|event| on_size(event)}
+    end
+
+    def on_update
+      @notebook.each_page do |page|
+        page.on_update
+      end
+    end
+
+    def on_size event
+      if @notebook
+        size = get_client_size
+        @notebook.set_dimensions(2, 2, (size.get_width - 4), (size.get_height - 4))
+      end
+    end
+
+    def create_tab
+      panel = LivePanel.new(@notebook)
+      @notebook.add_page(panel, panel.title, TRUE)
+      @notebook.get_current_page.set_index(@notebook.get_selection)
+    end
+
+    def close_tab
+      @notebook.get_current_page.close
+      @notebook.delete_page(@notebook.get_selection)
+    end
+
+    def get_current_page
+      @notebook.get_current_page
+    end
+  end
+
+end

data/lib/views.rb

@@ -0,0 +1,11 @@
+require 'rubygems'
+require 'wx'
+
+require 'views/colors.rb'
+
+require 'views/login_dialog.rb'
+require 'views/color_dialog.rb'
+
+require 'views/live_panel.rb'
+require 'views/main_panel.rb'
+require 'views/app_frame.rb'