module GraphQL::StaticValidation::RequiredArgumentsArePresent

def assert_required_args(ast_node, defn)
  args = defn.arguments(context.query.context)
  return if args.empty?
  present_argument_names = ast_node.arguments.map(&:name)
  required_argument_names = context.warden.arguments(defn)
    .select { |a| a.type.kind.non_null? && !a.default_value? && context.warden.get_argument(defn, a.name) }
    .map(&:name)
  missing_names = required_argument_names - present_argument_names
  if missing_names.any?
    add_error(GraphQL::StaticValidation::RequiredArgumentsArePresentError.new(
      "#{ast_node.class.name.split("::").last} '#{ast_node.name}' is missing required arguments: #{missing_names.join(", ")}",
      nodes: ast_node,
      class_name: ast_node.class.name.split("::").last,
      name: ast_node.name,
      arguments: "#{missing_names.join(", ")}"
    ))
  end
end

def on_directive(node, _parent)
  directive_defn = context.schema_directives[node.name]
  assert_required_args(node, directive_defn)
  super
end

def on_field(node, _parent)
  assert_required_args(node, field_definition)
  super
end