#!/usr/bin/ruby
# vim: et sw=2

VER="(7afb555)"
ME="bldx: "
$chdr="1;36"
$cinf="1;37"
$cadd="1;32"
$cdel="1;31"
$incl=[]
$excl=[]
$fld=nil
$opt=nil

def in_list(s,list)
  for i in list
    if i.match(s)
      return true
    end
  end
  return false
end

def regexfix2(str)
  tmp=[]
  for t in str.scan(/(_)|(\|)|([^_\|]*)/).flatten.compact
    tmp.push(t) if not t.empty?
  end
  res=""
  while t=tmp.shift
    case t
      when("_") then res+=(tmp.empty? or tmp[0]=="|") ? '([ ."\[\]\(\)\{\}]+|$)' : '[ ."\[\]\(\)\{\}]+'
      when("|") then res+="|"
      else res+="#{t}"
    end
  end
  return res
end

class Block
  Line=Struct.new :port, :del, :add

  def initialize(h)
    @hdr=h
    @fld=nil
    @sep=nil
    @flds=[]
    @ft=[]
    @la=[]
    @lst=[]
  end

  def hdr(h)
    @hdr=h
  end

  def fld(f,s)
    @fld=f
    @sep=s
    tmp=@fld.scan(/[^\t]*\t*/)
    tmp.delete_if { |t| t.empty? }
    tc=0
    for t in tmp
      @flds<<t.delete("\t")
      @ft<<tc
      tc+=t.count("\t")+t.count("^\t")/8
    end
  end

  def ignore?
    return true if not $incl.empty? and not in_list(@hdr,$incl)
    return true if in_list(@hdr,$excl)
    return true if not @flds.include?($fld)
    return false
  end

  def line(l,st)
    tmp=l.split(/\t+/)
    if tmp[0].empty?
      tc=/\t+/.match(l)[0].length
      return if !i=@ft.index(tc)
      n=@flds.index($fld)
      return if i>n
      i=1+(n-i)
      return if i>=tmp.length
      str=tmp[i]
      z=@la.last
    else
      port=tmp[0]
      i=@flds.index($fld)
      return if i>=tmp.length
      str=tmp[i]
      if !z=find(port)
        z=Line.new(port,[],[])
        @la<<z
      end
    end
    return if str.start_with?("!")
    arr=[]
    for v in str.split(",")
      if v.include?("-")
        t=v.split("-")
        t[0].to_i.upto(t[1].to_i) { |x| arr<<x.to_s }
      else
        arr<<v
      end
    end
    z.del+=arr if st=="-"
    z.add+=arr if st=="+"
    return
  end

  def list(l)
    tmp=l.split(/\t+/)
    if tmp[0].empty?
      tc=/\t+/.match(l)[0].length
      return if !i=@ft.index(tc)
      n=@flds.index($fld)
      return if i>n
      i=1+(n-i)
      return if i>=tmp.length
      str=tmp[i]
    else
      i=@flds.index($fld)
      return if i>=tmp.length
      str=tmp[i]
    end
    return if str.start_with?("!")
    arr=[]
    for v in str.split(",")
      if v.include?("-")
        t=v.split("-")
        t[0].to_i.upto(t[1].to_i) { |x| arr<<x.to_s }
      else
        arr<<v
      end
    end
    @lst+=arr
    return
  end

  def output(out)
    if $opt
      return if @lst.empty?
      @lst.sort! do |x,y|
        next -1 if x.start_with?("+") or x.start_with?("*")
        next 1 if y.start_with?("+") or y.start_with?("*")
        next x.to_i<=>y.to_i
      end
      @lst=combine(@lst)
      c=$chdr
      out.puts "\e[#{c}m%s\e[m" % @hdr
      out.puts "#{$fld}: %s" % @lst.join(",")
      return
    end
    return if prepare<1
    c=$chdr
    out.puts "\e[#{c}m%s\e[m" % @hdr
    for l in @la
      next if l.del.empty? and l.add.empty?
      c=$cinf
      out.puts "\e[#{c}mport: %s\e[m" % l.port
      c=$cdel
      out.puts "del: \e[#{c}m%s\e[m" % l.del.join(",") if not l.del.empty?
      c=$cadd
      out.puts "add: \e[#{c}m%s\e[m" % l.add.join(",") if not l.add.empty?
    end
    return
  end

  private

  def find(port)
    for l in @la
      return l if l.port==port
    end
    return nil
  end

  def combine(list)
    arr=[]
    u=nil
    for v in list
      if u and v.to_i==u+1
        u+=1
        next
      end
      if u and arr.last.to_i<u
        z=arr.pop
        arr<<"%s-%d" % [z,u]
        u=nil
      end
      if not (v.start_with?("+") or v.start_with?("*"))
        u=v.to_i
      end
      arr<<v
    end
    if u and arr.last.to_i<u
      z=arr.pop
      arr<<"%s-%d" % [z,u]
    end
    return arr
  end

  def prepare
    cnt=0
    for l in @la
      l.del.uniq!
      l.add.uniq!
      arr=[]
      for v in l.add
        arr<<v if l.del.include?(v)
      end
      for v in arr
        l.add.delete(v)
        l.del.delete(v)
      end
      cnt+=1 if not (l.del.empty? and l.add.empty?)
      l.del.sort! do |x,y|
        next -1 if x.start_with?("+") or x.start_with?("*")
        next 1 if y.start_with?("+") or y.start_with?("*")
        next x.to_i<=>y.to_i
      end
      l.add.sort! do |x,y|
        next -1 if x.start_with?("+") or x.start_with?("*")
        next 1 if y.start_with?("+") or y.start_with?("*")
        next x.to_i<=>y.to_i
      end
      l.del=combine(l.del)
      l.add=combine(l.add)
    end
    return cnt
  end
end

def usage
  me=ME.delete(":")
  puts "Usage: "+me+"<fld>"
  puts "  -B <pattern>  - include blocks matching RegEx <pattern>"
  puts "  -S <pattern>  - skip blocks matching RegEx <pattern>"
  puts "  -l            - make list from input (aggregation)"
  exit
end

def p_diff
  blocks=[]
  bl=fld=nil
  inbl=false
  igbl=false
  for l in $stdin
    l.chop!
    next if l.empty?
    st=l.slice!(0,1)
    if inbl
      if l.start_with?("-----")
        blocks<<bl if not igbl
        bl=fld=nil
        inbl=false
        igbl=false
        next
      end
      bl.line(l,st) if not igbl
      next
    end
    if l.start_with?("#")
      if bl
        bl.hdr(l)
        next
      end
      bl=Block.new(l)
      next
    end
    if bl
      if not l.start_with?("-----")
        fld=l
        next
      end
      if !fld
        bl=nil
        next
      end
      bl.fld(fld,l)
      inbl=true
      igbl=bl.ignore?
      next
    end
  end
  return blocks
end

def p_data
  blocks=[]
  bl=fld=nil
  inbl=false
  igbl=false
  for l in $stdin
    l.chop!
    next if l.empty?
    if inbl
      if l.start_with?("-----")
        blocks<<bl if not igbl
        bl=fld=nil
        inbl=false
        igbl=false
        next
      end
      bl.list(l) if not igbl
      next
    end
    if l.start_with?("#")
      if bl
        bl.hdr(l)
        next
      end
      bl=Block.new(l)
      next
    end
    if bl
      if not l.start_with?("-----")
        fld=l
        next
      end
      if !fld
        bl=nil
        next
      end
      bl.fld(fld,l)
      inbl=true
      igbl=bl.ignore?
      next
    end
  end
  return blocks
end

$istty=$stdout.isatty
i=0
while i<ARGV.length
  if ARGV[i]=="--"
    i+=1
    break
  end
  break if ARGV[i]=="-"
  break if not ARGV[i].start_with?("-")
  usage if ARGV[i]=="--help"
  if ARGV[i]=="--version"
    me=ME.delete(":")
    puts me+VER+" by Borg."
    exit
  end

  if ARGV[i]=="-l"
    i+=1
    $opt="list"
    next
  end
  if ARGV[i]=="-B"
    i+=1
    if !ARGV[i]
      $stderr.puts ME+"Need argument for '-B' option"
      exit
    end
    str=regexfix2(ARGV[i])
    $incl<<Regexp.new(str,true)
    i+=1
    next
  end
  if ARGV[i]=="-S"
    i+=1
    if !ARGV[i]
      $stderr.puts ME+"Need argument for '-S' option"
      exit
    end
    str=regexfix2(ARGV[i])
    $excl<<Regexp.new(str,true)
    i+=1
    next
  end
  $stderr.puts ME+"Unknown option: "+ARGV[i]
  exit
end

usage if i>=ARGV.length
$fld=ARGV[i]
if $opt
  blocks=p_data
else
  blocks=p_diff
end

exit if blocks.empty?
if $istty
  out=IO.popen("less -fqRFX","w")
else
  out=$stdout
end
begin
  blocks.each { |b| b.output(out) }
rescue Errno::EPIPE
end
out.close if out!=$stdout