From c0154e444b1bd27ebe46e20d367101d5655aec57 Mon Sep 17 00:00:00 2001
From: mufaddal motiwala <mufaddalmm.52@gmail.com>
Date: Mon, 13 Dec 2021 11:31:23 +0530
Subject: [PATCH] change API for fetching user owned entities

Signed-off-by: mufaddal motiwala <mufaddalmm.52@gmail.com>
---
 .../OwnedEntityPicker/OwnedEntityPicker.tsx   | 25 +++++--------------
 1 file changed, 6 insertions(+), 19 deletions(-)

diff --git a/plugins/scaffolder/src/components/fields/OwnedEntityPicker/OwnedEntityPicker.tsx b/plugins/scaffolder/src/components/fields/OwnedEntityPicker/OwnedEntityPicker.tsx
index b73158ac60..06b637a79c 100644
--- a/plugins/scaffolder/src/components/fields/OwnedEntityPicker/OwnedEntityPicker.tsx
+++ b/plugins/scaffolder/src/components/fields/OwnedEntityPicker/OwnedEntityPicker.tsx
@@ -13,18 +13,13 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { useApi } from '@backstage/core-plugin-api';
-import {
-  catalogApiRef,
-  formatEntityRefTitle,
-  useEntityOwnership,
-} from '@backstage/plugin-catalog-react';
+import { formatEntityRefTitle } from '@backstage/plugin-catalog-react';
 import { TextField } from '@material-ui/core';
 import FormControl from '@material-ui/core/FormControl';
 import Autocomplete from '@material-ui/lab/Autocomplete';
 import { FieldProps } from '@rjsf/core';
 import React from 'react';
-import { useAsync } from 'react-use';
+import { useOwnedEntities } from './useOwnedEntities';
 
 export const OwnedEntityPicker = ({
   onChange,
@@ -37,24 +32,16 @@ export const OwnedEntityPicker = ({
 }: FieldProps<string>) => {
   const allowedKinds = uiSchema['ui:options']?.allowedKinds as string[];
   const defaultKind = uiSchema['ui:options']?.defaultKind as string | undefined;
-  const catalogApi = useApi(catalogApiRef);
-  const { isOwnedEntity } = useEntityOwnership();
+  const { ownedEntities, loading } = useOwnedEntities(allowedKinds);
 
-  const { value: entities, loading } = useAsync(() =>
-    catalogApi.getEntities(
-      allowedKinds ? { filter: { kind: allowedKinds } } : undefined,
-    ),
-  );
-
-  const entityRefs = entities?.items
-    .map(e =>
-      isOwnedEntity(e) ? formatEntityRefTitle(e, { defaultKind }) : null,
-    )
+  const entityRefs = ownedEntities?.items
+    .map(e => formatEntityRefTitle(e, { defaultKind }))
     .filter(n => n);
 
   const onSelect = (_: any, value: string | null) => {
     onChange(value || '');
   };
+
   return (
     <FormControl
       margin="normal"