#!/usr/bin/ruby1.8

require 'rubygems'
require 'mechanize'
require 'iconv'
require 'rss/1.0'
require 'rss/2.0'
require 'open-uri'

def wrap(s)
  o=""
  until s.length<=16
    p=s[0..16]
    ok=false
    16.downto(0) do |i|
      if p[i..i]==" "
        o<<p[0..i-1]+"\n"
        s=s[i+1,s.length]
        ok=true
        break
      elsif p[i..i]=="-"
        o<<p[0..i]+"\n"
        s=s[i+1,s.length]
        ok=true
        break
      end
    end
    if not ok
      o<<p[0..15]+"\n"
      s=s[16,s.length]
    end
  end
  o<<s
end

class WeatherPage
  def initialize(day,name)
    @day=day
    @name=name
  end

  def scrape(page)
    col = @day.to_s
    @desc1 = page.search('//table[@class="wsPrognoseTabelle"]/tr[2]/td['+col+']/img')[0].attributes["alt"]
    @desc2 = page.search('//table[@class="wsPrognoseTabelle"]/tr[3]/td['+col+']/img')[0].attributes["alt"]
    @desc3 = page.search('//table[@class="wsPrognoseTabelle"]/tr[4]/td['+col+']/img')[0].attributes["alt"]
    @min = page.search('//table[@class="wsPrognoseTabelle"]/tr[5]/td['+col+']')[0].inner_html.gsub(/\s*$/,"")
    @max = page.search('//table[@class="wsPrognoseTabelle"]/tr[6]/td['+col+']')[0].inner_html.gsub(/\s*$/,"")
  end

  def desc
    if @desc1==@desc2 and @desc2==@desc3
      "<p style=\"text-align:left\">Ganzen Tag</p><br/>\n"+
      "<p style=\"text-align:right\">"+@desc1+"</p><br/>\n"+
      "<p> </p><br/>\n"+
      "<p> </p><br/>\n"+
      "<p> </p><br/>\n"+
      "<p> </p><br/>\n"
    elsif @desc1==@desc2
      "<p style=\"text-align:left\">Morgens, Nachmittags</p><br/>\n"+
      "<p style=\"text-align:right\">"+@desc1+"</p><br/>\n"+
      "<p style=\"text-align:left\">Abends</p><br/>\n"+
      "<p style=\"text-align:right\">"+@desc3+"</p><br/>\n"+
      "<p> </p><br/>\n"+
      "<p> </p><br/>\n"
    elsif @desc2==@desc3
      "<p style=\"text-align:left\">Morgens</p><br/>\n"+
      "<p style=\"text-align:right\">"+@desc1+"</p><br/>\n"+
      "<p style=\"text-align:left\">Nachmittags, Abends</p><br/>\n"+
      "<p style=\"text-align:right\">"+@desc2+"</p><br/>\n"+
      "<p> </p><br/>\n"+
      "<p> </p><br/>\n"
    elsif @desc1==@desc3
      "<p style=\"text-align:left\">Morgens, Abends</p><br/>\n"+
      "<p style=\"text-align:right\">"+@desc1+"</p><br/>\n"+
      "<p style=\"text-align:left\">Nachmittags</p><br/>\n"+
      "<p style=\"text-align:right\">"+@desc2+"</p><br/>\n"+
      "<p> </p><br/>\n"+
      "<p> </p><br/>\n"
    else
      "<p style=\"text-align:left\">Morgens</p><br/>\n"+
      "<p style=\"text-align:right\">"+@desc1+"</p><br/>\n"+
      "<p style=\"text-align:left\">Nachmittags</p><br/>\n"+
      "<p style=\"text-align:right\">"+@desc2+"</p><br/>\n"+
      "<p style=\"text-align:left\">Abends</p><br/>\n"+
      "<p style=\"text-align:right\">"+@desc3+"</p><br/>\n"
    end
  end

  def temp
   @min+"-"+@max+"C"
  end

  def get_page
    cache=@name.downcase+"_cache"
    cachetime=Time.new-(15*60)
    if File.exist?(cache) and (cachetime<=>File.mtime(cache))==-1
      # Cache is good, use it
      return IO.read(cache)
    end

    @url = "http://www.wdr.de/themen/wetter/nrw/prognose_wuppertal.jhtml"

    a = WWW::Mechanize.new

    a.get(@url) do |page|
      page.search("acronym").remove
      scrape(page)
    end

    @result = ""
    @result = "<p style=\"text-align:center\">Wetter "+@name+"</p><br/>\n"+desc+"<p style=\"text-align:center\">"+temp+"</p>"

    @result = Iconv.iconv("UTF-8", "ISO-8859-1", @result).to_s

    f=File.new(cache, "w")
    f.write @result
    f.close

    @result
  end
end

class BahnPage
  def initialize(station, arr_dep, time)
    @station = station
    @arr_dep = arr_dep 
    @time = time
  end

  def get_page
    cache="bahn_cache"
    cachetime=Time.new-(2*60)
    if File.exist?(cache) and (cachetime<=>File.mtime(cache))==-1
      # Cache is good, use it
      return IO.read(cache)
    end

    @url = "http://www.bahn.de/p/view/ankunft.shtml"

    a = WWW::Mechanize.new

    a.get(@url) do |page|
      results = page.form('formschnellsuche') do |f|
        f.field("input").value = @station
        f.field("boardType").value = @arr_dep
        f.checkboxes.each { |c| c.checked = (c.name=="GUIREQProduct_0") }
        f.field("time").value = @time
      end.submit

      @restime=""
      @row=1
      @seentimes=""
      until @restime==@time
        @restime=results.search('//td[@class="hafasContent"]/table[2]/tr['+@row.to_s+']/td[1]').innerHTML.gsub(/\s/,"")
        @seentimes << @restime+"\n"
        @row=@row+1
        raise "Time not found, "+@seentimes if @row==20
      end
      @row=(@row-1).to_s

      @train=results.search('//td[@class="hafasContent"]/table[2]/tr['+@row+']/td[3]/a').text.gsub(/\s/,"")
      if results.search('//td[@class="hafasContent"]/table[2]/tr[1]/th[5]').innerHTML=="Aktuelles"
        @status=results.search('//td[@class="hafasContent"]/table[2]/tr['+@row+']/td[6]/span').innerHTML.gsub(/&nbsp;/," ").gsub(/&#228;/,"ä").gsub(/&#246;/,"ö").strip
	@status=wrap(@status)
        @reason=results.search('//td[@class="hafasContent"]/table[2]/tr['+@row+']/td[7]').innerHTML.gsub(/&nbsp;/," ").gsub(/&#228;/,"ä").gsub(/&#246;/,"ö").strip
        @reason=wrap(@reason)
      else
        @status="Noch kein Info"
	t=Time.parse(@time)
	t=t-3600
        @reason="Info erst um "+t.strftime("%H:%M")
      end
    end

    @result = "<p style=\"text-align:left\">"+@time+"</p><p style=\"text-align:right\">"+@train+"</p><br/>\n"
    @result = @result+"<p style=\"text-align:center\">"+@status.gsub(/\n/,"</p><br/>\n<p style=\"text-align:center\">")+"</p><br/>\n"
    @result = @result+"<p> </p><br/>\n"
    @result = @result+"<p style=\"text-align:left\">"+@reason.gsub(/\n/,"</p><br/>\n<p style=\"text-align:left\">")+"</p><br/>\n"

    @result = Iconv.iconv("UTF-8", "ISO-8859-1", @result).to_s

    f=File.new(cache, "w")
    f.write @result
    f.close

    @result
  end
end

class RSSPage
  def initialize(url,name)
    @url=url
    @name=name
  end

  def get_page
    cache=@name.gsub(/\s/,"").downcase+"_cache"
    cachetime=Time.new-(30*60)
    if File.exist?(cache) and (cachetime<=>File.mtime(cache))==-1
      # Cache is good, use it
#      return IO.read(cache)
    end

    a = WWW::Mechanize.new { |agent|
      agent.user_agent_alias = "Windows Mozilla"
    }

    content=a.get_file(@url)
    rss = RSS::Parser.parse(content, false)

    @result = "<p style=\"text-align:center\">"+wrap(rss.items[0].title).gsub(/\n/,"</p><br/>\n<p style=\"text-align:center\">")+"</p><br/>\n"
    lines=0
    @result.scan("<br/>") do lines=lines+1 end
    lines.upto 6 do |l| @result << "<p> </p><br/>\n" end
    @result << "<p style=\"text-align:center\">"+@name+"</p>"

    f=File.new(cache, "w")
    f.write @result
    f.close

    @result
  end
end

t = Time.new

pages=[]
pages << BahnPage.new("Dortmund Hbf", "arr", "18:41") if t.wday==5 and t.hour>=17 and t.hour<19
pages << BahnPage.new("Dortmund Hbf", "dep", "07:14") if t.wday==1 and t.hour>=6 and t.hour<8
if pages.length==0
  pages << WeatherPage.new(1, "Heute")
  pages << WeatherPage.new(2, "Morgen") if t.hour>=9
  if t.hour>=8
    pages << RSSPage.new("http://www.spiegel.de/schlagzeilen/rss/0,5291,20,00.xml", "Spiegel Online")
    pages << RSSPage.new("http://newsrss.bbc.co.uk/rss/newsonline_world_edition/front_page/rss.xml", "BBC News")
    if t.wday>=1 and (t.wday<=4 or (t.wday==5 and t.hour<17))
      pages << RSSPage.new("http://rss.slashdot.org/Slashdot/slashdot", "Slashdot")
    end
  end
end


page=IO.read("pageno")
page="0" if page.nil?
page=page.to_i

page=0 if page>=pages.length

@result = pages[page].get_page

page=page+1

f=File.new("pageno", "w")
f.puts page
f.close

puts "Content-Type: application/xhtml+xml\n\n"
puts "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n"
puts "<!DOCTYPE html PUBLIC \"-//OPENWAVE//DTD XHTML Mobile 1.0//EN\" \"http://www.openwave.com/dtd/xhtml-mobile10.dtd\">\n"
puts "<html xmlns=\"http://www.w3.org/1999/xhtml\">\n"
puts "<head>\n"
puts "<meta name=\"expires\" content=\"90\" />\n"
puts "<!-- build-version: 0.72 -->\n"
puts "</head><body>\n"
puts @result
puts "</body>\n"
puts "</html>\n"