From 09dc09c6052cdf27131086d63695ff3cb14bd962 Mon Sep 17 00:00:00 2001
From: Lyzun Oleksandr <alex.nuzil@gmail.com>
Date: Tue, 21 Aug 2018 19:44:43 +0200
Subject: [PATCH] Prototype of GraphQL mutations for customer update

---
 .../Model/Resolver/CustomerUpdate.php         | 150 ++++++++++++++++++
 .../CustomerGraphQl/etc/schema.graphqls       |  11 ++
 2 files changed, 161 insertions(+)
 create mode 100644 app/code/Magento/CustomerGraphQl/Model/Resolver/CustomerUpdate.php

diff --git a/app/code/Magento/CustomerGraphQl/Model/Resolver/CustomerUpdate.php b/app/code/Magento/CustomerGraphQl/Model/Resolver/CustomerUpdate.php
new file mode 100644
index 0000000000000..96fe852f700cf
--- /dev/null
+++ b/app/code/Magento/CustomerGraphQl/Model/Resolver/CustomerUpdate.php
@@ -0,0 +1,150 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\CustomerGraphQl\Model\Resolver;
+
+use Magento\Authorization\Model\UserContextInterface;
+use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
+use Magento\CustomerGraphQl\Model\Resolver\Customer\CustomerDataProvider;
+use Magento\Framework\GraphQl\Config\Element\Field;
+use Magento\Framework\GraphQl\Exception\GraphQlAuthorizationException;
+use Magento\Framework\GraphQl\Query\Resolver\ContextInterface;
+use Magento\Framework\GraphQl\Query\Resolver\Value;
+use Magento\Framework\GraphQl\Query\Resolver\ValueFactory;
+use Magento\Framework\GraphQl\Query\ResolverInterface;
+use Magento\Store\Api\StoreResolverInterface;
+use Magento\Framework\Encryption\EncryptorInterface as Encryptor;
+use Magento\Customer\Model\CustomerRegistry;
+
+
+/**
+ * Customers field resolver, used for GraphQL request processing.
+ */
+class CustomerUpdate implements ResolverInterface
+{
+    /**
+     * @var CustomerDataProvider
+     */
+    private $customerResolver;
+
+    /**
+     * @var ValueFactory
+     */
+    private $valueFactory;
+
+    /**
+     * @var StoreResolverInterface
+     */
+    private $storeResolver;
+
+    /**
+     * @var \Magento\Newsletter\Model\SubscriberFactory
+     */
+    protected $subscriberFactory;
+
+    /**
+     * @var AuthenticationInterface
+     */
+    private $authentication;
+
+    /**
+     * @var CustomerRegistry
+     */
+    protected $customerRegistry;
+
+    /**
+     * @var Encryptor
+     */
+    protected $encryptor;
+
+    /**
+     * @param CustomerDataProvider $customerResolver
+     * @param ValueFactory $valueFactory
+     */
+    public function __construct(
+        CustomerDataProvider $customerResolver,
+        ValueFactory $valueFactory,
+        \Magento\Customer\Api\CustomerRepositoryInterface $customerRepository,
+        \Magento\Newsletter\Model\SubscriberFactory $subscriberFactory,
+        StoreResolverInterface $storeResolver,
+        Encryptor $encryptor,
+        CustomerRegistry $customerRegistry
+    ) {
+        $this->customerResolver = $customerResolver;
+        $this->valueFactory = $valueFactory;
+        $this->customerRepository = $customerRepository;
+        $this->subscriberFactory = $subscriberFactory;
+        $this->storeResolver = $storeResolver;
+        $this->encryptor = $encryptor;
+        $this->customerRegistry = $customerRegistry;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function resolve(
+        Field $field,
+        $context,
+        ResolveInfo $info,
+        array $value = null,
+        array $args = null
+    ) : Value {
+
+        /** @var ContextInterface $context */
+        if ((!$context->getUserId()) || $context->getUserType() == UserContextInterface::USER_TYPE_GUEST) {
+            throw new GraphQlAuthorizationException(
+                __(
+                    'Current customer does not have access to the resource "%1"',
+                    [\Magento\Customer\Model\Customer::ENTITY]
+                )
+            );
+        }
+
+        $customer = $this->customerRepository->getById($context->getUserId());
+
+        if (isset($args['email']) && $customer->getEmail() !== $args['email']) {
+            $customerSecure = $this->customerRegistry->retrieveSecureData($context->getUserId());
+            $hash = $customerSecure->getPasswordHash();
+            if (!$this->encryptor->validateHash($args['password'], $hash)) {
+                throw new GraphQlAuthorizationException(__('Invalid login or password.'));
+            }
+            $customer->setEmail($args['email']);
+        }
+
+        if (isset($args['firstname'])) {
+            $customer->setFirstname($args['firstname']);
+        }
+        if (isset($args['lastname'])){
+            $customer->setLastname($args['lastname']);
+        }
+
+        $customer->setStoreId($this->storeResolver->getCurrentStoreId());
+        $this->customerRepository->save($customer);
+
+        if (isset($args['is_subscribed'])) {
+
+            $checkSubscriber = $this->subscriberFactory->create()->loadByCustomerId($context->getUserId());
+
+            if ($args['is_subscribed'] === true && !$checkSubscriber->isSubscribed()) {
+                $this->subscriberFactory->create()->subscribeCustomerById($context->getUserId());
+            } elseif ($args['is_subscribed'] === false && $checkSubscriber->isSubscribed()) {
+                $this->subscriberFactory->create()->unsubscribeCustomerById($context->getUserId());
+            }
+        }
+
+
+        $data = $args;
+        $result = function () use ($data) {
+            return !empty($data) ? $data : [];
+        };
+
+        return $this->valueFactory->create($result);
+
+
+    }
+
+}
diff --git a/app/code/Magento/CustomerGraphQl/etc/schema.graphqls b/app/code/Magento/CustomerGraphQl/etc/schema.graphqls
index 687826cff001d..f74b4f82f77f8 100644
--- a/app/code/Magento/CustomerGraphQl/etc/schema.graphqls
+++ b/app/code/Magento/CustomerGraphQl/etc/schema.graphqls
@@ -51,3 +51,14 @@ type CustomerAddressRegion @doc(description: "CustomerAddressRegion defines the
     region_id: Int @doc(description: "Uniquely identifies the region")
 }
 
+type Mutation {
+    customerUpdate (firstname: String, lastname: String, email: String, password: String, is_subscribed: Boolean): customerItem @resolver(class: "\\Magento\\CustomerGraphQl\\Model\\Resolver\\CustomerUpdate") @doc(description:"Update customer personal information")
+}
+
+type customerItem {
+    firstname: String
+    lastname: String
+    email: String
+    password: String
+    is_subscribed: Boolean
+}