feat: support ai writer

Author: qinluhe

package.json

@@ -18,7 +18,7 @@
     "coverage": "pnpm run test:unit && pnpm run test:components"
   },
   "dependencies": {
-    "@appflowyinc/ai-chat": "0.0.15",
+    "@appflowyinc/ai-chat": "0.0.17",
     "@appflowyinc/editor": "^0.1.6",
     "@atlaskit/primitives": "^5.5.3",
     "@emoji-mart/data": "^1.1.2",

pnpm-lock.yaml

@@ -1687,7 +1687,9 @@
         "aiWriter": "AI Writer",
         "dateOrReminder": "Date or Reminder",
         "photoGallery": "Photo Gallery",
-        "file": "File"
+        "file": "File",
+        "continueWriting": "Continue Writing",
+        "askAIAnything": "Ask AI Anything"
       },
       "subPage": {
         "name": "Document",
@@ -2446,7 +2448,9 @@
     "toggleListShortForm": "Toggle",
     "quoteListShortForm": "Quote",
     "mathEquationShortForm": "Formula",
-    "codeBlockShortForm": "Code"
+    "codeBlockShortForm": "Code",
+    "improveWriting": "Improve writing",
+    "askAI": "Ask AI Anything"
   },
   "favorite": {
     "noFavorite": "No favorite page",

src/@types/translations/en.json

@@ -35,6 +35,7 @@ const wrapperCreator =
           databaseDoc={doc}
           rowDocMap={rowDocMap}
           readOnly={true}
+          workspaceId={''}
         >
           {children}
         </DatabaseContextProvider>

src/application/database-yjs/__tests__/selector.test.tsx

@@ -26,14 +26,15 @@ export interface DatabaseContextState {
   navigateToView?: (viewId: string, blockId?: string) => Promise<void>;
   onRendered?: (height: number) => void;
   showActions?: boolean;
+  workspaceId: string;
 }
 
 export const DatabaseContext = createContext<DatabaseContextState | null>(null);
 
 export const useDatabaseContext = () => {
   const context = useContext(DatabaseContext);
 
-  if (!context) {
+  if(!context) {
     throw new Error('DatabaseContext is not provided');
   }
 
@@ -89,7 +90,7 @@ export const useDatabaseView = () => {
   return viewId ? database?.get(YjsDatabaseKey.views)?.get(viewId) : undefined;
 };
 
-export function useDatabaseFields () {
+export function useDatabaseFields() {
   const database = useDatabase();
 
   return database.get(YjsDatabaseKey.fields);

src/application/database-yjs/context.ts

@@ -45,6 +45,20 @@ import {
 } from '@/application/slate-yjs/utils/yjs';
 
 export const CustomEditor = {
+  getEditorContent(editor: YjsEditor) {
+    const allNodes = editor.children ?? [];
+
+    return allNodes.map((node) => {
+      return CustomEditor.getBlockTextContent(node);
+    }).join('\n');
+  },
+
+  getSelectionContent(editor: YjsEditor) {
+    const range = editor.selection;
+
+    if(!range) return '';
+    return editor.string(range);
+  },
   // Get the text content of a block node, including the text content of its children and formula nodes
   getBlockTextContent(node: Node, depth: number = Infinity): string {
     if(Text.isText(node)) {
@@ -78,7 +92,7 @@ export const CustomEditor = {
   },
 
   setBlockData<T = BlockData>(editor: YjsEditor, blockId: string, updateData: T, select?: boolean) {
-    
+
     const block = getBlock(blockId, editor.sharedRoot);
     const oldData = dataStringTOJson(block.get(YjsEditorKey.block_data));
     const newData = {

src/application/slate-yjs/command/index.ts

@@ -991,6 +991,7 @@ export interface ViewMetaProps {
   cover?: ViewMetaCover;
   name?: string;
   viewId?: string;
+  workspaceId?: string;
   layout?: ViewLayout;
   visibleViewIds?: string[];
   extra?: ViewExtra | null;
@@ -1008,6 +1009,7 @@ export interface TextCount {
 
 export interface ViewComponentProps {
   doc: YDoc;
+  workspaceId: string;
   readOnly: boolean;
   navigateToView?: (viewId: string, blockId?: string) => Promise<void>;
   loadViewMeta?: LoadViewMeta;