feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

frontend/packages/studio/stores/bot-detail/src/utils/generate-image.ts

@@ -0,0 +1,165 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+ 
+import {
+  GenPicStatus,
+  PicType,
+  type GetPicTaskData,
+} from '@coze-arch/idl/playground_api';
+
+import {
+  type GenerateBackGroundModal,
+  type GenerateAvatarModal,
+  DotStatus,
+  GenerateType,
+} from '../types/generate-image';
+import { useBotSkillStore } from '../store/bot-skill';
+import getDotStatus from './get-dot-status';
+
+export const getInitBackgroundInfo = (
+  data: GetPicTaskData,
+  state: GenerateBackGroundModal,
+) => {
+  const { tasks = [] } = data;
+  const { backgroundImageInfoList = [] } = useBotSkillStore.getState();
+  // 当前渲染的背景图
+  const uri =
+    backgroundImageInfoList[0]?.mobile_background_image?.origin_image_uri;
+
+  const backgroundGifList = tasks.filter(
+    item => item.type && [PicType.BackgroundGif].includes(item.type),
+  );
+  const backgroundStaticList = tasks.filter(
+    item => item.type && [PicType.BackgroundStatic].includes(item.type),
+  );
+  const imageDotStatus = getDotStatus(
+    data,
+    PicType.BackgroundStatic,
+  ) as DotStatus;
+  const gifDotStatus = getDotStatus(data, PicType.BackgroundGif) as DotStatus;
+
+  // 动图相关state
+  state.gif.loading = backgroundGifList.some(
+    item => item.status === GenPicStatus.Generating,
+  );
+  state.gif.text =
+    backgroundGifList.find(item => item?.img_info?.prompt)?.img_info?.prompt
+      ?.ori_prompt ?? '';
+
+  state.gif.dotStatus = gifDotStatus;
+  const image = backgroundGifList.find(item => item.img_info?.ori_url);
+  // 第一帧信息
+  if (image) {
+    state.gif.image = {
+      img_info: {
+        tar_uri: image.img_info?.ori_uri,
+        tar_url: image.img_info?.ori_url,
+      },
+    };
+  }
+
+  // 静图相关state
+  state.image.loading = backgroundStaticList.some(
+    item => item.status === GenPicStatus.Generating,
+  );
+  state.image.dotStatus = imageDotStatus;
+  state.image.promptInfo =
+    backgroundStaticList.find(item => item?.img_info?.prompt?.ori_prompt)
+      ?.img_info?.prompt ?? {};
+
+  const lastImageTask =
+    tasks.find(item => item.type === PicType.BackgroundStatic) ?? {};
+  const lastGifTask =
+    tasks.find(item => item.type === PicType.BackgroundGif) ?? {};
+  // 当前选中的图片: 生成成功的 展示 成功的那个图， 否则找 背景图一致的
+  if (gifDotStatus === DotStatus.Success) {
+    state.selectedImage = lastGifTask;
+  } else if (imageDotStatus === DotStatus.Success) {
+    state.selectedImage = lastImageTask;
+  } else {
+    // 手动上传的 找不到
+    state.selectedImage =
+      tasks.find(item => item.img_info?.tar_uri === uri) ?? {};
+  }
+  // 当前tab：只有在 仅gif在状态不为done时 在gif tab
+  if (gifDotStatus !== DotStatus.None) {
+    state.activeKey = GenerateType.Gif;
+  }
+  // 当前正在生成的taskId
+  if (
+    gifDotStatus === DotStatus.Generating ||
+    imageDotStatus === DotStatus.Generating
+  ) {
+    state.generatingTaskId =
+      gifDotStatus === DotStatus.Generating
+        ? lastGifTask?.id
+        : lastImageTask?.id;
+  }
+};
+
+export const getInitAvatarInfo = (
+  data: GetPicTaskData,
+  state: GenerateAvatarModal,
+) => {
+  const { tasks = [] } = data || {};
+  const lastImageTask = tasks.find(
+    item => item.type === PicType.IconStatic,
+  ) || {
+    id: '',
+    img_info: {},
+  };
+  const lastGifTask = tasks.find(item => item.type === PicType.IconGif) || {
+    id: '',
+    img_info: {},
+  };
+  const gifDotStatus = getDotStatus(data, PicType.IconGif) as DotStatus;
+  const imageDotStatus = getDotStatus(data, PicType.IconStatic) as DotStatus;
+  if (
+    gifDotStatus === DotStatus.Success ||
+    imageDotStatus === DotStatus.Success
+  ) {
+    state.selectedImage =
+      gifDotStatus === DotStatus.Success ? lastGifTask : lastImageTask;
+  }
+
+  if (
+    gifDotStatus === DotStatus.Generating ||
+    imageDotStatus === DotStatus.Generating
+  ) {
+    state.generatingTaskId =
+      gifDotStatus === DotStatus.Generating
+        ? lastGifTask?.id
+        : lastImageTask?.id;
+  }
+  state.gif = {
+    dotStatus: gifDotStatus,
+    text: lastGifTask?.img_info?.prompt?.ori_prompt ?? '',
+    loading: gifDotStatus === DotStatus.Generating,
+    image: {
+      id: lastGifTask.img_info?.ori_uri ?? '',
+      img_info: {
+        tar_uri: lastGifTask.img_info?.ori_uri ?? '',
+        tar_url: lastGifTask.img_info?.ori_url ?? '',
+      },
+    },
+  };
+  state.image = {
+    dotStatus: imageDotStatus,
+    text: lastImageTask.img_info?.prompt?.ori_prompt ?? '',
+    loading: imageDotStatus === DotStatus.Generating,
+    textCustomizable: Boolean(lastImageTask.img_info?.prompt?.ori_prompt),
+  };
+};