From 96511604f64e5e0d5cb34dc790ab7117f6a57474 Mon Sep 17 00:00:00 2001
From: kosukeKK <koskacts@gmail.com>
Date: Thu, 7 Mar 2024 13:19:30 +0900
Subject: [PATCH] refactoring code

Signed-off-by: kosukeKK <koskacts@gmail.com>
---
 .../next/TemplateWizardPage/TemplateWizardPage.tsx    | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/plugins/scaffolder/src/next/TemplateWizardPage/TemplateWizardPage.tsx b/plugins/scaffolder/src/next/TemplateWizardPage/TemplateWizardPage.tsx
index 7f7cee4c7e..2ceb57f87e 100644
--- a/plugins/scaffolder/src/next/TemplateWizardPage/TemplateWizardPage.tsx
+++ b/plugins/scaffolder/src/next/TemplateWizardPage/TemplateWizardPage.tsx
@@ -15,6 +15,7 @@
  */
 import React from 'react';
 import { Navigate, useNavigate } from 'react-router-dom';
+import useAsync from 'react-use/lib/useAsync';
 import { stringifyEntityRef } from '@backstage/catalog-model';
 import {
   AnalyticsContext,
@@ -77,12 +78,10 @@ export const TemplateWizardPage = (props: TemplateWizardPageProps) => {
     name: templateName,
   });
 
-  const [editUrl, setEditURL] = React.useState('');
-  catalogApi.getEntityByRef(templateRef).then(data => {
-    const templateEditUrl =
-      data?.metadata.annotations?.['backstage.io/edit-url'];
-    if (templateEditUrl !== undefined) setEditURL(templateEditUrl);
-  });
+  const { value: editUrl } = useAsync(async () => {
+    const data = await catalogApi.getEntityByRef(templateRef);
+    return data?.metadata.annotations?.['backstage.io/edit-url'] || '';
+  }, [templateRef, catalogApi]);
 
   const scaffolderWizardContextMenuProps = {
     onEditorClicked: