From 8f0b8c48a9a3a84ebf603448e7619571ac53bba8 Mon Sep 17 00:00:00 2001
From: Robert <robertsolovev39@gmail.com>
Date: Tue, 12 Mar 2024 01:57:45 +0700
Subject: [PATCH] preserve image

---
 src/widgets/action-editor/ui/ActionEditor.tsx | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/widgets/action-editor/ui/ActionEditor.tsx b/src/widgets/action-editor/ui/ActionEditor.tsx
index af87c2c..5802aee 100644
--- a/src/widgets/action-editor/ui/ActionEditor.tsx
+++ b/src/widgets/action-editor/ui/ActionEditor.tsx
@@ -1,6 +1,7 @@
 import { useState } from "react";
 import { uploadFile } from "@/shared/lib/uploadFile";
 import type { CommentAction } from "@/entities/action/schema";
+import { getActionAttachmentType } from "@/entities/action/lib";
 
 import Tabs from "@mui/material/Tabs";
 import Tab from "@mui/material/Tab";
@@ -60,7 +61,13 @@ export default function ActionEditor({
   const [value, setValue] = useState(0);
   const [loading, setLoading] = useState(false);
 
-  const [imageUrl, setImageUrl] = useState<string | null>(null);
+  const attachmentType = initialAction
+    ? getActionAttachmentType(initialAction)
+    : "empty";
+
+  const [imageUrl, setImageUrl] = useState<string | null>(
+    attachmentType === "image" ? initialAction?.fileUrls[0]! : null,
+  );
 
   const [action, setAction] = useState<CommentAction>(
     initialAction ?? {