Rails でブロックまたはインライン文字列を受け取るメソッド

Question

たとえば content_tag の場合

content_tag :div, class: '...' do
  ...
end

content_tag :div, 'インライン文字列', class: '...'

のようにブロックもインライン文字列も受け取れますが
このようなメソッドはどうかけばよいのでしょうか

content_tag :div, class: を何度もかくのでショートハンド化したくて

ブロックを引数に取るとき
div(cls, option={}) do .. end

インライン文字列を引数に取るとき
div(cls, contents, option={})

という感じでよびだしたいのです

片方だけのメソッドは定義できるのですが両方に対応するにはどうすればよいのでしょうか

Answer 1

block_given?を使いましょう

def div(cls, contents = nil, option = {}, &block)
  if block_given?
    # ブロックを渡された時
  else
    # 渡されなかった時  
  end
end

Comments

[退会済みユーザー]

ブロックがあるときとないときで第２引数の扱いがかわってくるのですが
引数をスライドして取り出すというのは自動でやってくれるわけではないのでしょうか
ブロックがあるときとないときで 2 番目の引数を文字列としてとりだすかどうかは自分でかかないとだめってことですか？

[takumiabe]

ああ、それはそうなります。
https://github.com/rails/rails/blob/master/actionview/lib/action_view/helpers/url_helper.rb#L196-L206
これがすごく参考になると思います。(一番最初の多重代入でスライドしてますね)

[退会済みユーザー]

なるほどありがとうございます
地道にやるしかないんですね