class Tapioca::Dsl::Compilers::ActiveSupportTimeExt

def decorate
  return unless constant.respond_to?(:zone)
  root.create_path(constant) do |mod|
    return_type = if ::Time.zone
      "::ActiveSupport::TimeWithZone"
    else
      "::Time"
    end
    mod.create_method("current", return_type: return_type, class_method: true)
  end
end