From eda56f4d418c28da81fb63f185297a2ce42ef787 Mon Sep 17 00:00:00 2001
From: Shishir <75600200+shishir-intelli@users.noreply.github.com>
Date: Tue, 1 Aug 2023 17:14:00 +0530
Subject: [PATCH] Added UUID in TeamMemberRole.php for JSONAPI (3x) (#902)

---
 .../Storage/TeamMemberRoleStorageSchema.php   | 50 +++++++++++++++++++
 .../src/Entity/TeamMemberRole.php             | 30 +++++++++++
 2 files changed, 80 insertions(+)
 create mode 100644 modules/apigee_edge_teams/src/Entity/Storage/TeamMemberRoleStorageSchema.php

diff --git a/modules/apigee_edge_teams/src/Entity/Storage/TeamMemberRoleStorageSchema.php b/modules/apigee_edge_teams/src/Entity/Storage/TeamMemberRoleStorageSchema.php
new file mode 100644
index 000000000..618e73867
--- /dev/null
+++ b/modules/apigee_edge_teams/src/Entity/Storage/TeamMemberRoleStorageSchema.php
@@ -0,0 +1,50 @@
+<?php
+
+/**
+ * Copyright 2023 Google Inc.
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * version 2 as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
+ * MA 02110-1301, USA.
+ */
+
+namespace Drupal\apigee_edge_teams\Entity\Storage;
+
+use Drupal\Core\Entity\ContentEntityTypeInterface;
+use Drupal\Core\Entity\Sql\SqlContentEntityStorageSchema;
+
+/**
+ * Entity storage class for team member role entities.
+ */
+class TeamMemberRoleStorageSchema extends SqlContentEntityStorageSchema {
+
+  /**
+   * {@inheritdoc}
+   */
+  protected function getEntitySchema(ContentEntityTypeInterface $entity_type, $reset = FALSE) {
+    $schema = parent::getEntitySchema($entity_type, $reset);
+    // For JSONAPI uuid added in entity_keys, but because it, duplicate
+    // uuid field is generated, which cause error while installing team_member table.
+    if (!empty($schema['team_member_role']['fields']['uuid'])) {
+      foreach ($schema['team_member_role']['fields']['uuid'] as $key => $value) {
+        $schema['team_member_role']['fields']['uuid'][$key] = is_array($value) ? $value[0] : $value;
+      }
+      // Fix to remove duplicate UUID field in primary key.
+      if (!empty($schema['team_member_role']['unique keys']['team_member_role_field__uuid__value'][1]) && $schema['team_member_role']['unique keys']['team_member_role_field__uuid__value'][1] == 'uuid') {
+        unset($schema['team_member_role']['unique keys']['team_member_role_field__uuid__value'][1]);
+      }
+    }
+    return $schema;
+  }
+
+}
diff --git a/modules/apigee_edge_teams/src/Entity/TeamMemberRole.php b/modules/apigee_edge_teams/src/Entity/TeamMemberRole.php
index b40097013..b19eddfb7 100644
--- a/modules/apigee_edge_teams/src/Entity/TeamMemberRole.php
+++ b/modules/apigee_edge_teams/src/Entity/TeamMemberRole.php
@@ -20,11 +20,14 @@
 
 namespace Drupal\apigee_edge_teams\Entity;
 
+use Drupal\Core\Access\AccessResult;
+use Drupal\Core\Access\AccessResultAllowed;
 use Drupal\Core\Entity\ContentEntityBase;
 use Drupal\Core\Entity\EntityChangedTrait;
 use Drupal\Core\Entity\EntityTypeInterface;
 use Drupal\Core\Field\BaseFieldDefinition;
 use Drupal\Core\Field\FieldStorageDefinitionInterface;
+use Drupal\Core\Session\AccountInterface;
 use Drupal\user\UserInterface;
 
 /**
@@ -41,9 +44,11 @@
  *   data_table = "team_member_role_data",
  *   handlers = {
  *     "storage" = "Drupal\apigee_edge_teams\Entity\Storage\TeamMemberRoleStorage",
+ *     "storage_schema" = "Drupal\apigee_edge_teams\Entity\Storage\TeamMemberRoleStorageSchema",
  *   },
  *   entity_keys = {
  *     "id" = "uuid",
+ *     "uuid" = "uuid",
  *   },
  * )
  *
@@ -56,6 +61,31 @@ final class TeamMemberRole extends ContentEntityBase implements TeamMemberRoleIn
 
   use EntityChangedTrait;
 
+  /**
+   * {@inheritdoc}
+   */
+  public function access($operation, AccountInterface $account = NULL, $return_as_object = FALSE) {
+    if (!$account) {
+      // If we have hit this without an account return forbidden.
+      return AccessResult::forbidden();
+    }
+
+    $result = AccessResult::allowedIfHasPermissions($account, [
+      'administer team',
+      'manage team members',
+    ], 'OR');
+
+    if ($result->isNeutral()) {
+      $team = $this->getTeam();
+      $team_permission_handler = \Drupal::service('apigee_edge_teams.team_permissions');
+      $result = AccessResultAllowed::allowedIf(in_array('team_manage_members', $team_permission_handler->getDeveloperPermissionsByTeam($team, $account)))
+        ->addCacheableDependency($team)
+        ->cachePerUser();
+    }
+
+    return $result;
+  }
+
   /**
    * {@inheritdoc}
    */