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

VER="(6893dbd)"
ME="bldx: "
$chdr="1;36"
$cinf="1;37"
$cadd="1;32"
$cdel="1;31"
$chk_fields=["ip"]
$incl=[]
$excl=[]
$fld=nil
$opt=nil
$mode="normal"

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

def chk_net(str)
  st=str.split("/")
  return false if st.length>2
  pf=32
  if st.length==2
    return false if !st[1] or st[1].length>2
    return false if !/^[0-9]+$/.match(st[1])
    pf=st[1].to_i
  end
  return false if pf<1 or pf>32
  st=st[0].split(".")
  return false if st.length!=4
  4.times do
    n=st.shift
    return false if !n or n.length>3
    return false if !/^[0-9]+$/.match(n)
    n=n.to_i
    return false if n<0 or n>255
    (pf-=8; next) if pf>7
    bf=255<<(8-pf)
    return false if n!=(n&bf)
    pf=0
  end
  return true
end

class Block
  attr_reader :la, :name
  Line=Struct.new :port, :ln, :lp, :arr, :del, :add, :ln2, :lp2
  Line1=Struct.new :s0, :arr

  def initialize(h)
    @hdr=h
    @fld=nil
    @sep=nil
    @flds=[]
    @ft=[]
    @la=[]
    @ca=[]
    @arr=[]
    @name=/^#!? ([^ ]*)/.match(h)[1]
  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=" ")
    st1=l.split(/\t+/)
    if st1[0].empty?
      tc=/\t+/.match(l)[0].length
      return if !i=@ft.index(tc)
      n=@flds.index($fld)
      return if i>n
      k=1+(n-i)
      return if k>=st1.length
      str=st1[k]
      z=@la.last
      if i==1 and !z.ln2 and not (st1[1].start_with?("!") or st1[1].start_with?("-"))
        st2=st1[1].split(":")
        z.ln2=st2[0]
        z.lp2=st2[1]
      end
    else
      return if st1.length<2
      return if st1[0]=="-" or st1[1].start_with?("-")
      port=st1[0]
      st2=st1[1].split(":")
      ln=st2[0]
      lp=st2[1]
      i=@flds.index($fld)
      return if i>=st1.length
      str=st1[i]
      if !z=find(port)
        z=Line.new(port,ln,lp,[],[],[])
        @la<<z
      end
    end
    return if str=="?" or str.start_with?("-") or 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.arr.concat(arr) if st==" "
    z.del.concat(arr) if st=="-"
    z.add.concat(arr) if st=="+"
    return
  end

  def line1(l)
    tmp=l.split(/\t+/)
    z=nil
    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]
      return if str=="?" or str.start_with?("-") or str.start_with?("!")
      z=@la.last
      z.arr<<str
      return
    end
    i=@flds.index($fld)
    return if i>=tmp.length
    str=tmp[i]
    return if str=="?" or str.start_with?("-") or str.start_with?("!")
    z=Line1.new(tmp[0],[str]) if !z
    @la<<z
    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=="?" or str.start_with?("-") or 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
    @arr.concat(arr)
    @arr.uniq!
    return
  end

  def cmp(b)
    for l1 in @la
      next if l1.ln!=b.name and l1.ln2!=b.name
      for l2 in b.la
        next if l2.ln!=@name and l2.ln2!=@name
        next if l2.lp!=l1.port and l2.lp2!=l1.port
        arr=l1.arr-l2.arr
        if not arr.empty?
          @ca<<Line.new(l1.port,l1.ln,l1.lp,arr) if l1.ln==b.name
          @ca<<Line.new(l1.port,l1.ln2,l1.lp2,arr) if l1.ln2==b.name
        end
        break
      end
    end
    return
  end

  def chk
    if $fld=="ip"
      for l in @la
        c=Line1.new(l.s0,[])
        for t in l.arr
          c.arr<<t if not chk_net(t)
        end
        @ca<<c if not c.arr.empty?
      end
      return
    end
    return
  end

  def output_chk(out)
    i=@flds.index($fld)
    if i==0
      mcs=@flds[0].length
      for t in @ca
        c=t.s0.length
        mcs=c if c>mcs
      end
      mcs=7 if mcs<7
      sep="-"*mcs
      out.puts "\e[#{$chdr}m%s\e[m" % @hdr
      out.puts @flds[0],sep
      @ca.each { |t| out.puts t.s0 }
      out.puts sep
      return
    end
    col=[]
    col<<@flds[0].length/8+1
    col<<@flds[i].length/8+1
    lcs=@flds[i].length
    for t in @ca
      c=t.s0.length/8+1
      col[0]=c if c>col[0]
      for u in t.arr
        c=u.length
        lcs=c if c>lcs
        c=c/8+1
        col[1]=c if c>col[1]
      end
    end
    mcs=col[0]*8+lcs
    tmp=@flds[0]
    c=col[0]-(@flds[0].length/8)
    tab="\t"*c
    tmp+=tab+@flds[i]
    sep="-"*mcs
    out.puts "\e[#{$chdr}m%s\e[m" % @hdr
    out.puts tmp,sep
    for t in @ca
      tmp=t.s0
      c=col[0]-(t.s0.length/8)
      tab="\t"*c
      tmp+=tab+t.arr[0]
      out.puts tmp
      l=t.arr.length-1
      next if t.arr.length<1
      l.times do |i|
        tab="\t"*col[0]
        out.puts tab+t.arr[i]
      end
    end
    out.puts sep
    return
  end

  def output(out)
    if $opt=="list"
      return if @arr.empty?
      @arr.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
      arr=combine(@arr)
      c=$chdr
      out.puts "\e[#{c}m%s\e[m" % @hdr
      out.puts "#{$fld}: %s" % to_str(arr)
      return
    end
    if $opt=="cmp"
      return if @ca.empty?
      c=$chdr
      out.puts "\e[#{c}m%s\e[m" % @hdr
      for l in @ca
        next if l.arr.empty?
        arr=combine(l.arr)
        str=l.ln
        str+=":"+l.lp if l.lp
        c=$cinf
        out.puts "\e[#{c}mport: %s\e[m  link: %s" % [l.port,str]
        out.puts "#{$fld}: %s" % to_str(arr)
      end
      return
    end
    if $opt=="get"
      c=$chdr
      out.puts "\e[#{c}m%s\e[m" % @hdr
      for l in @la
        next if l.arr.empty?
        arr=combine(l.arr)
        str=l.ln
        str+=":"+l.lp if l.lp
        c=$cinf
        out.puts "\e[#{c}mport: %s\e[m  link: %s" % [l.port,str]
        out.puts "#{$fld}: %s" % to_str(arr)
      end
      return
    end
    if $opt=="chk"
      return if @ca.empty?
      output_chk(out)
      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?
      str=l.ln
      str+=":"+l.lp if l.lp
      c=$cinf
      out.puts "\e[#{c}mport: %s\e[m  link: %s" % [l.port,str]
      c=$cdel
      out.puts "del: \e[#{c}m%s\e[m" % to_str(l.del) if not l.del.empty?
      c=$cadd
      out.puts "add: \e[#{c}m%s\e[m" % to_str(l.add) 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 to_str(arr)
    if $mode=="huawei"
      tmp=[]
      for s in arr
        (tmp<<s; next) if not s.include?("-")
        s.sub!("-"," to ")
        tmp<<s
      end
      return tmp.join(" ")
    end
    return arr.join(",")
  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=l.del&l.add
      l.del=l.del-arr
      l.add=l.add-arr
      next if l.del.empty? and l.add.empty?
      cnt+=1
      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 <fld> (aggregation)"
  puts "  -cmp          - compare given list <fld> (links.txt)"
  puts "  -get          - get given list <fld> (links.txt)"
  puts "  -chk          - validate <fld>"
  exit
end

def p_diff(isdf)
  blocks=[]
  bl=fld=nil
  inbl=false
  igbl=false
  st=" "
  for l in $stdin
    l.chop!
    next if l.empty?
    st=l.slice!(0,1) if isdf
    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
      next if igbl
      bl.list(l) if $opt=="list"
      bl.line1(l) if $opt=="chk"
      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 do_cmp(blocks)
  c=blocks.length
  i=0
  while i<c
    b1=blocks[i]
    i+=1
    j=i
    while j<c
      b2=blocks[j]
      j+=1
      b1.cmp(b2)
      b2.cmp(b1)
    end
  end
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
    if $opt
      $stderr.puts ME+"Cannot mix '-l' with other options"
      exit
    end
    $opt="list"
    next
  end
  if ARGV[i]=="-cmp"
    i+=1
    if $opt
      $stderr.puts ME+"Cannot mix '-cmp' with other options"
      exit
    end
    $opt="cmp"
    next
  end
  if ARGV[i]=="-get"
    i+=1
    if $opt
      $stderr.puts ME+"Cannot mix '-get' with other options"
      exit
    end
    $opt="get"
    next
  end
  if ARGV[i]=="-chk"
    i+=1
    if $opt
      $stderr.puts ME+"Cannot mix '-chk' with other options"
      exit
    end
    $opt="chk"
    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
  if ARGV[i]=="-huawei"
    i+=1
    $mode="huawei"
    next
  end
  $stderr.puts ME+"Unknown option: "+ARGV[i]
  exit
end

usage if i>=ARGV.length
$fld=ARGV[i]
if $opt=="chk" and not $chk_fields.include?($fld)
  $stderr.puts ME+"unsupported field: "+$fld
  exit
end
case $opt
  when "cmp" then blocks=p_diff(false)
  when "get" then blocks=p_diff(false)
  when "list" then blocks=p_data
  when "chk" then blocks=p_data
  else blocks=p_diff(true)
end

exit if blocks.empty?
do_cmp(blocks) if $opt=="cmp"
if $opt=="chk"
  blocks.each { |b| b.chk }
end
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