mplext.rb

# mplextended (the equivalent of screen_away.pl for irssi) for weechat.
# Copyright (C) 2010, Mikkel Kroman
# 
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# Requires weechat-ruby (http://github.com/dominikh/weechat-ruby).
#
# 2010-09-03, mk <[email protected]>
#     version 1.0.3: minor changes, added timestamp format
# 2010-07-27, mk <[email protected]>
#     version 1.0.2: changed global variables to instance variables and added
#                    buffer clearing as soon as it has been printed
# 2010-07-23, mk <[email protected]>
#     version 1.0.1: refactored a tiny bit of the code
# 2010-07-22, mk <[email protected]>
#     version 1.0: screen and tmux (untested) support

require 'weechat'
include Weechat

@script = {
  name:        "mplext",
  author:      "Mikkel Kroman <[email protected]>",
  license:     "GPL3",
  version:     "1.0.3",
  description: "Set away status and log messages when detaching or reattaching from a terminal multiplexer."
}

@config = Script::Config.new({
  "color"    => [Color, "lightgreen"],
  "format"   => [String, "%H:%M:%S"],
  "message"  => [String, "not here.."],
  "interval" => [Integer, 5]
})

@sock, @away, @buffer = nil, false, []

def setup
  if ENV.include? 'STY'
    if %x{env LC_ALL=C screen -ls} =~ %r{Sockets? in (/.+)\.}
      @socket = File.join $1, ENV['STY']
    end
  elsif ENV.include? 'TMUX'
    @socket = ENV['TMUX'].split(?,).first
  end
  
  Weechat::Hooks::Print.new &method(:check_highlight)
  Weechat::Timer.new @config.interval.to_i * 1000, &method(:check_presence)
end

def check_presence remaining
  attached = File.executable? @socket

  if attached and @away
    Weechat.puts "Terminal multiplexer attached. Clearing away status."
    Weechat.exec "/away -all"
    print_buffer
    @away = false

  elsif not attached and not @away
    Weechat.puts "Terminal multiplexer detached. Setting away status."
    Weechat.exec "/away -all #{@config.message}"
    @away = true
  end
end

def check_highlight buffer, date, tags, display, highlighted, prefix, message
  if @away and highlighted
    time, channel, nick = Time.now.to_i, buffer.short_name, prefix.chop
    @buffer << [time, channel, nick, message]
  end
end

def print_buffer
  @buffer.each do |time, channel, nickname, message|
    stamp = Time.at(time).strftime @config.format
    format = @config.color, stamp, channel, nickname, message

    Weechat.puts "%s %s [%s] %s: %s" % format
  end.clear
end

Ah, yes, of course! I completely forgot that. Thanks!