#!/usr/local/bin/ruby
# tvguide-get.rb by anakata@vdnet.or.jp
# インターネットTVガイドから一週間分の番組表のURLを持ってくる
# 画像を取ってこないのでちと見栄え悪し → --get-iconで対応('00/05/02)
#
# '00/04/30  initial version.
# '00/05/02  ページ内のiconへのリンクを相対パスに書き換え
#            ページ内の外部へのリンクをCGI経由から直接リンクに書き換え
# '00/05/02  option追加(--span, --offset, --dir-to-save, --get-icon, --help)

require 'url-parse'
require 'http-access'
require 'getoptlong'

TVGUIDE_BASEURL = 'http://www.tvguide.or.jp/tableg/' # Gコード番組表
# TVGUIDE_BASEURL = 'http://www.tvguide.or.jp/table/' # 通常番組表
TVGUIDE_ICON_BASEURL = 'http://www.tvguide.or.jp/gif/'
ICON_LIST = 
['2.gif', 'bmode.gif', 'clear.gif', 'moji.gif', 
'n.gif', 'noscr.gif', 'sai.gif', 'shin.gif', 'shu.gif', 
'shuwa.gif', 'stereo.gif', 'taju.gif', 'w.gif', 'wide.gif']

#-----------------------------------------------------------------
# 下位手続きの定義

# page_get
# URLとファイル名を指定すると、URLの内容をHTTPで取得し、ファイルに保存する.
# なお、イテレータとして呼び出した場合には、取得した内容をブロックに渡し、
# その戻り値をファイルに保存する.
# URLが存在し、正常にファイルに保存できればtrue、さもなくばfalseを返す.
def page_get(urlstr, filename)
  url = URL.new(urlstr)
  http = HTTPAccess.new(url.host, url.port)
  if url.query
    http.request_get(url.path+'?'+url.query)
  else
    http.request_get(url.path)
  end
  http.get_response
  if http.code == '404'
    return false
  end

  begin
    f = File.new(filename, 'w')
    begin
      f.binmode
      data = ''
      http.get_data(65536) {|chunk| 
        if chunk != nil
          data << chunk
        end
      }
      if defined? yield
        f.print yield(data)
      else
        f.print data
      end
    ensure
      f.close
    end
    return true
  rescue
    return false
  end
end

# ディレクトリがないときは作成するchdir
# ただし再帰作成は行わない
def force_chdir(dir)
  if FileTest.exist?(dir)
    if FileTest.directory?(dir)
      Dir.chdir(dir)
    else
      puts "error: #{dir} is not directory."
      exit(1)
    end
  else
    begin
      Dir.mkdir(dir)
    rescue
      puts "error: #{dir} can't create."
      exit(1)
    end
    Dir.chdir(dir)
  end
end

#-----------------------------------------------------------------
# オプション取得・設定

parser = GetoptLong.new
parser.set_options(
  ['--span',        '-s', GetoptLong::REQUIRED_ARGUMENT],
  ['--offset',      '-o', GetoptLong::REQUIRED_ARGUMENT],
  ['--dir-to-save', '-d', GetoptLong::REQUIRED_ARGUMENT],
  ['--get-icon',    '-i', GetoptLong::NO_ARGUMENT],
  ['--help',        '-h', GetoptLong::NO_ARGUMENT]
)
begin
  parser.each_option do |name, arg|
    eval "$OPT_#{name.sub(/^--/, '').gsub(/-/, '_').upcase} = '#{arg}'"
  end
rescue
  exit(1)
end

if $OPT_HELP
print <<EOF
Usage: tvguide-get.rb [option...]
  save INTERNET TV GUIDE table page.
Options:
  -o OFFSET  --offset OFFSET   Set get offset(day)
  -s SPAN    --span SPAN       Set get span(day)
  -d DIR     --dir-to-save DIR specify output directory
  -i         --get-icon        get icon images too
  --help     -h                Output this help
Exec sample:
  tvguide-get.rb --get-icon --dir-to-save tvguide --offset -1 --span 15
EOF
  exit
end

if $OPT_SPAN 
  span = $OPT_SPAN.to_i
else
  span = 7
end

if $OPT_OFFSET 
  offset = $OPT_OFFSET.to_i
else
  offset = 0
end

if $OPT_DIR_TO_SAVE
  force_chdir($OPT_DIR_TO_SAVE)
end

#-----------------------------------------------------------------
# ページ取得

# icon取得(ファイル名決めうち)
if $OPT_GET_ICON
  curdir = Dir.pwd
  force_chdir('gif')
  ICON_LIST.each { |fn|
    page_get(TVGUIDE_ICON_BASEURL + fn, fn)
  }
  Dir.chdir(curdir)
end

# 各時間のページ取得
list = ''
now = Time.now
start_day = now + (offset * 24 * 3600)
stop_day = start_day + (span * 24 * 3600)
cur_day = start_day
while cur_day < stop_day
  m = sprintf("%02d", cur_day.mon)
  d = sprintf("%02d", cur_day.day)
  print "#{m}/#{d}:\n"
  list << "#{m}/#{d}:\n"
  for h in ['01','04','07','10','13','16','19','21','23']
    filename = "#{m}#{d}#{h}tb.htm"
    urlstr = "#{TVGUIDE_BASEURL}#{filename}"
    # ページ取得
    exist = page_get(urlstr, filename){ |data|
      # iconのURLを相対パスに置換
      # CGI経由のリンクを直接リンクに置換
      data.gsub(/src=[/]gif[/]/io, 'src=gif/').
           gsub(/HREF=[/]bin[/]go[.]to[?]/io, 'HREF=')
    }
    if exist
      list << "<A HREF=\"#{filename}\">#{h}h</A>\n"
    else
      list << "#{h}h\n"
    end
  end
  list << "<BR>\n"
  cur_day = cur_day + 24 * 3600
end

# index page出力
begin
 f = File.new('tvguide-index.html', 'w')
  begin
    f.print '<html><body>'
    f.print list
    f.print '</body>'
  ensure
    f.close
  end
rescue
end