Rubyでカレンダーを表示するプログラムを作成する

はじめに

以前からRubyの学習を進めており、アウトプットとしてカレンダーを出力するプログラムを作成しました。

今回は、プログラムの作成の過程についてまとめていきます。私のようなプログラミング学習初心者の一助となれば幸いです。

アウトプットイメージ

アウトプットイメージは、以下の添付画像の通り
Macのターミナルでcalコマンドを実行した結果と同じものを作成します。

追加要件として、以下も満たすことにします。

• 曜日は月曜日始まりにする（calコマンドの実行結果は日曜日始まり）
• rubyコマンドでファイルを実行する際、-mオプションで月を指定できるようにする
  • -mオプションで指定した月のカレンダーを出力する（今年）
• -mオプションの引数を指定しない場合、今月のカレンダーを表示する（今年）
• -mオプションの引数が不正な月の場合、次のエラーを出力する

➜ ruby calendar.rb -m 22
22 is neither a month number (1..12) nor a name

なお、今回は以下を利用してプログラムを作成しています。

• class Date
• library optparse (Ruby 3.1 リファレンスマニュアル)

作業手順

すべての内容を一度に考えるのではなく、下記の通り3ステップに分けて考えていくことにしました。

1. 月と年
2. 曜日
3. 日付

1. 月と年

カレンダーを出力する関数を定義し、引数には年と月をもたせます。
ただ、月は英語で表示させる必要があります。

require 'date'

def show_calender(year, month)
  date = Date.new(year, month)
  puts "    #{date.strftime('%B')} #{year}    "
end

show_calender(2024, 6)

期待する結果を出力できました。

2. 曜日

曜日は、月曜日始まりにして出力します。

require 'date'

def show_calender(year, month)
  date = Date.new(year, month)
  puts "    #{date.strftime('%B')} #{year}    "
  puts 'Mo Tu We Th Fr Sa Su'
end

show_calender(2024, 6)

出力結果が、少しカレンダーらしくなりました。

3.日付

最後は、日付部分になります。プログラムを作成するにあたり、さらに課題を分解して考えます。

1. 出力するカレンダーの月によって、日付の開始曜日が異なる
2. 日付は日曜日で改行する必要がある

出力するカレンダーの月によって、日付の開始曜日が異なる

Date#wdayを利用すれば、解決できそうです。

具体的には、以下の通りです。

1. 当該月のカレンダーの開始日付からwdayを利用して開始曜日を数字（0から6）で取得する
2. wdayの出力結果ごとに、スペースの挿入回数を調整して開始曜日の直下から開始日付を出力する

念のため、表で整理しておきます。

コードにすると、以下の通りとなります。

require 'date'

def show_calender(year, month)
  date = Date.new(year, month)
  puts "    #{date.strftime('%B')} #{year}    "
  puts 'Mo Tu We Th Fr Sa Su'

  if date == 0
    print '   ' * 6
  else
    print '   ' * (date.wday - 1)
  end
  print "#{date.day.to_s.rjust(2)}"
end

show_calender(2024, 6)

2024/6/1は、土曜日なので開始曜日の直下に開始日付が出力できています。

補足となりますが、rjustメソッドを利用して日付を出力する際に右寄せしています。

また、rjustメソッドは、Stringクラスのメソッドのためdate.dayの戻り値（Integer）をto_sメソッドでStringに変換する必要があります。

日付は日曜日で改行する必要がある

上記課題を取り組む前に、先に当該月の日付をすべて出力しておきます。

日付の出力は、以下の2ステップで行います。

1. 当該月の開始日と終了日の日付を取得する
2. 1.の日付範囲を繰り返し処理で出力する

当該月の開始日と終了日の取得に伴い、全体的にコードを書き直します。

require 'date'

def show_calender(year, month)
  start_date = Date.new(year, month)
  end_date = Date.new(year, month, -1)
  puts "    #{start_date.strftime('%B')} #{year}    "
  puts 'Mo Tu We Th Fr Sa Su'

  if start_date == 0
    print '   ' * 6
  else
    print '   ' * (start_date.wday - 1)
  end

  (start_date..end_date).each do |date|
    print "#{date.day.to_s.rjust(2)}"
  end
end

show_calender(2024, 6)

6月の日付がすべて出力されました、あともう少しです。

Dateクラスには、日曜日かどうかを判定する便利なメソッド（sunday?）が用意されているので、以下の1行を追加して日曜日で改行します。

puts if date.sunday?

アウトプットイメージと同じカレンダーが出力できました。

※本プログラムの作成には、putsメソッドとprintメソッドの違いを把握しておくことが必要です。

引用：プロを目指す人のためのRuby入門[改訂2版] 言語仕様からテスト駆動開発・デバッグ技法まで (Software Design plus)

-mオプションについて

次に、rubyコマンドで先ほどのcalender.rbファイルをターミナルで実行する際、-mオプションで月を指定できるようにします。

例えば、以下のようなコマンドを実行すると、2024年5月のカレンダーを出力します。

ruby calendar.rb -m 5

他の要件も忘れずに実装していきます。

• -mオプションの引数を指定しない場合、今月のカレンダーを表示する（今年）
• -mオプションの引数が不正な月の場合、次のエラーを出力する

library optparseとは

-mオプションの実装には、library optparseを利用します。

library optparseについて、公式ドキュメントの内容を理解していきます。

optparseを利用する流れは、基本的に以下のようです。

1. OptionParser オブジェクト opt を生成する
2. オプションを取り扱うブロックを opt に登録する
3. opt.parse(ARGV) でコマンドラインを実際に parse（解析） する

require 'optparse'
opt = OptionParser.new

opt.on('-a') {|v| p v }
opt.on('-b') {|v| p v }

opt.parse!(ARGV)
p ARGV

上記コードの実行結果は、以下の通りです。

ruby sample.rb -a foo bar -b baz
# => true
     true
     ["foo", "bar", "baz"]

ARGVは、argument vectorの略でオプション（-a, -b）とオプションの引数の値を配列に格納しています。

ARGVからオプションを取り除くには、opt.parse!(ARGV)と破壊的メソッドを利用すれば問題ないようです。

また、OptionParser#onメソッドのオプション定義で末尾に引数名（VAL）を書くと、オプションは引数を受け付けることの指定となります。

require 'optparse'
opt = OptionParser.new

opt.on('-a VAL') {|v| p v }         # <- " VAL" を追加

opt.parse!(ARGV)
p ARGV

実行結果は以下の通りです。ただ、-aだけでなくVALもオプションとして解析されているのか、ARGV配列は空になっています。

ruby sample.rb -a hoge

# => "hoge"
     []

-mオプションを実装する

それでは、-mオプションを実装していきます。

簡潔に言うと、-mオプションの引数の値をshow_calender関数の第二引数に渡す実装を行います。

具体的には、optparse利用時の最後のブロックの内容を実行するタイミングで

• ハッシュに、-mオプションの値を格納する

ができれば、show_calender関数の第二引数にハッシュの値を渡すだけです。

以下のコードを追加すれば、要件に沿ったプログラムの完成です。

require 'optparse'
opt = OptionParser.new

values = {}

values[:year] = Date.today.year
values[:month] = Date.today.month

opt.on('-m month') { |month| values[:month] = month.to_i }
opt.parse!(ARGV)

if values[:month] < 1 || values[:month] > 12
  puts "#{values[:month]} is neither a month number (1..12) nor a name"
else
  show_calender(values[:year], values[:month])
end

ruby calender.rb -m 5とターミナルで実行すれば、2024年5月のカレンダーが出力されました。

オプションなしでコマンドを実行すれば、コマンド実行時の月のカレンダーが出力されます。

1から12以外の数字をコマンドのオプション引数に指定すると、エラーも出力されるようになっています。

おわりに

初めて1からプログラムを作成しましたが、かなり大変でした。ただ、エラーを解決し期待する結果が出力できた瞬間は楽しかったです。

学びとしては、課題を細かく分解して一つずつ時間をかけて考えていくことが大切だと感じました。

引き続きプログラミングの学習を続けていきます。

参考