fix: yuque new doc not display in TOC

src/common/backend/services/interface.ts

@@ -67,12 +67,14 @@ export interface ServiceMeta {
   permission?: chrome.permissions.Permissions;
 }
 
+export interface UpdateTOCRequest {}
+
 export interface DocumentService<T = any> {
   getId(): string;
 
   getRepositories(): Promise<Repository[]>;
 
-  createDocument(request: CreateDocumentRequest): Promise<CompleteStatus | void>;
+  createDocument(request: T): Promise<CompleteStatus | void>;
 
   getUserInfo(): Promise<UserInfo>;
 

src/common/backend/services/yuque/interface.ts

@@ -1,4 +1,4 @@
-import { CompleteStatus, CreateDocumentRequest } from './../interface';
+import { CompleteStatus, CreateDocumentRequest, UpdateTOCRequest } from './../interface';
 import { Repository } from '../interface';
 
 export enum RepositoryType {
@@ -53,3 +53,8 @@ export interface YuqueCompleteStatus extends CompleteStatus {
 export interface YuqueCreateDocumentRequest extends CreateDocumentRequest {
   slug?: string;
 }
+
+export interface YuqueUpdateTOCRequest extends UpdateTOCRequest{
+  repositoryId: string;
+  documentId: number[];
+}

src/common/backend/services/yuque/service.ts

@@ -15,6 +15,7 @@ import {
   YuqueRepository,
   YuqueCompleteStatus,
   YuqueCreateDocumentRequest,
+  YuqueUpdateTOCRequest,
 } from './interface';
 
 const HOST = 'https://www.yuque.com';
@@ -100,6 +101,9 @@ export default class YuqueDocumentService implements DocumentService {
       }
     );
     const data = response;
+
+    await this.updateYuqueTOC({ repositoryId, documentId: [data.id] });
+
     return {
       href: `${HOST}/${repository.namespace}/${data.slug}`,
       repositoryId,
@@ -150,4 +154,24 @@ export default class YuqueDocumentService implements DocumentService {
       return [];
     }
   };
+
+  private updateYuqueTOC = async (info: YuqueUpdateTOCRequest) => {
+    const { repositoryId, documentId } = info;
+    const requestBody = {
+      action: 'prependNode',
+      action_mode: 'child',
+      doc_ids: documentId,
+      type: 'DOC',
+    };
+
+    try {
+      const response = await this.request.put(`repos/${repositoryId}/toc`, {
+        data: requestBody,
+			});
+      return response;
+    } catch (_error) {
+      return {};
+    }
+
+  };
 }