Pryのプロンプトに色付けする方法

本当は良くないんだけど、production環境で直接rails consoleを使って作業する事があるのでうっかりローカル環境と間違えないようにpryの色を変えておきたい。

検索するとPry.config.promptにprocを代入する方法がよく紹介されているけど現在は deprecated なので別の方法を使う。

Pry.config.prompt = Pry::Prompt.new(
  "custom",
  "my custom prompt",
  [proc { |obj, _, _|
    Pry::Helpers::Text.blue(RUBY_VERSION) + " " +
    Pry::Helpers::Text.green("pry(#{obj})> ")
  }]
)

これでrubyのバージョンを青文字で戦闘に表示してpry(main)の部分は緑色になる。production環境ではgreenをredに置き換えて赤色で表示しておいた。Rails.envなどを参照してdevelopmentやproductionを表示してもいい。

参考

Custom prompt with color · Issue #493 · pry/pry