salticid 0.9.0

data/lib/salticid/host/ip.rb

@@ -0,0 +1,19 @@
+module Salticid::Host::IP
+  # Host methods for IP addresses
+  
+  # Get all active IP addresses.
+  def ips
+    `ifconfig -a`.split("\n\n").map { |stanza|
+      stanza.split("\n").find { |line|
+        line =~ /inet addr:\s*([\d\.]{7..15})/
+        $1
+      }
+    }.compact
+  end
+
+  def private_ips
+  end
+
+  def public_ips
+  end
+end

data/lib/salticid/interface.rb

@@ -0,0 +1,192 @@
+class Salticid
+  class Interface
+    require 'ncurses'
+    require 'salticid/interface/ncurses'
+    require 'salticid/interface/resizable'
+    require 'salticid/interface/view'
+    require 'salticid/interface/tab_view'
+    require 'salticid/interface/host_view'
+
+    COLORS = {
+      :info => Ncurses::COLOR_WHITE,
+      :error => Ncurses::COLOR_RED,
+      :warn => Ncurses::COLOR_YELLOW,
+      :debug => Ncurses::COLOR_CYAN,
+      :finished => Ncurses::COLOR_GREEN
+    }
+    COLOR_PAIRS = {}
+
+    # Keys
+    KEY_SPACE = 32
+    KEY_ENTER = 13
+    KEY_SCROLL_UP = 258
+    KEY_SCROLL_DOWN = 259
+
+    def self.interfaces
+      @interfaces ||= []
+    end
+
+    def self.shutdown *args
+      @interfaces.each do |i|
+        i.shutdown *args
+      end
+    end
+
+    attr_reader :hosts, :salticid
+
+    def initialize(salticid)
+      self.class.interfaces << self
+
+      # Set up ncurses
+      Ncurses.initscr
+      Ncurses.cbreak
+      Ncurses.noecho
+      Ncurses.nonl
+      Ncurses.stdscr.intrflush false
+      Ncurses.stdscr.keypad true
+      Ncurses.start_color
+      Ncurses.use_default_colors
+
+      @salticid = salticid
+      @hosts = []
+      @tabs = TabView.new(
+        self,
+        :height => 1
+      )
+
+      @tabs.window.keypad true
+
+      colorize
+    end
+
+    # Add a new tab interface backed by source.
+    def add_tab(host)
+      @tabs.active.hide if tab
+
+      hv = HostView.new(
+        self,
+        :host => host,
+        :top => 1,
+        :height => Ncurses.LINES - 1
+      )
+      hv.on_state_change do |state|
+        @tabs.render
+      end
+      @tabs << hv
+      tab.show
+    end
+
+    def colorize
+      COLORS.each_with_index do |c, i|
+        Ncurses.init_pair i + 1, c.last, -1
+        COLOR_PAIRS[c.first] = Ncurses.COLOR_PAIR(i + 1)
+      end
+    end
+
+    def delete_tab(target)
+      begin
+        target.hide
+        @tabs.delete target
+      ensure
+        tab.show if tab
+      end
+    end
+
+    # Join the mainloop.
+    def join
+      @main.join
+    end
+
+    # List channels and topics
+    def list(*channels)
+      @connection.list *channels
+    end
+
+    # Mainloop
+    def main
+      @main = Thread.new do
+        Thread.current.priority = -1
+        trap("WINCH") { resize if Thread.current.alive? }
+
+        loop do
+          # Get character
+          if IO.select [$stdin], nil, nil, 1
+            char = @tabs.window.getch
+          else
+            Thread.pass
+            next
+          end
+
+          # Do stuff
+          case char
+          when ?\t
+            @tabs.scroll
+          when ?q
+            shutdown
+          when Ncurses::KEY_LEFT
+            @tabs.scroll -1
+          when Ncurses::KEY_RIGHT
+            @tabs.scroll 1
+          when Ncurses::KEY_PPAGE
+            tab.scroll -tab.height / 2
+          when Ncurses::KEY_NPAGE
+            tab.scroll tab.height / 2
+          when Ncurses::KEY_UP
+            tab.scroll -1 
+          when Ncurses::KEY_DOWN
+            tab.scroll 1
+          end
+        end
+      end
+    end
+
+    # Resize to fit display
+    def resize
+      # We need to nuke ncurses to pick up the new dimensions
+      Ncurses.def_prog_mode
+      Ncurses.endwin
+      Ncurses.reset_prog_mode
+      height, width = Ncurses.dimensions
+   
+      # Resize tabs 
+      @tabs.resize(
+       :width => width,
+       :height => height
+      )
+      @tabs.render
+    end
+
+    # Shut down interface
+    def shutdown(and_exit = true)
+      # Shut down views
+      @tabs.shutdown
+     
+      # Shut down ncurses
+      Ncurses.echo
+      Ncurses.nocbreak
+      Ncurses.nl
+      Ncurses.endwin
+
+      # Stop interface
+      @main.exit rescue nil
+
+      # Exit if okay
+      exit if and_exit
+    end
+
+    # Switch to a different conversation
+    def switch(target = nil)
+      if target
+        # Switch to a specific tab
+        @tabs.switch_to_label target
+      else
+        # Switch to the next tab
+        @tabs.scroll
+      end
+    end
+
+    def tab
+      @tabs.active
+    end
+  end
+end

data/lib/salticid/interface/conversation_view.rb

@@ -0,0 +1,150 @@
+module IRC
+  require ROOT + '/interface'
+  require ROOT + '/view'
+  class Interface
+    class ConversationView < View
+
+      attr_accessor :messages, :window
+
+      COLORS = [
+        Ncurses::COLOR_WHITE,
+        Ncurses::COLOR_RED,
+        Ncurses::COLOR_GREEN,
+        Ncurses::COLOR_YELLOW,
+        Ncurses::COLOR_BLUE,
+        Ncurses::COLOR_MAGENTA,
+        Ncurses::COLOR_CYAN
+      ] 
+
+      def initialize(interface, params = {})
+        @messages = []
+
+        @scroll_position = -1
+
+        super
+
+        @color_index = 0
+
+        @color_map = Hash.new do |hash, key|
+          if key
+            hash[key] = @color_index.modulo Ncurses.COLOR_PAIRS
+            @color_index += 1
+          else
+            nil
+          end
+        end
+
+        colorize
+      end
+
+      def <<(message)
+        # Scroll if at bottom
+        @scroll_position += 1 if @scroll_position == @messages.size - 1
+
+        # Add color
+        if message.respond_to? :user
+          @color_map[message.user.nick]
+        end
+
+        @messages << message
+        render
+      end
+
+      # Set up colors
+      def colorize
+        COLORS.each_with_index do |color, i|
+          Ncurses.init_pair i, color, Ncurses::COLOR_BLACK
+        end
+      end
+
+      def render
+        if @hidden
+          return
+        end
+
+        @window.erase
+
+        lines_left = @height
+        message_i = @scroll_position
+        while message_i >= 0
+          # Message
+          message = @messages[message_i]
+          message_i -= 1
+
+          IRC.log "displaying message #{message}"
+
+          # Time
+          time = message.time.strftime "%H:%M:%S"
+
+          case message
+          when PrivMsg
+            color = Ncurses.COLOR_PAIR(@color_map[message.user.nick])
+            user = message.user.to_s || ''
+            text = message.text
+          when Event
+            user = ''
+            text = message.text
+          when Command
+            user = message.user.to_s || ''
+            text = message.text
+          else
+            user = message.user.to_s || ''
+            text = message.command + ' ' + message.params.join(" ")
+          end
+          
+          offset = time.length + user.length + 3
+
+          lines = [text[0, @width - offset]]
+          if remaining = text[@width - offset..-1]
+            lines += remaining.lines(@width - offset - 1)
+          end
+
+          # Put lines in reverse
+          i = lines.size
+          while i > 0
+            i -= 1
+            line = lines[i]
+
+            lines_left -= 1
+            break if lines_left < 0
+
+            if i.zero?
+              # Put top line
+              @window.move lines_left, 0
+              @window.addstr time + ' '
+              @window.attron Ncurses::A_BOLD
+              @window.attron color if color
+              @window.addstr user
+              @window.attroff color if color
+              @window.attroff Ncurses::A_BOLD
+              @window.addstr ': '
+              @window.addstr line
+            else
+              # Put hanging line
+              @window.move lines_left, offset
+              @window.addstr line
+            end
+
+            unless @window.cursor[1] == 0
+              # Clear rest of line
+              @window.clrtoeol
+            end
+          end
+        end
+        @window.refresh
+      end
+
+      # Scrolls the window by delta messages
+      def scroll(delta)
+        @scroll_position += delta
+        if @scroll_position < 0
+          @scroll_position = 0
+        elsif @scroll_position >= @messages.size
+          @scroll_position = @messages.size - 1
+        end
+
+        render
+      end
+    end
+  end
+end

data/lib/salticid/interface/host_view.rb

@@ -0,0 +1,123 @@
+class Salticid
+  class Interface
+    class HostView < View
+
+      attr_accessor :messages, :window, :state
+
+      def initialize(interface, params = {})
+        @messages = []
+
+        @scroll_position = -1
+
+        super
+
+        @host = params[:host]
+        @host.on_log do |message|
+          self << message
+        end
+
+        @state = nil
+        @on_state_change = proc { |state| }
+      end
+
+      def <<(message)
+        # Scroll if at bottom
+        @scroll_position += 1 if @scroll_position == @messages.size - 1
+
+        @messages << message
+
+        if @state != message.severity
+          @state = message.severity
+          @on_state_change.call(@state)
+        end
+        
+        render
+      end
+
+      def on_state_change(&block)
+        @on_state_change = block
+      end
+
+      def render
+        return if @hidden
+
+        @window.erase
+
+        lines_left = @height
+        message_i = @scroll_position
+        while message_i >= 0
+          # Message
+          message = @messages[message_i]
+          message_i -= 1
+
+          # Time
+          time = message.time.strftime "%H:%M:%S"
+
+          text = message.text
+          color = Interface::COLOR_PAIRS[message.severity]
+          
+          offset = time.length + 1
+
+          width = @width - offset
+          lines = text.scan(/[^\n]{1,#{width}}/m)
+
+          # Put lines in reverse
+          i = lines.size
+          while i > 0
+            i -= 1
+            line = lines[i]
+
+            lines_left -= 1
+            break if lines_left < 0
+
+            if i.zero?
+              # Put top line
+              @window.move lines_left, 0
+              @window.addstr time + ' '
+              @window.attron Ncurses::A_BOLD
+              @window.attron color if color
+              @window.addstr line
+            else
+              # Put hanging line
+              @window.attron Ncurses::A_BOLD
+              @window.attron color if color
+              @window.move lines_left, offset
+              @window.addstr line
+            end
+            @window.attroff color if color
+            @window.attroff Ncurses::A_BOLD
+
+            unless @window.cursor[1] == 0
+              # Clear rest of line
+              @window.clrtoeol
+            end
+          end
+        end
+        @window.refresh
+      end
+
+      def to_s
+        @host.name
+      end
+
+      # Scrolls the window by delta messages
+      def scroll(delta)
+        @scroll_position += delta
+        if @scroll_position < 0
+          @scroll_position = 0
+        elsif @scroll_position >= @messages.size
+          @scroll_position = @messages.size - 1
+        end
+
+        render
+      end
+
+      def shutdown
+        @host.on_log do |message|
+          puts message.text
+        end
+        @host = nil
+      end
+    end
+  end
+end