class Tapioca::Dsl::Compilers::ActiveRecordRelations

def create_where_chain_methods(klass, return_type)
  WHERE_CHAIN_QUERY_METHODS.each do |method_name|
    case method_name
    when :not
      klass.create_method(
        method_name.to_s,
        parameters: [
          create_param("opts", type: "T.untyped"),
          create_rest_param("rest", type: "T.untyped"),
        ],
        return_type: return_type,
      )
    when :associated, :missing
      klass.create_method(
        method_name.to_s,
        parameters: [
          create_rest_param("args", type: "T.untyped"),
        ],
        return_type: return_type,
      )
    end
  end
end