Merge pull request #7 from 3kinglee/feat/end-syntax

Feat/end syntax & user avatar
AIOS-club · Feb 27, 2023 · c93a99c · c93a99c
2 parents c69f6b5 + 7124779
commit c93a99c
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 24 deletions.
diff --git a/extension/contentScript/index.js b/extension/contentScript/index.js
diff --git a/src/manifest.json b/src/manifest.json
@@ -1,6 +1,6 @@
 {
   "name": "ai-chat-export",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "manifest_version": 3,
   "author": "[email protected]",
   "description": "export ai-chat page as yaml",

diff --git a/src/utils.ts b/src/utils.ts
@@ -22,14 +22,18 @@ export const getChatObject = () => {
   const elements = document.querySelectorAll("[class*='min-h-[20px]']");
   if (elements.length <= 0) return;
 
+  const questionnerAvatar = getQuestionerAvatarUrl();
+
   const defaultJson = {
     $schema: "Dialog",
     Dialog: {
       title: "chatGPT对话",
       characters: [
         {
           name: "questioner",
-          avatar: "https://res.pandateacher.com/7KJKIMIE1677146652086.png",
+          avatar:
+            questionnerAvatar ||
+            "https://res.pandateacher.com/1O2BGDK91671292227563.jpg",
         },
         {
           name: "AI",
@@ -41,7 +45,7 @@ export const getChatObject = () => {
     },
   };
 
-  const contents = [];
+  let contents = [];
 
   elements.forEach((ele) => {
     const firstChild = ele.firstChild;
@@ -83,6 +87,20 @@ export const getChatObject = () => {
     contents.push(contentObj);
   });
 
+  const endContents = [{ CourseFinished: null }, { End: null }];
+  contents = [...contents, ...endContents];
   defaultJson.Dialog.contents = contents;
   return defaultJson;
 };
+
+export const getQuestionerAvatarUrl = () => {
+  let avatarUrl = Array.from(
+    document.querySelectorAll("[class*='w-[30px]']")[0]?.querySelectorAll("img")
+  ).find((item) => item.currentSrc.startsWith("https"))?.currentSrc;
+
+  if (/\?url=/.test(avatarUrl)) {
+    const url = avatarUrl.split("url=")[1];
+    avatarUrl && (avatarUrl = decodeURIComponent(url));
+  }
+  return avatarUrl;
+};