osd

#!/usr/bin/env ruby
require 'open-uri'
require "curses"
require 'sqlite3'
require 'thread'
require 'nokogiri'
require 'json'
require 'rexml/document'
scrlck = Mutex.new
include Curses

class Menu
  attr_reader :window
  attr_reader :items

  def initialize(x = 0, y = 0, width = 80, height = 25, items = [])
    @x, @y, @width, @height = x, y, width, height
    @items = items
    @index = 0
    @scroll = 0
  end

  def visible_first
    @scroll
  end

  def visible_last
    [@items.count - 1, @scroll + @height - 3].min
  end

  def visible_range
    visible_first..visible_last
  end

  def scroll
    @scroll
  end

  def scroll=(v)
    max = [0, @items.count + 2 - @height].max
    @scroll = [v, max].min
  end

  def scroll_to index
    while index > visible_last
      self.scroll += 1
    end
    while index < visible_first
      self.scroll -= 1
    end
  end

  def refresh
    @window.clear
    active = [1, 1]
    @items.each_with_index { |i, index|
      next unless visible_range.include? index
      @window.setpos index + 1 - @scroll, 1
      @window.attrset(A_NORMAL)
      @window.addstr " " * (@width - 2)

      @window.setpos index + 1 - @scroll, 1
      @window.attrset(index == @index ? A_STANDOUT : A_NORMAL)
      @window.addstr i
      active = [index + 1 - @scroll, 1] if index == @index
    }
    @window.attrset A_NORMAL
    @window.box '|', '-'
    @window.setpos *active
  end

  def run
    @window = Window.new @height, @width, @y, @x
    $refreshable.push @window
    refresh
    while ch = @window.getch
      case ch
      when 'w'
        @index -= 1
      when 's'
        @index += 1
      when 'x'
        exit
      when 27
        ch = @window.getch
        case ch
        when '['
          ch = @window.getch
          case ch
          when 'A'
            @index -= 1
          when 'B'
            @index += 1
          end
        end
      end
      @index += @items.count
      @index %= @items.count
      scroll_to @index
      refresh
    end
    $refreshable.remove @window
    @window.close
    @window = nil
  end
end


# RX is receive, incoming. TX is transmitting, outgoing.

init_screen
start_color
noecho
start_color
init_pair 1, COLOR_RED, COLOR_BLACK
init_pair 2, COLOR_GREEN, COLOR_BLACK
init_pair 3, COLOR_YELLOW, COLOR_BLACK
init_pair 4, COLOR_BLACK, COLOR_RED
init_pair 5, COLOR_BLACK, COLOR_GREEN
init_pair 6, COLOR_BLACK, COLOR_YELLOW

def clearrange(x, y, w, h)
  for yy in y...y+h
    setpos yy, x
    addstr " " * w
  end
end

timer = Thread.new do
  font = <<EOF.split("\n").map{|i|i.split('|')}
█▀▀█| ▄█ |█▀▀█|█▀▀█|█  █|█▀▀▀|█▀▀▀|█▀▀█|█▀▀█|█▀▀█
*  *|  * | ▄▀ |▄ ▀*|▀▀▀*|▀▀▀█|█▀▀█|   *|*▀▀*|▀▀▀█
▀▀▀▀|  ▀ |▀▀▀▀|▀▀▀▀|   ▀|▀▀▀▀|▀▀▀▀|   ▀|▀▀▀▀|▀▀▀▀
EOF
  left = 63
  blk = true
  while true
    scrlck.synchronize {
      begin
        attrset A_NORMAL
        clearrange 62, 0, 23, 3
        setpos 0, left
        attron(color_pair(2))

        now = Time.now
        digi = now.strftime('%H%M').split('')
        digi.each_with_index do |d,x|
          3.times do |i|
            setpos i, left + 5 * x + (x > 1 ? 2 : 0)
            
            font[i][d.to_i].split('').each do |st|
              if st == "*"
                attrset A_STANDOUT
                attron(color_pair(2))
                addstr " "
              else
                attrset A_NORMAL
                attron(color_pair(2))
                addstr st
              end
            end
          end
        end
        attrset A_NORMAL
        attron(color_pair(2))
        if blk = !blk
          setpos 0, left+10
          addstr "▄"
          setpos 1, left+10
          addstr "▄"
        end
        setpos 3, 62
        addstr now.strftime("%Y-%m-%d 星期" + "日一二三四五六"[now.wday])
        setpos 3, 82
        addstr now.strftime(":%S")

        attrset A_NORMAL
        setpos 4, 62
        addstr '周:一 二 三 四 五 六 日'
        setpos 5, 62
        monday = Date.today
        today = Date.today
        monday -= 1 while !monday.monday?
        addstr '%02d' % ((monday.cweek < 35 ? 52 : 0) + monday.cweek - 35)
        addstr ':'
        7.times do |i|
          if monday.day != today.day
            attrset A_NORMAL
          else
            attrset A_STANDOUT
          end
          attron(color_pair(2)) if i >= 5
          addstr monday.strftime("%d")

          attrset A_NORMAL
          addstr ' ' unless i == 6
          monday += 1
        end
      rescue
      end
    }
    sleep 1
  end
end

pm25 = Thread.new do
  top = 6
  getpm25 = lambda {
    body = open 'https://www.soupm25.com/ajax.php?mod=citydata&cityid=32' do |io| io.read end rescue nil
    result = {}
    if body
      result = JSON.parse body
    end
    result
  }
  setcolor = lambda { |i|
    if i > 200
      attron(color_pair(4))
    elsif i > 100
      attron(color_pair(1))
    elsif i > 50
      attron(color_pair(3))
    else
      attron(color_pair(2))
    end
  }
  resetcolor = lambda {
    attrset A_STANDOUT
  }
  levelstr = lambda { |i|
    if i > 300
      return "危险"
    elsif i > 200
      return "很不健康"
    elsif i > 100
      return "不健康"
    elsif i > 50
      return "中"
    else
      return "优"
    end
  }
  data = nil
  while true
    new_data = getpm25.()
    if new_data
      data = new_data
      open 'pm25', "wb" do |io| io.write Marshal.dump data end
    end
    if !data
      data = open 'pm25', 'rb' do |io| Marshal.load io.read end rescue nil
    end
    scrlck.synchronize {
      begin
        attrset A_STANDOUT
        clearrange 62, top + 0, 23, 1
        setpos top + 0, 62
        time = Time.at data['citydata']['lasttime'].to_i
        addstr time.strftime "空气质量    %m/%d %H:%M"
        
        attrset A_NORMAL
        clearrange 62, top + 1, 23, 2
        
        setpos top + 1, 62
        addstr "PM 2.5"
        addstr "*" if data['citydata']['aqitype'] != '1'
        setcolor.(data['citydata']['aqi25'].to_i)
        setpos top + 1, 70
        addstr " %s" % levelstr.(data['citydata']['aqi25'].to_i)
        setpos top + 1, 82
        addstr "%3d" % data['citydata']['aqi25'].to_i
        resetcolor.()

        attrset A_NORMAL
        setpos top + 2, 62
        addstr "*" if data['citydata']['aqitype'] == '1'
        addstr "PM 10"
        setcolor.(data['citydata']['aqi10'].to_i)
        setpos top + 2, 70
        addstr " %s" % levelstr.(data['citydata']['aqi10'].to_i)
        setpos top + 2, 82
        addstr "%3d" % data['citydata']['aqi10'].to_i
        resetcolor.()
      rescue
        attrset A_STANDOUT
        clearrange 62, top + 0, 23, 1
        setpos top + 0, 62
        addstr "空气质量         不可用"
        attrset A_NORMAL
        clearrange 62, top + 1, 23, 2
        setpos top + 1, 62
        addstr $!.message
      end
      refresh
    }
    sleep 60
  end
end


weather = Thread.new do
  top = 9
  getweather = lambda {|d=0|
    body = open 'http://php.weather.sina.com.cn/xml.php?city=%d5%f2%bd%ad&password=DJOYnieT8234jlsK&day='+d do |io| io.read end rescue nil
  }
  data = {}
  new_data = {}
  l = lambda { |x|
    new_data[x] = getweather.(x)
    if new_data[x]
      data[x] = REXML::Document.new new_data[x]
      open 'weather' + x, "wb" do |io| io.write new_data[x] end
    end
    if !data[x]
      data[x] = open 'weather' + x, 'rb' do |io| REXML::Document.new io.read end rescue nil
    end
  }
  pstatus = lambda { |t|
    case t
    when "晴", "多云"
      attron(color_pair(2))
    when /雨/, /雪/
      attron(color_pair(1))
    when "阴"
      attron(color_pair(3))
    end
    addstr t
  }
  ppower = pwind = lambda { |t,pw|
    case pw.split('-').last.to_i
    when 0..4
    when 5
      attron(color_pair(3))
    else
      attron(color_pair(1))
    end
    addstr t
  }
  while true
    3.times do |i| l.(i.to_s) end
    scrlck.synchronize {
      begin
        attrset A_STANDOUT
        clearrange 62, top + 0, 23, 1
        setpos top + 0, 62
        
        data['0'].to_s[/!-- published at ([0-9][0-9][0-9][0-9]-[0-9][0-9]-[0-9][0-9] [0-9][0-9]:[0-9][0-9]:[0-9][0-9]) --/]
        time = Time.parse($1, '%Y-%m-%d %H:%M:%S')
        addstr time.strftime "天气        %m/%d %H:%M"

        attrset A_NORMAL
        clearrange 62, top + 1, 23, 6
        ['0', '1', '2'].each do |day|
          d = data[day]
          date = Date.parse d.elements['//savedate_weather'].text
          setpos top + 2 + day.to_i * 2, 62
          addstr "日一二三四五六"[date.wday]
          setpos top + 1 + day.to_i * 2, 62
          addstr date.strftime "%d "
          

          status1 = d.elements['//status1'].text
          status2 = d.elements['//status2'].text
          if status1 == status2
            pstatus.(status1)
          else
            pstatus.(status1)
            attrset A_NORMAL
            addstr "转"
            pstatus.(status2)
          end

          attrset A_NORMAL
          setpos top + 1 + day.to_i * 2, 76
          addstr "#{"%3d" % d.elements['//temperature1'].text.to_i}~#{"%3d" % d.elements['//temperature2'].text.to_i}°C"

          attrset A_NORMAL
          setpos top + 2 + day.to_i * 2, 65
          wind1 = d.elements['//direction1'].text.gsub(/风$/,'')
          power1 = d.elements['//power1'].text
          wind2 = d.elements['//direction2'].text.gsub(/风$/,'')
          power2 = d.elements['//power2'].text
          if wind1 == wind2 && power1 == power2
            pwind.(wind1,power1)
            addstr "风"
            ppower.(power1,power1)
            attrset A_NORMAL
            addstr "级"
          elsif wind1 != wind2 && power1 == power2
            pwind.(wind1,power1)
            addstr "风"
            attrset A_NORMAL
            addstr "转"
            pwind.(wind2,power1)
            addstr "风"
            ppower.(power1,power1)
            attrset A_NORMAL
            addstr "级"
          elsif wind1 == wind2 && power1 != power2
            pwind.(wind1,power1)
            addstr "风"
            ppower.(power1,power1)
            attrset A_NORMAL
            addstr "转"
            ppower.(power2,power2)
            attrset A_NORMAL
            addstr "级"
          elsif wind1 != wind2 && power1 != power2
            pwind.(wind1,power1)
            addstr "风"
            ppower.(power1,power1)
            attrset A_NORMAL
            addstr "级转"
            pwind.(wind2,power2)
            addstr "风"
            ppower.(power2,power2)
            attrset A_NORMAL
            addstr "级"
          end
        end
        
      rescue
        attrset A_STANDOUT
        clearrange 62, top + 0, 23, 1
        setpos top + 0, 62
        addstr "天气             不可用"
        attrset A_NORMAL
        clearrange 62, top + 1, 23, 3
        setpos top + 1, 62
        addstr $!.message
        addstr $!.backtrace.inspect
      end
      refresh
    }
    sleep 60
  end
end

$status = {}

netcheck = Thread.new do
  while true
    result = open 'http://127.0.0.1:4567/ping' do |io| io.read end rescue nil
    if result
      result.split('&').each{|i|
        ifname, status = i.split('=')
        $status[ifname.to_sym] = status.to_i == 0 ? true : false
      }
    end
    sleep 10
  end
end

bwmon = Thread.new do

  zone = {
    "sum" => [],
    "lwan" => [ "pppoe-lwan" ],
    "gfw" => [ "tun+" ],
    "wan" => [ "eth0.3", "pppoe-wan_uni" ]
  }
  rzone = {}
  zone.each do |k, v|
    v.each do |i|
      rzone[i] = k
    end
  end

  top = 0
  limit = 9
  width = 60

  beforename = lambda { |i|
    if i == "wan"
      if $status[:wan]
        attron(color_pair(2))
      elsif $status[:wwan]
        attron(color_pair(3))
      else
        attron(color_pair(1))
      end
      return " 公网"
    elsif i == "lwan"
      if $status[:lwan]
        attron(color_pair(2))
      else
        attron(color_pair(1))
      end
      return " 校园"
    elsif i == "gfw"
      if $status[:gfw]
        attron(color_pair(2))
      else
        attron(color_pair(1))
      end
      return " 加密"
    elsif i == "sum"
      return "    Σ"
    end
    "%5s" % i
  }
  last_bw = nil
  updatetitle = lambda {
    if last_bw
      scrlck.synchronize {
        setpos top+0, 0
        attrset A_STANDOUT
        addstr " " * width
        setpos top+0, 0
        addstr "流量 (#{last_bw[0][1] - last_bw[0][0]} 秒平均) - #{Time.now.to_i - last_bw[0][1]} 秒前"

        zone.keys.each_with_index do |i, index|
          setpos top+0, width - (index + 1) * 7 + 1
          attrset A_STANDOUT
          addstr beforename.(i)
        end
        attrset A_NORMAL
        refresh
      }
    end
  }

  while true
    bw = (open 'http://127.0.0.1:4567/bwmon' do |io| eval io.read end rescue nil)

    unless bw && (bw.is_a?(Array) && bw.length > 0) && last_bw != bw
      5.times do
        updatetitle.call
        sleep 1
      end
      next
    end
    hosts = Hash[File.read('hosts').split("\n").map{|i|a=i.split(" ", 2);[a.first.downcase,a.last]}]
    arp = (open 'http://127.0.0.1:4567/arp' do |io| eval io.read end rescue nil)
    rarp = {}
    arp.each do |k, v| rarp[v] = k end if arp
    last_bw = bw

    bws = bw.group_by{|i| i[2]}.map do |k, v|
      pp = v.group_by{|i|[i[4], rzone[i[3]]]}
      sum = [0, 0]
      pp.keys.each do |group|
        pp[group] = pp[group].inject(0){|i,k|i += k[5]}
        sum[group[0]] += pp[group]
      end
      pp[[0, "sum"]] = sum[0]
      pp[[1, "sum"]] = sum[1]
      [k, pp]
    end

    bws = bws.sort_by {|i| i[1][[0, "sum"]] + i[1][[1, "sum"]]}.reverse

    sum = {}
    bws.each do |k,v|
      v.each do |group, value|
        sum[group] ||= 0
        sum[group] += value
      end
    end
    bws << ["sum", sum]
    timestart = Time.at(bw[0][0]).strftime("%H:%M:%S")
    timeend = Time.at(bw[0][1]).strftime("%H:%M:%S")
    timelen = bw[0][1] - bw[0][0]

    scrlck.synchronize {
      attrset A_NORMAL
      clearrange 0, top+0, width, 3 + limit * 2

      setpos top+0, 0
      attrset A_STANDOUT
      addstr " " * width
      setpos top+0, 0
      addstr "流量 (#{bw[0][1] - bw[0][0]} 秒平均) - #{Time.now.to_i - bw[0][1]} 秒前"

      zone.keys.each_with_index do |i, index|
        setpos top+0, width - (index + 1) * 7 + 1
        attrset A_STANDOUT
        addstr beforename.(i)
      end

      attrset A_NORMAL
      setpos top+limit * 2 + 1, 0
      attrset A_STANDOUT
      addstr " Σ "
      attrset A_NORMAL
      addstr " ... "
      if bws.length > limit
        addstr "(+ #{bws.length - limit})"
      end

      attron A_UNDERLINE
      clearrange 0, top+limit * 2, width, 1
      attroff A_UNDERLINE
      alwaysunderline = false
      bws.each_with_index do |i, index|
        next if index >= limit && i[0] != "sum"
        index = limit if i[0] == "sum"

        host = i[0]
        data = i[1]

        unless i[0] == "sum"
          setpos top+index * 2 + 1, 0
          addstr "#{host} - #{rarp[host]}"

          setpos top+index * 2 + 2, 2
          attron A_UNDERLINE if alwaysunderline || index + 1 == limit
          addstr hosts[host] if hosts[host]
          attroff A_UNDERLINE if alwaysunderline || index + 1 == limit
        end

        setpos top+index * 2 + 1, width - (zone.keys.count) * 7 - 1
        attron(color_pair(2))
        addstr "↓"
        attroff(color_pair(2))

        setpos top+index * 2 + 2, width - (zone.keys.count) * 7 - 1
        attron A_UNDERLINE if alwaysunderline || index + 1 == limit
        attron(color_pair(1))
        addstr "↑"
        attroff(color_pair(1))
        attroff A_UNDERLINE if alwaysunderline || index + 1 == limit

        2.times do |dir|
          zone.keys.each_with_index do |i, i2|
            setpos top+index * 2 + 1 + dir, width - (i2 + 1) * 7 + 1

            d = data[[dir, i]] || 0
            d /= 1.0 * timelen
            d *= 8.0

            unit = " "
            if d >= 1000
              d /= 1024.0
              unit = "K"
            end
            if d >= 1000
              d /= 1024.0
              unit = "M"
            end
            if d >= 1000
              d /= 1024.0
              unit = "G"
            end
            attron A_UNDERLINE if (alwaysunderline || index + 1 == limit) && dir == 1
            attron(color_pair(2)) if unit == "M"
            attron(color_pair(3)) if unit == "K"
            addstr "%5.1f%c" % [d, unit]
            attroff(color_pair(2)) if unit == "M"
            attroff(color_pair(3)) if unit == "K"
            attroff A_UNDERLINE if (alwaysunderline || index + 1 == limit) && dir == 1
          end
        end
      end
      refresh
    }
  end
end

statusbar = Thread.new do
  loop do
    scrlck.synchronize {
      attrset A_STANDOUT
      clearrange 0, 21, 85, 1
      setpos 21,0
      addstr "#{Time.now.strftime("%Y-%m-%d")}#{`uptime`.chomp}"
      refresh
    }
    sleep 1
  end
end
=begin
result = `iw wlan0 scan`

aps = result.split(/^BSS/).select { |i| i.length > 0 }
ssids = aps.map { |ap|
  ap[/^\tSSID: (.*)$/] ? $1 : nil
}.sort.uniq.map { |i|
  i.gsub(/\\x(..)/) { $1.to_i(16).chr }.force_encoding("utf-8")
}

p ssids

menu = Menu.new 0, 0, 20, 10, ssids
menu.run
=end

bwmon.join
statusbar.join
pm25.join
weather.join
timer.join

exit