diff options
Diffstat (limited to 'sample/cal.rb')
| -rw-r--r-- | sample/cal.rb | 262 |
1 files changed, 155 insertions, 107 deletions
diff --git a/sample/cal.rb b/sample/cal.rb index a5f4b4db18..97f75bcf1c 100644 --- a/sample/cal.rb +++ b/sample/cal.rb @@ -1,118 +1,166 @@ -#! /usr/local/bin/ruby - -# cal.rb (bsd compatible version): Written by Tadayoshi Funaba 1998 -# $Id: bsdcal.rb,v 1.2 1998/12/01 13:47:40 tadf Exp $ - -require 'date2' - -$tab = -{ - 'cn' => true, # China - 'de' => 2342032, # Germany (protestant states) - 'dk' => 2342032, # Denmark - 'es' => 2299161, # Spain - 'fi' => 2361390, # Finland - 'fr' => 2299227, # France - 'gb' => 2361222, # United Kingdom - 'gr' => 2423868, # Greece - 'hu' => 2301004, # Hungary - 'it' => 2299161, # Italy - 'jp' => true, # Japan - 'no' => 2342032, # Norway - 'pl' => 2299161, # Poland - 'pt' => 2299161, # Portugal - 'ru' => 2421639, # Russia - 'se' => 2361390, # Sweden - 'us' => 2361222, # United States - 'os' => false, # (old style) - 'ns' => true # (new style) -} - -$cc = 'gb' - -def usage - $stderr.puts 'usage: cal [-c iso3166] [-jy] [[month] year]' - exit 1 -end +#! /usr/bin/env ruby + +# cal.rb: Written by Tadayoshi Funaba 1998-2004,2006,2008 +# $Id: cal.rb,v 2.11 2008-01-06 08:42:17+09 tadf Exp $ + +require 'date' + +class Cal -def cal(m, y, gs) - for d in 1..31 - break if jd = Date.exist?(y, m, d, gs) + START = + { + 'cn' => Date::GREGORIAN, # China + 'de' => 2342032, # Germany (protestant states) + 'dk' => 2342032, # Denmark + 'es' => 2299161, # Spain + 'fi' => 2361390, # Finland + 'fr' => 2299227, # France + 'gb' => 2361222, # United Kingdom + 'gr' => 2423868, # Greece + 'hu' => 2301004, # Hungary + 'it' => 2299161, # Italy + 'jp' => Date::GREGORIAN, # Japan + 'no' => 2342032, # Norway + 'pl' => 2299161, # Poland + 'pt' => 2299161, # Portugal + 'ru' => 2421639, # Russia + 'se' => 2361390, # Sweden + 'us' => 2361222, # United States + 'os' => Date::JULIAN, # (old style) + 'ns' => Date::GREGORIAN # (new style) + } + + DEFAULT_START = 'gb' + + def initialize + opt_j; opt_m; opt_t; opt_y; opt_c end - fst = cur = Date.new(jd, gs) - ti = Date::MONTHNAMES[m] - ti << ' ' << y.to_s unless $yr - mo = ti.center((($w + 1) * 7) - 1) << "\n" - mo << ['S', 'M', 'Tu', 'W', 'Th', 'F', 'S']. - collect{|x| x.rjust($w)}.join(' ') << "\n" - mo << ' ' * (($w + 1) * fst.wday) - while cur.mon == fst.mon - mo << (if $jd then cur.yday else cur.mday end).to_s.rjust($w) - mo << (if (cur += 1).wday != 0 then "\s" else "\n" end) + + def opt_j(flag=false) @opt_j = flag end + def opt_m(flag=false) @opt_m = flag end + def opt_t(flag=false) @opt_t = flag end + def opt_y(flag=false) @opt_y = flag end + + def opt_c(arg=DEFAULT_START) @start = START[arg] end + + def set_params + @dw = if @opt_j then 3 else 2 end + @mw = (@dw + 1) * 7 - 1 + @mn = if @opt_j then 2 else 3 end + @tw = (@mw + 2) * @mn - 2 + @k = if @opt_m then 1 else 0 end + @da = if @opt_j then :yday else :mday end end - mo << "\n" * (6 - ((fst.wday + (cur - fst)) / 7)) - mo -end -def zip(xs) - yr = '' - until xs.empty? - ln = (if $jd then l, r, *xs = xs; [l, r] - else l, c, r, *xs = xs; [l, c, r] end). - collect{|x| x.split(/\n/no, -1)} - 8.times do - yr << ln.collect{|x| - x.shift.ljust((($w + 1) * 7) - 1)}.join(' ') << "\n" - end + def pict(y, m) + d = (1..31).detect{|x| Date.valid_date?(y, m, x, @start)} + fi = Date.new(y, m, d, @start) + fi -= (fi.jd - @k + 1) % 7 + + ve = (fi..fi + 6).collect{|cu| + %w(S M Tu W Th F S)[cu.wday] + } + ve += (fi..fi + 41).collect{|cu| + if cu.mon == m then cu.send(@da) end.to_s + } + + ve = ve.collect{|e| e.rjust(@dw)} + + gr = group(ve, 7) + gr = trans(gr) if @opt_t + ta = gr.collect{|xs| xs.join(' ')} + + ca = %w(January February March April May June July + August September October November December)[m - 1] + ca = ca + ' ' + y.to_s if !@opt_y + ca = ca.center(@mw) + + ta.unshift(ca) + end + + def group(xs, n) + (0..xs.size / n - 1).collect{|i| xs[i * n, n]} + end + + def trans(xs) + (0..xs[0].size - 1).collect{|i| xs.collect{|x| x[i]}} + end + + def stack(xs) + if xs.empty? then [] else xs[0] + stack(xs[1..-1]) end end - yr + + def block(xs, n) + stack(group(xs, n).collect{|ys| trans(ys).collect{|zs| zs.join(' ')}}) + end + + def unlines(xs) + xs.collect{|x| x + "\n"}.join + end + + def monthly(y, m) + unlines(pict(y, m)) + end + + def addmon(y, m, n) + y, m = (y * 12 + (m - 1) + n).divmod(12) + return y, m + 1 + end + + def yearly(y) + y.to_s.center(@tw) + "\n\n" + + unlines(block((0..11).collect{|n| pict(*addmon(y, 1, n))}, @mn)) + "\n" + end + + def print(y, m) + set_params + if @opt_y then yearly(y) else monthly(y, m) end + end + end -while /^-(.*)$/no =~ $*[0] - a = $1 - if /^c(.+)?$/no =~ a then - if $1 then - $cc = $1.downcase - elsif $*.length >= 2 then - $cc = $*[1].downcase - $*.shift - else - usage - end - else - a.scan(/./no) do |c| - case c - when 'j'; $jd = true - when 'y'; $yr = true - else usage +if __FILE__ == $0 + + require 'getoptlong' + + def usage + warn 'usage: cal [-c iso3166] [-jmty] [[month] year]' + exit 1 + end + + cal = Cal.new + + begin + GetoptLong.new(['-c', GetoptLong::REQUIRED_ARGUMENT], + ['-j', GetoptLong::NO_ARGUMENT], + ['-m', GetoptLong::NO_ARGUMENT], + ['-t', GetoptLong::NO_ARGUMENT], + ['-y', GetoptLong::NO_ARGUMENT]). + each do |opt, arg| + case opt + when '-c'; cal.opt_c(arg) || raise + when '-j'; cal.opt_j(true) + when '-m'; cal.opt_m(true) + when '-t'; cal.opt_t(true) + when '-y'; cal.opt_y(true) end end + rescue + usage end - $*.shift -end -usage if (gs = $tab[$cc]).nil? -case $*.length -when 0 - td = Date.today - m = td.mon - y = td.year -when 1 - y = $*[0].to_i - $yr = true -when 2 - m = $*[0].to_i - y = $*[1].to_i -else - usage -end -usage unless m.nil? or (1..12) === m -usage unless y >= -4712 -$w = if $jd then 3 else 2 end -unless $yr then - print cal(m, y, gs) -else - print y.to_s.center(((($w + 1) * 7) - 1) * - (if $jd then 2 else 3 end) + - (if $jd then 2 else 4 end)), "\n\n", - zip((1..12).collect{|m| cal(m, y, gs)}), "\n" + + y, m = ARGV.values_at(1, 0).compact.collect{|x| x.to_i} + cal.opt_y(true) if y && !m + + to = Date.today + y ||= to.year + m ||= to.mon + + usage unless m >= 1 && m <= 12 + usage unless y >= -4712 + + print cal.print(y, m) + end + +# See Bird & Wadler's Introduction to functional programming 4.5. |