From 28aaefa35551bd8ab2667db3d86368e9aca06484 Mon Sep 17 00:00:00 2001
From: Valerii Naida <vnayda@adobe.com>
Date: Wed, 19 Jun 2019 15:58:21 -0500
Subject: [PATCH] magento/graphql-ce#741: Add extension point to set custom
 parameters to Query Context object

---
 .../GraphQl/Model/Query/Resolver/Context.php  | 89 +++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 app/code/Magento/GraphQl/Model/Query/Resolver/Context.php

diff --git a/app/code/Magento/GraphQl/Model/Query/Resolver/Context.php b/app/code/Magento/GraphQl/Model/Query/Resolver/Context.php
new file mode 100644
index 0000000000000..2b8e3fabd6863
--- /dev/null
+++ b/app/code/Magento/GraphQl/Model/Query/Resolver/Context.php
@@ -0,0 +1,89 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\GraphQl\Model\Query\Resolver;
+
+use Magento\Framework\GraphQl\Query\Resolver\ContextInterface;
+
+/**
+ * Do not use this class. It was kept for backward compatibility.
+ *
+ * @deprecated \Magento\GraphQl\Model\Query\Context is used instead of this
+ */
+class Context extends \Magento\Framework\Model\AbstractExtensibleModel implements ContextInterface
+{
+    /**#@+
+     * Constants defined for type of context
+     */
+    const USER_TYPE_ID  = 'user_type';
+    const USER_ID = 'user_id';
+    /**#@-*/
+
+    /**
+     * Get extension attributes
+     *
+     * @return \Magento\Framework\GraphQl\Query\Resolver\ContextExtensionInterface
+     */
+    public function getExtensionAttributes() : \Magento\Framework\GraphQl\Query\Resolver\ContextExtensionInterface
+    {
+        return $this->_getExtensionAttributes();
+    }
+
+    /**
+     * Set extension attributes
+     *
+     * @param \Magento\Framework\GraphQl\Query\Resolver\ContextExtensionInterface $extensionAttributes
+     * @return ContextInterface
+     */
+    public function setExtensionAttributes(
+        \Magento\Framework\GraphQl\Query\Resolver\ContextExtensionInterface $extensionAttributes
+    ) : ContextInterface {
+        return $this->_setExtensionAttributes($extensionAttributes);
+    }
+
+    /**
+     * Get user id
+     *
+     * @return int
+     */
+    public function getUserId() : int
+    {
+        return (int) $this->getData(self::USER_ID);
+    }
+
+    /**
+     * Set user id
+     *
+     * @param int $userId
+     * @return ContextInterface
+     */
+    public function setUserId(int $userId) : ContextInterface
+    {
+        return $this->setData(self::USER_ID, $userId);
+    }
+
+    /**
+     * Get user type
+     *
+     * @return int
+     */
+    public function getUserType() : int
+    {
+        return (int) $this->getData(self::USER_TYPE_ID);
+    }
+
+    /**
+     * Set user type
+     *
+     * @param int $typeId
+     * @return ContextInterface
+     */
+    public function setUserType(int $typeId) : ContextInterface
+    {
+        return $this->setData(self::USER_TYPE_ID, $typeId);
+    }
+}