mikutter-rss-tl.rb

# -*- coding: utf-8 -*-

require 'rubygems'
require 'feed-normalizer'
require 'date'
require 'open-uri'


# IDからシンボルを作る
def sym(base, id)
  (base + id.to_s).to_sym
end


# 検索クラス
class Satoshi
  attr_reader :last_fetch_time

  ICON_COLORS = {
    :red => ["レッド", File.dirname(__FILE__) + "/red-icon-128.png"],
    :blue => ["ブルー", File.dirname(__FILE__) + "/blue-icon-128.png"],
    :black => ["ブラック", File.dirname(__FILE__) + "/black-icon-128.png"],
    :bronze => ["ブロンズ", File.dirname(__FILE__) + "/bronze-icon-128.png"],
    :green => ["グリーン", File.dirname(__FILE__) + "/green-icon-128.png"],
    :lightblue => ["ライトブルー", File.dirname(__FILE__) + "/lightblue-icon-128.png"],
    :mistred => ["ミストレッド", File.dirname(__FILE__) + "/mistred-icon-128.png"],
    :purple => ["パープル", File.dirname(__FILE__) + "/purple-icon-128.png"],
    :mikutter => ["みくったーちゃん", MUI::Skin.get("icon.png")],
  }

  # コンストラクタ
  def initialize(service, user_config, id)
    @service = service
    @last_config = Hash.new
    @result_queue = Array.new()
    @queue_lock = Mutex.new()
    @last_result_time = nil
    @last_fetch_time = Time.now
    @user_config = user_config
    @id = id
  end


  # インスタンス毎のコンフィグを読み込む
  def [](key)
    @user_config[sym(key.to_s, @id)]
  end


  # コンフィグの初期化
  def init_user_config()
    @user_config[sym("rss_url", @id)] ||= ""
    @user_config[sym("rss_reverse", @id)] ||= false

    @user_config[sym("rss_custom_style", @id)] ||= false
    @user_config[sym("rss_font_face", @id)] ||= 'Sans 10'
    @user_config[sym("rss_font_color", @id)] ||= [0, 0, 0]
    @user_config[sym("rss_background_color", @id)] ||= [65535, 65535, 65535]
    @user_config[sym("rss_loop", @id)] ||= false
    @user_config[sym("rss_icon", @id)] ||= :red
  end


  # 設定画面の生成
  def setting(plugin, prefix)
    id = @id

    plugin.settings prefix + "フィード" + id.to_s do
      input("URL", sym("rss_url", id))
      boolean("新しい記事を優先する", sym("rss_reverse", id))
      boolean("ループさせる", sym("rss_loop", id))

      select("アイコンの色", sym("rss_icon", id), ICON_COLORS.inject({}){ |result, kv|
        result[kv[0]] = kv[1][0] 
        result
      })

      settings "カスタムスタイル" do
        boolean("カスタムスタイルを使う", sym("rss_custom_style", id))
        fontcolor("フォント", sym("rss_font_face", id), sym("rss_font_color", id))
        color("背景色", sym("rss_background_color", id))
      end
    end
  end


  # 検索結果を取り出す
  def fetch()
    msg = nil

    @queue_lock.synchronize {
      if @user_config[sym("rss_reverse", @id)] then
        msg = @result_queue.pop
      else
        msg = @result_queue.shift
      end
    }

    if msg != nil then
      @last_fetch_time = Time.now
    end 

    # puts @urls.to_s + @result_queue.size.to_s

    return msg
  end


  # メッセージ保有してる？
  def empty?()
   @result_queue.empty?
  end


  # 検索する
  def search()
    begin
      url = @user_config[sym("rss_url", @id)]

      # 検索オプションが変わったら、キャッシュを破棄する
      is_reload = [sym("rss_url", @id)]
        .inject(false) { |result, key|
        result = result || (@user_config[key] != @last_config[key])

        @last_config[key] = @user_config[key]

        result
      }

      if is_reload then
        p "ID:" + @id.to_s + " setting changed"

        @queue_lock.synchronize {
          @result_queue.clear
          @last_result_time = nil
        }
      end

      # ループさせる
      if @user_config[sym("rss_loop", @id)] && empty? then
        @last_result_time = nil
      end

      if @user_config[sym("rss_url", @id)].empty? then
        return true
      end
 
      # RSSを読み込む
      feed = FeedNormalizer::FeedNormalizer.parse(open(@user_config[sym("rss_url", @id)]))

      drop_time = Time.now.to_i - (24 * 60 * 60 * @user_config["rss_drop"].to_i)

      entries = feed.entries.select { |entry|
        result_tmp = false

        if entry.last_updated != nil && entry.last_updated.to_i < drop_time then
            next
        elsif @last_result_time == nil then
          result_tmp = true
        elsif entry.last_updated != nil && @last_result_time < entry.last_updated then
          result_tmp = true
        else
          result_tmp = false
        end

        result_tmp
      }
  
      if entries.size == 0 then
        return true
      end
  
      msgs = entries.map { |entry| 
        # どうせタイムライン表示時に自動展開されちゃうので短縮はしない
        # links = MessageConverters.shrink_url([item.link.to_s])

        msg = Message.new(:message => ("【" + feed.title.force_encoding("utf-8") + "】\n" + entry.title.force_encoding("utf-8") + "\n\n[記事を読む]"), :system => true)
        msg[:rss_feed_url] = entry.url.force_encoding("utf-8")
        msg[:created] = entry.last_updated

        if feed.image.empty? then
          image_url = ICON_COLORS[@user_config[sym("rss_icon", @id)]][1]
        else
          image_url = feed.image
        end

        title = feed.title.force_encoding("utf-8")

        msg[:user] = User.new(:id => -3939,
                              :idname => "RSS",
                              :name => title,
                              :profile_image_url => image_url)

        if entry.last_updated != nil && (@last_result_time == nil || @last_result_time < entry.last_updated) then
          @last_result_time = entry.last_updated
        end

        msg
      }
  
      p "new message:" + msgs.size.to_s
      p "last time:" + $last_time.to_s
  
      @queue_lock.synchronize {
        @result_queue.concat(msgs.reverse)
      }
    rescue => e
      puts e
      puts e.backtrace

      return false 
    end

    return true
  end
end


Plugin.create :rss do 
  # グローバル変数の初期化
  $satoshis = []


  # カスタムスタイルを選択する
  def choice_style(message, key, default)
    if message[:rss] != nil && message[:rss][:rss_custom_style] then
      message[:rss][key]
    else
      default
    end
  end


  # 更新用ループ
  def search_loop(service)
    search_url(service) 

    Reserver.new(UserConfig[:rss_period]){
      search_loop(service)
    } 
  end
  

  # 混ぜ込みループ
  def insert_loop(service)
    begin

      # 混ぜ込むべきインスタンスを取得
      target_satoshi = $satoshis.select { |a| a != nil }
                                .sort { |a, b| a.last_fetch_time <=> b.last_fetch_time }
                                .find { |satoshi| !satoshi.empty?  }

      if target_satoshi != nil then
        msg = target_satoshi.fetch()

        msg[:modified] = Time.now
        msg[:rss] = target_satoshi
  
        # タイムラインに登録
        if defined?(timeline)
          timeline(:home_timeline) << [msg]
        else
          Plugin.call(:update, service, [msg])
        end

        # puts "last message :" + $result_queue.size.to_s
      end

      Reserver.new(UserConfig[:rss_insert_period]){
        insert_loop(service)
      } 

    rescue => e
      puts e
      puts e.backtrace
    end
  end


  # 検索
  def search_url(service)
    begin
      $satoshis.each { |satoshi|
        result = satoshi.search()

        if !result then
          msg = Message.new(:message => "フィードの取得に失敗しました。 " + satoshi[:rss_url], :system => true)

          msg[:user] = User.new(:id => -3939,
                                :idname => "RSS",
                                :name => "エラー",
                                :profile_image_url => MUI::Skin.get("icon.png"))

          # タイムラインに登録
          if defined?(timeline)
            timeline(:home_timeline) << [msg]
          else
            Plugin.call(:update, service, [msg])
          end
        end
      }
    rescue => e
      puts e
      puts e.backtrace
    end
  end


  # 起動時処理
  on_boot do |service|
    (0..5 - 1).each {|i|
      $satoshis << Satoshi.new(service, UserConfig, i + 1)
    }

    # コンフィグの初期化
    UserConfig[:rss_period] ||= 60
    UserConfig[:rss_insert_period] ||= 3
    UserConfig[:rss_drop] ||= 7

    $satoshis.each {|satoshi|
      satoshi.init_user_config
    }

    # 設定画面
    settings "RSS混ぜ込み" do
      adjustment("ポーリング間隔（秒）", :rss_period, 1, 6000)
      adjustment("混ぜ込み間隔（秒）", :rss_insert_period, 1, 600)
      adjustment("一定期間より前のフィードは流さない（日）", :rss_drop, 1, 365)
    end 

    $satoshis.each {|satoshi|
      if satoshi.equal?($satoshis[-1]) then
        satoshi.setting(self, "┗")
      else
        satoshi.setting(self, "┣")
      end
    }

    search_loop(service)
    insert_loop(service)
  end


  # 背景色決定
  filter_message_background_color do |message, color|
    begin
      color = choice_style(message.message, :rss_background_color, color)

      [message, color]

    rescue => e
      puts e
      puts e.backtrace
    end
  end


  # フォント色決定
  filter_message_font_color do |message, color|
    begin
      color = choice_style(message.message, :rss_font_color, color)

      [message, color]

    rescue => e
      puts e
      puts e.backtrace
    end
  end


  # フォント決定
  filter_message_font do |message, font|
    begin
      font = choice_style(message.message, :rss_font_face, font)

      [message, font]

    rescue => e
      puts e
      puts e.backtrace
    end
  end


  # リンクの処理
  Message::Entity.addlinkrule(:rss, /\[記事を読む\]/){ |segment|
    if segment[:message][:rss_feed_url] != nil then
      Gtk::TimeLine.openurl(segment[:message][:rss_feed_url])
    end
  }

end