chore: fix ai page user profile refresh (#7317 )

fix: editor stuck on image loading loop when uploading image in row document (#7313 )
* fix: editor stuck on image loading loop when uploading image in row document * test: editor stuck on image loading loop when uploading image in row document
2025-02-04 20:40:06 +08:00 · 2025-02-04 18:03:28 +08:00 · 2025-02-03 20:52:31 +08:00 · 2025-02-01 23:50:58 +08:00
1518 changed files with 31071 additions and 92934 deletions
@@ -65,7 +65,7 @@ runs:
            sudo wget -qO /etc/apt/trusted.gpg.d/dart_linux_signing_key.asc https://dl-ssl.google.com/linux/linux_signing_key.pub
            sudo wget -qO /etc/apt/sources.list.d/dart_stable.list https://storage.googleapis.com/download.dartlang.org/linux/debian/dart_stable.list
            sudo apt-get update
-            sudo apt-get install -y dart curl build-essential libssl-dev clang cmake ninja-build pkg-config libgtk-3-dev keybinder-3.0 libnotify-dev libcurl4-openssl-dev
+            sudo apt-get install -y dart curl build-essential libssl-dev clang cmake ninja-build pkg-config libgtk-3-dev keybinder-3.0 libnotify-dev
            ;;
          Windows)
            vcpkg integrate install
@@ -52,7 +52,7 @@ runs:
        sudo wget -qO /etc/apt/trusted.gpg.d/dart_linux_signing_key.asc https://dl-ssl.google.com/linux/linux_signing_key.pub
        sudo wget -qO /etc/apt/sources.list.d/dart_stable.list https://storage.googleapis.com/download.dartlang.org/linux/debian/dart_stable.list
        sudo apt-get update
-        sudo apt-get install -y dart curl build-essential libssl-dev clang cmake ninja-build pkg-config libgtk-3-dev keybinder-3.0 libnotify-dev network-manager libcurl4-openssl-dev
+        sudo apt-get install -y dart curl build-essential libssl-dev clang cmake ninja-build pkg-config libgtk-3-dev keybinder-3.0 libnotify-dev network-manager
      shell: bash

    - name: Enable Flutter Desktop
@@ -18,8 +18,8 @@ on:

 env:
  CARGO_TERM_COLOR: always
-  FLUTTER_VERSION: "3.27.4"
-  RUST_TOOLCHAIN: "1.84.0"
+  FLUTTER_VERSION: "3.22.3"
+  RUST_TOOLCHAIN: "1.81.0"
  CARGO_MAKE_VERSION: "0.37.18"
  CLOUD_VERSION: 0.6.54-amd64

@@ -25,10 +25,10 @@ on:

 env:
  CARGO_TERM_COLOR: always
-  FLUTTER_VERSION: "3.27.4"
-  RUST_TOOLCHAIN: "1.84.0"
+  FLUTTER_VERSION: "3.22.2"
+  RUST_TOOLCHAIN: "1.81.0"
  CARGO_MAKE_VERSION: "0.37.18"
-  CLOUD_VERSION: 0.9.49-amd64
+  CLOUD_VERSION: 0.6.54-amd64

 concurrency:
  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
@@ -174,7 +174,7 @@ jobs:
            sudo wget -qO /etc/apt/trusted.gpg.d/dart_linux_signing_key.asc https://dl-ssl.google.com/linux/linux_signing_key.pub
            sudo wget -qO /etc/apt/sources.list.d/dart_stable.list https://storage.googleapis.com/download.dartlang.org/linux/debian/dart_stable.list
            sudo apt-get update
-            sudo apt-get install -y dart curl build-essential libssl-dev clang cmake ninja-build pkg-config libgtk-3-dev keybinder-3.0 libnotify-dev libcurl4-openssl-dev
+            sudo apt-get install -y dart curl build-essential libssl-dev clang cmake ninja-build pkg-config libgtk-3-dev keybinder-3.0 libnotify-dev
          fi
        shell: bash

@@ -308,7 +308,7 @@ jobs:
          sudo wget -qO /etc/apt/trusted.gpg.d/dart_linux_signing_key.asc https://dl-ssl.google.com/linux/linux_signing_key.pub
          sudo wget -qO /etc/apt/sources.list.d/dart_stable.list https://storage.googleapis.com/download.dartlang.org/linux/debian/dart_stable.list
          sudo apt-get update
-          sudo apt-get install -y dart curl build-essential libssl-dev clang cmake ninja-build pkg-config libgtk-3-dev keybinder-3.0 libnotify-dev libcurl4-openssl-dev
+          sudo apt-get install -y dart curl build-essential libssl-dev clang cmake ninja-build pkg-config libgtk-3-dev keybinder-3.0 libnotify-dev
        shell: bash

      - name: Enable Flutter Desktop
@@ -18,15 +18,36 @@ on:
      - "!frontend/appflowy_web_app/**"

 env:
-  FLUTTER_VERSION: "3.27.4"
-  RUST_TOOLCHAIN: "1.84.0"
+  FLUTTER_VERSION: "3.22.3"
+  RUST_TOOLCHAIN: "1.81.0"

 concurrency:
  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
  cancel-in-progress: true

 jobs:
+  build-self-hosted:
+    if: github.event.pull_request.head.repo.full_name == github.repository
+    runs-on: self-hosted
+
+    steps:
+      - name: Checkout source code
+        uses: actions/checkout@v2
+
+      - name: Build AppFlowy
+        working-directory: frontend
+        run: |
+          cargo make --profile development-ios-arm64-sim appflowy-core-dev-ios
+          cargo make --profile development-ios-arm64-sim code_generation
+
+      - uses: futureware-tech/simulator-action@v3
+        id: simulator-action
+        with:
+          model: "iPhone 15"
+          shutdown_after_job: false
+
  integration-tests:
+    if: github.event.pull_request.head.repo.full_name != github.repository
    runs-on: macos-latest

    steps:
@@ -73,26 +94,26 @@ jobs:
          cargo make --profile development-ios-arm64-sim appflowy-core-dev-ios
          cargo make --profile development-ios-arm64-sim code_generation

-      # - uses: futureware-tech/simulator-action@v3
-      #   id: simulator-action
-      #   with:
-      #     model: "iPhone 15"
-      #     shutdown_after_job: false
+      - uses: futureware-tech/simulator-action@v3
+        id: simulator-action
+        with:
+          model: "iPhone 15"
+          shutdown_after_job: false

-      # - name: Run AppFlowy on simulator
-      #   working-directory: frontend/appflowy_flutter
-      #   run: |
-      #     flutter run -d ${{ steps.simulator-action.outputs.udid }} &
-      #     pid=$!
-      #     sleep 500
-      #     kill $pid
-      #   continue-on-error: true
+      - name: Run AppFlowy on simulator
+        working-directory: frontend/appflowy_flutter
+        run: |
+          flutter run -d ${{ steps.simulator-action.outputs.udid }} &
+          pid=$!
+          sleep 500
+          kill $pid
+        continue-on-error: true

-      # # Integration tests
-      # - name: Run integration tests
-      #   working-directory: frontend/appflowy_flutter
-      #   # The integration tests are flaky and sometimes fail with "Connection timed out":
-      #   # Don't block the CI. If the tests fail, the CI will still pass.
-      #   # Instead, we're using Code Magic to re-run the tests to check if they pass.
-      #   continue-on-error: true
-      #   run: flutter test integration_test/runner.dart -d ${{ steps.simulator-action.outputs.udid }}
+      # Integration tests
+      - name: Run integration tests
+        working-directory: frontend/appflowy_flutter
+        # The integration tests are flaky and sometimes fail with "Connection timed out":
+        # Don't block the CI. If the tests fail, the CI will still pass.
+        # Instead, we're using Code Magic to re-run the tests to check if they pass.
+        continue-on-error: true
+        run: flutter test integration_test/runner.dart -d ${{ steps.simulator-action.outputs.udid }}
@@ -6,8 +6,8 @@ on:
      - "*"

 env:
-  FLUTTER_VERSION: "3.27.4"
-  RUST_TOOLCHAIN: "1.84.0"
+  FLUTTER_VERSION: "3.22.0"
+  RUST_TOOLCHAIN: "1.81.0"

 jobs:
  create-release:
@@ -338,7 +338,7 @@ jobs:
          - {
            arch: x86_64,
            target: x86_64-unknown-linux-gnu,
-            os: ubuntu-22.04,
+            os: ubuntu-20.04,
            extra-build-args: "",
            flutter_profile: production-linux-x86_64,
          }
@@ -366,10 +366,9 @@ jobs:
        run: |
          sudo wget -qO /etc/apt/trusted.gpg.d/dart_linux_signing_key.asc https://dl-ssl.google.com/linux/linux_signing_key.pub
          sudo apt-get update
-          sudo apt-get install -y build-essential libsqlite3-dev libssl-dev clang cmake ninja-build pkg-config libgtk-3-dev libcurl4-openssl-dev
+          sudo apt-get install -y build-essential libsqlite3-dev libssl-dev clang cmake ninja-build pkg-config libgtk-3-dev
          sudo apt-get install keybinder-3.0
          sudo apt-get install -y alien libnotify-dev
-          sudo apt install libfuse2
          source $HOME/.cargo/env
          cargo install --force --locked cargo-make
          cargo install --force --locked duckscript_cli
@@ -18,8 +18,8 @@ on:

 env:
  CARGO_TERM_COLOR: always
-  CLOUD_VERSION: 0.9.49-amd64
-  RUST_TOOLCHAIN: "1.84.0"
+  CLOUD_VERSION: 0.8.3-amd64
+  RUST_TOOLCHAIN: "1.81.0"

 jobs:
  ubuntu-job:
@@ -106,7 +106,7 @@ jobs:
          af_cloud_test_ws_url: ws://localhost/ws/v1
          af_cloud_test_gotrue_url: http://localhost/gotrue
        run: |
-          DISABLE_CI_TEST_LOG="true" cargo test --no-default-features --features="dart" -- --skip local_ollama_test
+          DISABLE_CI_TEST_LOG="true" cargo test --no-default-features --features="dart"

      - name: rustfmt rust-lib
        run: cargo fmt --all -- --check
@@ -10,8 +10,8 @@ on:

 env:
  CARGO_TERM_COLOR: always
-  FLUTTER_VERSION: "3.27.4"
-  RUST_TOOLCHAIN: "1.84.0"
+  FLUTTER_VERSION: "3.22.0"
+  RUST_TOOLCHAIN: "1.81.0"

 jobs:
  tests:
@@ -1,126 +1,4 @@
 # Release Notes
-## Version 0.9.1 - 01/05/2025
-### Desktop
-#### New Features
- Added AppFlowy Prompt Library to AI Chat and Document's Ask AI
- Revamped the desktop in-app notification center
- Supported login with password, as well as forgot and change password options
- Supported copying link to invite members
- Improved the Settings' Members tab with new metadata: member avatar and joined time
-#### Bug Fixes
- Fixed data loss when using anonymous local
- Fixed crash when trying to delete an emoji
- Fixed Windows scaling issue
- Correctly displayed mention text by decoding web content
-### Mobile
-#### New Features
- Supported workspace search
- Improved UX for links in documents
- Supported changing password in Mobile Settings
- Added support for inviting members via links
-#### Bug Fixes
- Correctly displayed mention text by decoding web content
-
-## Version 0.9.0 - 30/04/2025
-### Desktop
-#### New Features
- Added AppFlowy Prompt Library to AI Chat and Document's Ask AI
- Revamped the desktop in-app notification center
- Supported login with password, as well as forgot and change password options
- Supported copying link to invite members
- Improved the Settings' Members tab with new metadata: member avatar and joined time
-#### Bug Fixes
- Fixed crash when trying to delete an emoji
- Fixed Windows scaling issue
- Correctly displayed mention text by decoding web content
-### Mobile
-#### New Features
- Supported workspace search
- Improved UX for links in documents
- Supported changing password in Mobile Settings
- Added support for inviting members via links
-#### Bug Fixes
- Correctly displayed mention text by decoding web content
-
-## Version 0.8.9 - 16/04/2025
-### Desktop
-#### New Features
- Supported pasting a link as a mention, providing a more condensed visualization of linked content
- Supported converting between link formats (e.g. transforming a mention into a bookmark)
- Improved the link editing experience with enhanced UX
- Added OTP (One-Time Password) support for sign-in authentication
- Added latest AI models: GPT-4.1, GPT-4.1-mini, and Claude 3.7 Sonnet
-#### Bug Fixes
- Fixed an issue where properties were not displaying in the row detail page
- Fixed a bug where Undo didn't work in the row detail page
- Fixed an issue where blocks didn't grow when the grid got bigger
- Fixed several bugs related to AI writers
-### Mobile
-#### New Features
- Added sign-in with OTP (One-Time Password)
-#### Bug Fixes
- Fixed an issue where the slash menu sometimes failed to display
- Updated the mention page block to handle page selection with more context.
-
-## Version 0.8.8 - 01/04/2025
-### New Features
- Added support for selecting AI models in AI writer
- Revamped link menu in toolbar
- Added support for using ":" to add emojis in documents
- Passed the history of past AI prompts and responses to AI writer
-### Bug Fixes
- Improved AI writer scrolling user experience
- Fixed issue where checklist items would disappear during reordering
- Fixed numbered lists generated by AI to maintain the same index as the input
-
-## Version 0.8.7 - 18/03/2025
-### New Features
- Made local AI free and integrated with Ollama
- Supported nested lists within callout and quote blocks
- Revamped the document's floating toolbar and added Turn Into
- Enabled custom icons in callout blocks
-### Bug Fixes
- Fixed occasional incorrect positioning of the slash menu
- Improved AI Chat and AI Writers with various bug fixes
- Adjusted the columns block to match the width of the editor
- Fixed a potential segfault caused by infinite recursion in the trash view
- Resolved an issue where the first added cover might be invisible
- Fixed adding cover images via Unsplash
-
-## Version 0.8.6 - 06/03/2025
-### Bug Fixes
- Fix the incorrect title positioning when adjusting the document width setting
- Enhance the user experience of the icon color picker for smoother interactions
- Add missing icons to the database to ensure completeness and consistency
- Resolve the issue with links not functioning correctly on Linux systems
- Improve the outline feature to work seamlessly within columns
- Center the bulleted list icon within columns for better visual alignment
- Enable dragging blocks under tables in the second column to enhance flexibility
- Disable the AI writer feature within tables to prevent conflicts and improve usability
- Automatically enable the header row when converting content from Markdown to ensure proper formatting
- Use the "Undo" function to revert the auto-formatting
-
-## Version 0.8.5 - 04/03/2025
-### New Features
- Columns in Documents: Arrange content side by side using drag-and-drop or the slash menu
- AI Writers: New AI assistants in documents with response formatting options (list, table, text with images, image-only), follow-up questions, contextual memory, and more
- Compact Mode for Databases: Enable compact mode for grid and kanban views (full-page and inline) to increase information density, displaying more data per screen
-### Bug Fixes
- Fixed an issue where callout blocks couldn’t be deleted when appearing as the first line in a document
- Fixed a bug preventing the relation field in databases from opening
- Fixed an issue where links in documents were unclickable on Linux
-
-## Version 0.8.4 - 18/02/2025
-### New Features
- Switch AI mode on mobile
- Support locking page
- Support uploading svg file as icon
- Support the slash, at, and plus menus on mobile
-### Bug Fixes
- Gallery not rendering in row page
- Save image should not copy the image (mobile)
- Support exporting more content to markdown
-
 ## Version 0.8.2 - 23/01/2025
 ### New Features
 - Customized database view icons
@@ -1,6 +1,6 @@
 <h1 align="center" style="border-bottom: none">
    <b>
-        <a href="https://www.appflowy.com">AppFlowy</a><br>
+        <a href="https://www.appflowy.io">AppFlowy.IO</a><br>
    </b>
    ⭐️  The Open Source Alternative To Notion  ⭐️ <br>
 </h1>
@@ -18,18 +18,18 @@ AppFlowy is the AI workspace where you achieve more without losing control of yo
 </p>

 <p align="center">
-    <a href="https://www.appflowy.com"><b>Website</b></a> •
+    <a href="https://www.appflowy.io"><b>Website</b></a> •
    <a href="https://forum.appflowy.io/"><b>Forum</b></a> •
    <a href="https://discord.gg/9Q2xaN37tV"><b>Discord</b></a> •
    <a href="https://www.reddit.com/r/AppFlowy"><b>Reddit</b></a> •
    <a href="https://twitter.com/appflowy"><b>Twitter</b></a>
 </p>

-<p align="center"><img src="https://appflowy.com/_next/static/media/tasks.796c753e.png" alt="AppFlowy Kanban Board for To-dos"  /></p>
-<p align="center"><img src="https://appflowy.com/_next/static/media/Grid.9e30484b.png" alt="AppFlowy Databases for Tasks and Projects"  /></p>
-<p align="center"><img src="https://appflowy.com/_next/static/media/sites.a8d5b2b9.png" alt="AppFlowy Sites for Beautiful documentation"  /></p>
-<p align="center"><img src="https://appflowy.com/_next/static/media/ai.e1460982.png" alt="AppFlowy AI" /></p>
-<p align="center"><img src="https://appflowy.com/_next/static/media/template.9ea13c3b.png" alt="AppFlowy Templates"  /></p>
+<p align="center"><img src="https://appflowy.io/_next/static/media/tasks.796c753e.png" alt="AppFlowy Kanban Board for To-dos"  /></p>
+<p align="center"><img src="https://appflowy.io/_next/static/media/Grid.9e30484b.png" alt="AppFlowy Databases for Tasks and Projects"  /></p>
+<p align="center"><img src="https://appflowy.io/_next/static/media/sites.a8d5b2b9.png" alt="AppFlowy Sites for Beautiful documentation"  /></p>
+<p align="center"><img src="https://appflowy.io/_next/static/media/ai.e1460982.png" alt="AppFlowy AI" /></p>
+<p align="center"><img src="https://appflowy.io/_next/static/media/template.9ea13c3b.png" alt="AppFlowy Templates"  /></p>

 <br></br>
 <p align="center" >
@@ -48,7 +48,7 @@ AppFlowy is the AI workspace where you achieve more without losing control of yo
    - [App Store](https://apps.apple.com/app/appflowy/id6457261352): iPhone
    - [Play Store](https://play.google.com/store/apps/details?id=io.appflowy.appflowy): Android 10 or above; ARMv7 is
      not supported
- [Self-hosting AppFlowy](https://appflowy.com/docs/self-host-appflowy-overview)
+- [Self-hosting AppFlowy](https://docs.appflowy.io/docs/guides/appflowy/self-hosting-appflowy)
 - [Source](https://docs.appflowy.io/docs/documentation/appflowy/from-source)

 ## Built With
@@ -78,7 +78,7 @@ report [here](https://github.com/AppFlowy-IO/AppFlowy/issues/new?assignees=&labe

 ## **Releases**

-Please see the [changelog](https://appflowy.com/what-is-new) for more details about a given release.
+Please see the [changelog](https://www.appflowy.io/whatsnew) for more details about a given release.

 ## Contributing

@@ -89,7 +89,9 @@ for details.

 If your Pull Request is accepted as it fixes a bug, adds functionality, or makes AppFlowy's codebase significantly
 easier to use or understand, **Congratulations!** If your administrative and managerial work behind the scenes sustains
-the community, **Congratulations!** You are now an official contributor to AppFlowy.
+the community, **Congratulations!** You are now an official contributor to AppFlowy. Get in touch with
+us ([link](https://tally.so/r/mKP5z3)) to receive the very special Contributor T-shirt!
+Proudly wear your T-shirt and show it to us by tagging [@appflowy](https://twitter.com/appflowy) on Twitter.

 ## Translations 🌎🗺

@@ -150,8 +152,8 @@ more information.

 ## Acknowledgments

-Special thanks to these amazing projects which help power AppFlowy:
+Special thanks to these amazing projects which help power AppFlowy.IO:

 - [cargo-make](https://github.com/sagiegurari/cargo-make)
 - [contrib.rocks](https://contrib.rocks)
- [flutter_chat_ui](https://pub.dev/packages/flutter_chat_ui)
+- [flutter_chat_ui](https://pub.dev/packages/flutter_chat_ui)
@@ -4,7 +4,7 @@ workflows:
    instance_type: mac_mini_m2
    max_build_duration: 30
    environment:
-      flutter: 3.27.4
+      flutter: 3.22.3
      xcode: latest
      cocoapods: default

@@ -1,16 +1,16 @@
 #https://github.com/sagiegurari/cargo-make

 extend = [
-  { path = "scripts/makefile/desktop.toml" },
-  { path = "scripts/makefile/mobile.toml" },
-  { path = "scripts/makefile/protobuf.toml" },
-  { path = "scripts/makefile/tests.toml" },
-  { path = "scripts/makefile/docker.toml" },
-  { path = "scripts/makefile/env.toml" },
-  { path = "scripts/makefile/flutter.toml" },
-  { path = "scripts/makefile/tool.toml" },
-  { path = "scripts/makefile/tauri.toml" },
-  { path = "scripts/makefile/web.toml" },
+    { path = "scripts/makefile/desktop.toml" },
+    { path = "scripts/makefile/mobile.toml" },
+    { path = "scripts/makefile/protobuf.toml" },
+    { path = "scripts/makefile/tests.toml" },
+    { path = "scripts/makefile/docker.toml" },
+    { path = "scripts/makefile/env.toml" },
+    { path = "scripts/makefile/flutter.toml" },
+    { path = "scripts/makefile/tool.toml" },
+    { path = "scripts/makefile/tauri.toml" },
+    { path = "scripts/makefile/web.toml" },
 ]

 [config]
@@ -26,7 +26,7 @@ CARGO_MAKE_EXTEND_WORKSPACE_MAKEFILE = true
 CARGO_MAKE_CRATE_FS_NAME = "dart_ffi"
 CARGO_MAKE_CRATE_NAME = "dart-ffi"
 LIB_NAME = "dart_ffi"
-APPFLOWY_VERSION = "0.9.1"
+APPFLOWY_VERSION = "0.8.2"
 FLUTTER_DESKTOP_FEATURES = "dart"
 PRODUCT_NAME = "AppFlowy"
 MACOSX_DEPLOYMENT_TARGET = "11.0"
@@ -228,31 +228,30 @@ script = ['''
    echo APP_ENVIRONMENT: ${APP_ENVIRONMENT}
    echo BUILD_ARCHS: ${BUILD_ARCHS}
    echo BUILD_VERSION: ${BUILD_VERSION}
-    echo RUST_VERSION: "$(rustc --version)"
    ''']
 script_runner = "@shell"

 [tasks.setup-crate-type]
 private = true
 script = [
-  """
-      toml = readfile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml
-      val = replace ${toml} "staticlib" ${CRATE_TYPE}
-      result = writefile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml ${val}
-      assert ${result}
-      """,
+    """
+        toml = readfile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml
+        val = replace ${toml} "staticlib" ${CRATE_TYPE}
+        result = writefile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml ${val}
+        assert ${result}
+        """,
 ]
 script_runner = "@duckscript"

 [tasks.restore-crate-type]
 private = true
 script = [
-  """
-      toml = readfile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml
-      val = replace ${toml} ${CRATE_TYPE} "staticlib"
-      result = writefile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml ${val}
-      assert ${result}
-      """,
+    """
+        toml = readfile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml
+        val = replace ${toml} ${CRATE_TYPE} "staticlib"
+        result = writefile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml ${val}
+        assert ${result}
+        """,
 ]
 script_runner = "@duckscript"

@@ -280,24 +279,24 @@ TEST_COMPILE_TARGET = "x86_64-pc-windows-msvc"
 [tasks.setup-test-crate-type]
 private = true
 script = [
-  """
-      toml = readfile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml
-      val = replace ${toml} "staticlib" ${TEST_CRATE_TYPE}
-      result = writefile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml ${val}
-      assert ${result}
-      """,
+    """
+        toml = readfile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml
+        val = replace ${toml} "staticlib" ${TEST_CRATE_TYPE}
+        result = writefile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml ${val}
+        assert ${result}
+        """,
 ]
 script_runner = "@duckscript"

 [tasks.restore-test-crate-type]
 private = true
 script = [
-  """
-      toml = readfile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml
-      val = replace ${toml} ${TEST_CRATE_TYPE} "staticlib"
-      result = writefile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml ${val}
-      assert ${result}
-      """,
+    """
+        toml = readfile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml
+        val = replace ${toml} ${TEST_CRATE_TYPE} "staticlib"
+        result = writefile ${CARGO_MAKE_WORKSPACE_WORKING_DIRECTORY}/rust-lib/${CARGO_MAKE_CRATE_NAME}/Cargo.toml ${val}
+        assert ${result}
+        """,
 ]
 script_runner = "@duckscript"

@@ -4,7 +4,6 @@ analyzer:
  exclude:
    - "**/*.g.dart"
    - "**/*.freezed.dart"
-    - "packages/**/*.dart"

 linter:
  rules:
@@ -43,8 +43,6 @@
 		This is used by the Flutter tool to generate GeneratedPluginRegistrant.java
 		-->
    <meta-data android:name="flutterEmbedding" android:value="2" />
-    <meta-data android:name="io.flutter.embedding.android.EnableImpeller"
-      android:value="false" />
  </application>
  <uses-permission android:name="android.permission.INTERNET" />
  <uses-permission android:name="android.permission.ACCESS_NETWORK_STATE" />
@@ -1,4 +0,0 @@
-<?xml version="1.0" encoding="utf-8"?><!-- Uploaded to: SVG Repo, www.svgrepo.com, Generator: SVG Repo Mixer Tools -->
-<svg width="800px" height="800px" viewBox="0 0 16 16" fill="none" xmlns="http://www.w3.org/2000/svg">
-<path fill-rule="evenodd" clip-rule="evenodd" d="M16 8C16 12.4183 12.4183 16 8 16C3.58172 16 0 12.4183 0 8C0 3.58172 3.58172 0 8 0C12.4183 0 16 3.58172 16 8ZM9.25 3.75C9.25 4.44036 8.69036 5 8 5C7.30964 5 6.75 4.44036 6.75 3.75C6.75 3.05964 7.30964 2.5 8 2.5C8.69036 2.5 9.25 3.05964 9.25 3.75ZM12 8H9.41901L11.2047 13H9.081L8 9.97321L6.91901 13H4.79528L6.581 8H4V6H12V8Z" fill="#000000"/>
-</svg>
@@ -8,7 +8,7 @@
 "{""data"":""Juan"",""last_modified"":1726063423,""created_at"":1726063423,""field_type"":0}","{""created_at"":1726122510,""reminder_id"":"""",""include_time"":false,""is_range"":true,""last_modified"":1726122515,""data"":""1725604115"",""end_timestamp"":""1725776915"",""field_type"":2}","{""field_type"":1,""created_at"":1726063677,""last_modified"":1726063677,""data"":""93100""}","{""field_type"":1,""data"":""4.86"",""created_at"":1726062597,""last_modified"":1726062597}",,"{""last_modified"":1726062377,""field_type"":5,""data"":""Yes"",""created_at"":1726062377}","{""last_modified"":1726064412,""field_type"":7,""data"":""{\""options\"":[{\""id\"":\""tTDq\"",\""name\"":\""complete onboarding\"",\""color\"":\""Purple\""},{\""id\"":\""E8Ds\"",\""name\"":\""contact support\"",\""color\"":\""Purple\""},{\""id\"":\""RoGN\"",\""name\"":\""get started\"",\""color\"":\""Purple\""}],\""selected_option_ids\"":[\""tTDq\"",\""E8Ds\""]}"",""created_at"":1726064396}",,"{""created_at"":1726065278,""field_type"":3,""data"":""qnja"",""last_modified"":1726065278}","{""data"":""R9I7,yORP,1i4f"",""field_type"":4,""created_at"":1726105126,""last_modified"":1726105127}","{""data"":""1726122515"",""field_type"":8}","{""data"":""1726060541"",""field_type"":9}"
 "{""data"":""Alex"",""created_at"":1726063432,""last_modified"":1726063432,""field_type"":0}","{""reminder_id"":"""",""data"":""1725292800"",""include_time"":true,""last_modified"":1726122448,""created_at"":1726122422,""is_range"":true,""end_timestamp"":""1725551940"",""field_type"":2}","{""field_type"":1,""last_modified"":1726063683,""created_at"":1726063683,""data"":""3560""}","{""created_at"":1726062561,""data"":""1.96"",""last_modified"":1726062561,""field_type"":1}","{""last_modified"":1726063952,""created_at"":1726063931,""data"":""al3x1343@protonmail.com"",""field_type"":6}","{""last_modified"":1726062375,""field_type"":5,""created_at"":1726062375,""data"":""Yes""}","{""data"":""{\""options\"":[{\""id\"":\""qNyr\"",\""name\"":\""finish reading book\"",\""color\"":\""Purple\""}],\""selected_option_ids\"":[]}"",""created_at"":1726064616,""last_modified"":1726064616,""field_type"":7}",,"{""data"":""qnja"",""created_at"":1726065272,""last_modified"":1726065272,""field_type"":3}","{""created_at"":1726105180,""last_modified"":1726105180,""field_type"":4,""data"":""R9I7,1i4f""}","{""field_type"":8,""data"":""1726122448""}","{""field_type"":9,""data"":""1726060541""}"
 "{""last_modified"":1726063478,""created_at"":1726063436,""field_type"":0,""data"":""Alexander""}",,"{""field_type"":1,""last_modified"":1726063691,""created_at"":1726063691,""data"":""2073""}","{""field_type"":1,""data"":""0.5"",""last_modified"":1726062577,""created_at"":1726062577}","{""last_modified"":1726063991,""field_type"":6,""created_at"":1726063991,""data"":""alexandernotthedra@gmail.com""}","{""field_type"":5,""last_modified"":1726062378,""created_at"":1726062377,""data"":""No""}",,,"{""created_at"":1726065291,""data"":""GSf_"",""last_modified"":1726065291,""field_type"":3}","{""last_modified"":1726105142,""created_at"":1726105133,""data"":""SEUo"",""field_type"":4}","{""field_type"":8,""data"":""1726105142""}","{""field_type"":9,""data"":""1726060542""}"
-"{""field_type"":0,""created_at"":1726063454,""last_modified"":1726063454,""data"":""George""}","{""created_at"":1726122467,""end_timestamp"":""1726468070"",""include_time"":false,""is_range"":true,""reminder_id"":"""",""field_type"":2,""data"":""1726295270"",""last_modified"":1726122470}",,,"{""field_type"":6,""data"":""george.aq@appflowy.io"",""last_modified"":1726064104,""created_at"":1726064016}","{""last_modified"":1726062376,""created_at"":1726062376,""field_type"":5,""data"":""Yes""}","{""data"":""{\""options\"":[{\""id\"":\""s_dQ\"",\""name\"":\""bug triage\"",\""color\"":\""Purple\""},{\""id\"":\""-Zfo\"",\""name\"":\""fix bugs\"",\""color\"":\""Purple\""},{\""id\"":\""wsDN\"",\""name\"":\""attend meetings\"",\""color\"":\""Purple\""}],\""selected_option_ids\"":[\""s_dQ\"",\""-Zfo\""]}"",""last_modified"":1726064468,""created_at"":1726064424,""field_type"":7}","{""data"":[""{\""id\"":\""8a77f84d-64e9-4e67-b902-fa23980459ec\"",\""name\"":\""BQdTmxpRI6f.png\"",\""url\"":\""https://file-examples.com/wp-content/storage/2017/10/file_example_PNG_500kB.png\"",\""upload_type\"":\""NetworkMedia\"",\""file_type\"":\""Image\""}""],""field_type"":14,""created_at"":1726122956,""last_modified"":1726122956}","{""field_type"":3,""data"":""qnja"",""created_at"":1726065313,""last_modified"":1726065313}","{""data"":""R9I7,yORP"",""field_type"":4,""last_modified"":1726105198,""created_at"":1726105187}","{""data"":""1726122956"",""field_type"":8}","{""data"":""1726060543"",""field_type"":9}"
+"{""field_type"":0,""created_at"":1726063454,""last_modified"":1726063454,""data"":""George""}","{""created_at"":1726122467,""end_timestamp"":""1726468070"",""include_time"":false,""is_range"":true,""reminder_id"":"""",""field_type"":2,""data"":""1726295270"",""last_modified"":1726122470}",,,"{""field_type"":6,""data"":""george.aq@appflowy.io"",""last_modified"":1726064104,""created_at"":1726064016}","{""last_modified"":1726062376,""created_at"":1726062376,""field_type"":5,""data"":""Yes""}","{""data"":""{\""options\"":[{\""id\"":\""s_dQ\"",\""name\"":\""bug triage\"",\""color\"":\""Purple\""},{\""id\"":\""-Zfo\"",\""name\"":\""fix bugs\"",\""color\"":\""Purple\""},{\""id\"":\""wsDN\"",\""name\"":\""attend meetings\"",\""color\"":\""Purple\""}],\""selected_option_ids\"":[\""s_dQ\"",\""-Zfo\""]}"",""last_modified"":1726064468,""created_at"":1726064424,""field_type"":7}","{""data"":[""{\""id\"":\""8a77f84d-64e9-4e67-b902-fa23980459ec\"",\""name\"":\""BQdTmxpRI6f.png\"",\""url\"":\""https://static.cdninstagram.com/rsrc.php/v3/ym/r/BQdTmxpRI6f.png\"",\""upload_type\"":\""NetworkMedia\"",\""file_type\"":\""Image\""}""],""field_type"":14,""created_at"":1726122956,""last_modified"":1726122956}","{""field_type"":3,""data"":""qnja"",""created_at"":1726065313,""last_modified"":1726065313}","{""data"":""R9I7,yORP"",""field_type"":4,""last_modified"":1726105198,""created_at"":1726105187}","{""data"":""1726122956"",""field_type"":8}","{""data"":""1726060543"",""field_type"":9}"
 "{""field_type"":0,""last_modified"":1726063467,""data"":""Joanna"",""created_at"":1726063467}","{""include_time"":false,""end_timestamp"":""1727072893"",""is_range"":true,""last_modified"":1726122493,""created_at"":1726122483,""data"":""1726554493"",""field_type"":2,""reminder_id"":""""}","{""last_modified"":1726065463,""data"":""16470"",""field_type"":1,""created_at"":1726065463}","{""created_at"":1726062626,""field_type"":1,""last_modified"":1726062626,""data"":""-5.36""}","{""last_modified"":1726064069,""data"":""joannastrawberry29+hello@gmail.com"",""created_at"":1726064069,""field_type"":6}",,"{""field_type"":7,""created_at"":1726064444,""last_modified"":1726064460,""data"":""{\""options\"":[{\""id\"":\""ZxJz\"",\""name\"":\""post on Twitter\"",\""color\"":\""Purple\""},{\""id\"":\""upwi\"",\""name\"":\""watch Youtube videos\"",\""color\"":\""Purple\""}],\""selected_option_ids\"":[\""upwi\""]}""}",,"{""created_at"":1726065317,""last_modified"":1726065317,""field_type"":3,""data"":""qnja""}","{""field_type"":4,""last_modified"":1726105173,""data"":""uRAO,yORP"",""created_at"":1726105170}","{""data"":""1726122493"",""field_type"":8}","{""data"":""1726060545"",""field_type"":9}"
 "{""last_modified"":1726063457,""created_at"":1726063457,""data"":""George"",""field_type"":0}","{""include_time"":true,""reminder_id"":"""",""field_type"":2,""is_range"":true,""created_at"":1726122521,""end_timestamp"":""1725829200"",""data"":""1725822900"",""last_modified"":1726122535}","{""last_modified"":1726065493,""field_type"":1,""data"":""9500"",""created_at"":1726065493}","{""last_modified"":1726062680,""created_at"":1726062680,""field_type"":1,""data"":""1.7""}","{""data"":""plgeorgebball@gmail.com"",""field_type"":6,""last_modified"":1726064087,""created_at"":1726064036}",,"{""last_modified"":1726064513,""data"":""{\""options\"":[{\""id\"":\""zy0x\"",\""name\"":\""game vs celtics\"",\""color\"":\""Purple\""},{\""id\"":\""WJsv\"",\""name\"":\""training\"",\""color\"":\""Purple\""},{\""id\"":\""w-f8\"",\""name\"":\""game vs spurs\"",\""color\"":\""Purple\""},{\""id\"":\""p1VQ\"",\""name\"":\""game vs knicks\"",\""color\"":\""Purple\""},{\""id\"":\""VjUA\"",\""name\"":\""recovery\"",\""color\"":\""Purple\""},{\""id\"":\""sQ8X\"",\""name\"":\""don't get injured\"",\""color\"":\""Purple\""}],\""selected_option_ids\"":[]}"",""created_at"":1726064486,""field_type"":7}",,"{""field_type"":3,""last_modified"":1726065310,""data"":""qnja"",""created_at"":1726065310}","{""created_at"":1726105205,""field_type"":4,""last_modified"":1726105249,""data"":""R9I7,1i4f,yORP,SEUo""}","{""data"":""1726122535"",""field_type"":8}","{""field_type"":9,""data"":""1726060546""}"
 "{""data"":""Judy"",""created_at"":1726063475,""field_type"":0,""last_modified"":1726063487}","{""end_timestamp"":"""",""reminder_id"":"""",""data"":""1726640950"",""field_type"":2,""include_time"":false,""created_at"":1726122550,""last_modified"":1726122550,""is_range"":false}",,,"{""created_at"":1726063882,""field_type"":6,""last_modified"":1726064000,""data"":""judysmithjr@outlook.com""}","{""last_modified"":1726062712,""field_type"":5,""data"":""Yes"",""created_at"":1726062712}","{""created_at"":1726064549,""field_type"":7,""data"":""{\""options\"":[{\""id\"":\""j8cC\"",\""name\"":\""finish training\"",\""color\"":\""Purple\""},{\""id\"":\""SmSk\"",\""name\"":\""brainwash\"",\""color\"":\""Purple\""},{\""id\"":\""mnf5\"",\""name\"":\""welcome to ba sing se\"",\""color\"":\""Purple\""},{\""id\"":\""hcrj\"",\""name\"":\""don't mess up\"",\""color\"":\""Purple\""}],\""selected_option_ids\"":[\""j8cC\"",\""SmSk\"",\""mnf5\"",\""hcrj\""]}"",""last_modified"":1726064591}",,,"{""field_type"":4,""last_modified"":1726105152,""created_at"":1726105152,""data"":""R9I7""}","{""field_type"":8,""data"":""1726122550""}","{""field_type"":9,""data"":""1726060549""}"
@@ -1,12 +0,0 @@
-output: dist/
-releases:
-  - name: dev
-    jobs:
-      - name: release-dev-linux-deb
-        package:
-          platform: linux
-          target: deb
-      - name: release-dev-linux-rpm
-        package:
-          platform: linux
-          target: rpm
@@ -1,36 +0,0 @@
-----BEGIN PUBLIC KEY-----
-MIIGQzCCBDUGByqGSM44BAEwggQoAoICAQDlkozRmUnVH1MJFqOamAmUYu0YruaT
-rrt6rCIZ0LFrfNnmHA4LOQEcXwBTTyn5sBmkPq+lb/rjmERKhmvl1rfo6q7tJ8mG
-4TWqSu0tOJQ6QxexnNW4yhzK/r9MS5MQus4Al+y2hQLaAMOUIOnaWIrC9OHy7xyw
-+sVipECVKyQqipS4shGUSqbcN+ocQuTB+I0MtIjBii0DGSEY3pxQrfNWjHFL7iTV
-KiTn3YOWPJQvv3FvEDrN+5xU5JZpD97ZhXaJpLUyOQaNvcPaOELPWcOSJwqHOpf5
-b5N/VZ8SGbHNdxy9d5sSChBgtuAOihEhSp6SjFQ9eVHOf4NyJwSEMmi0gpdpqm4Z
-QRJUnM2zIi0p9twR9FRYXzrxOs6yGCQEY+xFG93ShTLTj3zMrIyFqBsqEwFyJiJW
-YWe/zp0V7UlLP+jXO9u9eghNmly7QVqD2P0qs/1V0jZFRuLWpsv4inau/qMZ5EhG
-G4xCJZXfN1pkehy6e05/h+vs5anK3Wa/H8AtY6cK4CpzAanELvn3AH7VLbAhLswu
-6d5CV+DoFgxCWMzGBSdmCYU+2wRLaL8Q9TZHDR+pvQlunEFdfFoGES9WjBPhAsVA
-6Mq22U8XSje9yHI3X9Eqe/7a+ajSgcGmB7oQ11+4xf5h2PtubRW/JL0KMjxCxMTp
-q1md6Ndx/ptBUwIdAIOyiKb2YcTLWAOt+LAlRXMsY1+W4pTXJfV6RcMCggIAPxbd
-0HNj2O/aQhJxNZDMBIcx6+cZ+LKch7qLcaEpVqWHvDSnR2eOJJzWn0RoKK+Vuix/
-4T8texSQkWxAeFFdo6kyrR9XNL7hqEFFq8o9VpmvRzvG6h/bBgh3AHAQE3p/8Wrb
-K13IhnlWqd0MjFufSphm63o0gaWl95j+6KeUoKQnioetu9HiMtFKx0d/KYqTQJg7
-hvR6VNCU2oShfXR3ce7RnUYwD37+djrUjUkoAZkZq2KoxBiKyeoSIeqAme19tKcO
-s6b17mhALELuJ+NtDwlDunyiCDUYX9lTPijHwKeIFtBs38+OtRk3aIqmWTQdbsCz
-Axp+kUMA5ESBME/RBNCSPHuDvtA3wfWvNbA5DXfZLwCgNSxhekq8XntIsRzfJ4v4
-uPzKFcVM3+sUUfSF04HHC9ol+PpLqXUyMnskiizqxFPq7H+6tyFZ7X2HiG6TjcfV
-Wthmv+JyfcABjVnk2qFH7GagENbdtYmfUox13LhE59Sh5chaJnCFtCDp8NClWgZn
-ixCOFQ9EgTLaH6MovTvWpEgG2MfBCu5SMUHi2qSflorqpRFH+rA7NZSnyz3wm7NB
-+fJSOP0IjEkOh7MafU6Z61oK9WY/Fc+F1zIENVv8PUc3p75y/4RAp4xzyKcTilaN
-C9U/3MRr3QmWwY7ejtZx6xdOxsvWBRDRSNbDdIkDggIGAAKCAgEAt1DHYZoeXY0r
-vYXmxdNO6zfnbz1GGZHXpakzm9h4BrxPDP5J8DQ9ZeVVKg5+cU9AyMO3cZHp7wkx
-k6IB+ZDUpqO1D3lWriRl2fI8cS4edI0fzpnW1nyhhFD4MbKmP+v27aH+DhZ4Up3y
-GMmJTLmKiYx1EgZp7Sx77PBYDVMsKKd3h9+Hjp2YtUTfD2lleAmC+wcQGZiNtGw/
-eKpsmUVnWrepOdntWTtCQi1OvfcHaF2QmgktCq+68hbDNYWaXmzVIiQqrdv/zzOG
-hCFIrRGWemrxL0iFG4Pzc4UfOINsISQLcUxRuF6pQWPxF8O/mWKfzAeqWxmIujUM
-EoSEuI3yQ8VjlYpW/8FSK7UhgnHBHOpCJWPWs/vQXAnaUR2PYyzuIzhVEhFs8YA8
-iBIKnixIC2hu0YbEk3TBr/TRcbd7mDw9Mq7NT88xzdU13+Wh+4zhdX3rtBHYzBtI
-7GaONGUNyY4h0duoyLpH6dxevaeKN6/bEdzYESjoE58QA88CpnAZGhJVphAba4cb
-w6GTDhK3RlPWh6hRqJwLDILGtnJS3UKeBDRmKMqNuqmHqPjyAAvt9JBO8lzjoLgf
-1cDsXHNWBVwA2jsX2CukNJPlY1Fa3MWhdaUXmy6QGMSisr1sptvBt1Phry8T2u+P
-Y29SB4jvwqls268rP0cWqy4WXwlVwuc=
-----END PUBLIC KEY-----
@@ -23,24 +23,24 @@ void main() {
      final expandFinder = find.byFlowySvg(FlowySvgs.hamburger_s_s);

      // Is expanded by default
-      expect(collapseFinder, findsNothing);
-      expect(expandFinder, findsOneWidget);
-
-      // Collapse hidden groups
-      await tester.tap(expandFinder);
-      await tester.pumpAndSettle();
-
-      // Is collapsed
      expect(collapseFinder, findsOneWidget);
      expect(expandFinder, findsNothing);

-      // Expand hidden groups
+      // Collapse hidden groups
      await tester.tap(collapseFinder);
      await tester.pumpAndSettle();

-      // Is expanded
+      // Is collapsed
      expect(collapseFinder, findsNothing);
      expect(expandFinder, findsOneWidget);
+
+      // Expand hidden groups
+      await tester.tap(expandFinder);
+      await tester.pumpAndSettle();
+
+      // Is expanded
+      expect(collapseFinder, findsOneWidget);
+      expect(expandFinder, findsNothing);
    });

    testWidgets('hide first group, and show it again', (tester) async {
@@ -48,9 +48,6 @@ void main() {
      await tester.tapAnonymousSignInButton();
      await tester.createNewPageWithNameUnderParent(layout: ViewLayoutPB.Board);

-      final expandFinder = find.byFlowySvg(FlowySvgs.hamburger_s_s);
-      await tester.tapButton(expandFinder);
-
      // Tap the options of the first group
      final optionsFinder = find
          .descendant(
@@ -12,38 +12,24 @@ import 'uncategorized/uncategorized_test_runner.dart'
 import 'workspace/workspace_test_runner.dart' as workspace_test_runner;

 Future<void> main() async {
-  // don't remove this test, it can prevent the test from failing.
-  {
-    preset_af_cloud_env_test.main();
-    data_migration_test_runner.main();
+  preset_af_cloud_env_test.main();

-    // uncategorized
-    uncategorized_test_runner.main();
+  data_migration_test_runner.main();

-    // workspace
-    workspace_test_runner.main();
-  }
+  // uncategorized
+  uncategorized_test_runner.main();

-  // sidebar
-  // don't remove this test, it can prevent the test from failing.
-  {
-    preset_af_cloud_env_test.main();
-    sidebar_move_page_test.main();
-    sidebar_rename_untitled_test.main();
-    sidebar_icon_test.main();
-  }
-
-  // database
-  // don't remove this test, it can prevent the test from failing.
-  {
-    preset_af_cloud_env_test.main();
-    database_test_runner.main();
-  }
+  // workspace
+  workspace_test_runner.main();

  // document
-  // don't remove this test, it can prevent the test from failing.
-  {
-    preset_af_cloud_env_test.main();
-    document_test_runner.main();
-  }
+  document_test_runner.main();
+
+  // sidebar
+  sidebar_move_page_test.main();
+  sidebar_rename_untitled_test.main();
+  sidebar_icon_test.main();
+
+  // database
+  database_test_runner.main();
 }
@@ -15,6 +15,7 @@ void main() {
        cloudType: AuthenticatorType.appflowyCloudSelfHost,
      );

+      await tester.tapContinousAnotherWay();
      await tester.tapAnonymousSignInButton();
      await tester.expectToSeeHomePageWithGetStartedPage();

@@ -30,6 +31,12 @@ void main() {
      await tester.enterUserName('local_user');

      // Scroll to sign-in
+      await tester.scrollUntilVisible(
+        find.byType(AccountSignInOutButton),
+        100,
+        scrollable: find.findSettingsScrollable(),
+      );
+
      await tester.tapButton(find.byType(AccountSignInOutButton));

      // sign up with Google
@@ -1,3 +1,5 @@
+import 'anon_user_data_migration_test.dart' as anon_user_test;
+
 void main() async {
-  // anon_user_test.main();
+  anon_user_test.main();
 }
@@ -1,19 +1,12 @@
-import 'package:appflowy/ai/service/ai_entities.dart';
 import 'package:appflowy/env/cloud_env.dart';
 import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/ai/operations/ai_writer_entities.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/plugins.dart';
-import 'package:appflowy_backend/protobuf/flowy-ai/entities.pbenum.dart';
 import 'package:appflowy_backend/protobuf/flowy-folder/view.pb.dart';
-import 'package:appflowy_editor/appflowy_editor.dart';
 import 'package:easy_localization/easy_localization.dart';
-import 'package:flutter/services.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';

-import '../../../shared/ai_test_op.dart';
 import '../../../shared/constants.dart';
-import '../../../shared/mock/mock_ai.dart';
 import '../../../shared/util.dart';

 void main() {
@@ -24,7 +17,6 @@ void main() {
        (tester) async {
      await tester.initializeAppFlowy(
        cloudType: AuthenticatorType.appflowyCloudSelfHost,
-        aiRepositoryBuilder: () => MockAIRepository(),
      );
      await tester.tapGoogleLoginInButton();
      await tester.expectToSeeHomePageWithGetStartedPage();
@@ -41,7 +33,7 @@ void main() {
      await tester.editor.tapSlashMenuItemWithName(
        LocaleKeys.document_slashMenu_name_aiWriter.tr(),
      );
-      expect(find.byType(AiWriterBlockComponent), findsOneWidget);
+      expect(find.byType(AIWriterBlockComponent), findsOneWidget);

      // switch to another page
      await tester.openPage(Constants.gettingStartedPageName);
@@ -49,186 +41,7 @@ void main() {
      await tester.openPage(pageName);

      // expect the ai writer block is not in the document
-      expect(find.byType(AiWriterBlockComponent), findsNothing);
-    });
-
-    testWidgets('Improve writing', (tester) async {
-      await tester.initializeAppFlowy(
-        cloudType: AuthenticatorType.appflowyCloudSelfHost,
-      );
-      await tester.tapGoogleLoginInButton();
-      await tester.expectToSeeHomePageWithGetStartedPage();
-
-      const pageName = 'Document';
-      await tester.createNewPageInSpace(
-        spaceName: Constants.generalSpaceName,
-        layout: ViewLayoutPB.Document,
-        pageName: pageName,
-      );
-
-      await tester.editor.tapLineOfEditorAt(0);
-
-      // insert a paragraph
-      final text = 'I have an apple';
-      await tester.editor.tapLineOfEditorAt(0);
-      await tester.ime.insertText(text);
-      await tester.editor.updateSelection(
-        Selection(
-          start: Position(path: [0]),
-          end: Position(path: [0], offset: text.length),
-        ),
-      );
-
-      await tester.pumpAndSettle();
-      await tester.tapButton(find.byType(ImproveWritingButton));
-
-      final editorState = tester.editor.getCurrentEditorState();
-      final document = editorState.document;
-
-      expect(document.root.children.length, 3);
-      expect(document.root.children[1].type, ParagraphBlockKeys.type);
-      expect(
-        document.root.children[1].delta!.toPlainText(),
-        'I have an apple and a banana',
-      );
-    });
-
-    testWidgets('fix grammar', (tester) async {
-      await tester.initializeAppFlowy(
-        cloudType: AuthenticatorType.appflowyCloudSelfHost,
-      );
-      await tester.tapGoogleLoginInButton();
-      await tester.expectToSeeHomePageWithGetStartedPage();
-
-      const pageName = 'Document';
-      await tester.createNewPageInSpace(
-        spaceName: Constants.generalSpaceName,
-        layout: ViewLayoutPB.Document,
-        pageName: pageName,
-      );
-
-      await tester.editor.tapLineOfEditorAt(0);
-
-      // insert a paragraph
-      final text = 'We didn’t had enough money';
-      await tester.editor.tapLineOfEditorAt(0);
-      await tester.ime.insertText(text);
-      await tester.editor.updateSelection(
-        Selection(
-          start: Position(path: [0]),
-          end: Position(path: [0], offset: text.length),
-        ),
-      );
-
-      await tester.pumpAndSettle();
-      await tester.selectAIWriter(AiWriterCommand.fixSpellingAndGrammar);
-
-      final editorState = tester.editor.getCurrentEditorState();
-      final document = editorState.document;
-
-      expect(document.root.children.length, 3);
-      expect(document.root.children[1].type, ParagraphBlockKeys.type);
-      expect(
-        document.root.children[1].delta!.toPlainText(),
-        'We didn’t have enough money',
-      );
-    });
-
-    testWidgets('ask ai', (tester) async {
-      await tester.initializeAppFlowy(
-        cloudType: AuthenticatorType.appflowyCloudDevelop,
-        aiRepositoryBuilder: () => MockAIRepository(
-          validator: _CompletionHistoryValidator(),
-        ),
-      );
-      await tester.tapGoogleLoginInButton();
-      await tester.expectToSeeHomePageWithGetStartedPage();
-
-      const pageName = 'Document';
-      await tester.createNewPageInSpace(
-        spaceName: Constants.generalSpaceName,
-        layout: ViewLayoutPB.Document,
-        pageName: pageName,
-      );
-
-      await tester.editor.tapLineOfEditorAt(0);
-
-      // insert a paragraph
-      final text = 'What is TPU?';
-      await tester.editor.tapLineOfEditorAt(0);
-      await tester.ime.insertText(text);
-      await tester.editor.updateSelection(
-        Selection(
-          start: Position(path: [0]),
-          end: Position(path: [0], offset: text.length),
-        ),
-      );
-
-      await tester.pumpAndSettle();
-      await tester.selectAIWriter(AiWriterCommand.userQuestion);
-
-      await tester.enterTextInPromptTextField("Explain the concept of TPU");
-      await tester.simulateKeyEvent(LogicalKeyboardKey.enter);
-      await tester.pumpAndSettle();
-
-      // await tester.selectModel("GPT-4o-mini");
-
-      await tester.enterTextInPromptTextField("How about GPU?");
-      await tester.simulateKeyEvent(LogicalKeyboardKey.enter);
-      await tester.pumpAndSettle();
+      expect(find.byType(AIWriterBlockComponent), findsNothing);
    });
  });
 }
-
-class _CompletionHistoryValidator extends StreamCompletionValidator {
-  static const _expectedResponses = {
-    1: "What is TPU?",
-    3: [
-      "What is TPU?",
-      "Explain the concept of TPU",
-      "TPU is a tensor processing unit that is designed to accelerate machine",
-    ],
-    5: [
-      "What is TPU?",
-      "Explain the concept of TPU",
-      "TPU is a tensor processing unit that is designed to accelerate machine",
-      "How about GPU?",
-      "GPU is a graphics processing unit that is designed to accelerate machine learning tasks.",
-    ],
-  };
-
-  @override
-  bool validate(
-    String text,
-    String? objectId,
-    CompletionTypePB completionType,
-    PredefinedFormat? format,
-    List<String> sourceIds,
-    List<AiWriterRecord> history,
-  ) {
-    assert(completionType == CompletionTypePB.UserQuestion);
-    if (history.isEmpty) return false;
-
-    final expectedMessages = _expectedResponses[history.length];
-    if (expectedMessages == null) return false;
-
-    if (expectedMessages is String) {
-      _assertMessage(history[0].content, expectedMessages);
-      return true;
-    } else if (expectedMessages is List<String>) {
-      for (var i = 0; i < expectedMessages.length; i++) {
-        _assertMessage(history[i].content, expectedMessages[i]);
-      }
-      return true;
-    }
-
-    return false;
-  }
-
-  void _assertMessage(String actual, String expected) {
-    assert(
-      actual.trim() == expected,
-      "expected '$expected', but got '${actual.trim()}'",
-    );
-  }
-}
@@ -57,7 +57,7 @@ void main() {
      // move the checkbox to the child of the block at path [9]
      await tester.editor.dragBlock(
        [10],
-        const Offset(120, -20),
+        const Offset(80, -30),
      );

      // wait for the move animation to complete
@@ -8,13 +8,11 @@ import '../../shared/util.dart';
 void main() {
  IntegrationTestWidgetsFlutterBinding.ensureInitialized();

-  group('Preset cloud env', () {
-    testWidgets('use self-hosted cloud', (tester) async {
+  group('Empty', () {
+    testWidgets('set appflowy cloud', (tester) async {
      await tester.initializeAppFlowy(
        cloudType: AuthenticatorType.appflowyCloudSelfHost,
      );
-
-      await tester.pumpAndSettle();
    });
  });
 }
@@ -39,8 +39,8 @@ void main() {
      );
      await tester.tapButton(find.byType(AccountSignInOutButton));

-      tester.expectToSeeText(LocaleKeys.button_yes.tr());
-      await tester.tapButtonWithName(LocaleKeys.button_yes.tr());
+      tester.expectToSeeText(LocaleKeys.button_ok.tr());
+      await tester.tapButtonWithName(LocaleKeys.button_ok.tr());

      // Go to the sign in page again
      await tester.pumpAndSettle(const Duration(seconds: 5));
@@ -57,6 +57,12 @@ void main() {
      await tester.openSettings();
      await tester.openSettingsPage(SettingsPage.account);

+      // Scroll to sign-in
+      await tester.scrollUntilVisible(
+        find.byType(AccountSignInOutButton),
+        100,
+        scrollable: find.findSettingsScrollable(),
+      );
      await tester.tapButton(find.byType(AccountSignInOutButton));

      tester.expectToSeeGoogleLoginButton();
@@ -80,8 +80,8 @@ void main() {
      await tester.tapGoogleLoginInButton();
      await tester.expectToSeeHomePageWithGetStartedPage();

-      expect(find.byType(AppFlowyEditorPage), findsOneWidget);
-      expect(find.text('Blank page'), findsNothing);
+      expect(find.byType(AppFlowyEditorPage), findsNothing);
+      expect(find.text('Blank page'), findsOne);
    });
  });
 }
@@ -6,7 +6,7 @@ import 'package:appflowy/generated/locale_keys.g.dart';
 import 'package:appflowy/shared/icon_emoji_picker/icon_picker.dart';
 import 'package:appflowy/workspace/presentation/command_palette/command_palette.dart';
 import 'package:appflowy/workspace/presentation/command_palette/widgets/search_field.dart';
-import 'package:appflowy/workspace/presentation/command_palette/widgets/search_result_cell.dart';
+import 'package:appflowy/workspace/presentation/command_palette/widgets/search_result_tile.dart';
 import 'package:appflowy_backend/protobuf/flowy-folder/view.pbenum.dart';
 import 'package:appflowy_editor/appflowy_editor.dart';
 import 'package:easy_localization/easy_localization.dart';
@@ -44,12 +44,12 @@ void main() {
      await tester.pumpAndSettle(const Duration(milliseconds: 200));

      // Expect two search results "ViewOna" and "ViewOne" (Distance 1 to ViewOna)
-      expect(find.byType(SearchResultCell), findsNWidgets(2));
+      expect(find.byType(SearchResultTile), findsNWidgets(2));

      // The score should be higher for "ViewOna" thus it should be shown first
      final secondDocumentWidget = tester
-          .widget(find.byType(SearchResultCell).first) as SearchResultCell;
-      expect(secondDocumentWidget.item.displayName, secondDocument);
+          .widget(find.byType(SearchResultTile).first) as SearchResultTile;
+      expect(secondDocumentWidget.result.data, secondDocument);

      // Change search to "ViewOne"
      await tester.enterText(searchFieldFinder, firstDocument);
@@ -57,9 +57,9 @@ void main() {

      // The score should be higher for "ViewOne" thus it should be shown first
      final firstDocumentWidget = tester.widget(
-        find.byType(SearchResultCell).first,
-      ) as SearchResultCell;
-      expect(firstDocumentWidget.item.displayName, firstDocument);
+        find.byType(SearchResultTile).first,
+      ) as SearchResultTile;
+      expect(firstDocumentWidget.result.data, firstDocument);
    });

    testWidgets('Displaying icons in search results', (tester) async {
@@ -89,11 +89,11 @@ void main() {
      );
      await tester.enterText(searchFieldFinder, 'Page-$randomValue');
      await tester.pumpAndSettle(const Duration(milliseconds: 200));
-      expect(find.byType(SearchResultCell), findsNWidgets(2));
+      expect(find.byType(SearchResultTile), findsNWidgets(2));

      /// check results
      final svgs = find.descendant(
-        of: find.byType(SearchResultCell),
+        of: find.byType(SearchResultTile),
        matching: find.byType(FlowySvg),
      );
      expect(svgs, findsNWidgets(2));
@@ -1,5 +1,5 @@
 import 'package:appflowy/workspace/presentation/command_palette/command_palette.dart';
-import 'package:appflowy/workspace/presentation/command_palette/widgets/search_recent_view_cell.dart';
+import 'package:appflowy/workspace/presentation/command_palette/widgets/recent_view_tile.dart';
 import 'package:appflowy/workspace/presentation/command_palette/widgets/recent_views_list.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';
@@ -27,12 +27,11 @@ void main() {
      expect(find.byType(RecentViewsList), findsOneWidget);

      // Expect three recent history items
-      expect(find.byType(SearchRecentViewCell), findsNWidgets(3));
+      expect(find.byType(RecentViewTile), findsNWidgets(3));

      // Expect the first item to be the last viewed document
      final firstDocumentWidget =
-          tester.widget(find.byType(SearchRecentViewCell).first)
-              as SearchRecentViewCell;
+          tester.widget(find.byType(RecentViewTile).first) as RecentViewTile;
      expect(firstDocumentWidget.view.name, secondDocument);
    });
  });
@@ -15,7 +15,6 @@ void main() {
      await tester.initializeAppFlowy();
      await tester.tapAnonymousSignInButton();

-      // create a database and add a linked database view
      await tester.createNewPageWithNameUnderParent(layout: ViewLayoutPB.Grid);
      await tester.tapCreateLinkedDatabaseViewButton(DatabaseLayoutPB.Grid);

@@ -30,11 +29,6 @@ void main() {
      await tester.tapHidePropertyButton();
      tester.noFieldWithName('New field 1');

-      // create another field, New field 1 to be hidden still
-      await tester.tapNewPropertyButton();
-      await tester.dismissFieldEditor();
-      tester.noFieldWithName('New field 1');
-
      // go back to inline database view, expect field to be shown
      await tester.tapTabBarLinkedViewByViewName('Untitled');
      tester.findFieldWithName('New field 1');
@@ -66,40 +60,5 @@ void main() {
      await tester.tapDatabaseSortButton();
      await tester.tapCreateSortByFieldType(FieldType.RichText, "New field 1");
    });
-
-    testWidgets('field cell width', (tester) async {
-      await tester.initializeAppFlowy();
-      await tester.tapAnonymousSignInButton();
-
-      // create a database and add a linked database view
-      await tester.createNewPageWithNameUnderParent(layout: ViewLayoutPB.Grid);
-      await tester.tapCreateLinkedDatabaseViewButton(DatabaseLayoutPB.Grid);
-
-      // create a field
-      await tester.scrollToRight(find.byType(GridPage));
-      await tester.tapNewPropertyButton();
-      await tester.renameField('New field 1');
-      await tester.dismissFieldEditor();
-
-      // check the width of the field
-      expect(tester.getFieldWidth('New field 1'), 150);
-
-      // change the width of the field
-      await tester.changeFieldWidth('New field 1', 200);
-      expect(tester.getFieldWidth('New field 1'), 205);
-
-      // create another field, New field 1 to be same width
-      await tester.tapNewPropertyButton();
-      await tester.dismissFieldEditor();
-      expect(tester.getFieldWidth('New field 1'), 205);
-
-      // go back to inline database view, expect New field 1 to be 150px
-      await tester.tapTabBarLinkedViewByViewName('Untitled');
-      expect(tester.getFieldWidth('New field 1'), 150);
-
-      // go back to linked database view, expect New field 1 to be 205px
-      await tester.tapTabBarLinkedViewByViewName('Grid');
-      expect(tester.getFieldWidth('New field 1'), 205);
-    });
  });
 }
@@ -1,10 +1,5 @@
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/plugins/database/grid/presentation/grid_page.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/plugins.dart';
 import 'package:appflowy_backend/protobuf/flowy-database2/setting_entities.pbenum.dart';
 import 'package:appflowy_backend/protobuf/flowy-folder/view.pb.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:flutter_animate/flutter_animate.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';

@@ -78,37 +73,5 @@ void main() {

      await tester.pumpAndSettle();
    });
-
-    testWidgets('insert grid in column', (tester) async {
-      await tester.initializeAppFlowy();
-      await tester.tapAnonymousSignInButton();
-
-      /// create page and show slash menu
-      await tester.createNewPageWithNameUnderParent(name: 'test page');
-      await tester.editor.tapLineOfEditorAt(0);
-      await tester.editor.showSlashMenu();
-      await tester.pumpAndSettle();
-
-      /// create a column
-      await tester.editor.tapSlashMenuItemWithName(
-        LocaleKeys.document_slashMenu_name_twoColumns.tr(),
-      );
-      final actionList = find.byType(BlockActionList);
-      expect(actionList, findsNWidgets(2));
-      final position = tester.getCenter(actionList.last);
-
-      /// tap the second child of column
-      await tester.tapAt(position.copyWith(dx: position.dx + 50));
-
-      /// create a grid
-      await tester.editor.showSlashMenu();
-      await tester.pumpAndSettle();
-      await tester.editor.tapSlashMenuItemWithName(
-        LocaleKeys.document_slashMenu_name_grid.tr(),
-      );
-
-      final grid = find.byType(GridPageContent);
-      expect(grid, findsOneWidget);
-    });
  });
 }
@@ -27,9 +27,8 @@ void main() {
      await tester.pumpAndSettle();

      // click the align center
-      await tester.tapButtonWithFlowySvgData(FlowySvgs.toolbar_alignment_m);
-      await tester
-          .tapButtonWithFlowySvgData(FlowySvgs.toolbar_text_align_center_m);
+      await tester.tapButtonWithFlowySvgData(FlowySvgs.toolbar_align_left_s);
+      await tester.tapButtonWithFlowySvgData(FlowySvgs.toolbar_align_center_s);

      // expect to see the align center
      final editorState = tester.editor.getCurrentEditorState();
@@ -37,15 +36,13 @@ void main() {
      expect(first.attributes[blockComponentAlign], 'center');

      // click the align right
-      await tester.tapButtonWithFlowySvgData(FlowySvgs.toolbar_alignment_m);
-      await tester
-          .tapButtonWithFlowySvgData(FlowySvgs.toolbar_text_align_right_m);
+      await tester.tapButtonWithFlowySvgData(FlowySvgs.toolbar_align_center_s);
+      await tester.tapButtonWithFlowySvgData(FlowySvgs.toolbar_align_right_s);
      expect(first.attributes[blockComponentAlign], 'right');

      // click the align left
-      await tester.tapButtonWithFlowySvgData(FlowySvgs.toolbar_alignment_m);
-      await tester
-          .tapButtonWithFlowySvgData(FlowySvgs.toolbar_text_align_left_m);
+      await tester.tapButtonWithFlowySvgData(FlowySvgs.toolbar_align_right_s);
+      await tester.tapButtonWithFlowySvgData(FlowySvgs.toolbar_align_left_s);
      expect(first.attributes[blockComponentAlign], 'left');
    });

@@ -78,7 +75,7 @@ void main() {
        [
          LogicalKeyboardKey.control,
          LogicalKeyboardKey.shift,
-          LogicalKeyboardKey.keyC,
+          LogicalKeyboardKey.keyE,
        ],
        tester: tester,
        withKeyUp: true,
@@ -1,12 +1,10 @@
-import 'dart:async';
 import 'dart:io';

 import 'package:appflowy/generated/locale_keys.g.dart';
+import 'package:appflowy/plugins/document/presentation/editor_plugins/block_menu/block_menu_button.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/copy_and_paste/clipboard_service.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/image/custom_image_block_component/custom_image_block_component.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/link_preview/custom_link_preview.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/link_preview/link_preview_menu.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/link_preview/paste_as/paste_as_menu.dart';
 import 'package:appflowy/startup/startup.dart';
 import 'package:appflowy_editor/appflowy_editor.dart';
 import 'package:appflowy_editor_plugins/appflowy_editor_plugins.dart';
@@ -322,14 +320,8 @@ void main() {
      (tester) async {
        const url = 'https://appflowy.io';
        await tester.pasteContent(plainText: url, (editorState) async {
-          final pasteAsMenu = find.byType(PasteAsMenu);
-          expect(pasteAsMenu, findsOneWidget);
-          final bookmarkButton = find.text(
-            LocaleKeys.document_plugins_linkPreview_typeSelection_bookmark.tr(),
-          );
-          await tester.tapButton(bookmarkButton);
          // the second one is the paragraph node
-          expect(editorState.document.root.children.length, 1);
+          expect(editorState.document.root.children.length, 2);
          final node = editorState.getNodeAtPath([0])!;
          expect(node.type, LinkPreviewBlockKeys.type);
          expect(node.attributes[LinkPreviewBlockKeys.url], url);
@@ -341,20 +333,19 @@ void main() {
        await tester.hoverOnWidget(
          find.byType(CustomLinkPreviewWidget),
          onHover: () async {
-            /// show menu
-            final menu = find.byType(CustomLinkPreviewMenu);
-            expect(menu, findsOneWidget);
-            await tester.tapButton(menu);
-
-            final convertToLinkButton = find.text(
-              LocaleKeys.document_plugins_linkPreview_linkPreviewMenu_toUrl
-                  .tr(),
-            );
+            final convertToLinkButton = find.byWidgetPredicate((widget) {
+              return widget is MenuBlockButton &&
+                  widget.tooltip ==
+                      LocaleKeys.document_plugins_urlPreview_convertToLink.tr();
+            });
            expect(convertToLinkButton, findsOneWidget);
-            await tester.tapButton(convertToLinkButton);
+            await tester.tap(convertToLinkButton);
+            await tester.pumpAndSettle();
          },
        );

+        await tester.pumpAndSettle();
+
        final editorState = tester.editor.getCurrentEditorState();
        final textNode = editorState.getNodeAtPath([0])!;
        expect(textNode.type, ParagraphBlockKeys.type);
@@ -372,19 +363,14 @@ void main() {
      (tester) async {
        const url = 'https://appflowy.io';
        await tester.pasteContent(plainText: url, (editorState) async {
-          final pasteAsMenu = find.byType(PasteAsMenu);
-          expect(pasteAsMenu, findsOneWidget);
-          final bookmarkButton = find.text(
-            LocaleKeys.document_plugins_linkPreview_typeSelection_bookmark.tr(),
-          );
-          await tester.tapButton(bookmarkButton);
          // the second one is the paragraph node
-          expect(editorState.document.root.children.length, 1);
+          expect(editorState.document.root.children.length, 2);
          final node = editorState.getNodeAtPath([0])!;
          expect(node.type, LinkPreviewBlockKeys.type);
          expect(node.attributes[LinkPreviewBlockKeys.url], url);
        });

+        await tester.editor.tapLineOfEditorAt(0);
        await tester.simulateKeyEvent(
          LogicalKeyboardKey.keyZ,
          isControlPressed:
@@ -472,7 +458,7 @@ void main() {
    });

    testWidgets('paste the image url', (tester) async {
-      const plainText = 'http://example.com/1.jpg';
+      const plainText = 'https://appflowy.io/1.jpg';
      final image = await rootBundle.load('assets/test/images/sample.jpeg');
      final bytes = image.buffer.asUint8List();
      await tester.pasteContent(plainText: plainText, image: ('jpeg', bytes),
@@ -483,6 +469,16 @@ void main() {
      });
    });

+    testWidgets('paste image url without extension', (tester) async {
+      const plainText =
+          'https://images.unsplash.com/photo-1469474968028-56623f02e42e?ixlib=rb-4.0.3&q=85&fm=jpg&crop=entropy&cs=srgb&dl=david-marcu-78A265wPiO4-unsplash.jpg&w=640';
+      await tester.pasteContent(plainText: plainText, (editorState) {
+        final node = editorState.getNodeAtPath([0])!;
+        expect(node.type, ImageBlockKeys.type);
+        expect(node.attributes[ImageBlockKeys.url], isNotEmpty);
+      });
+    });
+
    const testMarkdownText = '''
 # I'm h1
 ## I'm h2
@@ -525,7 +521,7 @@ void main() {

 extension on WidgetTester {
  Future<void> pasteContent(
-    FutureOr<void> Function(EditorState editorState) test, {
+    void Function(EditorState editorState) test, {
    Future<void> Function(EditorState editorState)? beforeTest,
    String? plainText,
    String? html,
@@ -562,6 +558,6 @@ extension on WidgetTester {
    );
    await pumpAndSettle(const Duration(milliseconds: 1000));

-    await test(editor.getCurrentEditorState());
+    test(editor.getCurrentEditorState());
  }
 }
@@ -13,8 +13,6 @@ void main() {
      await tester.initializeAppFlowy();

      await tester.tapAnonymousSignInButton();
-      final finder = find.text(gettingStarted, findRichText: true);
-      await tester.pumpUntilFound(finder, timeout: const Duration(seconds: 2));

      // create a new document
      const pageName = 'Test Document';
@@ -1,453 +0,0 @@
-import 'dart:io';
-
-import 'package:appflowy/generated/flowy_svgs.g.dart';
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/copy_and_paste/clipboard_service.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/desktop_toolbar/link/link_hover_menu.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/link_embed/link_embed_block_component.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/link_embed/link_embed_menu.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/link_preview/custom_link_preview_block_component.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/link_preview/link_preview_menu.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/link_preview/paste_as/paste_as_menu.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/mention/mention_block.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/mention/mention_link_block.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/mention/mention_link_error_preview.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/mention/mention_link_preview.dart';
-import 'package:appflowy/startup/startup.dart';
-import 'package:appflowy_editor/appflowy_editor.dart';
-import 'package:appflowy_editor_plugins/appflowy_editor_plugins.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:flutter/material.dart';
-import 'package:flutter/services.dart';
-import 'package:flutter_test/flutter_test.dart';
-import 'package:integration_test/integration_test.dart';
-
-import '../../shared/util.dart';
-
-void main() {
-  IntegrationTestWidgetsFlutterBinding.ensureInitialized();
-
-  const avaliableLink = 'https://appflowy.io/',
-      unavailableLink = 'www.thereIsNoting.com';
-
-  Future<void> preparePage(WidgetTester tester, {String? pageName}) async {
-    await tester.initializeAppFlowy();
-    await tester.tapAnonymousSignInButton();
-    await tester.createNewPageWithNameUnderParent(name: pageName);
-    await tester.editor.tapLineOfEditorAt(0);
-  }
-
-  Future<void> pasteLink(WidgetTester tester, String link) async {
-    await getIt<ClipboardService>()
-        .setData(ClipboardServiceData(plainText: link));
-
-    /// paste the link
-    await tester.simulateKeyEvent(
-      LogicalKeyboardKey.keyV,
-      isControlPressed: Platform.isLinux || Platform.isWindows,
-      isMetaPressed: Platform.isMacOS,
-    );
-    await tester.pumpAndSettle(Duration(seconds: 1));
-  }
-
-  Future<void> pasteAs(
-    WidgetTester tester,
-    String link,
-    PasteMenuType type, {
-    Duration waitTime = const Duration(milliseconds: 500),
-  }) async {
-    await pasteLink(tester, link);
-    final convertToMentionButton = find.text(type.title);
-    await tester.tapButton(convertToMentionButton);
-    await tester.pumpAndSettle(waitTime);
-  }
-
-  void checkUrl(Node node, String link) {
-    expect(node.type, ParagraphBlockKeys.type);
-    expect(node.delta!.toJson(), [
-      {
-        'insert': link,
-        'attributes': {'href': link},
-      }
-    ]);
-  }
-
-  void checkMention(Node node, String link) {
-    final delta = node.delta!;
-    final insert = (delta.first as TextInsert).text;
-    final attributes = delta.first.attributes;
-    expect(insert, MentionBlockKeys.mentionChar);
-    final mention =
-        attributes?[MentionBlockKeys.mention] as Map<String, dynamic>;
-    expect(mention[MentionBlockKeys.type], MentionType.externalLink.name);
-    expect(mention[MentionBlockKeys.url], avaliableLink);
-  }
-
-  void checkBookmark(Node node, String link) {
-    expect(node.type, LinkPreviewBlockKeys.type);
-    expect(node.attributes[LinkPreviewBlockKeys.url], link);
-  }
-
-  void checkEmbed(Node node, String link) {
-    expect(node.type, LinkPreviewBlockKeys.type);
-    expect(node.attributes[LinkEmbedKeys.previewType], LinkEmbedKeys.embed);
-    expect(node.attributes[LinkPreviewBlockKeys.url], link);
-  }
-
-  group('Paste as URL', () {
-    Future<void> pasteAndTurnInto(
-      WidgetTester tester,
-      String link,
-      String title,
-    ) async {
-      await pasteLink(tester, link);
-      final convertToLinkButton = find
-          .text(LocaleKeys.document_plugins_linkPreview_typeSelection_URL.tr());
-      await tester.tapButton(convertToLinkButton);
-
-      /// hover link and turn into mention
-      await tester.hoverOnWidget(
-        find.byType(LinkHoverTrigger),
-        onHover: () async {
-          final turnintoButton = find.byFlowySvg(FlowySvgs.turninto_m);
-          await tester.tapButton(turnintoButton);
-          final convertToButton = find.text(title);
-          await tester.tapButton(convertToButton);
-          await tester.pumpAndSettle(Duration(seconds: 1));
-        },
-      );
-    }
-
-    testWidgets('paste a link', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteLink(tester, link);
-      final convertToLinkButton = find
-          .text(LocaleKeys.document_plugins_linkPreview_typeSelection_URL.tr());
-      await tester.tapButton(convertToLinkButton);
-      final node = tester.editor.getNodeAtPath([0]);
-      checkUrl(node, link);
-    });
-
-    testWidgets('paste a link and turn into mention', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAndTurnInto(
-        tester,
-        link,
-        LinkConvertMenuCommand.toMention.title,
-      );
-
-      /// check metion values
-      final node = tester.editor.getNodeAtPath([0]);
-      checkMention(node, link);
-    });
-
-    testWidgets('paste a link and turn into bookmark', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAndTurnInto(
-        tester,
-        link,
-        LinkConvertMenuCommand.toBookmark.title,
-      );
-
-      /// check metion values
-      final node = tester.editor.getNodeAtPath([0]);
-      checkBookmark(node, link);
-    });
-
-    testWidgets('paste a link and turn into embed', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAndTurnInto(
-        tester,
-        link,
-        LinkConvertMenuCommand.toEmbed.title,
-      );
-
-      /// check metion values
-      final node = tester.editor.getNodeAtPath([0]);
-      checkEmbed(node, link);
-    });
-  });
-
-  group('Paste as Mention', () {
-    Future<void> pasteAsMention(WidgetTester tester, String link) =>
-        pasteAs(tester, link, PasteMenuType.mention);
-
-    String getMentionLink(Node node) {
-      final insert = node.delta?.first as TextInsert?;
-      final mention = insert?.attributes?[MentionBlockKeys.mention]
-          as Map<String, dynamic>?;
-      return mention?[MentionBlockKeys.url] ?? '';
-    }
-
-    Future<void> hoverMentionAndClick(
-      WidgetTester tester,
-      String command,
-    ) async {
-      final mentionLink = find.byType(MentionLinkBlock);
-      expect(mentionLink, findsOneWidget);
-      await tester.hoverOnWidget(
-        mentionLink,
-        onHover: () async {
-          final errorPreview = find.byType(MentionLinkErrorPreview);
-          expect(errorPreview, findsOneWidget);
-          final convertButton = find.byFlowySvg(FlowySvgs.turninto_m);
-          await tester.tapButton(convertButton);
-          final menuButton = find.text(command);
-          await tester.tapButton(menuButton);
-        },
-      );
-    }
-
-    testWidgets('paste a link as mention', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsMention(tester, link);
-      final node = tester.editor.getNodeAtPath([0]);
-      checkMention(node, link);
-    });
-
-    testWidgets('paste as mention and copy link', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsMention(tester, link);
-      final mentionLink = find.byType(MentionLinkBlock);
-      expect(mentionLink, findsOneWidget);
-      await tester.hoverOnWidget(
-        mentionLink,
-        onHover: () async {
-          final preview = find.byType(MentionLinkPreview);
-          if (!preview.hasFound) {
-            final copyButton = find.byFlowySvg(FlowySvgs.toolbar_link_m);
-            await tester.tapButton(copyButton);
-          } else {
-            final moreOptionButton = find.byFlowySvg(FlowySvgs.toolbar_more_m);
-            await tester.tapButton(moreOptionButton);
-            final copyButton =
-                find.text(MentionLinktMenuCommand.copyLink.title);
-            await tester.tapButton(copyButton);
-          }
-        },
-      );
-      final clipboardContent = await getIt<ClipboardService>().getData();
-      expect(clipboardContent.plainText, link);
-    });
-
-    testWidgets('paste as error mention and turninto url', (tester) async {
-      String link = unavailableLink;
-      await preparePage(tester);
-      await pasteAsMention(tester, link);
-      Node node = tester.editor.getNodeAtPath([0]);
-      link = getMentionLink(node);
-      await hoverMentionAndClick(
-        tester,
-        MentionLinktErrorMenuCommand.toURL.title,
-      );
-      node = tester.editor.getNodeAtPath([0]);
-      checkUrl(node, link);
-    });
-
-    testWidgets('paste as error mention and turninto embed', (tester) async {
-      String link = unavailableLink;
-      await preparePage(tester);
-      await pasteAsMention(tester, link);
-      Node node = tester.editor.getNodeAtPath([0]);
-      link = getMentionLink(node);
-      await hoverMentionAndClick(
-        tester,
-        MentionLinktErrorMenuCommand.toEmbed.title,
-      );
-      node = tester.editor.getNodeAtPath([0]);
-      checkEmbed(node, link);
-    });
-
-    testWidgets('paste as error mention and turninto bookmark', (tester) async {
-      String link = unavailableLink;
-      await preparePage(tester);
-      await pasteAsMention(tester, link);
-      Node node = tester.editor.getNodeAtPath([0]);
-      link = getMentionLink(node);
-      await hoverMentionAndClick(
-        tester,
-        MentionLinktErrorMenuCommand.toBookmark.title,
-      );
-      node = tester.editor.getNodeAtPath([0]);
-      checkBookmark(node, link);
-    });
-
-    testWidgets('paste as error mention and remove link', (tester) async {
-      String link = unavailableLink;
-      await preparePage(tester);
-      await pasteAsMention(tester, link);
-      Node node = tester.editor.getNodeAtPath([0]);
-      link = getMentionLink(node);
-      await hoverMentionAndClick(
-        tester,
-        MentionLinktErrorMenuCommand.removeLink.title,
-      );
-      node = tester.editor.getNodeAtPath([0]);
-      expect(node.type, ParagraphBlockKeys.type);
-      expect(node.delta!.toJson(), [
-        {'insert': link},
-      ]);
-    });
-  });
-
-  group('Paste as Bookmark', () {
-    Future<void> pasteAsBookmark(WidgetTester tester, String link) =>
-        pasteAs(tester, link, PasteMenuType.bookmark);
-
-    Future<void> hoverAndClick(
-      WidgetTester tester,
-      LinkPreviewMenuCommand command,
-    ) async {
-      final bookmark = find.byType(CustomLinkPreviewBlockComponent);
-      expect(bookmark, findsOneWidget);
-      await tester.hoverOnWidget(
-        bookmark,
-        onHover: () async {
-          final menuButton = find.byFlowySvg(FlowySvgs.toolbar_more_m);
-          await tester.tapButton(menuButton);
-          final commandButton = find.text(command.title);
-          await tester.tapButton(commandButton);
-        },
-      );
-    }
-
-    testWidgets('paste a link as bookmark', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsBookmark(tester, link);
-      final node = tester.editor.getNodeAtPath([0]);
-      checkBookmark(node, link);
-    });
-
-    testWidgets('paste a link as bookmark and convert to mention',
-        (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsBookmark(tester, link);
-      await hoverAndClick(tester, LinkPreviewMenuCommand.convertToMention);
-      final node = tester.editor.getNodeAtPath([0]);
-      checkMention(node, link);
-    });
-
-    testWidgets('paste a link as bookmark and convert to url', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsBookmark(tester, link);
-      await hoverAndClick(tester, LinkPreviewMenuCommand.convertToUrl);
-      final node = tester.editor.getNodeAtPath([0]);
-      checkUrl(node, link);
-    });
-
-    testWidgets('paste a link as bookmark and convert to embed',
-        (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsBookmark(tester, link);
-      await hoverAndClick(tester, LinkPreviewMenuCommand.convertToEmbed);
-      final node = tester.editor.getNodeAtPath([0]);
-      checkEmbed(node, link);
-    });
-
-    testWidgets('paste a link as bookmark and copy link', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsBookmark(tester, link);
-      await hoverAndClick(tester, LinkPreviewMenuCommand.copyLink);
-      final clipboardContent = await getIt<ClipboardService>().getData();
-      expect(clipboardContent.plainText, link);
-    });
-
-    testWidgets('paste a link as bookmark and replace link', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsBookmark(tester, link);
-      await hoverAndClick(tester, LinkPreviewMenuCommand.replace);
-      await tester.simulateKeyEvent(
-        LogicalKeyboardKey.keyA,
-        isControlPressed: Platform.isLinux || Platform.isWindows,
-        isMetaPressed: Platform.isMacOS,
-      );
-      await tester.simulateKeyEvent(LogicalKeyboardKey.delete);
-      await tester.enterText(find.byType(TextFormField), unavailableLink);
-      await tester.tapButton(find.text(LocaleKeys.button_replace.tr()));
-      final node = tester.editor.getNodeAtPath([0]);
-      checkBookmark(node, unavailableLink);
-    });
-
-    testWidgets('paste a link as bookmark and remove link', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsBookmark(tester, link);
-      await hoverAndClick(tester, LinkPreviewMenuCommand.removeLink);
-      final node = tester.editor.getNodeAtPath([0]);
-      expect(node.type, ParagraphBlockKeys.type);
-      expect(node.delta!.toJson(), [
-        {'insert': link},
-      ]);
-    });
-  });
-  group('Paste as Embed', () {
-    Future<void> pasteAsEmbed(WidgetTester tester, String link) =>
-        pasteAs(tester, link, PasteMenuType.embed);
-
-    Future<void> hoverAndConvert(
-      WidgetTester tester,
-      LinkEmbedConvertCommand command,
-    ) async {
-      final embed = find.byType(LinkEmbedBlockComponent);
-      expect(embed, findsOneWidget);
-      await tester.hoverOnWidget(
-        embed,
-        onHover: () async {
-          final menuButton = find.byFlowySvg(FlowySvgs.turninto_m);
-          await tester.tapButton(menuButton);
-          final commandButton = find.text(command.title);
-          await tester.tapButton(commandButton);
-        },
-      );
-    }
-
-    testWidgets('paste a link as embed', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsEmbed(tester, link);
-      final node = tester.editor.getNodeAtPath([0]);
-      checkEmbed(node, link);
-    });
-
-    testWidgets('paste a link as bookmark and convert to mention',
-        (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsEmbed(tester, link);
-      await hoverAndConvert(tester, LinkEmbedConvertCommand.toMention);
-      final node = tester.editor.getNodeAtPath([0]);
-      checkMention(node, link);
-    });
-
-    testWidgets('paste a link as bookmark and convert to url', (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsEmbed(tester, link);
-      await hoverAndConvert(tester, LinkEmbedConvertCommand.toURL);
-      final node = tester.editor.getNodeAtPath([0]);
-      checkUrl(node, link);
-    });
-
-    testWidgets('paste a link as bookmark and convert to bookmark',
-        (tester) async {
-      final link = avaliableLink;
-      await preparePage(tester);
-      await pasteAsEmbed(tester, link);
-      await hoverAndConvert(tester, LinkEmbedConvertCommand.toBookmark);
-      final node = tester.editor.getNodeAtPath([0]);
-      checkBookmark(node, link);
-    });
-  });
-}
@@ -76,12 +76,13 @@ void main() {
        LocaleKeys.document_slashMenu_name_heading1.tr(): HeadingBlockKeys.type,
        LocaleKeys.document_slashMenu_name_heading2.tr(): HeadingBlockKeys.type,
        LocaleKeys.document_slashMenu_name_heading3.tr(): HeadingBlockKeys.type,
-        LocaleKeys.editor_bulletedListShortForm.tr():
+        LocaleKeys.document_slashMenu_name_bulletedList.tr():
            BulletedListBlockKeys.type,
-        LocaleKeys.editor_numberedListShortForm.tr():
+        LocaleKeys.document_slashMenu_name_numberedList.tr():
            NumberedListBlockKeys.type,
        LocaleKeys.document_slashMenu_name_quote.tr(): QuoteBlockKeys.type,
-        LocaleKeys.editor_checkbox.tr(): TodoListBlockKeys.type,
+        LocaleKeys.document_slashMenu_name_todoList.tr():
+            TodoListBlockKeys.type,
        LocaleKeys.document_slashMenu_name_callout.tr(): CalloutBlockKeys.type,
        LocaleKeys.document_slashMenu_name_text.tr(): ParagraphBlockKeys.type,
      };
@@ -116,12 +117,13 @@ void main() {
        LocaleKeys.document_slashMenu_name_heading1.tr(): HeadingBlockKeys.type,
        LocaleKeys.document_slashMenu_name_heading2.tr(): HeadingBlockKeys.type,
        LocaleKeys.document_slashMenu_name_heading3.tr(): HeadingBlockKeys.type,
-        LocaleKeys.editor_bulletedListShortForm.tr():
+        LocaleKeys.document_slashMenu_name_bulletedList.tr():
            BulletedListBlockKeys.type,
-        LocaleKeys.editor_numberedListShortForm.tr():
+        LocaleKeys.document_slashMenu_name_numberedList.tr():
            NumberedListBlockKeys.type,
        LocaleKeys.document_slashMenu_name_quote.tr(): QuoteBlockKeys.type,
-        LocaleKeys.editor_checkbox.tr(): TodoListBlockKeys.type,
+        LocaleKeys.document_slashMenu_name_todoList.tr():
+            TodoListBlockKeys.type,
        LocaleKeys.document_slashMenu_name_callout.tr(): CalloutBlockKeys.type,
        LocaleKeys.document_slashMenu_name_text.tr(): ParagraphBlockKeys.type,
      };
@@ -1,6 +1,5 @@
 import 'package:appflowy_editor/appflowy_editor.dart';
 import 'package:flutter/material.dart';
-import 'package:flutter/services.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';

@@ -48,41 +47,5 @@ void main() {

      expect(editorState.selection!.start.offset, 0);
    });
-
-    testWidgets('select and delete text', (tester) async {
-      await tester.initializeAppFlowy();
-      await tester.tapAnonymousSignInButton();
-
-      /// create a new document
-      await tester.createNewPageWithNameUnderParent();
-
-      /// input text
-      final editor = tester.editor;
-      final editorState = editor.getCurrentEditorState();
-
-      const inputText = 'Test for text selection and deletion';
-      final texts = inputText.split(' ');
-      await editor.tapLineOfEditorAt(0);
-      await tester.ime.insertText(inputText);
-
-      /// selecte and delete
-      int index = 0;
-      while (texts.isNotEmpty) {
-        final text = texts.removeAt(0);
-        await tester.editor.updateSelection(
-          Selection(
-            start: Position(path: [0], offset: index),
-            end: Position(path: [0], offset: index + text.length),
-          ),
-        );
-        await tester.simulateKeyEvent(LogicalKeyboardKey.delete);
-        index++;
-      }
-
-      /// excpete the text value is correct
-      final node = editorState.getNodeAtPath([0])!;
-      final nodeText = node.delta?.toPlainText() ?? '';
-      expect(nodeText, ' ' * (index - 1));
-    });
  });
 }
@@ -13,7 +13,6 @@ import 'document_with_multi_image_block_test.dart'
    as document_with_multi_image_block_test;
 import 'document_with_simple_table_test.dart'
    as document_with_simple_table_test;
-import 'document_link_preview_test.dart' as document_link_preview_test;

 void main() {
  IntegrationTestWidgetsFlutterBinding.ensureInitialized();
@@ -29,5 +28,4 @@ void main() {
  document_find_menu_test.main();
  document_toolbar_test.main();
  document_with_simple_table_test.main();
-  document_link_preview_test.main();
 }
@@ -1,19 +1,5 @@
-import 'package:appflowy/generated/flowy_svgs.g.dart';
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/copy_and_paste/clipboard_service.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/desktop_toolbar/desktop_floating_toolbar.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/desktop_toolbar/link/link_create_menu.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/desktop_toolbar/link/link_edit_menu.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/desktop_toolbar/link/link_hover_menu.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/plugins.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/toolbar_item/custom_link_toolbar_item.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/toolbar_item/more_option_toolbar_item.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/toolbar_item/text_suggestions_toolbar_item.dart';
-import 'package:appflowy/startup/startup.dart';
 import 'package:appflowy_editor/appflowy_editor.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:flutter/material.dart';
-import 'package:flutter/services.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';

@@ -22,33 +8,24 @@ import '../../shared/util.dart';
 void main() {
  IntegrationTestWidgetsFlutterBinding.ensureInitialized();

-  Future<void> selectText(WidgetTester tester, String text) async {
-    await tester.editor.updateSelection(
-      Selection.single(
-        path: [0],
-        startOffset: 0,
-        endOffset: text.length,
-      ),
-    );
-  }
-
-  Future<void> prepareForToolbar(WidgetTester tester, String text) async {
-    await tester.initializeAppFlowy();
-    await tester.tapAnonymousSignInButton();
-
-    await tester.createNewPageWithNameUnderParent();
-
-    await tester.editor.tapLineOfEditorAt(0);
-    await tester.ime.insertText(text);
-    await selectText(tester, text);
-  }
-
  group('document toolbar:', () {
    testWidgets('font family', (tester) async {
-      await prepareForToolbar(tester, 'font family');
+      await tester.initializeAppFlowy();
+      await tester.tapAnonymousSignInButton();
+
+      await tester.createNewPageWithNameUnderParent();
+
+      await tester.editor.tapLineOfEditorAt(0);
+      const text = 'font family';
+      await tester.ime.insertText(text);
+      await tester.editor.updateSelection(
+        Selection.single(
+          path: [0],
+          startOffset: 0,
+          endOffset: text.length,
+        ),
+      );

-      // tap more options button
-      await tester.tapButtonWithFlowySvgData(FlowySvgs.toolbar_more_m);
      // tap the font family button
      final fontFamilyButton = find.byKey(kFontFamilyToolbarItemKey);
      await tester.tapButton(fontFamilyButton);
@@ -69,302 +46,5 @@ void main() {
        abel,
      );
    });
-
-    testWidgets('heading 1~3', (tester) async {
-      const text = 'heading';
-      await prepareForToolbar(tester, text);
-
-      Future<void> testChangeHeading(
-        FlowySvgData svg,
-        String title,
-        int level,
-      ) async {
-        /// tap suggestions item
-        final suggestionsButton = find.byKey(kSuggestionsItemKey);
-        await tester.tapButton(suggestionsButton);
-
-        /// tap item
-        await tester.ensureVisible(find.byFlowySvg(svg));
-        await tester.tapButton(find.byFlowySvg(svg));
-
-        /// check the type of node is [HeadingBlockKeys.type]
-        await selectText(tester, text);
-        final editorState = tester.editor.getCurrentEditorState();
-        final selection = editorState.selection!;
-        final node = editorState.getNodeAtPath(selection.start.path)!,
-            nodeLevel = node.attributes[HeadingBlockKeys.level]!;
-        expect(node.type, HeadingBlockKeys.type);
-        expect(nodeLevel, level);
-
-        /// show toolbar again
-        await selectText(tester, text);
-
-        /// the text of suggestions item should be changed
-        expect(
-          find.descendant(of: suggestionsButton, matching: find.text(title)),
-          findsOneWidget,
-        );
-      }
-
-      await testChangeHeading(
-        FlowySvgs.type_h1_m,
-        LocaleKeys.document_toolbar_h1.tr(),
-        1,
-      );
-
-      await testChangeHeading(
-        FlowySvgs.type_h2_m,
-        LocaleKeys.document_toolbar_h2.tr(),
-        2,
-      );
-      await testChangeHeading(
-        FlowySvgs.type_h3_m,
-        LocaleKeys.document_toolbar_h3.tr(),
-        3,
-      );
-    });
-
-    testWidgets('toggle 1~3', (tester) async {
-      const text = 'toggle';
-      await prepareForToolbar(tester, text);
-
-      Future<void> testChangeToggle(
-        FlowySvgData svg,
-        String title,
-        int? level,
-      ) async {
-        /// tap suggestions item
-        final suggestionsButton = find.byKey(kSuggestionsItemKey);
-        await tester.tapButton(suggestionsButton);
-
-        /// tap item
-        await tester.ensureVisible(find.byFlowySvg(svg));
-        await tester.tapButton(find.byFlowySvg(svg));
-
-        /// check the type of node is [HeadingBlockKeys.type]
-        await selectText(tester, text);
-        final editorState = tester.editor.getCurrentEditorState();
-        final selection = editorState.selection!;
-        final node = editorState.getNodeAtPath(selection.start.path)!,
-            nodeLevel = node.attributes[ToggleListBlockKeys.level];
-        expect(node.type, ToggleListBlockKeys.type);
-        expect(nodeLevel, level);
-
-        /// show toolbar again
-        await selectText(tester, text);
-
-        /// the text of suggestions item should be changed
-        expect(
-          find.descendant(of: suggestionsButton, matching: find.text(title)),
-          findsOneWidget,
-        );
-      }
-
-      await testChangeToggle(
-        FlowySvgs.type_toggle_list_m,
-        LocaleKeys.editor_toggleListShortForm.tr(),
-        null,
-      );
-
-      await testChangeToggle(
-        FlowySvgs.type_toggle_h1_m,
-        LocaleKeys.editor_toggleHeading1ShortForm.tr(),
-        1,
-      );
-
-      await testChangeToggle(
-        FlowySvgs.type_toggle_h2_m,
-        LocaleKeys.editor_toggleHeading2ShortForm.tr(),
-        2,
-      );
-
-      await testChangeToggle(
-        FlowySvgs.type_toggle_h3_m,
-        LocaleKeys.editor_toggleHeading3ShortForm.tr(),
-        3,
-      );
-    });
-
-    testWidgets('toolbar will not rebuild after click item', (tester) async {
-      const text = 'Test rebuilding';
-      await prepareForToolbar(tester, text);
-      Finder toolbar = find.byType(DesktopFloatingToolbar);
-      Element toolbarElement = toolbar.evaluate().first;
-      final elementHashcode = toolbarElement.hashCode;
-      final boldButton = find.byFlowySvg(FlowySvgs.toolbar_bold_m),
-          underlineButton = find.byFlowySvg(FlowySvgs.toolbar_underline_m),
-          italicButton = find.byFlowySvg(FlowySvgs.toolbar_inline_italic_m);
-
-      /// tap format buttons
-      await tester.tapButton(boldButton);
-      await tester.tapButton(underlineButton);
-      await tester.tapButton(italicButton);
-      toolbar = find.byType(DesktopFloatingToolbar);
-      toolbarElement = toolbar.evaluate().first;
-
-      /// check if the toolbar is not rebuilt
-      expect(elementHashcode, toolbarElement.hashCode);
-      final editorState = tester.editor.getCurrentEditorState();
-
-      /// check text formats
-      expect(
-        editorState
-            .getDeltaAttributeValueInSelection(AppFlowyRichTextKeys.bold),
-        true,
-      );
-      expect(
-        editorState
-            .getDeltaAttributeValueInSelection(AppFlowyRichTextKeys.italic),
-        true,
-      );
-      expect(
-        editorState
-            .getDeltaAttributeValueInSelection(AppFlowyRichTextKeys.underline),
-        true,
-      );
-    });
-  });
-
-  group('document toolbar: link', () {
-    String? getLinkFromNode(Node node) {
-      for (final insert in node.delta!) {
-        final link = insert.attributes?.href;
-        if (link != null) return link;
-      }
-      return null;
-    }
-
-    bool isPageLink(Node node) {
-      for (final insert in node.delta!) {
-        final isPage = insert.attributes?.isPage;
-        if (isPage == true) return true;
-      }
-      return false;
-    }
-
-    String getNodeText(Node node) {
-      for (final insert in node.delta!) {
-        if (insert is TextInsert) return insert.text;
-      }
-      return '';
-    }
-
-    testWidgets('insert link and remove link', (tester) async {
-      const text = 'insert link', link = 'https://test.appflowy.cloud';
-      await prepareForToolbar(tester, text);
-
-      final toolbar = find.byType(DesktopFloatingToolbar);
-      expect(toolbar, findsOneWidget);
-
-      /// tap link button to show CreateLinkMenu
-      final linkButton = find.byFlowySvg(FlowySvgs.toolbar_link_m);
-      await tester.tapButton(linkButton);
-      final createLinkMenu = find.byType(LinkCreateMenu);
-      expect(createLinkMenu, findsOneWidget);
-
-      /// test esc to close
-      await tester.simulateKeyEvent(LogicalKeyboardKey.escape);
-      expect(toolbar, findsNothing);
-
-      /// show toolbar again
-      await tester.editor.tapLineOfEditorAt(0);
-      await selectText(tester, text);
-      await tester.tapButton(linkButton);
-
-      /// insert link
-      final textField = find.descendant(
-        of: createLinkMenu,
-        matching: find.byType(TextFormField),
-      );
-      await tester.enterText(textField, link);
-      await tester.pumpAndSettle();
-      await tester.simulateKeyEvent(LogicalKeyboardKey.enter);
-      Node node = tester.editor.getNodeAtPath([0]);
-      expect(getLinkFromNode(node), link);
-      await tester.simulateKeyEvent(LogicalKeyboardKey.escape);
-
-      /// hover link
-      await tester.hoverOnWidget(find.byType(LinkHoverTrigger));
-      final hoverMenu = find.byType(LinkHoverMenu);
-      expect(hoverMenu, findsOneWidget);
-
-      /// copy link
-      final copyButton = find.descendant(
-        of: hoverMenu,
-        matching: find.byFlowySvg(FlowySvgs.toolbar_link_m),
-      );
-      await tester.tapButton(copyButton);
-      final clipboardContent = await getIt<ClipboardService>().getData();
-      final plainText = clipboardContent.plainText;
-      expect(plainText, link);
-
-      /// remove link
-      await tester.hoverOnWidget(find.byType(LinkHoverTrigger));
-      await tester.tapButton(find.byFlowySvg(FlowySvgs.toolbar_link_unlink_m));
-      node = tester.editor.getNodeAtPath([0]);
-      expect(getLinkFromNode(node), null);
-    });
-
-    testWidgets('insert link and edit link', (tester) async {
-      const text = 'edit link',
-          link = 'https://test.appflowy.cloud',
-          afterText = '$text after';
-      await prepareForToolbar(tester, text);
-
-      /// tap link button to show CreateLinkMenu
-      final linkButton = find.byFlowySvg(FlowySvgs.toolbar_link_m);
-      await tester.tapButton(linkButton);
-
-      /// search for page and select it
-      final textField = find.descendant(
-        of: find.byType(LinkCreateMenu),
-        matching: find.byType(TextFormField),
-      );
-      await tester.enterText(textField, gettingStarted);
-      await tester.pumpAndSettle();
-      await tester.simulateKeyEvent(LogicalKeyboardKey.enter);
-      await tester.simulateKeyEvent(LogicalKeyboardKey.escape);
-
-      Node node = tester.editor.getNodeAtPath([0]);
-      expect(isPageLink(node), true);
-      expect(getLinkFromNode(node) == link, false);
-
-      /// hover link
-      await tester.hoverOnWidget(find.byType(LinkHoverTrigger));
-
-      /// click edit button to show LinkEditMenu
-      final editButton = find.byFlowySvg(FlowySvgs.toolbar_link_edit_m);
-      await tester.tapButton(editButton);
-      final linkEditMenu = find.byType(LinkEditMenu);
-      expect(linkEditMenu, findsOneWidget);
-
-      /// change the link text
-      final titleField = find.descendant(
-        of: linkEditMenu,
-        matching: find.byType(TextFormField),
-      );
-      await tester.enterText(titleField, afterText);
-      await tester.pumpAndSettle();
-      await tester.tapButton(
-        find.descendant(of: linkEditMenu, matching: find.text(gettingStarted)),
-      );
-      final linkField = find.ancestor(
-        of: find.text(LocaleKeys.document_toolbar_linkInputHint.tr()),
-        matching: find.byType(TextFormField),
-      );
-      await tester.enterText(linkField, link);
-      await tester.pumpAndSettle();
-      await tester.simulateKeyEvent(LogicalKeyboardKey.enter);
-
-      /// apply the change
-      final applyButton =
-          find.text(LocaleKeys.settings_appearance_documentSettings_apply.tr());
-      await tester.tapButton(applyButton);
-
-      node = tester.editor.getNodeAtPath([0]);
-      expect(isPageLink(node), false);
-      expect(getLinkFromNode(node), link);
-      expect(getNodeText(node), afterText);
-    });
  });
 }
@@ -237,11 +237,11 @@ void main() {
      await tester.tapButton(find.byType(MentionDateBlock));
      await tester.pumpAndSettle();

-      // set date to be fifteenth of the last month
+      // set date to be fifteenth of the next month
      await tester.tap(
        find.descendant(
          of: find.byType(DesktopAppFlowyDatePicker),
-          matching: find.byFlowySvg(FlowySvgs.arrow_left_s),
+          matching: find.byFlowySvg(FlowySvgs.arrow_right_s),
        ),
      );
      await tester.pumpAndSettle();
@@ -275,9 +275,9 @@ void main() {
        timeFormat: UserTimeFormatPB.TwentyFourHour,
      );
      final now = DateTime.now();
-      final fifteenthOfLastMonth = DateTime(now.year, now.month - 1, 15);
+      final fifteenthOfNextMonth = DateTime(now.year, now.month + 1, 15);
      final formattedDate =
-          dateTimeSettings.dateFormat.formatDate(fifteenthOfLastMonth, false);
+          dateTimeSettings.dateFormat.formatDate(fifteenthOfNextMonth, false);

      expect(find.byType(MentionDateBlock), findsOneWidget);
      expect(find.text('@$formattedDate'), findsOneWidget);
@@ -378,11 +378,11 @@ void main() {
      await tester.tapButton(find.byType(MentionDateBlock));
      await tester.pumpAndSettle();

-      // set date to be fifteenth of the last month
+      // set date to be fifteenth of the next month
      await tester.tap(
        find.descendant(
          of: find.byType(DesktopAppFlowyDatePicker),
-          matching: find.byFlowySvg(FlowySvgs.arrow_left_s),
+          matching: find.byFlowySvg(FlowySvgs.arrow_right_s),
        ),
      );
      await tester.pumpAndSettle();
@@ -416,9 +416,9 @@ void main() {
        timeFormat: UserTimeFormatPB.TwentyFourHour,
      );
      final now = DateTime.now();
-      final fifteenthOfLastMonth = DateTime(now.year, now.month - 1, 15);
+      final fifteenthOfNextMonth = DateTime(now.year, now.month + 1, 15);
      final formattedDate =
-          dateTimeSettings.dateFormat.formatDate(fifteenthOfLastMonth, false);
+          dateTimeSettings.dateFormat.formatDate(fifteenthOfNextMonth, false);

      expect(find.byType(MentionDateBlock), findsOneWidget);
      expect(find.text('@$formattedDate'), findsOneWidget);
@@ -1,11 +1,9 @@
-import 'package:appflowy/generated/flowy_svgs.g.dart';
 import 'package:appflowy/generated/locale_keys.g.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/plugins.dart';
 import 'package:appflowy_editor/appflowy_editor.dart';
 import 'package:easy_localization/easy_localization.dart';
 import 'package:flowy_infra_ui/style_widget/button.dart';
 import 'package:flutter/material.dart';
-import 'package:flutter/services.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';

@@ -34,15 +32,9 @@ void main() {
        Selection.single(path: [0], startOffset: 0, endOffset: formula.length),
      );

-      // tap the more options button
-      final moreOptionButton = find.findFlowyTooltip(
-        LocaleKeys.document_toolbar_moreOptions.tr(),
-      );
-      await tester.tapButton(moreOptionButton);
-
      // tap the inline math equation button
-      final inlineMathEquationButton = find.text(
-        LocaleKeys.document_toolbar_equation.tr(),
+      final inlineMathEquationButton = find.findFlowyTooltip(
+        LocaleKeys.document_plugins_createInlineMathEquation.tr(),
      );
      await tester.tapButton(inlineMathEquationButton);

@@ -85,15 +77,10 @@ void main() {
        Selection.single(path: [0], startOffset: 0, endOffset: formula.length),
      );

-      // tap the more options button
-      final moreOptionButton = find.findFlowyTooltip(
-        LocaleKeys.document_toolbar_moreOptions.tr(),
-      );
-      await tester.tapButton(moreOptionButton);
-
      // tap the inline math equation button
-      final inlineMathEquationButton =
-          find.byFlowySvg(FlowySvgs.type_formula_m);
+      var inlineMathEquationButton = find.findFlowyTooltip(
+        LocaleKeys.document_plugins_createInlineMathEquation.tr(),
+      );
      await tester.tapButton(inlineMathEquationButton);

      // expect to see the math equation block
@@ -105,7 +92,17 @@ void main() {
        Selection.single(path: [0], startOffset: 0, endOffset: 1),
      );

-      await tester.tapButton(moreOptionButton);
+      // expect to the see the inline math equation button is highlighted
+      inlineMathEquationButton = find.descendant(
+        of: find.findFlowyTooltip(
+          LocaleKeys.document_plugins_createInlineMathEquation.tr(),
+        ),
+        matching: find.byType(SVGIconItemWidget),
+      );
+      expect(
+        tester.widget<SVGIconItemWidget>(inlineMathEquationButton).isHighlight,
+        isTrue,
+      );

      // cancel the format
      await tester.tapButton(inlineMathEquationButton);
@@ -136,15 +133,10 @@ void main() {
        Selection.single(path: [0], startOffset: 0, endOffset: formula.length),
      );

-      // tap the more options button
-      final moreOptionButton = find.findFlowyTooltip(
-        LocaleKeys.document_toolbar_moreOptions.tr(),
-      );
-      await tester.tapButton(moreOptionButton);
-
      // tap the inline math equation button
-      final inlineMathEquationButton =
-          find.byFlowySvg(FlowySvgs.type_formula_m);
+      final inlineMathEquationButton = find.findFlowyTooltip(
+        LocaleKeys.document_plugins_createInlineMathEquation.tr(),
+      );
      await tester.tapButton(inlineMathEquationButton);

      // expect to see the math equation block
@@ -171,55 +163,5 @@ void main() {
        lessThan(5),
      );
    });
-
-    testWidgets('insert inline math equation by shortcut', (tester) async {
-      await tester.initializeAppFlowy();
-      await tester.tapAnonymousSignInButton();
-
-      // create a new document
-      await tester.createNewPageWithNameUnderParent(
-        name: 'insert inline math equation by shortcut',
-      );
-
-      // tap the first line of the document
-      await tester.editor.tapLineOfEditorAt(0);
-      // insert a inline page
-      const formula = 'E = MC ^ 2';
-      await tester.ime.insertText(formula);
-      await tester.editor.updateSelection(
-        Selection.single(path: [0], startOffset: 0, endOffset: formula.length),
-      );
-
-      // mock key event
-      await tester.simulateKeyEvent(
-        LogicalKeyboardKey.keyE,
-        isShiftPressed: true,
-        isControlPressed: true,
-      );
-
-      // expect to see the math equation block
-      final inlineMathEquation = find.byType(InlineMathEquation);
-      expect(inlineMathEquation, findsOneWidget);
-
-      await tester.editor.tapLineOfEditorAt(0);
-      const text = 'Hello World';
-      await tester.ime.insertText(text);
-
-      final inlineText = find.textContaining(text, findRichText: true);
-      expect(inlineText, findsOneWidget);
-
-      // the text should be in the same line with the math equation
-      final inlineMathEquationPosition = tester.getRect(inlineMathEquation);
-      final textPosition = tester.getRect(inlineText);
-      // allow 5px difference
-      expect(
-        (textPosition.top - inlineMathEquationPosition.top).abs(),
-        lessThan(5),
-      );
-      expect(
-        (textPosition.bottom - inlineMathEquationPosition.bottom).abs(),
-        lessThan(5),
-      );
-    });
  });
 }
@@ -48,7 +48,7 @@ void main() {
      await tester.editor.showSlashMenu();
      await tester.editor.tapSlashMenuItemWithName(
        LocaleKeys.document_slashMenu_name_photoGallery.tr(),
-        offset: 100,
+        offset: 80,
      );
      expect(find.byType(MultiImageBlockComponent), findsOneWidget);
      expect(find.byType(MultiImagePlaceholder), findsOneWidget);
@@ -146,7 +146,7 @@ void main() {
      await tester.editor.showSlashMenu();
      await tester.editor.tapSlashMenuItemWithName(
        LocaleKeys.document_slashMenu_name_photoGallery.tr(),
-        offset: 100,
+        offset: 80,
      );
      expect(find.byType(MultiImageBlockComponent), findsOneWidget);
      expect(find.byType(MultiImagePlaceholder), findsOneWidget);
@@ -176,7 +176,6 @@ Future<void> insertOutlineInDocument(WidgetTester tester) async {
  await tester.editor.showSlashMenu();
  await tester.editor.tapSlashMenuItemWithName(
    LocaleKeys.document_slashMenu_name_outline.tr(),
-    offset: 180,
  );
  await tester.pumpAndSettle();
 }
@@ -1,4 +1,3 @@
-import 'package:appflowy/generated/flowy_svgs.g.dart';
 import 'package:appflowy/generated/locale_keys.g.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/plugins.dart';
 import 'package:appflowy_editor/appflowy_editor.dart';
@@ -86,10 +85,16 @@ void main() {
        ),
      );

-      await tester.tapButton(find.byFlowySvg(FlowySvgs.toolbar_text_format_m));
+      await tester.tapButton(find.byType(HeadingPopup));
+      await tester.pumpAndSettle();
+
+      expect(
+        find.byType(HeadingButton),
+        findsNWidgets(3),
+      );

      // tap the H1 button
-      await tester.tapButton(find.byFlowySvg(FlowySvgs.type_h1_m).at(0));
+      await tester.tapButton(find.byType(HeadingButton).at(0));
      await tester.pumpAndSettle();

      final editorState = tester.editor.getCurrentEditorState();
@@ -1,3 +1,5 @@
+import 'dart:io';
+
 import 'package:appflowy/plugins/base/emoji/emoji_picker.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/base/emoji_picker_button.dart';
 import 'package:appflowy/shared/icon_emoji_picker/flowy_icon_emoji_picker.dart';
@@ -5,9 +7,11 @@ import 'package:appflowy/shared/icon_emoji_picker/recent_icons.dart';
 import 'package:appflowy/workspace/presentation/widgets/view_title_bar.dart';
 import 'package:appflowy_backend/protobuf/flowy-folder/view.pb.dart';
 import 'package:flowy_infra_ui/style_widget/text_field.dart';
-import 'package:flowy_svg/flowy_svg.dart';
+import 'package:flutter/services.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';
+import 'package:path/path.dart' as p;
+import 'package:path_provider/path_provider.dart';

 import '../../shared/base.dart';
 import '../../shared/common_operations.dart';
@@ -211,12 +215,17 @@ void main() {
    }
  });

-  testWidgets('Update page custom image icon in title bar', (tester) async {
+  testWidgets('Update page custom icon in title bar', (tester) async {
    await tester.initializeAppFlowy();
    await tester.tapAnonymousSignInButton();

    /// prepare local image
-    final iconData = await tester.prepareImageIcon();
+    final imagePath = await rootBundle.load('assets/test/images/sample.jpeg');
+    final tempDirectory = await getTemporaryDirectory();
+    final localImagePath = p.join(tempDirectory.path, 'sample.jpeg');
+    final imageFile = File(localImagePath)
+      ..writeAsBytesSync(imagePath.buffer.asUint8List());
+    final iconData = EmojiIconData.custom(imageFile.path);

    // create document, board, grid and calendar views
    for (final value in ViewLayoutPB.values) {
@@ -250,97 +259,4 @@ void main() {
      );
    }
  });
-
-  testWidgets('Update page custom svg icon in title bar', (tester) async {
-    await tester.initializeAppFlowy();
-    await tester.tapAnonymousSignInButton();
-
-    /// prepare local image
-    final iconData = await tester.prepareSvgIcon();
-
-    // create document, board, grid and calendar views
-    for (final value in ViewLayoutPB.values) {
-      if (value == ViewLayoutPB.Chat) {
-        continue;
-      }
-
-      await tester.createNewPageWithNameUnderParent(
-        name: value.name,
-        parentName: gettingStarted,
-        layout: value,
-      );
-
-      // update its icon
-      await tester.updatePageIconInTitleBarByName(
-        name: value.name,
-        layout: value,
-        icon: iconData,
-      );
-
-      tester.expectViewHasIcon(
-        value.name,
-        value,
-        iconData,
-      );
-
-      tester.expectViewTitleHasIcon(
-        value.name,
-        value,
-        iconData,
-      );
-    }
-  });
-
-  testWidgets('Update page custom svg icon in title bar by pasting a link',
-      (tester) async {
-    await tester.initializeAppFlowy();
-    await tester.tapAnonymousSignInButton();
-
-    /// prepare local image
-    const testIconLink =
-        'https://beta.appflowy.cloud/api/file_storage/008e6f23-516b-4d8d-b1fe-2b75c51eee26/v1/blob/6bdf8dff%2D0e54%2D4d35%2D9981%2Dcde68bef1141/BGpLnRtb3AGBNgSJsceu70j83zevYKrMLzqsTIJcBeI=.svg';
-
-    /// create document, board, grid and calendar views
-    for (final value in ViewLayoutPB.values) {
-      if (value == ViewLayoutPB.Chat) {
-        continue;
-      }
-
-      await tester.createNewPageWithNameUnderParent(
-        name: value.name,
-        parentName: gettingStarted,
-        layout: value,
-      );
-
-      /// update its icon
-      await tester.updatePageIconInTitleBarByPasteALink(
-        name: value.name,
-        layout: value,
-        iconLink: testIconLink,
-      );
-
-      /// check if there is a svg in page
-      final pageName = tester.findPageName(
-        value.name,
-        layout: value,
-      );
-      final imageInPage = find.descendant(
-        of: pageName,
-        matching: find.byType(SvgPicture),
-      );
-      expect(imageInPage, findsOneWidget);
-
-      /// check if there is a svg in title
-      final imageInTitle = find.descendant(
-        of: find.byType(ViewTitleBar),
-        matching: find.byWidgetPredicate((w) {
-          if (w is! SvgPicture) return false;
-          final loader = w.bytesLoader;
-          if (loader is! SvgFileLoader) return false;
-          return loader.file.path.endsWith('.svg');
-        }),
-      );
-      expect(imageInTitle, findsOneWidget);
-    }
-  });
 }
@@ -1,166 +1,42 @@
 import 'dart:io';

-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/plugins/emoji/emoji_handler.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:flowy_infra_ui/flowy_infra_ui.dart';
+import 'package:appflowy/workspace/presentation/settings/widgets/emoji_picker/emoji_picker.dart';
+import 'package:appflowy_editor/appflowy_editor.dart';
+import 'package:appflowy_editor/src/editor/editor_component/service/editor.dart';
 import 'package:flutter/services.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';
+
+import '../../shared/keyboard.dart';
 import '../../shared/util.dart';

 void main() {
  IntegrationTestWidgetsFlutterBinding.ensureInitialized();
-
-  Future<void> prepare(WidgetTester tester) async {
-    await tester.initializeAppFlowy();
-    await tester.tapAnonymousSignInButton();
-    await tester.createNewPageWithNameUnderParent();
-    await tester.editor.tapLineOfEditorAt(0);
-  }
-
  // May be better to move this to an existing test but unsure what it fits with
  group('Keyboard shortcuts related to emojis', () {
    testWidgets('cmd/ctrl+alt+e shortcut opens the emoji picker',
        (tester) async {
-      await prepare(tester);
+      await tester.initializeAppFlowy();
+      await tester.tapAnonymousSignInButton();

-      expect(find.byType(EmojiHandler), findsNothing);
+      final Finder editor = find.byType(AppFlowyEditor);
+      await tester.tap(editor);
+      await tester.pumpAndSettle();

-      await tester.simulateKeyEvent(
-        LogicalKeyboardKey.keyE,
-        isAltPressed: true,
-        isMetaPressed: Platform.isMacOS,
-        isControlPressed: !Platform.isMacOS,
+      expect(find.byType(EmojiSelectionMenu), findsNothing);
+
+      await FlowyTestKeyboard.simulateKeyDownEvent(
+        [
+          Platform.isMacOS
+              ? LogicalKeyboardKey.meta
+              : LogicalKeyboardKey.control,
+          LogicalKeyboardKey.alt,
+          LogicalKeyboardKey.keyE,
+        ],
+        tester: tester,
      );
-      await tester.pumpAndSettle(Duration(seconds: 1));
-      expect(find.byType(EmojiHandler), findsOneWidget);

-      /// press backspace to hide the emoji picker
-      await tester.simulateKeyEvent(LogicalKeyboardKey.backspace);
-      expect(find.byType(EmojiHandler), findsNothing);
-    });
-
-    testWidgets('insert emoji by slash menu', (tester) async {
-      await prepare(tester);
-      await tester.editor.showSlashMenu();
-
-      /// show emoji picler
-      await tester.editor.tapSlashMenuItemWithName(
-        LocaleKeys.document_slashMenu_name_emoji.tr(),
-        offset: 100,
-      );
-      await tester.pumpAndSettle(Duration(seconds: 1));
-      expect(find.byType(EmojiHandler), findsOneWidget);
-      await tester.simulateKeyEvent(LogicalKeyboardKey.enter);
-      final firstNode =
-          tester.editor.getCurrentEditorState().getNodeAtPath([0])!;
-
-      /// except the emoji is in document
-      expect(firstNode.delta!.toPlainText().contains('😀'), true);
-    });
-  });
-
-  group('insert emoji by colon', () {
-    Future<void> createNewDocumentAndShowEmojiList(
-      WidgetTester tester, {
-      String? search,
-    }) async {
-      await prepare(tester);
-      await tester.ime.insertText(':${search ?? 'a'}');
-      await tester.pumpAndSettle(Duration(seconds: 1));
-    }
-
-    testWidgets('insert with click', (tester) async {
-      await createNewDocumentAndShowEmojiList(tester);
-
-      /// emoji list is showing
-      final emojiHandler = find.byType(EmojiHandler);
-      expect(emojiHandler, findsOneWidget);
-      final emojiButtons =
-          find.descendant(of: emojiHandler, matching: find.byType(FlowyButton));
-      final firstTextFinder = find.descendant(
-        of: emojiButtons.first,
-        matching: find.byType(FlowyText),
-      );
-      final emojiText =
-          (firstTextFinder.evaluate().first.widget as FlowyText).text;
-
-      /// click first emoji item
-      await tester.tapButton(emojiButtons.first);
-      final firstNode =
-          tester.editor.getCurrentEditorState().getNodeAtPath([0])!;
-
-      /// except the emoji is in document
-      expect(emojiText.contains(firstNode.delta!.toPlainText()), true);
-    });
-
-    testWidgets('insert with arrow and enter', (tester) async {
-      await createNewDocumentAndShowEmojiList(tester);
-
-      /// emoji list is showing
-      final emojiHandler = find.byType(EmojiHandler);
-      expect(emojiHandler, findsOneWidget);
-      final emojiButtons =
-          find.descendant(of: emojiHandler, matching: find.byType(FlowyButton));
-
-      /// tap arrow down and arrow up
-      await tester.simulateKeyEvent(LogicalKeyboardKey.arrowUp);
-      await tester.simulateKeyEvent(LogicalKeyboardKey.arrowDown);
-
-      final firstTextFinder = find.descendant(
-        of: emojiButtons.first,
-        matching: find.byType(FlowyText),
-      );
-      final emojiText =
-          (firstTextFinder.evaluate().first.widget as FlowyText).text;
-
-      /// tap enter
-      await tester.simulateKeyEvent(LogicalKeyboardKey.enter);
-      final firstNode =
-          tester.editor.getCurrentEditorState().getNodeAtPath([0])!;
-
-      /// except the emoji is in document
-      expect(emojiText.contains(firstNode.delta!.toPlainText()), true);
-    });
-
-    testWidgets('insert with searching', (tester) async {
-      await createNewDocumentAndShowEmojiList(tester, search: 's');
-
-      /// search for `smiling eyes`, IME is not working, use keyboard input
-      final searchText = [
-        LogicalKeyboardKey.keyM,
-        LogicalKeyboardKey.keyI,
-        LogicalKeyboardKey.keyL,
-        LogicalKeyboardKey.keyI,
-        LogicalKeyboardKey.keyN,
-        LogicalKeyboardKey.keyG,
-        LogicalKeyboardKey.space,
-        LogicalKeyboardKey.keyE,
-        LogicalKeyboardKey.keyY,
-        LogicalKeyboardKey.keyE,
-        LogicalKeyboardKey.keyS,
-      ];
-
-      for (final key in searchText) {
-        await tester.simulateKeyEvent(key);
-      }
-
-      /// tap enter
-      await tester.simulateKeyEvent(LogicalKeyboardKey.enter);
-      final firstNode =
-          tester.editor.getCurrentEditorState().getNodeAtPath([0])!;
-
-      /// except the emoji is in document
-      expect(firstNode.delta!.toPlainText().contains('😄'), true);
-    });
-
-    testWidgets('start searching with sapce', (tester) async {
-      await createNewDocumentAndShowEmojiList(tester, search: ' ');
-
-      /// emoji list is showing
-      final emojiHandler = find.byType(EmojiHandler);
-      expect(emojiHandler, findsNothing);
+      expect(find.byType(EmojiSelectionMenu), findsOneWidget);
    });
  });
 }
@@ -1,12 +1,13 @@
 import 'dart:io';

+import 'package:flutter/material.dart';
+import 'package:flutter/services.dart';
+
 import 'package:appflowy/generated/locale_keys.g.dart';
 import 'package:appflowy/workspace/application/settings/prelude.dart';
 import 'package:appflowy/workspace/presentation/home/menu/sidebar/sidebar.dart';
 import 'package:appflowy/workspace/presentation/settings/settings_dialog.dart';
 import 'package:easy_localization/easy_localization.dart';
-import 'package:flutter/material.dart';
-import 'package:flutter/services.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';

@@ -37,7 +38,7 @@ void main() {
          LocaleKeys.settings_workspacePage_appearance_options_light.tr(),
        ),
      );
-      await tester.pumpAndSettle(const Duration(milliseconds: 250));
+      await tester.pumpAndSettle();

      themeMode = tester.widget<MaterialApp>(appFinder).themeMode;
      expect(themeMode, ThemeMode.light);
@@ -47,7 +48,7 @@ void main() {
          LocaleKeys.settings_workspacePage_appearance_options_dark.tr(),
        ),
      );
-      await tester.pumpAndSettle(const Duration(milliseconds: 250));
+      await tester.pumpAndSettle();

      themeMode = tester.widget<MaterialApp>(appFinder).themeMode;
      expect(themeMode, ThemeMode.dark);
@@ -65,11 +66,10 @@ void main() {
        ],
        tester: tester,
      );
-      await tester.pumpAndSettle(const Duration(milliseconds: 500));
+      await tester.pumpAndSettle();

-      // disable it temporarily. It works on macOS but not on Linux.
-      // themeMode = tester.widget<MaterialApp>(appFinder).themeMode;
-      // expect(themeMode, ThemeMode.light);
+      themeMode = tester.widget<MaterialApp>(appFinder).themeMode;
+      expect(themeMode, ThemeMode.light);
    });

    testWidgets('show or hide home menu', (tester) async {
@@ -1,16 +1,12 @@
-import 'dart:convert';
 import 'dart:io';

 import 'package:appflowy/plugins/shared/share/share_button.dart';
-import 'package:appflowy_backend/protobuf/flowy-folder/view.pbenum.dart';
-import 'package:archive/archive.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';
 import 'package:path/path.dart' as p;

 import '../../shared/mock/mock_file_picker.dart';
 import '../../shared/util.dart';
-import '../document/document_with_database_test.dart';

 void main() {
  IntegrationTestWidgetsFlutterBinding.ensureInitialized();
@@ -22,7 +18,7 @@ void main() {

      // mock the file picker
      final path = await mockSaveFilePath(
-        p.join(context.applicationDataDirectory, 'test.zip'),
+        p.join(context.applicationDataDirectory, 'test.md'),
      );
      // click the share button and select markdown
      await tester.tapShareButton();
@@ -32,14 +28,10 @@ void main() {
      tester.expectToExportSuccess();

      final file = File(path);
-      expect(file.existsSync(), true);
-      final archive = ZipDecoder().decodeBytes(file.readAsBytesSync());
-      for (final entry in archive) {
-        if (entry.isFile && entry.name.endsWith('.md')) {
-          final markdown = utf8.decode(entry.content);
-          expect(markdown, expectedMarkdown);
-        }
-      }
+      final isExist = file.existsSync();
+      expect(isExist, true);
+      final markdown = file.readAsStringSync();
+      expect(markdown, expectedMarkdown);
    });

    testWidgets(
@@ -65,7 +57,7 @@ void main() {
        final path = await mockSaveFilePath(
          p.join(
            context.applicationDataDirectory,
-            '${shareButtonState.view.name}.zip',
+            '${shareButtonState.view.name}.md',
          ),
        );

@@ -77,44 +69,10 @@ void main() {
        tester.expectToExportSuccess();

        final file = File(path);
-        expect(file.existsSync(), true);
-        final archive = ZipDecoder().decodeBytes(file.readAsBytesSync());
-        for (final entry in archive) {
-          if (entry.isFile && entry.name.endsWith('.md')) {
-            final markdown = utf8.decode(entry.content);
-            expect(markdown, expectedMarkdown);
-          }
-        }
+        final isExist = file.existsSync();
+        expect(isExist, true);
      },
    );
-
-    testWidgets('share the markdown with database', (tester) async {
-      final context = await tester.initializeAppFlowy();
-      await tester.tapAnonymousSignInButton();
-      await insertLinkedDatabase(tester, ViewLayoutPB.Grid);
-
-      // mock the file picker
-      final path = await mockSaveFilePath(
-        p.join(context.applicationDataDirectory, 'test.zip'),
-      );
-      // click the share button and select markdown
-      await tester.tapShareButton();
-      await tester.tapMarkdownButton();
-
-      // expect to see the success dialog
-      tester.expectToExportSuccess();
-
-      final file = File(path);
-      expect(file.existsSync(), true);
-      final archive = ZipDecoder().decodeBytes(file.readAsBytesSync());
-      bool hasCsvFile = false;
-      for (final entry in archive) {
-        if (entry.isFile && entry.name.endsWith('.csv')) {
-          hasCsvFile = true;
-        }
-      }
-      expect(hasCsvFile, true);
-    });
  });
 }

@@ -1,8 +1,13 @@
 import 'dart:io';

+import 'package:appflowy/startup/startup.dart';
+import 'package:appflowy/startup/tasks/prelude.dart';
+import 'package:appflowy/workspace/application/settings/prelude.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';

+import '../../shared/util.dart';
+
 void main() {
  IntegrationTestWidgetsFlutterBinding.ensureInitialized();

@@ -86,24 +91,23 @@ void main() {
    //   }
    // });

-    // Disable this test because it failed after executing.
-    // testWidgets('reset to default location', (tester) async {
-    //   await tester.initializeAppFlowy();
+    testWidgets('reset to default location', (tester) async {
+      await tester.initializeAppFlowy();

-    //   await tester.tapAnonymousSignInButton();
+      await tester.tapAnonymousSignInButton();

-    //   // home and readme document
-    //   await tester.expectToSeeHomePageWithGetStartedPage();
+      // home and readme document
+      await tester.expectToSeeHomePageWithGetStartedPage();

-    //   // open settings and restore the location
-    //   await tester.openSettings();
-    //   await tester.openSettingsPage(SettingsPage.manageData);
-    //   await tester.restoreLocation();
+      // open settings and restore the location
+      await tester.openSettings();
+      await tester.openSettingsPage(SettingsPage.manageData);
+      await tester.restoreLocation();

-    //   expect(
-    //     await appFlowyApplicationDataDirectory().then((value) => value.path),
-    //     await getIt<ApplicationDataStorage>().getPath(),
-    //   );
-    // });
+      expect(
+        await appFlowyApplicationDataDirectory().then((value) => value.path),
+        await getIt<ApplicationDataStorage>().getPath(),
+      );
+    });
  });
 }
@@ -13,6 +13,7 @@ void main() {
  hotkeys_test.main();
  emoji_shortcut_test.main();
  hotkeys_test.main();
+  emoji_shortcut_test.main();
  share_markdown_test.main();
  import_files_test.main();
  zoom_in_out_test.main();
@@ -1,56 +0,0 @@
-import 'package:appflowy/mobile/presentation/inline_actions/mobile_inline_actions_menu.dart';
-import 'package:appflowy/mobile/presentation/inline_actions/mobile_inline_actions_menu_group.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/mention/mention_page_block.dart';
-import 'package:flutter_test/flutter_test.dart';
-import 'package:integration_test/integration_test.dart';
-
-import '../../shared/util.dart';
-
-void main() {
-  IntegrationTestWidgetsFlutterBinding.ensureInitialized();
-
-  const title = 'Test At Menu';
-
-  group('at menu', () {
-    testWidgets('show at menu', (tester) async {
-      await tester.launchInAnonymousMode();
-      await tester.createPageAndShowAtMenu(title);
-      final menuWidget = find.byType(MobileInlineActionsMenu);
-      expect(menuWidget, findsOneWidget);
-    });
-
-    testWidgets('search by at menu', (tester) async {
-      await tester.launchInAnonymousMode();
-      await tester.createPageAndShowAtMenu(title);
-      const searchText = gettingStarted;
-      await tester.ime.insertText(searchText);
-      final actionWidgets = find.byType(MobileInlineActionsWidget);
-      expect(actionWidgets, findsNWidgets(2));
-    });
-
-    testWidgets('tap at menu', (tester) async {
-      await tester.launchInAnonymousMode();
-      await tester.createPageAndShowAtMenu(title);
-      const searchText = gettingStarted;
-      await tester.ime.insertText(searchText);
-      final actionWidgets = find.byType(MobileInlineActionsWidget);
-      await tester.tap(actionWidgets.last);
-      expect(find.byType(MentionPageBlock), findsOneWidget);
-    });
-
-    testWidgets('create subpage with at menu', (tester) async {
-      await tester.launchInAnonymousMode();
-      await tester.createNewDocumentOnMobile(title);
-      await tester.editor.tapLineOfEditorAt(0);
-      const subpageName = 'Subpage';
-      await tester.ime.insertText('[[$subpageName');
-      await tester.pumpAndSettle();
-      final actionWidgets = find.byType(MobileInlineActionsWidget);
-      await tester.tapButton(actionWidgets.first);
-      final firstNode =
-          tester.editor.getCurrentEditorState().getNodeAtPath([0]);
-      assert(firstNode != null);
-      expect(firstNode!.delta?.toPlainText().contains('['), false);
-    });
-  });
-}
@@ -1,11 +1,8 @@
 import 'package:integration_test/integration_test.dart';

-import 'at_menu_test.dart' as at_menu;
-import 'at_menu_test.dart' as at_menu_test;
 import 'page_style_test.dart' as page_style_test;
 import 'plus_menu_test.dart' as plus_menu_test;
 import 'simple_table_test.dart' as simple_table_test;
-import 'slash_menu_test.dart' as slash_menu;
 import 'title_test.dart' as title_test;
 import 'toolbar_test.dart' as toolbar_test;

@@ -16,9 +13,6 @@ void main() {
  title_test.main();
  page_style_test.main();
  plus_menu_test.main();
-  at_menu_test.main();
  simple_table_test.main();
  toolbar_test.main();
-  slash_menu.main();
-  at_menu.main();
 }
@@ -1,12 +1,15 @@
-import 'package:appflowy/generated/flowy_svgs.g.dart';
+import 'dart:io';
+
 import 'package:appflowy/mobile/presentation/base/view_page/app_bar_buttons.dart';
 import 'package:appflowy/mobile/presentation/presentation.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/header/emoji_icon_widget.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/page_style/_page_style_icon.dart';
 import 'package:appflowy/shared/icon_emoji_picker/flowy_icon_emoji_picker.dart';
-import 'package:flutter/material.dart';
+import 'package:flutter/services.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';
+import 'package:path/path.dart' as p;
+import 'package:path_provider/path_provider.dart';

 import '../../shared/emoji.dart';
 import '../../shared/util.dart';
@@ -15,11 +18,16 @@ void main() {
  IntegrationTestWidgetsFlutterBinding.ensureInitialized();

  group('document title:', () {
-    testWidgets('update page custom image icon in title bar', (tester) async {
+    testWidgets('update page custom icon in title bar', (tester) async {
      await tester.launchInAnonymousMode();

      /// prepare local image
-      final iconData = await tester.prepareImageIcon();
+      final imagePath = await rootBundle.load('assets/test/images/sample.jpeg');
+      final tempDirectory = await getTemporaryDirectory();
+      final localImagePath = p.join(tempDirectory.path, 'sample.jpeg');
+      final imageFile = File(localImagePath)
+        ..writeAsBytesSync(imagePath.buffer.asUint8List());
+      final iconData = EmojiIconData.custom(imageFile.path);

      /// create an empty page
      await tester
@@ -42,63 +50,16 @@ void main() {

      /// check result
      final documentPage = find.byType(MobileDocumentScreen);
-      final rawEmojiIconFinder = find
+      final rawEmojiIconWidget = find
          .descendant(
            of: documentPage,
            matching: find.byType(RawEmojiIconWidget),
          )
-          .last;
-      final rawEmojiIconWidget =
-          rawEmojiIconFinder.evaluate().first.widget as RawEmojiIconWidget;
+          .evaluate()
+          .first
+          .widget as RawEmojiIconWidget;
      final iconDataInWidget = rawEmojiIconWidget.emoji;
      expect(iconDataInWidget.type, FlowyIconType.custom);
-      final imageFinder =
-          find.descendant(of: rawEmojiIconFinder, matching: find.byType(Image));
-      expect(imageFinder, findsOneWidget);
-    });
-
-    testWidgets('update page custom svg icon in title bar', (tester) async {
-      await tester.launchInAnonymousMode();
-
-      /// prepare local image
-      final iconData = await tester.prepareSvgIcon();
-
-      /// create an empty page
-      await tester
-          .tapButton(find.byKey(BottomNavigationBarItemType.add.valueKey));
-
-      /// show Page style page
-      await tester.tapButton(find.byType(MobileViewPageLayoutButton));
-      final pageStyleIcon = find.byType(PageStyleIcon);
-      final iconInPageStyleIcon = find.descendant(
-        of: pageStyleIcon,
-        matching: find.byType(RawEmojiIconWidget),
-      );
-      expect(iconInPageStyleIcon, findsNothing);
-
-      /// show icon picker
-      await tester.tapButton(pageStyleIcon);
-
-      /// upload custom icon
-      await tester.pickImage(iconData);
-
-      /// check result
-      final documentPage = find.byType(MobileDocumentScreen);
-      final rawEmojiIconFinder = find
-          .descendant(
-            of: documentPage,
-            matching: find.byType(RawEmojiIconWidget),
-          )
-          .last;
-      final rawEmojiIconWidget =
-          rawEmojiIconFinder.evaluate().first.widget as RawEmojiIconWidget;
-      final iconDataInWidget = rawEmojiIconWidget.emoji;
-      expect(iconDataInWidget.type, FlowyIconType.custom);
-      final svgFinder = find.descendant(
-        of: rawEmojiIconFinder,
-        matching: find.byType(SvgPicture),
-      );
-      expect(svgFinder, findsOneWidget);
    });
  });
 }
@@ -1,9 +1,6 @@
 import 'dart:async';

 import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/mobile/presentation/inline_actions/mobile_inline_actions_menu.dart';
-import 'package:appflowy/mobile/presentation/inline_actions/mobile_inline_actions_menu_group.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/mention/mention_page_block.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/plugins.dart';
 import 'package:appflowy_editor/appflowy_editor.dart';
 import 'package:easy_localization/easy_localization.dart';
@@ -88,32 +85,5 @@ void main() {
        equals(Selection.collapsed(Position(path: [2]))),
      );
    });
-
-    const title = 'Test Plus Menu';
-    testWidgets('show plus menu', (tester) async {
-      await tester.launchInAnonymousMode();
-      await tester.createPageAndShowPlusMenu(title);
-      final menuWidget = find.byType(MobileInlineActionsMenu);
-      expect(menuWidget, findsOneWidget);
-    });
-
-    testWidgets('search by plus menu', (tester) async {
-      await tester.launchInAnonymousMode();
-      await tester.createPageAndShowPlusMenu(title);
-      const searchText = gettingStarted;
-      await tester.ime.insertText(searchText);
-      final actionWidgets = find.byType(MobileInlineActionsWidget);
-      expect(actionWidgets, findsNWidgets(2));
-    });
-
-    testWidgets('tap plus menu', (tester) async {
-      await tester.launchInAnonymousMode();
-      await tester.createPageAndShowPlusMenu(title);
-      const searchText = gettingStarted;
-      await tester.ime.insertText(searchText);
-      final actionWidgets = find.byType(MobileInlineActionsWidget);
-      await tester.tap(actionWidgets.last);
-      expect(find.byType(MentionPageBlock), findsOneWidget);
-    });
  });
 }
@@ -1,84 +0,0 @@
-import 'package:appflowy/mobile/presentation/selection_menu/mobile_selection_menu_item.dart';
-import 'package:appflowy/mobile/presentation/selection_menu/mobile_selection_menu_item_widget.dart';
-import 'package:appflowy/mobile/presentation/selection_menu/mobile_selection_menu_widget.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/slash_menu/slash_menu_items/mobile_items.dart';
-import 'package:flutter/cupertino.dart';
-import 'package:flutter_test/flutter_test.dart';
-import 'package:integration_test/integration_test.dart';
-
-import '../../shared/util.dart';
-
-void main() {
-  IntegrationTestWidgetsFlutterBinding.ensureInitialized();
-
-  const title = 'Test Slash Menu';
-
-  group('slash menu', () {
-    testWidgets('show slash menu', (tester) async {
-      await tester.launchInAnonymousMode();
-      await tester.createPageAndShowSlashMenu(title);
-      final menuWidget = find.byType(MobileSelectionMenuWidget);
-      expect(menuWidget, findsOneWidget);
-      final items =
-          (menuWidget.evaluate().first.widget as MobileSelectionMenuWidget)
-              .items;
-      int i = 0;
-      for (final item in items) {
-        final localItem = mobileItems[i];
-        expect(item.name, localItem.name);
-        i++;
-      }
-    });
-
-    testWidgets('search by slash menu', (tester) async {
-      await tester.launchInAnonymousMode();
-      await tester.createPageAndShowSlashMenu(title);
-      const searchText = 'Heading';
-      await tester.ime.insertText(searchText);
-      final itemWidgets = find.byType(MobileSelectionMenuItemWidget);
-      int number = 0;
-      for (final item in mobileItems) {
-        if (item is MobileSelectionMenuItem) {
-          for (final childItem in item.children) {
-            if (childItem.name
-                .toLowerCase()
-                .contains(searchText.toLowerCase())) {
-              number++;
-            }
-          }
-        } else {
-          if (item.name.toLowerCase().contains(searchText.toLowerCase())) {
-            number++;
-          }
-        }
-      }
-      expect(itemWidgets, findsNWidgets(number));
-    });
-
-    testWidgets('tap to show submenu', (tester) async {
-      await tester.launchInAnonymousMode();
-      await tester.createNewDocumentOnMobile(title);
-      await tester.editor.tapLineOfEditorAt(0);
-      final listview = find.descendant(
-        of: find.byType(MobileSelectionMenuWidget),
-        matching: find.byType(ListView),
-      );
-      for (final item in mobileItems) {
-        if (item is! MobileSelectionMenuItem) continue;
-        await tester.editor.showSlashMenu();
-        await tester.scrollUntilVisible(
-          find.text(item.name),
-          50,
-          scrollable: listview,
-          duration: const Duration(milliseconds: 250),
-        );
-        await tester.tap(find.text(item.name));
-        final childrenLength = ((listview.evaluate().first.widget as ListView)
-                .childrenDelegate as SliverChildListDelegate)
-            .children
-            .length;
-        expect(childrenLength, item.children.length);
-      }
-    });
-  });
-}
@@ -1,11 +1,12 @@
 import 'package:appflowy/generated/flowy_svgs.g.dart';
 import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/mobile/presentation/bottom_sheet/bottom_sheet_edit_link_widget.dart';
 import 'package:appflowy/mobile/presentation/editor/mobile_editor_screen.dart';
 import 'package:appflowy/mobile/presentation/mobile_bottom_navigation_bar.dart';
+import 'package:appflowy/plugins/document/presentation/editor_plugins/mobile_toolbar_v3/appflowy_mobile_toolbar_item.dart';
+import 'package:appflowy/plugins/document/presentation/editor_plugins/mobile_toolbar_v3/util.dart';
 import 'package:appflowy_editor/appflowy_editor.dart';
 import 'package:easy_localization/easy_localization.dart';
-import 'package:flutter/material.dart';
+import 'package:flowy_infra_ui/style_widget/text_field.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:integration_test/integration_test.dart';

@@ -14,125 +15,102 @@ import '../../shared/util.dart';
 void main() {
  IntegrationTestWidgetsFlutterBinding.ensureInitialized();

-  Future<void> createNeaPage(WidgetTester tester) async {
-    final createPageButton =
-        find.byKey(BottomNavigationBarItemType.add.valueKey);
-    await tester.tapButton(createPageButton);
-    expect(find.byType(MobileDocumentScreen), findsOneWidget);
-    final editor = find.byType(AppFlowyEditor);
-    expect(editor, findsOneWidget);
-  }
-
-  const testLink = 'https://appflowy.io/';
-
-  group('links', () {
+  group('toolbar menu:', () {
    testWidgets('insert links', (tester) async {
      await tester.launchInAnonymousMode();

-      await createNeaPage(tester);
-      await tester.editor.tapLineOfEditorAt(0);
+      final createPageButton = find.byKey(
+        BottomNavigationBarItemType.add.valueKey,
+      );
+      await tester.tapButton(createPageButton);
+      expect(find.byType(MobileDocumentScreen), findsOneWidget);
+
+      final editor = find.byType(AppFlowyEditor);
+      expect(editor, findsOneWidget);
      final editorState = tester.editor.getCurrentEditorState();

+      /// move cursor to content
+      final root = editorState.document.root;
+      final lastNode = root.children.lastOrNull;
+      await editorState.updateSelectionWithReason(
+        Selection.collapsed(Position(path: lastNode!.path)),
+      );
+      await tester.pumpAndSettle();
+
      /// insert two lines of text
-      const strFirst = 'FirstLine', strSecond = 'SecondLine';
-      await tester.ime.insertText(strFirst);
+      const strFirst = 'FirstLine',
+          strSecond = 'SecondLine',
+          link = 'google.com';
+      await editorState.insertTextAtCurrentSelection(strFirst);
+      await tester.pumpAndSettle();
      await editorState.insertNewLine();
-      await tester.ime.insertText(strSecond);
+      await tester.pumpAndSettle();
+      await editorState.insertTextAtCurrentSelection(strSecond);
+      await tester.pumpAndSettle();
      final firstLine = find.text(strFirst, findRichText: true),
          secondLine = find.text(strSecond, findRichText: true);
      expect(firstLine, findsOneWidget);
      expect(secondLine, findsOneWidget);

      /// select the first line
-      await tester.doubleTapAt(tester.getCenter(firstLine));
+      await tester.longPress(firstLine);
      await tester.pumpAndSettle();

+      /// find aa item and tap it
+      final aaItem = find.byWidgetPredicate(
+        (widget) =>
+            widget is AppFlowyMobileToolbarIconItem &&
+            widget.icon == FlowySvgs.m_toolbar_aa_m,
+      );
+      expect(aaItem, findsOneWidget);
+      await tester.tapButton(aaItem);
+
      /// find link button and tap it
-      final linkButton = find.byFlowySvg(FlowySvgs.toolbar_link_m);
+      final linkButton = find.byWidgetPredicate(
+        (widget) =>
+            widget is MobileToolbarMenuItemWrapper &&
+            widget.icon == FlowySvgs.m_toolbar_link_m,
+      );
+      expect(linkButton, findsOneWidget);
      await tester.tapButton(linkButton);

      /// input the link
-      final textFormField = find.byType(TextFormField);
-      expect(textFormField, findsNWidgets(2));
-      final linkField = textFormField.last;
-      await tester.enterText(linkField, testLink);
+      final linkField = find.byWidgetPredicate(
+        (w) =>
+            w is FlowyTextField &&
+            w.hintText == LocaleKeys.document_inlineLink_url_placeholder.tr(),
+      );
+      await tester.enterText(linkField, link);
      await tester.pumpAndSettle();
-      await tester.tapButton(find.byFlowySvg(FlowySvgs.toolbar_link_earth_m));

-      /// apply the link
+      /// complete inputting
      await tester.tapButton(find.text(LocaleKeys.button_done.tr()));

      /// do it again
      /// select the second line
-      await tester.doubleTapAt(tester.getCenter(secondLine));
+      await tester.longPress(secondLine);
      await tester.pumpAndSettle();
+      await tester.tapButton(aaItem);
      await tester.tapButton(linkButton);
-      await tester.enterText(linkField, testLink);
+      await tester.enterText(linkField, link);
      await tester.pumpAndSettle();
-      await tester.tapButton(find.byFlowySvg(FlowySvgs.toolbar_link_earth_m));
      await tester.tapButton(find.text(LocaleKeys.button_done.tr()));

      final firstNode = editorState.getNodeAtPath([0]);
      final secondNode = editorState.getNodeAtPath([1]);

      Map commonDeltaJson(String insert) => {
-            'insert': insert,
-            'attributes': {'href': testLink, 'is_page_link': false},
+            "insert": insert,
+            "attributes": {"href": link},
          };

      expect(
        firstNode?.delta?.toJson(),
-        [commonDeltaJson(strFirst)],
+        commonDeltaJson(strFirst),
      );
      expect(
        secondNode?.delta?.toJson(),
-        [commonDeltaJson(strSecond)],
-      );
-    });
-
-    testWidgets('change a link', (tester) async {
-      await tester.launchInAnonymousMode();
-
-      await createNeaPage(tester);
-      await tester.editor.tapLineOfEditorAt(0);
-      final editorState = tester.editor.getCurrentEditorState();
-      const testText = 'TestText';
-      await tester.ime.insertText(testText);
-      final textFinder = find.text(testText, findRichText: true);
-
-      /// select the first line
-      await tester.doubleTapAt(tester.getCenter(textFinder));
-      await tester.pumpAndSettle();
-
-      /// find link button and tap it
-      final linkButton = find.byFlowySvg(FlowySvgs.toolbar_link_m);
-      await tester.tapButton(linkButton);
-
-      /// input the link
-      final textFormField = find.byType(TextFormField);
-      expect(textFormField, findsNWidgets(2));
-      final linkField = textFormField.last;
-      await tester.enterText(linkField, testLink);
-      await tester.pumpAndSettle();
-      await tester.tapButton(find.byFlowySvg(FlowySvgs.toolbar_link_earth_m));
-
-      /// apply the link
-      await tester.tapButton(find.text(LocaleKeys.button_done.tr()));
-
-      /// show edit link menu
-      await tester.longPress(textFinder);
-      await tester.pumpAndSettle();
-      final linkEditMenu = find.byType(MobileBottomSheetEditLinkWidget);
-      expect(linkEditMenu, findsOneWidget);
-
-      /// remove the link
-      await tester.tapButton(find.byFlowySvg(FlowySvgs.toolbar_link_unlink_m));
-      final node = editorState.getNodeAtPath([0]);
-      expect(
-        node?.delta?.toJson(),
-        [
-          {'insert': testText},
-        ],
+        commonDeltaJson(strSecond),
      );
    });
  });
@@ -1,59 +0,0 @@
-import 'package:appflowy/generated/flowy_svgs.g.dart';
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/mobile/presentation/search/mobile_search_ask_ai_entrance.dart';
-import 'package:appflowy/mobile/presentation/search/mobile_search_result.dart';
-import 'package:appflowy/mobile/presentation/search/mobile_search_textfield.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:flutter/material.dart';
-import 'package:flutter_test/flutter_test.dart';
-import 'package:integration_test/integration_test.dart';
-
-import '../../shared/util.dart';
-
-void main() {
-  IntegrationTestWidgetsFlutterBinding.ensureInitialized();
-
-  group('Search test', () {
-    testWidgets('tap to search page', (tester) async {
-      await tester.launchInAnonymousMode();
-      final searchButton = find.byFlowySvg(FlowySvgs.m_home_search_icon_m);
-      await tester.tapButton(searchButton);
-
-      ///check for UI element
-      expect(find.byType(MobileSearchAskAiEntrance), findsNothing);
-      expect(find.byType(MobileSearchRecentList), findsOneWidget);
-      expect(find.byType(MobileSearchResultList), findsNothing);
-
-      /// search for something
-      final searchTextField = find.descendant(
-        of: find.byType(MobileSearchTextfield),
-        matching: find.byType(TextFormField),
-      );
-      final query = '$gettingStarted searching';
-      await tester.enterText(searchTextField, query);
-      await tester.pumpAndSettle();
-
-      expect(find.byType(MobileSearchRecentList), findsNothing);
-      expect(find.byType(MobileSearchResultList), findsOneWidget);
-      expect(
-        find.text(LocaleKeys.search_noResultForSearching.tr()),
-        findsOneWidget,
-      );
-
-      /// clear text
-      final clearButton = find.byFlowySvg(FlowySvgs.clear_s);
-      await tester.tapButton(clearButton);
-      expect(find.byType(MobileSearchRecentList), findsOneWidget);
-      expect(find.byType(MobileSearchResultList), findsNothing);
-
-      /// tap cancel button
-      final cancelButton = find.text(LocaleKeys.button_cancel.tr());
-      expect(cancelButton, findsNothing);
-      await tester.enterText(searchTextField, query);
-      await tester.pumpAndSettle();
-      expect(cancelButton, findsOneWidget);
-      await tester.tapButton(cancelButton);
-      expect(cancelButton, findsNothing);
-    });
-  });
-}
@@ -1,41 +0,0 @@
-import 'package:appflowy/ai/ai.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/ai/ai_writer_toolbar_item.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/ai/operations/ai_writer_entities.dart';
-import 'package:flutter/material.dart';
-import 'package:flutter_test/flutter_test.dart';
-
-import 'util.dart';
-
-extension AppFlowyAITest on WidgetTester {
-  Future<void> selectAIWriter(AiWriterCommand command) async {
-    await tapButton(find.byType(AiWriterToolbarActionList));
-    await tapButton(find.text(command.i18n));
-    await pumpAndSettle();
-  }
-
-  Future<void> selectModel(String modelName) async {
-    await tapButton(find.byType(SelectModelMenu));
-    await tapButton(find.text(modelName));
-    await pumpAndSettle();
-  }
-
-  Future<void> enterTextInPromptTextField(String text) async {
-    // Wait for the text field to be visible
-    await pumpAndSettle();
-
-    // Find the ExtendedTextField widget
-    final textField = find.descendant(
-      of: find.byType(PromptInputTextField),
-      matching: find.byType(TextField),
-    );
-    expect(textField, findsOneWidget, reason: 'ExtendedTextField not found');
-
-    final widget = element(textField).widget as TextField;
-    expect(widget.enabled, isTrue, reason: 'TextField is not enabled');
-
-    await tap(textField);
-
-    testTextInput.enterText(text);
-    await pumpAndSettle(const Duration(milliseconds: 300));
-  }
-}
@@ -28,8 +28,8 @@ extension AppFlowyAuthTest on WidgetTester {

    await tapButton(find.byType(AccountSignInOutButton));

-    expectToSeeText(LocaleKeys.button_yes.tr());
-    await tapButtonWithName(LocaleKeys.button_yes.tr());
+    expectToSeeText(LocaleKeys.button_ok.tr());
+    await tapButtonWithName(LocaleKeys.button_ok.tr());
  }

  Future<void> tapSignInAsGuest() async {
@@ -1,7 +1,6 @@
 import 'dart:async';
 import 'dart:io';

-import 'package:appflowy/ai/service/appflowy_ai_service.dart';
 import 'package:appflowy/env/cloud_env.dart';
 import 'package:appflowy/env/cloud_env_test.dart';
 import 'package:appflowy/startup/entry_point.dart';
@@ -21,8 +20,6 @@ import 'package:path/path.dart' as p;
 import 'package:path_provider/path_provider.dart';
 import 'package:universal_platform/universal_platform.dart';

-import 'mock/mock_ai.dart';
-
 class FlowyTestContext {
  FlowyTestContext({required this.applicationDataDirectory});

@@ -36,14 +33,13 @@ extension AppFlowyTestBase on WidgetTester {
    // use to specify the application data directory, if not specified, a temporary directory will be used.
    String? dataDirectory,
    Size windowSize = const Size(1600, 1200),
-    String? email,
    AuthenticatorType? cloudType,
-    AIRepository Function()? aiRepositoryBuilder,
+    String? email,
  }) async {
    if (Platform.isLinux || Platform.isWindows || Platform.isMacOS) {
+      // Set the window size
      await binding.setSurfaceSize(windowSize);
    }
-    //cloudType = AuthenticatorType.appflowyCloudDevelop;

    mockHotKeyManagerHandlers();
    final applicationDataDirectory = dataDirectory ??
@@ -54,12 +50,43 @@ extension AppFlowyTestBase on WidgetTester {
    await FlowyRunner.run(
      AppFlowyApplication(),
      IntegrationMode.integrationTest,
-      rustEnvsBuilder: () => _buildRustEnvs(cloudType),
-      didInitGetItCallback: () => _initializeCloudServices(
-        cloudType: cloudType,
-        email: email,
-        aiRepositoryBuilder: aiRepositoryBuilder,
-      ),
+      rustEnvsBuilder: () {
+        final rustEnvs = <String, String>{};
+        if (cloudType != null) {
+          switch (cloudType) {
+            case AuthenticatorType.local:
+              break;
+            case AuthenticatorType.appflowyCloudSelfHost:
+              rustEnvs["GOTRUE_ADMIN_EMAIL"] = "admin@example.com";
+              rustEnvs["GOTRUE_ADMIN_PASSWORD"] = "password";
+              break;
+            default:
+              throw Exception("not supported");
+          }
+        }
+        return rustEnvs;
+      },
+      didInitGetItCallback: () {
+        return Future(
+          () async {
+            if (cloudType != null) {
+              switch (cloudType) {
+                case AuthenticatorType.local:
+                  await useLocalServer();
+                  break;
+                case AuthenticatorType.appflowyCloudSelfHost:
+                  await useTestSelfHostedAppFlowyCloud();
+                  getIt.unregister<AuthService>();
+                  getIt.registerFactory<AuthService>(
+                    () => AppFlowyCloudMockAuthService(email: email),
+                  );
+                default:
+                  throw Exception("not supported");
+              }
+            }
+          },
+        );
+      },
    );

    await waitUntilSignInPageShow();
@@ -68,76 +95,6 @@ extension AppFlowyTestBase on WidgetTester {
    );
  }

-  Map<String, String> _buildRustEnvs(AuthenticatorType? cloudType) {
-    final rustEnvs = <String, String>{};
-    if (cloudType != null) {
-      switch (cloudType) {
-        case AuthenticatorType.local:
-          break;
-        case AuthenticatorType.appflowyCloudSelfHost:
-        case AuthenticatorType.appflowyCloudDevelop:
-          rustEnvs["GOTRUE_ADMIN_EMAIL"] = "admin@example.com";
-          rustEnvs["GOTRUE_ADMIN_PASSWORD"] = "password";
-          break;
-        default:
-          throw Exception("Unsupported cloud type: $cloudType");
-      }
-    }
-    return rustEnvs;
-  }
-
-  Future<void> _initializeCloudServices({
-    required AuthenticatorType? cloudType,
-    String? email,
-    AIRepository Function()? aiRepositoryBuilder,
-  }) async {
-    if (cloudType == null) return;
-
-    switch (cloudType) {
-      case AuthenticatorType.local:
-        await useLocalServer();
-        break;
-      case AuthenticatorType.appflowyCloudSelfHost:
-        await _setupAppFlowyCloud(
-          useLocal: false,
-          email: email,
-          aiRepositoryBuilder: aiRepositoryBuilder,
-        );
-        break;
-      case AuthenticatorType.appflowyCloudDevelop:
-        await _setupAppFlowyCloud(
-          useLocal: integrationMode().isDevelop,
-          email: email,
-          aiRepositoryBuilder: aiRepositoryBuilder,
-        );
-        break;
-      default:
-        throw Exception("Unsupported cloud type: $cloudType");
-    }
-  }
-
-  Future<void> _setupAppFlowyCloud({
-    required bool useLocal,
-    String? email,
-    AIRepository Function()? aiRepositoryBuilder,
-  }) async {
-    if (useLocal) {
-      await useAppFlowyCloudDevelop("http://localhost");
-    } else {
-      await useSelfHostedAppFlowyCloud(TestEnv.afCloudUrl);
-    }
-
-    getIt.unregister<AuthService>();
-    getIt.unregister<AIRepository>();
-
-    getIt.registerFactory<AuthService>(
-      () => AppFlowyCloudMockAuthService(email: email),
-    );
-    getIt.registerFactory<AIRepository>(
-      aiRepositoryBuilder ?? () => MockAIRepository(),
-    );
-  }
-
  void mockHotKeyManagerHandlers() {
    TestDefaultBinaryMessengerBinding.instance.defaultBinaryMessenger
        .setMockMethodCallHandler(const MethodChannel('hotkey_manager'),
@@ -318,6 +275,10 @@ extension AppFlowyFinderTestBase on CommonFinders {
  }
 }

+Future<void> useTestSelfHostedAppFlowyCloud() async {
+  await useSelfHostedAppFlowyCloudWithURL(TestEnv.afCloudUrl);
+}
+
 Future<String> mockApplicationDataStorage({
  // use to append after the application data directory
  String? pathExtension,
@@ -50,8 +50,6 @@ import 'package:flutter/gestures.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter/services.dart';
 import 'package:flutter_test/flutter_test.dart';
-import 'package:path/path.dart' as p;
-import 'package:path_provider/path_provider.dart';
 import 'package:universal_platform/universal_platform.dart';

 import 'emoji.dart';
@@ -67,10 +65,12 @@ extension CommonOperations on WidgetTester {
    } else {
      // cloud version
      final anonymousButton = find.byType(SignInAnonymousButtonV2);
-      await tapButton(anonymousButton, warnIfMissed: true);
+      await tapButton(anonymousButton);
    }

-    await pumpAndSettle(const Duration(milliseconds: 200));
+    if (Platform.isWindows) {
+      await pumpAndSettle(const Duration(milliseconds: 200));
+    }
  }

  Future<void> tapContinousAnotherWay() async {
@@ -615,7 +615,7 @@ extension CommonOperations on WidgetTester {
        );
    final distanceY = getCenter(to).dy - getCenter(from).dx;
    await drag(from, Offset(0, distanceY));
-    await pumpAndSettle(const Duration(seconds: 1));
+    await pumpAndSettle();
  }

  // tap the button with [FlowySvgData]
@@ -677,25 +677,6 @@ extension CommonOperations on WidgetTester {
    await pumpAndSettle();
  }

-  Future<void> updatePageIconInTitleBarByPasteALink({
-    required String name,
-    required ViewLayoutPB layout,
-    required String iconLink,
-  }) async {
-    await openPage(
-      name,
-      layout: layout,
-    );
-    final title = find.descendant(
-      of: find.byType(ViewTitleBar),
-      matching: find.text(name),
-    );
-    await tapButton(title);
-    await tapButton(find.byType(EmojiPickerButton));
-    await pasteImageLinkAsIcon(iconLink);
-    await pumpAndSettle();
-  }
-
  Future<void> openNotificationHub({int tabIndex = 0}) async {
    final finder = find.descendant(
      of: find.byType(NotificationButton),
@@ -954,45 +935,6 @@ extension CommonOperations on WidgetTester {
      ),
    );
  }
-
-  Future<EmojiIconData> prepareImageIcon() async {
-    final imagePath = await rootBundle.load('assets/test/images/sample.jpeg');
-    final tempDirectory = await getTemporaryDirectory();
-    final localImagePath = p.join(tempDirectory.path, 'sample.jpeg');
-    final imageFile = File(localImagePath)
-      ..writeAsBytesSync(imagePath.buffer.asUint8List());
-    return EmojiIconData.custom(imageFile.path);
-  }
-
-  Future<EmojiIconData> prepareSvgIcon() async {
-    final imagePath = await rootBundle.load('assets/test/images/sample.svg');
-    final tempDirectory = await getTemporaryDirectory();
-    final localImagePath = p.join(tempDirectory.path, 'sample.svg');
-    final imageFile = File(localImagePath)
-      ..writeAsBytesSync(imagePath.buffer.asUint8List());
-    return EmojiIconData.custom(imageFile.path);
-  }
-
-  /// create new page and show slash menu
-  Future<void> createPageAndShowSlashMenu(String title) async {
-    await createNewDocumentOnMobile(title);
-    await editor.tapLineOfEditorAt(0);
-    await editor.showSlashMenu();
-  }
-
-  /// create new page and show at menu
-  Future<void> createPageAndShowAtMenu(String title) async {
-    await createNewDocumentOnMobile(title);
-    await editor.tapLineOfEditorAt(0);
-    await editor.showAtMenu();
-  }
-
-  /// create new page and show plus menu
-  Future<void> createPageAndShowPlusMenu(String title) async {
-    await createNewDocumentOnMobile(title);
-    await editor.tapLineOfEditorAt(0);
-    await editor.showPlusMenu();
-  }
 }

 extension SettingsFinder on CommonFinders {
@@ -1,9 +1,17 @@
 import 'dart:io';

+import 'package:appflowy/plugins/database/application/field/filter_entities.dart';
+import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/choicechip/select_option/condition_list.dart';
+import 'package:appflowy/plugins/database/widgets/cell/desktop_row_detail/desktop_row_detail_checklist_cell.dart';
+import 'package:appflowy/shared/icon_emoji_picker/flowy_icon_emoji_picker.dart';
+import 'package:appflowy/shared/icon_emoji_picker/icon_picker.dart';
+import 'package:flutter/gestures.dart';
+import 'package:flutter/material.dart';
+import 'package:flutter/services.dart';
+
 import 'package:appflowy/generated/flowy_svgs.g.dart';
 import 'package:appflowy/generated/locale_keys.g.dart';
 import 'package:appflowy/plugins/database/application/calculations/calculation_type_ext.dart';
-import 'package:appflowy/plugins/database/application/field/filter_entities.dart';
 import 'package:appflowy/plugins/database/board/presentation/board_page.dart';
 import 'package:appflowy/plugins/database/board/presentation/widgets/board_column_header.dart';
 import 'package:appflowy/plugins/database/calendar/application/calendar_bloc.dart';
@@ -19,11 +27,10 @@ import 'package:appflowy/plugins/database/grid/presentation/widgets/common/type_
 import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/choicechip/checkbox.dart';
 import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/choicechip/checklist.dart';
 import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/choicechip/choicechip.dart';
-import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/choicechip/date.dart';
-import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/choicechip/select_option/condition_list.dart';
 import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/choicechip/select_option/option_list.dart';
 import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/choicechip/select_option/select_option.dart';
 import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/choicechip/text.dart';
+import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/choicechip/date.dart';
 import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/create_filter_list.dart';
 import 'package:appflowy/plugins/database/grid/presentation/widgets/filter/disclosure_button.dart';
 import 'package:appflowy/plugins/database/grid/presentation/widgets/footer/grid_footer.dart';
@@ -37,7 +44,6 @@ import 'package:appflowy/plugins/database/grid/presentation/widgets/toolbar/filt
 import 'package:appflowy/plugins/database/grid/presentation/widgets/toolbar/sort_button.dart';
 import 'package:appflowy/plugins/database/tab_bar/desktop/tab_bar_add_button.dart';
 import 'package:appflowy/plugins/database/tab_bar/desktop/tab_bar_header.dart';
-import 'package:appflowy/plugins/database/widgets/cell/desktop_row_detail/desktop_row_detail_checklist_cell.dart';
 import 'package:appflowy/plugins/database/widgets/cell/editable_cell_skeleton/checkbox.dart';
 import 'package:appflowy/plugins/database/widgets/cell/editable_cell_skeleton/checklist.dart';
 import 'package:appflowy/plugins/database/widgets/cell/editable_cell_skeleton/date.dart';
@@ -70,8 +76,6 @@ import 'package:appflowy/plugins/database/widgets/setting/database_setting_actio
 import 'package:appflowy/plugins/database/widgets/setting/database_settings_list.dart';
 import 'package:appflowy/plugins/database/widgets/setting/setting_button.dart';
 import 'package:appflowy/plugins/database/widgets/setting/setting_property_list.dart';
-import 'package:appflowy/shared/icon_emoji_picker/flowy_icon_emoji_picker.dart';
-import 'package:appflowy/shared/icon_emoji_picker/icon_picker.dart';
 import 'package:appflowy/util/field_type_extension.dart';
 import 'package:appflowy/workspace/presentation/widgets/date_picker/widgets/clear_date_button.dart';
 import 'package:appflowy/workspace/presentation/widgets/date_picker/widgets/date_type_option_button.dart';
@@ -86,9 +90,6 @@ import 'package:easy_localization/easy_localization.dart';
 import 'package:flowy_infra_ui/flowy_infra_ui.dart';
 import 'package:flowy_infra_ui/style_widget/text_input.dart';
 import 'package:flowy_infra_ui/widget/buttons/primary_button.dart';
-import 'package:flutter/gestures.dart';
-import 'package:flutter/material.dart';
-import 'package:flutter/services.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:path/path.dart' as p;
 // Non-exported member of the table_calendar library
@@ -942,31 +943,6 @@ extension AppFlowyDatabaseTest on WidgetTester {
    await pumpAndSettle(const Duration(milliseconds: 200));
  }

-  Future<void> changeFieldWidth(String fieldName, double width) async {
-    final field = find.byWidgetPredicate(
-      (widget) => widget is GridFieldCell && widget.fieldInfo.name == fieldName,
-    );
-    await hoverOnWidget(
-      field,
-      onHover: () async {
-        final dragHandle = find.descendant(
-          of: field,
-          matching: find.byType(DragToExpandLine),
-        );
-        await drag(dragHandle, Offset(width - getSize(field).width, 0));
-        await pumpAndSettle(const Duration(milliseconds: 200));
-      },
-    );
-  }
-
-  double getFieldWidth(String fieldName) {
-    final field = find.byWidgetPredicate(
-      (widget) => widget is GridFieldCell && widget.fieldInfo.name == fieldName,
-    );
-
-    return getSize(field).width;
-  }
-
  Future<void> findDateEditor(dynamic matcher) async {
    final finder = find.byType(DateCellEditor);
    expect(finder, matcher);
@@ -1488,7 +1464,6 @@ extension AppFlowyDatabaseTest on WidgetTester {
    );

    await tapButton(button);
-    await tapButtonWithName(LocaleKeys.button_delete.tr());
  }

  Future<void> dragDropRescheduleCalendarEvent() async {
@@ -1596,7 +1571,7 @@ extension AppFlowyDatabaseTest on WidgetTester {
        of: textField,
        matching: find.byWidgetPredicate(
          (widget) =>
-              widget is FlowySvg && widget.svg == FlowySvgs.close_filled_s,
+              widget is FlowySvg && widget.svg == FlowySvgs.close_filled_m,
        ),
      ),
    );
@@ -307,11 +307,9 @@ class EditorOperations {
  Future<void> openTurnIntoMenu(Path path) async {
    await hoverAndClickOptionMenuButton(path);
    await tester.tapButton(
-      find
-          .findTextInFlowyText(
-            LocaleKeys.document_plugins_optionAction_turnInto.tr(),
-          )
-          .first,
+      find.findTextInFlowyText(
+        LocaleKeys.document_plugins_optionAction_turnInto.tr(),
+      ),
    );
    await tester.pumpUntilFound(find.byType(TurnIntoOptionMenu));
  }
@@ -1,24 +1,20 @@
 import 'dart:convert';
-import 'dart:io';

-import 'package:appflowy/plugins/document/presentation/editor_plugins/copy_and_paste/clipboard_service.dart';
 import 'package:appflowy/shared/icon_emoji_picker/flowy_icon_emoji_picker.dart';
 import 'package:appflowy/shared/icon_emoji_picker/icon_color_picker.dart';
 import 'package:appflowy/shared/icon_emoji_picker/icon_picker.dart';
 import 'package:appflowy/shared/icon_emoji_picker/icon_uploader.dart';
 import 'package:appflowy/shared/icon_emoji_picker/tab.dart';
-import 'package:appflowy/startup/startup.dart';
 import 'package:appflowy/workspace/presentation/home/menu/sidebar/space/space_icon_popup.dart';
+import 'package:cross_file/cross_file.dart';
 import 'package:desktop_drop/desktop_drop.dart';
 import 'package:flowy_infra_ui/style_widget/primary_rounded_button.dart';
 import 'package:flowy_svg/flowy_svg.dart';
 import 'package:flutter/material.dart';
-import 'package:flutter/services.dart';
 import 'package:flutter_emoji_mart/flutter_emoji_mart.dart';
 import 'package:flutter_test/flutter_test.dart';

 import 'base.dart';
-import 'common_operations.dart';

 extension EmojiTestExtension on WidgetTester {
  Future<void> tapEmoji(String emoji) async {
@@ -94,7 +90,7 @@ extension EmojiTestExtension on WidgetTester {
    final dropTargetWidget = dropTarget.evaluate().first.widget as DropTarget;
    dropTargetWidget.onDragDone?.call(
      DropDoneDetails(
-        files: [DropItemFile(icon.emoji)],
+        files: [XFile(icon.emoji)],
        localPosition: Offset.zero,
        globalPosition: Offset.zero,
      ),
@@ -108,37 +104,4 @@ extension EmojiTestExtension on WidgetTester {
    );
    await tapButton(confirmButton);
  }
-
-  Future<void> pasteImageLinkAsIcon(String link) async {
-    final pickTab = find.byType(PickerTab);
-    expect(pickTab, findsOneWidget);
-    await pumpAndSettle();
-
-    /// switch to custom tab
-    final iconTab = find.descendant(
-      of: pickTab,
-      matching: find.text(PickerTabType.custom.tr),
-    );
-    expect(iconTab, findsOneWidget);
-    await tapButton(iconTab);
-
-    // mock the clipboard
-    await getIt<ClipboardService>()
-        .setData(ClipboardServiceData(plainText: link));
-
-    // paste the link
-    await simulateKeyEvent(
-      LogicalKeyboardKey.keyV,
-      isControlPressed: Platform.isLinux || Platform.isWindows,
-      isMetaPressed: Platform.isMacOS,
-    );
-    await pumpAndSettle(const Duration(seconds: 5));
-
-    /// confirm to upload
-    final confirmButton = find.descendant(
-      of: find.byType(IconUploader),
-      matching: find.byType(PrimaryRoundedButton),
-    );
-    await tapButton(confirmButton);
-  }
 }
@@ -8,7 +8,6 @@ import 'package:appflowy/plugins/document/presentation/banner.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/header/document_cover_widget.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/header/emoji_icon_widget.dart';
 import 'package:appflowy/shared/appflowy_network_image.dart';
-import 'package:appflowy/shared/appflowy_network_svg.dart';
 import 'package:appflowy/shared/icon_emoji_picker/flowy_icon_emoji_picker.dart';
 import 'package:appflowy/shared/icon_emoji_picker/icon_picker.dart';
 import 'package:appflowy/workspace/application/sidebar/folder/folder_bloc.dart';
@@ -253,19 +252,16 @@ extension Expectation on WidgetTester {
      );
      expect(icon, findsOneWidget);
    } else if (type == FlowyIconType.custom) {
-      final isSvg = data.emoji.endsWith('.svg');
      if (isURL(data.emoji)) {
        final image = find.descendant(
          of: pageName,
-          matching: isSvg
-              ? find.byType(FlowyNetworkSvg)
-              : find.byType(FlowyNetworkImage),
+          matching: find.byType(FlowyNetworkImage),
        );
        expect(image, findsOneWidget);
      } else {
        final image = find.descendant(
          of: pageName,
-          matching: isSvg ? find.byType(SvgPicture) : find.byType(Image),
+          matching: find.byType(Image),
        );
        expect(image, findsOneWidget);
      }
@@ -294,26 +290,16 @@ extension Expectation on WidgetTester {
      );
      expect(icon, findsOneWidget);
    } else if (type == FlowyIconType.custom) {
-      final isSvg = data.emoji.endsWith('.svg');
      if (isURL(data.emoji)) {
        final image = find.descendant(
          of: find.byType(ViewTitleBar),
-          matching: isSvg
-              ? find.byType(FlowyNetworkSvg)
-              : find.byType(FlowyNetworkImage),
+          matching: find.byType(FlowyNetworkImage),
        );
        expect(image, findsOneWidget);
      } else {
        final image = find.descendant(
          of: find.byType(ViewTitleBar),
-          matching: isSvg
-              ? find.byWidgetPredicate((w) {
-                  if (w is! SvgPicture) return false;
-                  final loader = w.bytesLoader;
-                  if (loader is! SvgFileLoader) return false;
-                  return loader.file.path.endsWith('.svg');
-                })
-              : find.byType(Image),
+          matching: find.byType(Image),
        );
        expect(image, findsOneWidget);
      }
@@ -1,128 +0,0 @@
-import 'dart:async';
-
-import 'package:appflowy/ai/service/ai_entities.dart';
-import 'package:appflowy/ai/service/appflowy_ai_service.dart';
-import 'package:appflowy/ai/service/error.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/ai/operations/ai_writer_entities.dart';
-import 'package:appflowy_backend/protobuf/flowy-ai/entities.pbenum.dart';
-import 'package:mocktail/mocktail.dart';
-
-final _mockAiMap = <CompletionTypePB, Map<String, List<String>>>{
-  CompletionTypePB.ImproveWriting: {
-    "I have an apple": [
-      "I",
-      "have",
-      "an",
-      "apple",
-      "and",
-      "a",
-      "banana",
-    ],
-  },
-  CompletionTypePB.SpellingAndGrammar: {
-    "We didn’t had enough money": [
-      "We",
-      "didn’t",
-      "have",
-      "enough",
-      "money",
-    ],
-  },
-  CompletionTypePB.UserQuestion: {
-    "Explain the concept of TPU": [
-      "TPU",
-      "is",
-      "a",
-      "tensor",
-      "processing",
-      "unit",
-      "that",
-      "is",
-      "designed",
-      "to",
-      "accelerate",
-      "machine",
-    ],
-    "How about GPU?": [
-      "GPU",
-      "is",
-      "a",
-      "graphics",
-      "processing",
-      "unit",
-      "that",
-      "is",
-      "designed",
-      "to",
-      "accelerate",
-      "machine",
-      "learning",
-      "tasks",
-    ],
-  },
-};
-
-abstract class StreamCompletionValidator {
-  bool validate(
-    String text,
-    String? objectId,
-    CompletionTypePB completionType,
-    PredefinedFormat? format,
-    List<String> sourceIds,
-    List<AiWriterRecord> history,
-  );
-}
-
-class MockCompletionStream extends Mock implements CompletionStream {}
-
-class MockAIRepository extends Mock implements AppFlowyAIService {
-  MockAIRepository({this.validator});
-  StreamCompletionValidator? validator;
-
-  @override
-  Future<(String, CompletionStream)?> streamCompletion({
-    String? objectId,
-    required String text,
-    PredefinedFormat? format,
-    List<String> sourceIds = const [],
-    List<AiWriterRecord> history = const [],
-    required CompletionTypePB completionType,
-    required Future<void> Function() onStart,
-    required Future<void> Function(String text) processMessage,
-    required Future<void> Function(String text) processAssistMessage,
-    required Future<void> Function() onEnd,
-    required void Function(AIError error) onError,
-    required void Function(LocalAIStreamingState state)
-        onLocalAIStreamingStateChange,
-  }) async {
-    if (validator != null) {
-      if (!validator!.validate(
-        text,
-        objectId,
-        completionType,
-        format,
-        sourceIds,
-        history,
-      )) {
-        throw Exception('Invalid completion');
-      }
-    }
-    final stream = MockCompletionStream();
-    unawaited(
-      Future(() async {
-        await onStart();
-        final lines = _mockAiMap[completionType]?[text.trim()];
-
-        if (lines == null) {
-          throw Exception('No mock ai found for $text and $completionType');
-        }
-
-        for (final line in lines) {
-          await processMessage('$line ');
-        }
-        await onEnd();
-      }),
-    );
-    return ('mock_id', stream);
-  }
-}
@@ -0,0 +1,81 @@
+import 'dart:async';
+import 'dart:convert';
+
+import 'package:appflowy/ai/service/error.dart';
+import 'package:appflowy/ai/service/openai_client.dart';
+import 'package:appflowy/ai/service/text_completion.dart';
+import 'package:http/http.dart' as http;
+import 'package:mocktail/mocktail.dart';
+
+class MyMockClient extends Mock implements http.Client {
+  @override
+  Future<http.StreamedResponse> send(http.BaseRequest request) async {
+    final requestType = request.method;
+    final requestUri = request.url;
+
+    if (requestType == 'POST' &&
+        requestUri == OpenAIRequestType.textCompletion.uri) {
+      final responseHeaders = <String, String>{
+        'content-type': 'text/event-stream',
+      };
+      final responseBody = Stream.fromIterable([
+        utf8.encode(
+          '{ "choices": [{"text": "Lorem ipsum dolor sit amet, consectetuer adipiscing elit. Aenean commodo ligula ", "index": 0, "logprobs": null, "finish_reason": null}]}',
+        ),
+        utf8.encode('\n'),
+        utf8.encode('[DONE]'),
+      ]);
+
+      // Return a mocked response with the expected data
+      return http.StreamedResponse(responseBody, 200, headers: responseHeaders);
+    }
+
+    // Return an error response for any other request
+    return http.StreamedResponse(const Stream.empty(), 404);
+  }
+}
+
+class MockOpenAIRepository extends HttpOpenAIRepository {
+  MockOpenAIRepository() : super(apiKey: 'dummyKey', client: MyMockClient());
+
+  @override
+  Future<void> getStreamedCompletions({
+    required String prompt,
+    required Future<void> Function() onStart,
+    required Future<void> Function(TextCompletionResponse response) onProcess,
+    required Future<void> Function() onEnd,
+    required void Function(AIError error) onError,
+    String? suffix,
+    int maxTokens = 2048,
+    double temperature = 0.3,
+    bool useAction = false,
+  }) async {
+    final request = http.Request('POST', OpenAIRequestType.textCompletion.uri);
+    final response = await client.send(request);
+
+    String previousSyntax = '';
+    if (response.statusCode == 200) {
+      await for (final chunk in response.stream
+          .transform(const Utf8Decoder())
+          .transform(const LineSplitter())) {
+        await onStart();
+        final data = chunk.trim().split('data: ');
+        if (data[0] != '[DONE]') {
+          final response = TextCompletionResponse.fromJson(
+            json.decode(data[0]),
+          );
+          if (response.choices.isNotEmpty) {
+            final text = response.choices.first.text;
+            if (text == previousSyntax && text == '\n') {
+              continue;
+            }
+            await onProcess(response);
+            previousSyntax = response.choices.first.text;
+          }
+        } else {
+          await onEnd();
+        }
+      }
+    }
+  }
+}
@@ -9,6 +9,7 @@ import 'package:appflowy/workspace/presentation/settings/settings_dialog.dart';
 import 'package:appflowy/workspace/presentation/settings/widgets/settings_menu_element.dart';
 import 'package:appflowy/workspace/presentation/widgets/toggle/toggle.dart';
 import 'package:easy_localization/easy_localization.dart';
+import 'package:flowy_infra_ui/style_widget/text_field.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter_test/flutter_test.dart';

@@ -78,14 +79,14 @@ extension AppFlowySettings on WidgetTester {
    // Enable editing username
    final editUsernameFinder = find.descendant(
      of: find.byType(AccountUserProfile),
-      matching: find.byFlowySvg(FlowySvgs.toolbar_link_edit_m),
+      matching: find.byFlowySvg(FlowySvgs.edit_s),
    );
    await tap(editUsernameFinder, warnIfMissed: false);
    await pumpAndSettle();

    final userNameFinder = find.descendant(
      of: find.byType(AccountUserProfile),
-      matching: find.byType(TextField),
+      matching: find.byType(FlowyTextField),
    );
    await enterText(userNameFinder, name);
    await pumpAndSettle();
@@ -1,5 +1,5 @@
 PODS:
-  - app_links (0.0.2):
+  - app_links (0.0.1):
    - Flutter
  - appflowy_backend (0.0.1):
    - Flutter
@@ -8,35 +8,35 @@ PODS:
    - ReachabilitySwift
  - device_info_plus (0.0.1):
    - Flutter
-  - DKImagePickerController/Core (4.3.9):
+  - DKImagePickerController/Core (4.3.4):
    - DKImagePickerController/ImageDataManager
    - DKImagePickerController/Resource
-  - DKImagePickerController/ImageDataManager (4.3.9)
-  - DKImagePickerController/PhotoGallery (4.3.9):
+  - DKImagePickerController/ImageDataManager (4.3.4)
+  - DKImagePickerController/PhotoGallery (4.3.4):
    - DKImagePickerController/Core
    - DKPhotoGallery
-  - DKImagePickerController/Resource (4.3.9)
-  - DKPhotoGallery (0.0.19):
-    - DKPhotoGallery/Core (= 0.0.19)
-    - DKPhotoGallery/Model (= 0.0.19)
-    - DKPhotoGallery/Preview (= 0.0.19)
-    - DKPhotoGallery/Resource (= 0.0.19)
+  - DKImagePickerController/Resource (4.3.4)
+  - DKPhotoGallery (0.0.17):
+    - DKPhotoGallery/Core (= 0.0.17)
+    - DKPhotoGallery/Model (= 0.0.17)
+    - DKPhotoGallery/Preview (= 0.0.17)
+    - DKPhotoGallery/Resource (= 0.0.17)
    - SDWebImage
    - SwiftyGif
-  - DKPhotoGallery/Core (0.0.19):
+  - DKPhotoGallery/Core (0.0.17):
    - DKPhotoGallery/Model
    - DKPhotoGallery/Preview
    - SDWebImage
    - SwiftyGif
-  - DKPhotoGallery/Model (0.0.19):
+  - DKPhotoGallery/Model (0.0.17):
    - SDWebImage
    - SwiftyGif
-  - DKPhotoGallery/Preview (0.0.19):
+  - DKPhotoGallery/Preview (0.0.17):
    - DKPhotoGallery/Model
    - DKPhotoGallery/Resource
    - SDWebImage
    - SwiftyGif
-  - DKPhotoGallery/Resource (0.0.19):
+  - DKPhotoGallery/Resource (0.0.17):
    - SDWebImage
    - SwiftyGif
  - file_picker (0.0.1):
@@ -65,34 +65,31 @@ PODS:
    - FlutterMacOS
  - permission_handler_apple (9.3.0):
    - Flutter
-  - ReachabilitySwift (5.2.4)
-  - saver_gallery (0.0.1):
-    - Flutter
-  - SDWebImage (5.21.0):
-    - SDWebImage/Core (= 5.21.0)
-  - SDWebImage/Core (5.21.0)
-  - Sentry/HybridSDK (8.46.0)
-  - sentry_flutter (8.14.2):
+  - ReachabilitySwift (5.0.0)
+  - SDWebImage (5.14.2):
+    - SDWebImage/Core (= 5.14.2)
+  - SDWebImage/Core (5.14.2)
+  - Sentry/HybridSDK (8.35.1)
+  - sentry_flutter (8.8.0):
    - Flutter
    - FlutterMacOS
-    - Sentry/HybridSDK (= 8.46.0)
+    - Sentry/HybridSDK (= 8.35.1)
  - share_plus (0.0.1):
    - Flutter
  - shared_preferences_foundation (0.0.1):
    - Flutter
    - FlutterMacOS
-  - sqflite_darwin (0.0.4):
+  - sqflite (0.0.3):
    - Flutter
    - FlutterMacOS
  - super_native_extensions (0.0.1):
    - Flutter
-  - SwiftyGif (5.4.5)
-  - Toast (4.1.1)
+  - SwiftyGif (5.4.3)
+  - Toast (4.0.0)
  - url_launcher_ios (0.0.1):
    - Flutter
  - webview_flutter_wkwebview (0.0.1):
    - Flutter
-    - FlutterMacOS

 DEPENDENCIES:
  - app_links (from `.symlinks/plugins/app_links/ios`)
@@ -111,14 +108,13 @@ DEPENDENCIES:
  - package_info_plus (from `.symlinks/plugins/package_info_plus/ios`)
  - path_provider_foundation (from `.symlinks/plugins/path_provider_foundation/darwin`)
  - permission_handler_apple (from `.symlinks/plugins/permission_handler_apple/ios`)
-  - saver_gallery (from `.symlinks/plugins/saver_gallery/ios`)
  - sentry_flutter (from `.symlinks/plugins/sentry_flutter/ios`)
  - share_plus (from `.symlinks/plugins/share_plus/ios`)
  - shared_preferences_foundation (from `.symlinks/plugins/shared_preferences_foundation/darwin`)
-  - sqflite_darwin (from `.symlinks/plugins/sqflite_darwin/darwin`)
+  - sqflite (from `.symlinks/plugins/sqflite/darwin`)
  - super_native_extensions (from `.symlinks/plugins/super_native_extensions/ios`)
  - url_launcher_ios (from `.symlinks/plugins/url_launcher_ios/ios`)
-  - webview_flutter_wkwebview (from `.symlinks/plugins/webview_flutter_wkwebview/darwin`)
+  - webview_flutter_wkwebview (from `.symlinks/plugins/webview_flutter_wkwebview/ios`)

 SPEC REPOS:
  trunk:
@@ -163,55 +159,52 @@ EXTERNAL SOURCES:
    :path: ".symlinks/plugins/path_provider_foundation/darwin"
  permission_handler_apple:
    :path: ".symlinks/plugins/permission_handler_apple/ios"
-  saver_gallery:
-    :path: ".symlinks/plugins/saver_gallery/ios"
  sentry_flutter:
    :path: ".symlinks/plugins/sentry_flutter/ios"
  share_plus:
    :path: ".symlinks/plugins/share_plus/ios"
  shared_preferences_foundation:
    :path: ".symlinks/plugins/shared_preferences_foundation/darwin"
-  sqflite_darwin:
-    :path: ".symlinks/plugins/sqflite_darwin/darwin"
+  sqflite:
+    :path: ".symlinks/plugins/sqflite/darwin"
  super_native_extensions:
    :path: ".symlinks/plugins/super_native_extensions/ios"
  url_launcher_ios:
    :path: ".symlinks/plugins/url_launcher_ios/ios"
  webview_flutter_wkwebview:
-    :path: ".symlinks/plugins/webview_flutter_wkwebview/darwin"
+    :path: ".symlinks/plugins/webview_flutter_wkwebview/ios"

 SPEC CHECKSUMS:
-  app_links: 3da4c36b46cac3bf24eb897f1a6ce80bda109874
+  app_links: c5161ac5ab5383ad046884568b4b91cb52df5d91
  appflowy_backend: 78f6a053f756e6bc29bcc5a2106cbe77b756e97a
  connectivity_plus: 481668c94744c30c53b8895afb39159d1e619bdf
-  device_info_plus: 21fcca2080fbcd348be798aa36c3e5ed849eefbe
-  DKImagePickerController: 946cec48c7873164274ecc4624d19e3da4c1ef3c
-  DKPhotoGallery: b3834fecb755ee09a593d7c9e389d8b5d6deed60
+  device_info_plus: 71ffc6ab7634ade6267c7a93088ed7e4f74e5896
+  DKImagePickerController: b512c28220a2b8ac7419f21c491fc8534b7601ac
+  DKPhotoGallery: fdfad5125a9fdda9cc57df834d49df790dbb4179
  file_picker: 9b3292d7c8bc68c8a7bf8eb78f730e49c8efc517
  flowy_infra_ui: 931b73a18b54a392ab6152eebe29a63a30751f53
  Flutter: e0871f40cf51350855a761d2e70bf5af5b9b5de7
  fluttertoast: 76fea30fcf04176325f6864c87306927bd7d2038
  image_picker_ios: 7fe1ff8e34c1790d6fff70a32484959f563a928a
-  integration_test: 4a889634ef21a45d28d50d622cf412dc6d9f586e
+  integration_test: d5929033778cc4991a187e4e1a85396fa4f59b3a
  irondash_engine_context: 8e58ca8e0212ee9d1c7dc6a42121849986c88486
  keyboard_height_plugin: ef70a8181b29f27670e9e2450814ca6b6dc05b05
  open_filex: 432f3cd11432da3e39f47fcc0df2b1603854eff1
  package_info_plus: af8e2ca6888548050f16fa2f1938db7b5a5df499
  path_provider_foundation: 080d55be775b7414fd5a5ef3ac137b97b097e564
  permission_handler_apple: 4ed2196e43d0651e8ff7ca3483a069d469701f2d
-  ReachabilitySwift: 32793e867593cfc1177f5d16491e3a197d2fccda
-  saver_gallery: af2d0c762dafda254e0ad025ef0dabd6506cd490
-  SDWebImage: f84b0feeb08d2d11e6a9b843cb06d75ebf5b8868
-  Sentry: da60d980b197a46db0b35ea12cb8f39af48d8854
-  sentry_flutter: 27892878729f42701297c628eb90e7c6529f3684
+  ReachabilitySwift: 985039c6f7b23a1da463388634119492ff86c825
+  SDWebImage: b9a731e1d6307f44ca703b3976d18c24ca561e84
+  Sentry: 1fe34e9c2cbba1e347623610d26db121dcb569f1
+  sentry_flutter: e24b397f9a61fa5bbefd8279c3b2242ca86faa90
  share_plus: 50da8cb520a8f0f65671c6c6a99b3617ed10a58a
  shared_preferences_foundation: 9e1978ff2562383bd5676f64ec4e9aa8fa06a6f7
-  sqflite_darwin: 20b2a3a3b70e43edae938624ce550a3cbf66a3d0
+  sqflite: c35dad70033b8862124f8337cc994a809fcd9fa3
  super_native_extensions: b763c02dc3a8fd078389f410bf15149179020cb4
-  SwiftyGif: 706c60cf65fa2bc5ee0313beece843c8eb8194d4
-  Toast: 1f5ea13423a1e6674c4abdac5be53587ae481c4e
+  SwiftyGif: 6c3eafd0ce693cad58bb63d2b2fb9bacb8552780
+  Toast: 91b396c56ee72a5790816f40d3a94dd357abc196
  url_launcher_ios: 694010445543906933d732453a59da0a173ae33d
-  webview_flutter_wkwebview: 44d4dee7d7056d5ad185d25b38404436d56c547c
+  webview_flutter_wkwebview: 45a041c7831641076618876de3ba75c712860c6b

 PODFILE CHECKSUM: d0d9b4ff572d8695c38eb3f9b490f55cdfc57eca

@@ -1,7 +1,7 @@
 import UIKit
 import Flutter

-@main
+@UIApplicationMain
@objc class AppDelegate: FlutterAppDelegate {
  override func application(
    _ application: UIApplication,
@@ -1,78 +1,75 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
 <plist version="1.0">
-	<dict>
-		<key>CADisableMinimumFrameDurationOnPhone</key>
-		<true />
-		<key>CFBundleDevelopmentRegion</key>
-		<string>$(DEVELOPMENT_LANGUAGE)</string>
-		<key>CFBundleExecutable</key>
-		<string>$(EXECUTABLE_NAME)</string>
-		<key>CFBundleIdentifier</key>
-		<string>$(PRODUCT_BUNDLE_IDENTIFIER)</string>
-		<key>CFBundleInfoDictionaryVersion</key>
-		<string>6.0</string>
-		<key>CFBundleLocalizations</key>
-		<array>
-			<string>en</string>
-		</array>
-		<key>CFBundleName</key>
-		<string>AppFlowy</string>
-		<key>CFBundlePackageType</key>
-		<string>APPL</string>
-		<key>CFBundleShortVersionString</key>
-		<string>$(FLUTTER_BUILD_NAME)</string>
-		<key>CFBundleSignature</key>
-		<string>????</string>
-		<key>CFBundleURLTypes</key>
-		<array>
-			<dict>
-				<key>CFBundleURLName</key>
-				<string></string>
-				<key>CFBundleURLSchemes</key>
-				<array>
-					<string>appflowy-flutter</string>
-				</array>
-			</dict>
-		</array>
-		<key>CFBundleVersion</key>
-		<string>$(FLUTTER_BUILD_NUMBER)</string>
-		<key>FLTEnableImpeller</key>
-		<false />
-		<key>LSRequiresIPhoneOS</key>
-		<true />
-		<key>NSAppTransportSecurity</key>
+<dict>
+	<key>CADisableMinimumFrameDurationOnPhone</key>
+	<true/>
+	<key>CFBundleDevelopmentRegion</key>
+	<string>$(DEVELOPMENT_LANGUAGE)</string>
+	<key>CFBundleExecutable</key>
+	<string>$(EXECUTABLE_NAME)</string>
+	<key>CFBundleIdentifier</key>
+	<string>$(PRODUCT_BUNDLE_IDENTIFIER)</string>
+	<key>CFBundleInfoDictionaryVersion</key>
+	<string>6.0</string>
+	<key>CFBundleLocalizations</key>
+	<array>
+		<string>en</string>
+	</array>
+	<key>CFBundleName</key>
+	<string>AppFlowy</string>
+	<key>CFBundlePackageType</key>
+	<string>APPL</string>
+	<key>CFBundleShortVersionString</key>
+	<string>$(FLUTTER_BUILD_NAME)</string>
+	<key>CFBundleSignature</key>
+	<string>????</string>
+	<key>CFBundleURLTypes</key>
+	<array>
 		<dict>
-			<key>NSAllowsArbitraryLoads</key>
-			<true />
+			<key>CFBundleURLName</key>
+			<string></string>
+			<key>CFBundleURLSchemes</key>
+			<array>
+				<string>appflowy-flutter</string>
+			</array>
 		</dict>
-		<key>NSPhotoLibraryUsageDescription</key>
-		<string>AppFlowy needs access to your photos to let you add images to your documents</string>
-		<key>NSPhotoLibraryAddUsageDescription</key>
-		<string>AppFlowy needs access to your photos to let you add images to your photo library</string>
-		<key>UIApplicationSupportsIndirectInputEvents</key>
-		<true />
-		<key>NSCameraUsageDescription</key>
-		<string>AppFlowy needs access to your camera to let you add images to your documents from
-			camera</string>
-		<key>UILaunchStoryboardName</key>
-		<string>LaunchScreen</string>
-		<key>UIMainStoryboardFile</key>
-		<string>Main</string>
-		<key>UISupportedInterfaceOrientations</key>
-		<array>
-			<string>UIInterfaceOrientationPortrait</string>
-		</array>
-		<key>UISupportedInterfaceOrientations~ipad</key>
-		<array>
-			<string>UIInterfaceOrientationLandscapeLeft</string>
-			<string>UIInterfaceOrientationLandscapeRight</string>
-			<string>UIInterfaceOrientationPortrait</string>
-			<string>UIInterfaceOrientationPortraitUpsideDown</string>
-		</array>
-		<key>UISupportsDocumentBrowser</key>
-		<true />
-		<key>UIViewControllerBasedStatusBarAppearance</key>
-		<false />
+	</array>
+	<key>CFBundleVersion</key>
+	<string>$(FLUTTER_BUILD_NUMBER)</string>
+	<key>FLTEnableImpeller</key>
+	<false/>
+	<key>LSRequiresIPhoneOS</key>
+	<true/>
+	<key>NSAppTransportSecurity</key>
+	<dict>
+		<key>NSAllowsArbitraryLoads</key>
+		<true/>
 	</dict>
-</plist>
+	<key>NSPhotoLibraryUsageDescription</key>
+	<string>AppFlowy needs access to your photos to let you add images to your documents</string>
+	<key>UIApplicationSupportsIndirectInputEvents</key>
+	<true/>
+	<key>NSCameraUsageDescription</key>
+	<string>AppFlowy needs access to your camera to let you add images to your documents from camera</string>
+	<key>UILaunchStoryboardName</key>
+	<string>LaunchScreen</string>
+	<key>UIMainStoryboardFile</key>
+	<string>Main</string>
+	<key>UISupportedInterfaceOrientations</key>
+	<array>
+		<string>UIInterfaceOrientationPortrait</string>
+	</array>
+	<key>UISupportedInterfaceOrientations~ipad</key>
+	<array>
+		<string>UIInterfaceOrientationLandscapeLeft</string>
+		<string>UIInterfaceOrientationLandscapeRight</string>
+		<string>UIInterfaceOrientationPortrait</string>
+		<string>UIInterfaceOrientationPortraitUpsideDown</string>
+	</array>
+	<key>UISupportsDocumentBrowser</key>
+	<true/>
+	<key>UIViewControllerBasedStatusBarAppearance</key>
+	<false/>
+</dict>
+</plist>
@@ -1,20 +1,12 @@
-export 'service/ai_entities.dart';
-export 'service/ai_prompt_input_bloc.dart';
-export 'service/ai_prompt_selector_cubit.dart';
-export 'service/appflowy_ai_service.dart';
-export 'service/error.dart';
-export 'service/ai_model_state_notifier.dart';
-export 'service/select_model_bloc.dart';
 export 'widgets/loading_indicator.dart';
 export 'widgets/prompt_input/action_buttons.dart';
-export 'widgets/prompt_input/desktop_prompt_input.dart';
+export 'widgets/prompt_input/desktop_input_text_field.dart';
 export 'widgets/prompt_input/file_attachment_list.dart';
 export 'widgets/prompt_input/layout_define.dart';
 export 'widgets/prompt_input/mention_page_bottom_sheet.dart';
 export 'widgets/prompt_input/mention_page_menu.dart';
-export 'widgets/prompt_input/prompt_input_text_controller.dart';
+export 'widgets/prompt_input/mentioned_page_text_span.dart';
 export 'widgets/prompt_input/predefined_format_buttons.dart';
 export 'widgets/prompt_input/select_sources_bottom_sheet.dart';
 export 'widgets/prompt_input/select_sources_menu.dart';
-export 'widgets/prompt_input/select_model_menu.dart';
 export 'widgets/prompt_input/send_button.dart';
@@ -0,0 +1,34 @@
+import 'package:appflowy_backend/protobuf/flowy-ai/protobuf.dart';
+import 'package:appflowy_result/appflowy_result.dart';
+
+import 'error.dart';
+import 'text_completion.dart';
+
+abstract class AIRepository {
+  Future<void> getStreamedCompletions({
+    required String prompt,
+    required Future<void> Function() onStart,
+    required Future<void> Function(TextCompletionResponse response) onProcess,
+    required Future<void> Function() onEnd,
+    required void Function(AIError error) onError,
+    String? suffix,
+    int maxTokens = 2048,
+    double temperature = 0.3,
+    bool useAction = false,
+  });
+
+  Future<void> streamCompletion({
+    String? objectId,
+    required String text,
+    required CompletionTypePB completionType,
+    required Future<void> Function() onStart,
+    required Future<void> Function(String text) onProcess,
+    required Future<void> Function() onEnd,
+    required void Function(AIError error) onError,
+  });
+
+  Future<FlowyResult<List<String>, AIError>> generateImage({
+    required String prompt,
+    int n = 1,
+  });
+}
@@ -1,217 +0,0 @@
-import 'package:appflowy/generated/flowy_svgs.g.dart';
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy_backend/protobuf/flowy-ai/protobuf.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:equatable/equatable.dart';
-import 'package:json_annotation/json_annotation.dart';
-
-part 'ai_entities.g.dart';
-
-class AIStreamEventPrefix {
-  static const data = 'data:';
-  static const error = 'error:';
-  static const metadata = 'metadata:';
-  static const start = 'start:';
-  static const finish = 'finish:';
-  static const comment = 'comment:';
-  static const aiResponseLimit = 'AI_RESPONSE_LIMIT';
-  static const aiImageResponseLimit = 'AI_IMAGE_RESPONSE_LIMIT';
-  static const aiMaxRequired = 'AI_MAX_REQUIRED:';
-  static const localAINotReady = 'LOCAL_AI_NOT_READY';
-  static const localAIDisabled = 'LOCAL_AI_DISABLED';
-}
-
-enum AiType {
-  cloud,
-  local;
-
-  bool get isCloud => this == cloud;
-  bool get isLocal => this == local;
-}
-
-class PredefinedFormat extends Equatable {
-  const PredefinedFormat({
-    required this.imageFormat,
-    required this.textFormat,
-  });
-
-  final ImageFormat imageFormat;
-  final TextFormat? textFormat;
-
-  PredefinedFormatPB toPB() {
-    return PredefinedFormatPB(
-      imageFormat: switch (imageFormat) {
-        ImageFormat.text => ResponseImageFormatPB.TextOnly,
-        ImageFormat.image => ResponseImageFormatPB.ImageOnly,
-        ImageFormat.textAndImage => ResponseImageFormatPB.TextAndImage,
-      },
-      textFormat: switch (textFormat) {
-        TextFormat.paragraph => ResponseTextFormatPB.Paragraph,
-        TextFormat.bulletList => ResponseTextFormatPB.BulletedList,
-        TextFormat.numberedList => ResponseTextFormatPB.NumberedList,
-        TextFormat.table => ResponseTextFormatPB.Table,
-        _ => null,
-      },
-    );
-  }
-
-  @override
-  List<Object?> get props => [imageFormat, textFormat];
-}
-
-enum ImageFormat {
-  text,
-  image,
-  textAndImage;
-
-  bool get hasText => this == text || this == textAndImage;
-
-  FlowySvgData get icon {
-    return switch (this) {
-      ImageFormat.text => FlowySvgs.ai_text_s,
-      ImageFormat.image => FlowySvgs.ai_image_s,
-      ImageFormat.textAndImage => FlowySvgs.ai_text_image_s,
-    };
-  }
-
-  String get i18n {
-    return switch (this) {
-      ImageFormat.text => LocaleKeys.chat_changeFormat_textOnly.tr(),
-      ImageFormat.image => LocaleKeys.chat_changeFormat_imageOnly.tr(),
-      ImageFormat.textAndImage =>
-        LocaleKeys.chat_changeFormat_textAndImage.tr(),
-    };
-  }
-}
-
-enum TextFormat {
-  paragraph,
-  bulletList,
-  numberedList,
-  table;
-
-  FlowySvgData get icon {
-    return switch (this) {
-      TextFormat.paragraph => FlowySvgs.ai_paragraph_s,
-      TextFormat.bulletList => FlowySvgs.ai_list_s,
-      TextFormat.numberedList => FlowySvgs.ai_number_list_s,
-      TextFormat.table => FlowySvgs.ai_table_s,
-    };
-  }
-
-  String get i18n {
-    return switch (this) {
-      TextFormat.paragraph => LocaleKeys.chat_changeFormat_text.tr(),
-      TextFormat.bulletList => LocaleKeys.chat_changeFormat_bullet.tr(),
-      TextFormat.numberedList => LocaleKeys.chat_changeFormat_number.tr(),
-      TextFormat.table => LocaleKeys.chat_changeFormat_table.tr(),
-    };
-  }
-}
-
-enum AiPromptCategory {
-  @JsonValue("other")
-  other,
-  @JsonValue("development")
-  development,
-  @JsonValue("writing")
-  writing,
-  @JsonValue("healthAndFitness")
-  healthAndFitness,
-  @JsonValue("business")
-  business,
-  @JsonValue("marketing")
-  marketing,
-  @JsonValue("travel")
-  travel,
-  @JsonValue("contentSeo")
-  contentSeo,
-  @JsonValue("emailMarketing")
-  emailMarketing,
-  @JsonValue("paidAds")
-  paidAds,
-  @JsonValue("prCommunication")
-  prCommunication,
-  @JsonValue("recruiting")
-  recruiting,
-  @JsonValue("sales")
-  sales,
-  @JsonValue("socialMedia")
-  socialMedia,
-  @JsonValue("strategy")
-  strategy,
-  @JsonValue("caseStudies")
-  caseStudies,
-  @JsonValue("salesCopy")
-  salesCopy,
-  @JsonValue("education")
-  education,
-  @JsonValue("work")
-  work,
-  @JsonValue("podcastProduction")
-  podcastProduction,
-  @JsonValue("copyWriting")
-  copyWriting,
-  @JsonValue("customerSuccess")
-  customerSuccess;
-
-  String get i18n {
-    return switch (this) {
-      other => LocaleKeys.ai_customPrompt_others.tr(),
-      development => LocaleKeys.ai_customPrompt_development.tr(),
-      writing => LocaleKeys.ai_customPrompt_writing.tr(),
-      healthAndFitness => LocaleKeys.ai_customPrompt_healthAndFitness.tr(),
-      business => LocaleKeys.ai_customPrompt_business.tr(),
-      marketing => LocaleKeys.ai_customPrompt_marketing.tr(),
-      travel => LocaleKeys.ai_customPrompt_travel.tr(),
-      contentSeo => LocaleKeys.ai_customPrompt_contentSeo.tr(),
-      emailMarketing => LocaleKeys.ai_customPrompt_emailMarketing.tr(),
-      paidAds => LocaleKeys.ai_customPrompt_paidAds.tr(),
-      prCommunication => LocaleKeys.ai_customPrompt_prCommunication.tr(),
-      recruiting => LocaleKeys.ai_customPrompt_recruiting.tr(),
-      sales => LocaleKeys.ai_customPrompt_sales.tr(),
-      socialMedia => LocaleKeys.ai_customPrompt_socialMedia.tr(),
-      strategy => LocaleKeys.ai_customPrompt_strategy.tr(),
-      caseStudies => LocaleKeys.ai_customPrompt_caseStudies.tr(),
-      salesCopy => LocaleKeys.ai_customPrompt_salesCopy.tr(),
-      education => LocaleKeys.ai_customPrompt_education.tr(),
-      work => LocaleKeys.ai_customPrompt_work.tr(),
-      podcastProduction => LocaleKeys.ai_customPrompt_podcastProduction.tr(),
-      copyWriting => LocaleKeys.ai_customPrompt_copyWriting.tr(),
-      customerSuccess => LocaleKeys.ai_customPrompt_customerSuccess.tr(),
-    };
-  }
-}
-
-@JsonSerializable()
-class AiPrompt extends Equatable {
-  const AiPrompt({
-    required this.id,
-    required this.name,
-    required this.content,
-    required this.category,
-    required this.example,
-    required this.isFeatured,
-  });
-
-  factory AiPrompt.fromJson(Map<String, dynamic> json) =>
-      _$AiPromptFromJson(json);
-
-  Map<String, dynamic> toJson() => _$AiPromptToJson(this);
-
-  final String id;
-  final String name;
-  final String content;
-  @JsonKey(
-    unknownEnumValue: AiPromptCategory.other,
-    defaultValue: AiPromptCategory.other,
-  )
-  final AiPromptCategory category;
-  @JsonKey(defaultValue: "")
-  final String example;
-  @JsonKey(defaultValue: false)
-  final bool isFeatured;
-
-  @override
-  List<Object?> get props => [id, name, content, category, example, isFeatured];
-}
@@ -1,214 +0,0 @@
-import 'package:appflowy/ai/ai.dart';
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/plugins/ai_chat/application/ai_model_switch_listener.dart';
-import 'package:appflowy/workspace/application/settings/ai/local_llm_listener.dart';
-import 'package:appflowy_backend/dispatch/dispatch.dart';
-import 'package:appflowy_backend/log.dart';
-import 'package:appflowy_backend/protobuf/flowy-ai/entities.pb.dart';
-import 'package:appflowy_result/appflowy_result.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:universal_platform/universal_platform.dart';
-
-typedef OnModelStateChangedCallback = void Function(AIModelState state);
-typedef OnAvailableModelsChangedCallback = void Function(
-  List<AIModelPB>,
-  AIModelPB?,
-);
-
-/// Represents the state of an AI model
-class AIModelState {
-  const AIModelState({
-    required this.type,
-    required this.hintText,
-    required this.tooltip,
-    required this.isEditable,
-    required this.localAIEnabled,
-  });
-  final AiType type;
-
-  /// The text displayed as placeholder/hint in the input field
-  /// Shows different messages based on AI state (enabled, initializing, disabled)
-  final String hintText;
-
-  /// Optional tooltip text that appears on hover
-  /// Provides additional context about the current state of the AI
-  /// Null when no tooltip should be shown
-  final String? tooltip;
-
-  final bool isEditable;
-  final bool localAIEnabled;
-}
-
-class AIModelStateNotifier {
-  AIModelStateNotifier({required this.objectId})
-      : _localAIListener =
-            UniversalPlatform.isDesktop ? LocalAIStateListener() : null,
-        _aiModelSwitchListener = AIModelSwitchListener(objectId: objectId) {
-    _startListening();
-    _init();
-  }
-
-  final String objectId;
-  final LocalAIStateListener? _localAIListener;
-  final AIModelSwitchListener _aiModelSwitchListener;
-
-  LocalAIPB? _localAIState;
-  ModelSelectionPB? _modelSelection;
-
-  AIModelState _currentState = _defaultState();
-  List<AIModelPB> _availableModels = [];
-  AIModelPB? _selectedModel;
-
-  final List<OnModelStateChangedCallback> _stateChangedCallbacks = [];
-  final List<OnAvailableModelsChangedCallback>
-      _availableModelsChangedCallbacks = [];
-
-  /// Starts platform-specific listeners
-  void _startListening() {
-    if (UniversalPlatform.isDesktop) {
-      _localAIListener?.start(
-        stateCallback: (state) async {
-          _localAIState = state;
-          _updateAll();
-        },
-      );
-    }
-
-    _aiModelSwitchListener.start(
-      onUpdateSelectedModel: (model) async {
-        _selectedModel = model;
-        _updateAll();
-        if (model.isLocal && UniversalPlatform.isDesktop) {
-          await _loadLocalState();
-          _updateAll();
-        }
-      },
-    );
-  }
-
-  Future<void> _init() async {
-    await Future.wait([
-      if (UniversalPlatform.isDesktop) _loadLocalState(),
-      _loadModelSelection(),
-    ]);
-    _updateAll();
-  }
-
-  /// Register callbacks for state or available-models changes
-  void addListener({
-    OnModelStateChangedCallback? onStateChanged,
-    OnAvailableModelsChangedCallback? onAvailableModelsChanged,
-  }) {
-    if (onStateChanged != null) {
-      _stateChangedCallbacks.add(onStateChanged);
-    }
-    if (onAvailableModelsChanged != null) {
-      _availableModelsChangedCallbacks.add(onAvailableModelsChanged);
-    }
-  }
-
-  /// Remove previously registered callbacks
-  void removeListener({
-    OnModelStateChangedCallback? onStateChanged,
-    OnAvailableModelsChangedCallback? onAvailableModelsChanged,
-  }) {
-    if (onStateChanged != null) {
-      _stateChangedCallbacks.remove(onStateChanged);
-    }
-    if (onAvailableModelsChanged != null) {
-      _availableModelsChangedCallbacks.remove(onAvailableModelsChanged);
-    }
-  }
-
-  Future<void> dispose() async {
-    _stateChangedCallbacks.clear();
-    _availableModelsChangedCallbacks.clear();
-    await _localAIListener?.stop();
-    await _aiModelSwitchListener.stop();
-  }
-
-  /// Returns current AIModelState
-  AIModelState getState() => _currentState;
-
-  /// Returns available models and the selected model
-  (List<AIModelPB>, AIModelPB?) getModelSelection() =>
-      (_availableModels, _selectedModel);
-
-  void _updateAll() {
-    _currentState = _computeState();
-    for (final cb in _stateChangedCallbacks) {
-      cb(_currentState);
-    }
-    for (final cb in _availableModelsChangedCallbacks) {
-      cb(_availableModels, _selectedModel);
-    }
-  }
-
-  Future<void> _loadModelSelection() async {
-    await AIEventGetSourceModelSelection(
-      ModelSourcePB(source: objectId),
-    ).send().fold(
-      (ms) {
-        _modelSelection = ms;
-        _availableModels = ms.models;
-        _selectedModel = ms.selectedModel;
-      },
-      (e) => Log.error("Failed to fetch models: \$e"),
-    );
-  }
-
-  Future<void> _loadLocalState() async {
-    await AIEventGetLocalAIState().send().fold(
-          (s) => _localAIState = s,
-          (e) => Log.error("Failed to fetch local AI state: \$e"),
-        );
-  }
-
-  static AIModelState _defaultState() => AIModelState(
-        type: AiType.cloud,
-        hintText: LocaleKeys.chat_inputMessageHint.tr(),
-        tooltip: null,
-        isEditable: true,
-        localAIEnabled: false,
-      );
-
-  /// Core logic computing the state from local and selection data
-  AIModelState _computeState() {
-    if (UniversalPlatform.isMobile) return _defaultState();
-
-    if (_modelSelection == null || _localAIState == null) {
-      return _defaultState();
-    }
-
-    if (!_selectedModel!.isLocal) {
-      return _defaultState();
-    }
-
-    final enabled = _localAIState!.enabled;
-    final running = _localAIState!.isReady;
-    final hintKey = enabled
-        ? (running
-            ? LocaleKeys.chat_inputLocalAIMessageHint
-            : LocaleKeys.settings_aiPage_keys_localAIInitializing)
-        : LocaleKeys.settings_aiPage_keys_localAIDisabled;
-    final tooltipKey = enabled
-        ? (running
-            ? null
-            : LocaleKeys.settings_aiPage_keys_localAINotReadyTextFieldPrompt)
-        : LocaleKeys.settings_aiPage_keys_localAIDisabledTextFieldPrompt;
-
-    return AIModelState(
-      type: AiType.local,
-      hintText: hintKey.tr(),
-      tooltip: tooltipKey?.tr(),
-      isEditable: running,
-      localAIEnabled: enabled,
-    );
-  }
-}
-
-extension AIModelPBExtension on AIModelPB {
-  bool get isDefault => name == 'Auto';
-  String get i18n =>
-      isDefault ? LocaleKeys.chat_switchModel_autoModel.tr() : name;
-}
@@ -1,200 +0,0 @@
-import 'package:bloc/bloc.dart';
-import 'package:collection/collection.dart';
-import 'package:flutter/widgets.dart';
-import 'package:freezed_annotation/freezed_annotation.dart';
-
-import 'ai_entities.dart';
-import 'appflowy_ai_service.dart';
-
-part 'ai_prompt_selector_cubit.freezed.dart';
-
-class AiPromptSelectorCubit extends Cubit<AiPromptSelectorState> {
-  AiPromptSelectorCubit({
-    AppFlowyAIService? aiService,
-  })  : _aiService = aiService ?? AppFlowyAIService(),
-        super(AiPromptSelectorState.loading()) {
-    filterTextController.addListener(_filterTextChanged);
-    _loadPrompts();
-  }
-
-  final AppFlowyAIService _aiService;
-  final filterTextController = TextEditingController();
-  final List<AiPrompt> availablePrompts = [];
-
-  @override
-  Future<void> close() async {
-    filterTextController.dispose();
-    await super.close();
-  }
-
-  void _loadPrompts() async {
-    availablePrompts.addAll(await _aiService.getBuiltInPrompts());
-    final featuredPrompts =
-        availablePrompts.where((prompt) => prompt.isFeatured);
-    final visiblePrompts = _getFilteredPrompts(featuredPrompts);
-    emit(
-      AiPromptSelectorState.ready(
-        visiblePrompts: visiblePrompts,
-        showCustomPrompts: false,
-        isFeaturedCategorySelected: true,
-        selectedPromptId: visiblePrompts.firstOrNull?.id,
-        selectedCategory: null,
-        favoritePrompts: [],
-      ),
-    );
-  }
-
-  void selectFeaturedCategory() {
-    state.maybeMap(
-      ready: (readyState) {
-        final prompts = availablePrompts.where((prompt) => prompt.isFeatured);
-        final visiblePrompts = _getFilteredPrompts(prompts);
-        emit(
-          readyState.copyWith(
-            visiblePrompts: visiblePrompts,
-            isFeaturedCategorySelected: true,
-            selectedPromptId: visiblePrompts.firstOrNull?.id,
-          ),
-        );
-      },
-      orElse: () {},
-    );
-  }
-
-  void selectCategory(AiPromptCategory? category) {
-    state.maybeMap(
-      ready: (readyState) {
-        final unfilteredPrompts = category == null
-            ? availablePrompts
-            : availablePrompts.where((prompt) => prompt.category == category);
-        final prompts = _getFilteredPrompts(unfilteredPrompts);
-
-        // Check if the selected prompt is still visible
-        String? selectedPromptId = prompts.firstOrNull?.id;
-        if (readyState.selectedPromptId != null &&
-            prompts.any((prompt) => prompt.id == readyState.selectedPromptId)) {
-          selectedPromptId = readyState.selectedPromptId;
-        }
-
-        emit(
-          readyState.copyWith(
-            visiblePrompts: prompts,
-            isFeaturedCategorySelected: false,
-            selectedCategory: category,
-            selectedPromptId: selectedPromptId,
-          ),
-        );
-      },
-      orElse: () {},
-    );
-  }
-
-  void selectPrompt(String promptId) {
-    state.maybeMap(
-      ready: (readyState) {
-        final selectedPrompt = readyState.visiblePrompts
-            .firstWhereOrNull((prompt) => prompt.id == promptId);
-        if (selectedPrompt != null) {
-          emit(
-            readyState.copyWith(selectedPromptId: selectedPrompt.id),
-          );
-        }
-      },
-      orElse: () {},
-    );
-  }
-
-  void toggleFavorite(String promptId) {
-    state.maybeMap(
-      ready: (readyState) {
-        final favoritePrompts = [...readyState.favoritePrompts];
-        if (favoritePrompts.contains(promptId)) {
-          favoritePrompts.remove(promptId);
-        } else {
-          favoritePrompts.add(promptId);
-        }
-        emit(
-          readyState.copyWith(favoritePrompts: favoritePrompts),
-        );
-
-        // TODO(RS): Save the updated favorite prompts to local storage or database
-      },
-      orElse: () {},
-    );
-  }
-
-  void reset() {
-    filterTextController.clear();
-    state.maybeMap(
-      ready: (readyState) {
-        emit(
-          readyState.copyWith(
-            visiblePrompts: availablePrompts,
-            selectedPromptId: null,
-            selectedCategory: null,
-          ),
-        );
-      },
-      orElse: () {},
-    );
-  }
-
-  void _filterTextChanged() {
-    state.maybeMap(
-      ready: (readyState) {
-        final unfilteredPrompts = readyState.selectedCategory == null
-            ? readyState.isFeaturedCategorySelected
-                ? availablePrompts.where((prompt) => prompt.isFeatured)
-                : availablePrompts
-            : availablePrompts.where(
-                (prompt) => prompt.category == readyState.selectedCategory,
-              );
-        final filteredPrompts = _getFilteredPrompts(unfilteredPrompts);
-
-        emit(
-          readyState.copyWith(visiblePrompts: filteredPrompts),
-        );
-      },
-      orElse: () {},
-    );
-  }
-
-  List<AiPrompt> _getFilteredPrompts(Iterable<AiPrompt> prompts) {
-    final filterText = filterTextController.value.text.trim().toLowerCase();
-
-    return prompts.where((prompt) {
-      final content = "${prompt.name} ${prompt.name}".toLowerCase();
-      return content.contains(filterText);
-    }).toList();
-  }
-}
-
-@freezed
-class AiPromptSelectorState with _$AiPromptSelectorState {
-  const AiPromptSelectorState._();
-
-  const factory AiPromptSelectorState.loading() = _AiPromptSelectorLoadingState;
-
-  const factory AiPromptSelectorState.ready({
-    required List<AiPrompt> visiblePrompts,
-    required List<String> favoritePrompts,
-    required bool showCustomPrompts,
-    required bool isFeaturedCategorySelected,
-    required AiPromptCategory? selectedCategory,
-    required String? selectedPromptId,
-  }) = _AiPromptSelectorReadyState;
-
-  bool get isLoading => this is _AiPromptSelectorLoadingState;
-  bool get isReady => this is _AiPromptSelectorReadyState;
-
-  AiPrompt? get selectedPrompt => maybeMap(
-        ready: (state) => state.visiblePrompts
-            .firstWhereOrNull((prompt) => prompt.id == state.selectedPromptId),
-        orElse: () => null,
-      );
-
-  AiPromptCategory? get selectedCategory => maybeMap(
-        ready: (state) => state.selectedCategory,
-        orElse: () => null,
-      );
-}
@@ -1,236 +1,195 @@
 import 'dart:async';
-import 'dart:convert';
 import 'dart:ffi';
 import 'dart:isolate';

 import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/plugins/document/presentation/editor_plugins/ai/operations/ai_writer_entities.dart';
-import 'package:appflowy/shared/list_extension.dart';
 import 'package:appflowy_backend/dispatch/dispatch.dart';
-import 'package:appflowy_backend/log.dart';
 import 'package:appflowy_backend/protobuf/flowy-ai/entities.pb.dart';
 import 'package:appflowy_result/appflowy_result.dart';
 import 'package:easy_localization/easy_localization.dart';
 import 'package:fixnum/fixnum.dart' as fixnum;
-import 'package:flutter/services.dart';

-import 'ai_entities.dart';
+import 'ai_client.dart';
 import 'error.dart';
+import 'text_completion.dart';

-enum LocalAIStreamingState {
-  notReady,
-  disabled,
-}
+enum AskAIAction {
+  summarize,
+  fixSpelling,
+  improveWriting,
+  makeItLonger;

-abstract class AIRepository {
-  Future<(String, CompletionStream)?> streamCompletion({
-    String? objectId,
-    required String text,
-    PredefinedFormat? format,
-    List<String> sourceIds = const [],
-    List<AiWriterRecord> history = const [],
-    required CompletionTypePB completionType,
-    required Future<void> Function() onStart,
-    required Future<void> Function(String text) processMessage,
-    required Future<void> Function(String text) processAssistMessage,
-    required Future<void> Function() onEnd,
-    required void Function(AIError error) onError,
-    required void Function(LocalAIStreamingState state)
-        onLocalAIStreamingStateChange,
-  });
+  String get toInstruction => switch (this) {
+        summarize => 'Tl;dr',
+        fixSpelling => 'Correct this to standard English:',
+        improveWriting => 'Rewrite this in your own words:',
+        makeItLonger => 'Make this text longer:',
+      };

-  Future<List<AiPrompt>> getBuiltInPrompts();
+  String prompt(String input) => switch (this) {
+        summarize => '$input\n\n$toInstruction',
+        _ => "$toInstruction\n\n$input",
+      };

-  void updateFavoritePrompts(List<String> promptIds);
+  static AskAIAction from(int index) => switch (index) {
+        0 => summarize,
+        1 => fixSpelling,
+        2 => improveWriting,
+        3 => makeItLonger,
+        _ => fixSpelling
+      };
+
+  String get name => switch (this) {
+        summarize => LocaleKeys.document_plugins_smartEditSummarize.tr(),
+        fixSpelling => LocaleKeys.document_plugins_smartEditFixSpelling.tr(),
+        improveWriting =>
+          LocaleKeys.document_plugins_smartEditImproveWriting.tr(),
+        makeItLonger => LocaleKeys.document_plugins_smartEditMakeLonger.tr(),
+      };
 }

 class AppFlowyAIService implements AIRepository {
  @override
-  Future<(String, CompletionStream)?> streamCompletion({
-    String? objectId,
-    required String text,
-    PredefinedFormat? format,
-    List<String> sourceIds = const [],
-    List<AiWriterRecord> history = const [],
-    required CompletionTypePB completionType,
+  Future<FlowyResult<List<String>, AIError>> generateImage({
+    required String prompt,
+    int n = 1,
+  }) {
+    throw UnimplementedError();
+  }
+
+  @override
+  Future<void> getStreamedCompletions({
+    required String prompt,
    required Future<void> Function() onStart,
-    required Future<void> Function(String text) processMessage,
-    required Future<void> Function(String text) processAssistMessage,
+    required Future<void> Function(TextCompletionResponse response) onProcess,
    required Future<void> Function() onEnd,
    required void Function(AIError error) onError,
-    required void Function(LocalAIStreamingState state)
-        onLocalAIStreamingStateChange,
-  }) async {
-    final stream = AppFlowyCompletionStream(
-      onStart: onStart,
-      processMessage: processMessage,
-      processAssistMessage: processAssistMessage,
-      processError: onError,
-      onLocalAIStreamingStateChange: onLocalAIStreamingStateChange,
-      onEnd: onEnd,
-    );
+    String? suffix,
+    int maxTokens = 2048,
+    double temperature = 0.3,
+    bool useAction = false,
+  }) {
+    throw UnimplementedError();
+  }

-    final records = history.map((record) => record.toPB()).toList();
+  @override
+  Future<CompletionStream> streamCompletion({
+    String? objectId,
+    required String text,
+    required CompletionTypePB completionType,
+    required Future<void> Function() onStart,
+    required Future<void> Function(String text) onProcess,
+    required Future<void> Function() onEnd,
+    required void Function(AIError error) onError,
+  }) async {
+    final stream = CompletionStream(
+      onStart,
+      onProcess,
+      onEnd,
+      onError,
+    );
+    final List<String> ragIds = [];
+    if (objectId != null) {
+      ragIds.add(objectId);
+    }

    final payload = CompleteTextPB(
      text: text,
      completionType: completionType,
-      format: format?.toPB(),
      streamPort: fixnum.Int64(stream.nativePort),
-      objectId: objectId ?? '',
-      ragIds: [
-        if (objectId != null) objectId,
-        ...sourceIds,
-      ].unique(),
-      history: records,
+      objectId: objectId ?? "",
+      ragIds: ragIds,
    );

-    return AIEventCompleteText(payload).send().fold(
-      (task) => (task.taskId, stream),
-      (error) {
-        Log.error(error);
-        return null;
+    // ignore: unawaited_futures
+    AIEventCompleteText(payload).send();
+    return stream;
+  }
+}
+
+CompletionTypePB completionTypeFromInt(AskAIAction action) {
+  switch (action) {
+    case AskAIAction.summarize:
+      return CompletionTypePB.MakeShorter;
+    case AskAIAction.fixSpelling:
+      return CompletionTypePB.SpellingAndGrammar;
+    case AskAIAction.improveWriting:
+      return CompletionTypePB.ImproveWriting;
+    case AskAIAction.makeItLonger:
+      return CompletionTypePB.MakeLonger;
+  }
+}
+
+class CompletionStream {
+  CompletionStream(
+    Future<void> Function() onStart,
+    Future<void> Function(String text) onProcess,
+    Future<void> Function() onEnd,
+    void Function(AIError error) onError,
+  ) {
+    _port.handler = _controller.add;
+    _subscription = _controller.stream.listen(
+      (event) async {
+        if (event == "AI_RESPONSE_LIMIT") {
+          onError(
+            AIError(
+              message: LocaleKeys.sideBar_aiResponseLimit.tr(),
+              code: AIErrorCode.aiResponseLimitExceeded,
+            ),
+          );
+        }
+
+        if (event == "AI_IMAGE_RESPONSE_LIMIT") {
+          onError(
+            AIError(
+              message: LocaleKeys.sideBar_aiImageResponseLimit.tr(),
+              code: AIErrorCode.aiImageResponseLimitExceeded,
+            ),
+          );
+        }
+
+        if (event.startsWith("AI_MAX_REQUIRED:")) {
+          final msg = event.substring(16);
+          onError(
+            AIError(
+              message: msg,
+            ),
+          );
+        }
+
+        if (event.startsWith("start:")) {
+          await onStart();
+        }
+
+        if (event.startsWith("data:")) {
+          await onProcess(event.substring(5));
+        }
+
+        if (event.startsWith("finish:")) {
+          await onEnd();
+        }
+
+        if (event.startsWith("error:")) {
+          onError(AIError(message: event.substring(6)));
+        }
      },
    );
  }

-  @override
-  Future<List<AiPrompt>> getBuiltInPrompts() async {
-    final prompts = <AiPrompt>[];
-
-    try {
-      final jsonString =
-          await rootBundle.loadString('assets/built_in_prompts.json');
-      // final data = await rootBundle.load('assets/built_in_prompts.json');
-      // final jsonString = utf8.decode(data.buffer.asUint8List());
-      final jsonData = json.decode(jsonString) as Map<String, dynamic>;
-      final promptJson = jsonData['prompts'] as List<dynamic>;
-      prompts.addAll(
-        promptJson
-            .map((e) => AiPrompt.fromJson(e as Map<String, dynamic>))
-            .toList(),
-      );
-    } catch (e) {
-      Log.error(e);
-    }
-
-    return prompts;
-  }
-
-  @override
-  void updateFavoritePrompts(List<String> promptIds) {}
-}
-
-abstract class CompletionStream {
-  CompletionStream({
-    required this.onStart,
-    required this.processMessage,
-    required this.processAssistMessage,
-    required this.processError,
-    required this.onLocalAIStreamingStateChange,
-    required this.onEnd,
-  });
-
-  final Future<void> Function() onStart;
-  final Future<void> Function(String text) processMessage;
-  final Future<void> Function(String text) processAssistMessage;
-  final void Function(AIError error) processError;
-  final void Function(LocalAIStreamingState state)
-      onLocalAIStreamingStateChange;
-  final Future<void> Function() onEnd;
-}
-
-class AppFlowyCompletionStream extends CompletionStream {
-  AppFlowyCompletionStream({
-    required super.onStart,
-    required super.processMessage,
-    required super.processAssistMessage,
-    required super.processError,
-    required super.onEnd,
-    required super.onLocalAIStreamingStateChange,
-  }) {
-    _startListening();
-  }
-
  final RawReceivePort _port = RawReceivePort();
  final StreamController<String> _controller = StreamController.broadcast();
  late StreamSubscription<String> _subscription;
  int get nativePort => _port.sendPort.nativePort;

-  void _startListening() {
-    _port.handler = _controller.add;
-    _subscription = _controller.stream.listen(
-      (event) async {
-        await _handleEvent(event);
-      },
-    );
-  }
-
  Future<void> dispose() async {
    await _controller.close();
    await _subscription.cancel();
    _port.close();
  }

-  Future<void> _handleEvent(String event) async {
-    // Check simple matches first
-    if (event == AIStreamEventPrefix.aiResponseLimit) {
-      processError(
-        AIError(
-          message: LocaleKeys.ai_textLimitReachedDescription.tr(),
-          code: AIErrorCode.aiResponseLimitExceeded,
-        ),
-      );
-      return;
-    }
-
-    if (event == AIStreamEventPrefix.aiImageResponseLimit) {
-      processError(
-        AIError(
-          message: LocaleKeys.ai_imageLimitReachedDescription.tr(),
-          code: AIErrorCode.aiImageResponseLimitExceeded,
-        ),
-      );
-      return;
-    }
-
-    // Otherwise, parse out prefix:content
-    if (event.startsWith(AIStreamEventPrefix.aiMaxRequired)) {
-      processError(
-        AIError(
-          message: event.substring(AIStreamEventPrefix.aiMaxRequired.length),
-          code: AIErrorCode.other,
-        ),
-      );
-    } else if (event.startsWith(AIStreamEventPrefix.start)) {
-      await onStart();
-    } else if (event.startsWith(AIStreamEventPrefix.data)) {
-      await processMessage(
-        event.substring(AIStreamEventPrefix.data.length),
-      );
-    } else if (event.startsWith(AIStreamEventPrefix.comment)) {
-      await processAssistMessage(
-        event.substring(AIStreamEventPrefix.comment.length),
-      );
-    } else if (event.startsWith(AIStreamEventPrefix.finish)) {
-      await onEnd();
-    } else if (event.startsWith(AIStreamEventPrefix.localAIDisabled)) {
-      onLocalAIStreamingStateChange(
-        LocalAIStreamingState.disabled,
-      );
-    } else if (event.startsWith(AIStreamEventPrefix.localAINotReady)) {
-      onLocalAIStreamingStateChange(
-        LocalAIStreamingState.notReady,
-      );
-    } else if (event.startsWith(AIStreamEventPrefix.error)) {
-      processError(
-        AIError(
-          message: event.substring(AIStreamEventPrefix.error.length),
-          code: AIErrorCode.other,
-        ),
-      );
-    } else {
-      Log.debug('Unknown AI event: $event');
-    }
+  StreamSubscription<String> listen(
+    void Function(String event)? onData,
+  ) {
+    return _controller.stream.listen(onData);
  }
 }
@@ -7,7 +7,7 @@ part 'error.g.dart';
 class AIError with _$AIError {
  const factory AIError({
    required String message,
-    required AIErrorCode code,
+    @Default(AIErrorCode.other) AIErrorCode code,
  }) = _AIError;

  factory AIError.fromJson(Map<String, Object?> json) =>
@@ -0,0 +1,173 @@
+import 'dart:async';
+import 'dart:convert';
+
+import 'package:appflowy_backend/protobuf/flowy-ai/entities.pbenum.dart';
+import 'package:appflowy_result/appflowy_result.dart';
+import 'package:http/http.dart' as http;
+
+import 'ai_client.dart';
+import 'error.dart';
+import 'text_completion.dart';
+
+enum OpenAIRequestType {
+  textCompletion,
+  textEdit,
+  imageGenerations;
+
+  Uri get uri {
+    switch (this) {
+      case OpenAIRequestType.textCompletion:
+        return Uri.parse('https://api.openai.com/v1/completions');
+      case OpenAIRequestType.textEdit:
+        return Uri.parse('https://api.openai.com/v1/chat/completions');
+      case OpenAIRequestType.imageGenerations:
+        return Uri.parse('https://api.openai.com/v1/images/generations');
+    }
+  }
+}
+
+class HttpOpenAIRepository implements AIRepository {
+  const HttpOpenAIRepository({
+    required this.client,
+    required this.apiKey,
+  });
+
+  final http.Client client;
+  final String apiKey;
+
+  Map<String, String> get headers => {
+        'Authorization': 'Bearer $apiKey',
+        'Content-Type': 'application/json',
+      };
+
+  @override
+  Future<void> getStreamedCompletions({
+    required String prompt,
+    required Future<void> Function() onStart,
+    required Future<void> Function(TextCompletionResponse response) onProcess,
+    required Future<void> Function() onEnd,
+    required void Function(AIError error) onError,
+    String? suffix,
+    int maxTokens = 2048,
+    double temperature = 0.3,
+    bool useAction = false,
+  }) async {
+    final parameters = {
+      'model': 'gpt-3.5-turbo-instruct',
+      'prompt': prompt,
+      'suffix': suffix,
+      'max_tokens': maxTokens,
+      'temperature': temperature,
+      'stream': true,
+    };
+
+    final request = http.Request('POST', OpenAIRequestType.textCompletion.uri);
+    request.headers.addAll(headers);
+    request.body = jsonEncode(parameters);
+
+    final response = await client.send(request);
+
+    // NEED TO REFACTOR.
+    // WHY OPENAI USE TWO LINES TO INDICATE THE START OF THE STREAMING RESPONSE?
+    // AND WHY OPENAI USE [DONE] TO INDICATE THE END OF THE STREAMING RESPONSE?
+    int syntax = 0;
+    var previousSyntax = '';
+    if (response.statusCode == 200) {
+      await for (final chunk in response.stream
+          .transform(const Utf8Decoder())
+          .transform(const LineSplitter())) {
+        syntax += 1;
+        if (!useAction) {
+          if (syntax == 3) {
+            await onStart();
+            continue;
+          } else if (syntax < 3) {
+            continue;
+          }
+        } else {
+          if (syntax == 2) {
+            await onStart();
+            continue;
+          } else if (syntax < 2) {
+            continue;
+          }
+        }
+        final data = chunk.trim().split('data: ');
+        if (data.length > 1) {
+          if (data[1] != '[DONE]') {
+            final response = TextCompletionResponse.fromJson(
+              json.decode(data[1]),
+            );
+            if (response.choices.isNotEmpty) {
+              final text = response.choices.first.text;
+              if (text == previousSyntax && text == '\n') {
+                continue;
+              }
+              await onProcess(response);
+              previousSyntax = response.choices.first.text;
+            }
+          } else {
+            await onEnd();
+          }
+        }
+      }
+    } else {
+      final body = await response.stream.bytesToString();
+      onError(
+        AIError.fromJson(json.decode(body)['error']),
+      );
+    }
+    return;
+  }
+
+  @override
+  Future<FlowyResult<List<String>, AIError>> generateImage({
+    required String prompt,
+    int n = 1,
+  }) async {
+    final parameters = {
+      'prompt': prompt,
+      'n': n,
+      'size': '512x512',
+    };
+
+    try {
+      final response = await client.post(
+        OpenAIRequestType.imageGenerations.uri,
+        headers: headers,
+        body: json.encode(parameters),
+      );
+
+      if (response.statusCode == 200) {
+        final data = json.decode(
+          utf8.decode(response.bodyBytes),
+        )['data'] as List;
+        final urls = data
+            .map((e) => e.values)
+            .expand((e) => e)
+            .map((e) => e.toString())
+            .toList();
+        return FlowyResult.success(urls);
+      } else {
+        return FlowyResult.failure(
+          AIError.fromJson(json.decode(response.body)['error']),
+        );
+      }
+    } catch (error) {
+      return FlowyResult.failure(AIError(message: error.toString()));
+    }
+  }
+
+  @override
+  Future<void> streamCompletion({
+    String? objectId,
+    required String text,
+    required CompletionTypePB completionType,
+    required Future<void> Function() onStart,
+    required Future<void> Function(String text) onProcess,
+    required Future<void> Function() onEnd,
+    required void Function(AIError error) onError,
+  }) {
+    throw UnimplementedError();
+  }
+}
@@ -1,92 +0,0 @@
-import 'dart:async';
-
-import 'package:appflowy/ai/service/ai_model_state_notifier.dart';
-import 'package:appflowy_backend/dispatch/dispatch.dart';
-import 'package:appflowy_backend/protobuf/flowy-ai/entities.pbserver.dart';
-import 'package:flutter_bloc/flutter_bloc.dart';
-import 'package:freezed_annotation/freezed_annotation.dart';
-
-part 'select_model_bloc.freezed.dart';
-
-class SelectModelBloc extends Bloc<SelectModelEvent, SelectModelState> {
-  SelectModelBloc({
-    required AIModelStateNotifier aiModelStateNotifier,
-  })  : _aiModelStateNotifier = aiModelStateNotifier,
-        super(SelectModelState.initial(aiModelStateNotifier)) {
-    on<SelectModelEvent>(
-      (event, emit) {
-        event.when(
-          selectModel: (model) {
-            AIEventUpdateSelectedModel(
-              UpdateSelectedModelPB(
-                source: _aiModelStateNotifier.objectId,
-                selectedModel: model,
-              ),
-            ).send();
-
-            emit(state.copyWith(selectedModel: model));
-          },
-          didLoadModels: (models, selectedModel) {
-            emit(
-              SelectModelState(
-                models: models,
-                selectedModel: selectedModel,
-              ),
-            );
-          },
-        );
-      },
-    );
-
-    _aiModelStateNotifier.addListener(
-      onAvailableModelsChanged: _onAvailableModelsChanged,
-    );
-  }
-
-  final AIModelStateNotifier _aiModelStateNotifier;
-
-  @override
-  Future<void> close() async {
-    _aiModelStateNotifier.removeListener(
-      onAvailableModelsChanged: _onAvailableModelsChanged,
-    );
-    await super.close();
-  }
-
-  void _onAvailableModelsChanged(
-    List<AIModelPB> models,
-    AIModelPB? selectedModel,
-  ) {
-    if (!isClosed) {
-      add(SelectModelEvent.didLoadModels(models, selectedModel));
-    }
-  }
-}
-
-@freezed
-class SelectModelEvent with _$SelectModelEvent {
-  const factory SelectModelEvent.selectModel(
-    AIModelPB model,
-  ) = _SelectModel;
-
-  const factory SelectModelEvent.didLoadModels(
-    List<AIModelPB> models,
-    AIModelPB? selectedModel,
-  ) = _DidLoadModels;
-}
-
-@freezed
-class SelectModelState with _$SelectModelState {
-  const factory SelectModelState({
-    required List<AIModelPB> models,
-    required AIModelPB? selectedModel,
-  }) = _SelectModelState;
-
-  factory SelectModelState.initial(AIModelStateNotifier notifier) {
-    final (models, selectedModel) = notifier.getModelSelection();
-    return SelectModelState(
-      models: models,
-      selectedModel: selectedModel,
-    );
-  }
-}
@@ -0,0 +1,27 @@
+import 'package:freezed_annotation/freezed_annotation.dart';
+
+part 'text_completion.freezed.dart';
+part 'text_completion.g.dart';
+
+@freezed
+class TextCompletionChoice with _$TextCompletionChoice {
+  factory TextCompletionChoice({
+    required String text,
+    required int index,
+    // ignore: invalid_annotation_target
+    @JsonKey(name: 'finish_reason') String? finishReason,
+  }) = _TextCompletionChoice;
+
+  factory TextCompletionChoice.fromJson(Map<String, Object?> json) =>
+      _$TextCompletionChoiceFromJson(json);
+}
+
+@freezed
+class TextCompletionResponse with _$TextCompletionResponse {
+  const factory TextCompletionResponse({
+    required List<TextCompletionChoice> choices,
+  }) = _TextCompletionResponse;
+
+  factory TextCompletionResponse.fromJson(Map<String, Object?> json) =>
+      _$TextCompletionResponseFromJson(json);
+}
@@ -1,183 +0,0 @@
-import 'package:appflowy/ai/ai.dart';
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy_ui/appflowy_ui.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:flutter/material.dart';
-import 'package:flutter_bloc/flutter_bloc.dart';
-
-class AiPromptCategoryList extends StatefulWidget {
-  const AiPromptCategoryList({
-    super.key,
-    required this.padding,
-  });
-
-  final EdgeInsetsGeometry padding;
-
-  @override
-  State<AiPromptCategoryList> createState() => _AiPromptCategoryListState();
-}
-
-class _AiPromptCategoryListState extends State<AiPromptCategoryList> {
-  bool isSearching = false;
-  @override
-  Widget build(BuildContext context) {
-    final theme = AppFlowyTheme.of(context);
-
-    return Padding(
-      padding: EdgeInsets.all(
-        theme.spacing.l,
-      ),
-      child: TextFieldTapRegion(
-        groupId: "ai_prompt_category_list",
-        child: Column(
-          crossAxisAlignment: CrossAxisAlignment.stretch,
-          spacing: theme.spacing.s,
-          children: [
-            _buildFeaturedCategory(context),
-            const AFDivider(),
-            Expanded(
-              child: ListView(
-                padding: EdgeInsets.zero,
-                children: [
-                  _buildCategoryItem(context, null),
-                  _buildCategoryItem(context, AiPromptCategory.development),
-                  _buildCategoryItem(context, AiPromptCategory.writing),
-                  _buildCategoryItem(context, AiPromptCategory.business),
-                  _buildCategoryItem(context, AiPromptCategory.marketing),
-                  _buildCategoryItem(
-                    context,
-                    AiPromptCategory.healthAndFitness,
-                  ),
-                  _buildCategoryItem(context, AiPromptCategory.travel),
-                  _buildCategoryItem(context, AiPromptCategory.contentSeo),
-                  _buildCategoryItem(context, AiPromptCategory.emailMarketing),
-                  _buildCategoryItem(context, AiPromptCategory.paidAds),
-                  _buildCategoryItem(context, AiPromptCategory.prCommunication),
-                  _buildCategoryItem(context, AiPromptCategory.recruiting),
-                  _buildCategoryItem(context, AiPromptCategory.sales),
-                  _buildCategoryItem(context, AiPromptCategory.socialMedia),
-                  _buildCategoryItem(context, AiPromptCategory.strategy),
-                  _buildCategoryItem(context, AiPromptCategory.caseStudies),
-                  _buildCategoryItem(context, AiPromptCategory.salesCopy),
-                  _buildCategoryItem(context, AiPromptCategory.education),
-                  _buildCategoryItem(context, AiPromptCategory.work),
-                  _buildCategoryItem(
-                    context,
-                    AiPromptCategory.podcastProduction,
-                  ),
-                  _buildCategoryItem(context, AiPromptCategory.copyWriting),
-                  _buildCategoryItem(context, AiPromptCategory.customerSuccess),
-                  _buildCategoryItem(context, AiPromptCategory.other),
-                ],
-              ),
-            ),
-          ],
-        ),
-      ),
-    );
-  }
-
-  Widget _buildCategoryItem(
-    BuildContext context,
-    AiPromptCategory? category,
-  ) {
-    return AiPromptCategoryItem(
-      category: category,
-      onSelect: () {
-        context.read<AiPromptSelectorCubit>().selectCategory(category);
-      },
-    );
-  }
-
-  Widget _buildFeaturedCategory(BuildContext context) {
-    final theme = AppFlowyTheme.of(context);
-    final isSelected = context.watch<AiPromptSelectorCubit>().state.maybeMap(
-          ready: (state) => state.isFeaturedCategorySelected,
-          orElse: () => false,
-        );
-
-    return AFBaseButton(
-      onTap: () {
-        if (!isSelected) {
-          context.read<AiPromptSelectorCubit>().selectFeaturedCategory();
-        }
-      },
-      builder: (context, isHovering, disabled) {
-        return Text(
-          LocaleKeys.ai_customPrompt_featured.tr(),
-          style: AppFlowyTheme.of(context).textStyle.body.standard(
-                color: theme.textColorScheme.primary,
-              ),
-          overflow: TextOverflow.ellipsis,
-        );
-      },
-      borderRadius: theme.borderRadius.m,
-      padding: EdgeInsets.symmetric(
-        vertical: theme.spacing.s,
-        horizontal: theme.spacing.m,
-      ),
-      borderColor: (context, isHovering, disabled, isFocused) =>
-          Colors.transparent,
-      backgroundColor: (context, isHovering, disabled) {
-        if (isSelected) {
-          return theme.fillColorScheme.themeSelect;
-        }
-        if (isHovering) {
-          return theme.fillColorScheme.primaryAlpha5;
-        }
-        return Colors.transparent;
-      },
-    );
-  }
-}
-
-class AiPromptCategoryItem extends StatelessWidget {
-  const AiPromptCategoryItem({
-    super.key,
-    required this.category,
-    required this.onSelect,
-  });
-
-  final AiPromptCategory? category;
-  final VoidCallback onSelect;
-
-  @override
-  Widget build(BuildContext context) {
-    final theme = AppFlowyTheme.of(context);
-    final isSelected = context.watch<AiPromptSelectorCubit>().state.maybeMap(
-          ready: (state) =>
-              !state.isFeaturedCategorySelected &&
-              state.selectedCategory == category,
-          orElse: () => false,
-        );
-
-    return AFBaseButton(
-      onTap: onSelect,
-      builder: (context, isHovering, disabled) {
-        return Text(
-          category?.i18n ?? LocaleKeys.ai_customPrompt_all.tr(),
-          style: AppFlowyTheme.of(context).textStyle.body.standard(
-                color: theme.textColorScheme.primary,
-              ),
-          overflow: TextOverflow.ellipsis,
-        );
-      },
-      borderRadius: theme.borderRadius.m,
-      padding: EdgeInsets.symmetric(
-        vertical: theme.spacing.s,
-        horizontal: theme.spacing.m,
-      ),
-      borderColor: (context, isHovering, disabled, isFocused) =>
-          Colors.transparent,
-      backgroundColor: (context, isHovering, disabled) {
-        if (isSelected) {
-          return theme.fillColorScheme.themeSelect;
-        }
-        if (isHovering) {
-          return theme.fillColorScheme.primaryAlpha5;
-        }
-        return Colors.transparent;
-      },
-    );
-  }
-}
@@ -1,129 +0,0 @@
-import 'package:appflowy/ai/ai.dart';
-import 'package:appflowy/generated/flowy_svgs.g.dart';
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy_ui/appflowy_ui.dart';
-import 'package:collection/collection.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:flutter/material.dart';
-import 'package:flutter_bloc/flutter_bloc.dart';
-
-import 'ai_prompt_category_list.dart';
-import 'ai_prompt_preview.dart';
-import 'ai_prompt_visible_list.dart';
-
-Future<AiPrompt?> showAiPromptModal(
-  BuildContext context, {
-  required AiPromptSelectorCubit aiPromptSelectorCubit,
-}) async {
-  return showDialog<AiPrompt?>(
-    context: context,
-    builder: (context) {
-      return BlocProvider.value(
-        value: aiPromptSelectorCubit,
-        child: const AiPromptModal(),
-      );
-    },
-  );
-}
-
-class AiPromptModal extends StatelessWidget {
-  const AiPromptModal({super.key});
-
-  @override
-  Widget build(BuildContext context) {
-    final theme = AppFlowyTheme.of(context);
-
-    return AFModal(
-      constraints: const BoxConstraints(
-        maxWidth: 1200,
-        maxHeight: 800,
-      ),
-      child: Column(
-        children: [
-          AFModalHeader(
-            leading: Text(
-              LocaleKeys.ai_customPrompt_browsePrompts.tr(),
-              style: theme.textStyle.heading4.prominent(
-                color: theme.textColorScheme.primary,
-              ),
-            ),
-            trailing: [
-              AFGhostButton.normal(
-                onTap: () => Navigator.of(context).pop(),
-                padding: EdgeInsets.all(theme.spacing.s),
-                builder: (context, isHovering, disabled) {
-                  return Center(
-                    child: FlowySvg(
-                      FlowySvgs.toast_close_s,
-                      size: Size.square(20),
-                    ),
-                  );
-                },
-              ),
-            ],
-          ),
-          Expanded(
-            child: AFModalBody(
-              child: BlocBuilder<AiPromptSelectorCubit, AiPromptSelectorState>(
-                builder: (context, state) {
-                  final theme = AppFlowyTheme.of(context);
-
-                  return state.map(
-                    loading: (_) {
-                      return const Center(
-                        child: CircularProgressIndicator(),
-                      );
-                    },
-                    ready: (_) {
-                      return Row(
-                        crossAxisAlignment: CrossAxisAlignment.stretch,
-                        children: [
-                          Expanded(
-                            child: AiPromptCategoryList(
-                              padding: EdgeInsetsDirectional.only(
-                                top: theme.spacing.l,
-                                end: theme.spacing.l,
-                              ),
-                            ),
-                          ),
-                          Expanded(
-                            flex: 2,
-                            child: AiPromptVisibleList(),
-                          ),
-                          Expanded(
-                            flex: 3,
-                            child: BlocBuilder<AiPromptSelectorCubit,
-                                AiPromptSelectorState>(
-                              builder: (context, state) {
-                                final selectedPrompt = state.maybeMap(
-                                  ready: (state) {
-                                    return state.visiblePrompts
-                                        .firstWhereOrNull(
-                                      (prompt) =>
-                                          prompt.id == state.selectedPromptId,
-                                    );
-                                  },
-                                  orElse: () => null,
-                                );
-                                if (selectedPrompt == null) {
-                                  return const SizedBox.shrink();
-                                }
-                                return AiPromptPreview(
-                                  prompt: selectedPrompt,
-                                );
-                              },
-                            ),
-                          ),
-                        ],
-                      );
-                    },
-                  );
-                },
-              ),
-            ),
-          ),
-        ],
-      ),
-    );
-  }
-}
@@ -1,189 +0,0 @@
-import 'package:appflowy/ai/ai.dart';
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/plugins/ai_chat/presentation/message/ai_markdown_text.dart';
-import 'package:appflowy_ui/appflowy_ui.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:flowy_infra_ui/widget/spacing.dart';
-import 'package:flutter/material.dart';
-
-class AiPromptPreview extends StatelessWidget {
-  const AiPromptPreview({
-    super.key,
-    required this.prompt,
-  });
-
-  final AiPrompt prompt;
-
-  @override
-  Widget build(BuildContext context) {
-    final theme = AppFlowyTheme.of(context);
-
-    return SelectionArea(
-      child: Column(
-        children: [
-          Padding(
-            padding: EdgeInsets.only(
-              top: theme.spacing.l,
-              left: theme.spacing.l,
-              right: theme.spacing.l,
-            ),
-            child: SelectionContainer.disabled(
-              child: Row(
-                children: [
-                  Expanded(
-                    child: Text(
-                      prompt.name,
-                      style: theme.textStyle.headline.standard(
-                        color: theme.textColorScheme.primary,
-                      ),
-                    ),
-                  ),
-                  HSpace(theme.spacing.s),
-                  AFFilledTextButton.primary(
-                    text: LocaleKeys.ai_customPrompt_usePrompt.tr(),
-                    onTap: () {
-                      Navigator.of(context).pop(prompt);
-                    },
-                  ),
-                ],
-              ),
-            ),
-          ),
-          VSpace(theme.spacing.xs),
-          Expanded(
-            child: ListView(
-              padding: EdgeInsets.all(
-                theme.spacing.l,
-              ),
-              children: [
-                SelectionContainer.disabled(
-                  child: Text(
-                    LocaleKeys.ai_customPrompt_prompt.tr(),
-                    style: theme.textStyle.heading4.standard(
-                      color: theme.textColorScheme.primary,
-                    ),
-                  ),
-                ),
-                VSpace(theme.spacing.xs),
-                _PromptContent(
-                  prompt: prompt,
-                ),
-                VSpace(theme.spacing.xl),
-                if (prompt.example.isNotEmpty) ...[
-                  SelectionContainer.disabled(
-                    child: Text(
-                      LocaleKeys.ai_customPrompt_example.tr(),
-                      style: theme.textStyle.heading4.standard(
-                        color: theme.textColorScheme.primary,
-                      ),
-                    ),
-                  ),
-                  VSpace(theme.spacing.xs),
-                  _PromptExample(
-                    prompt: prompt,
-                  ),
-                  VSpace(theme.spacing.xl),
-                ],
-              ],
-            ),
-          ),
-        ],
-      ),
-    );
-  }
-}
-
-class _PromptContent extends StatelessWidget {
-  const _PromptContent({
-    required this.prompt,
-  });
-
-  final AiPrompt prompt;
-
-  @override
-  Widget build(BuildContext context) {
-    final theme = AppFlowyTheme.of(context);
-    final textSpans = _buildTextSpans(context, prompt.content);
-
-    return Container(
-      padding: EdgeInsets.all(theme.spacing.l),
-      decoration: BoxDecoration(
-        color: theme.fillColorScheme.quaternary,
-        borderRadius: BorderRadius.circular(theme.borderRadius.m),
-      ),
-      child: Text.rich(
-        TextSpan(
-          style: theme.textStyle.body.standard(
-            color: theme.textColorScheme.primary,
-          ),
-          children: textSpans,
-        ),
-      ),
-    );
-  }
-
-  List<TextSpan> _buildTextSpans(BuildContext context, String text) {
-    final theme = AppFlowyTheme.of(context);
-    final spans = <TextSpan>[];
-
-    final parts = _splitPromptText(text);
-    for (final part in parts) {
-      if (part.startsWith('[') && part.endsWith(']')) {
-        spans.add(
-          TextSpan(
-            text: part,
-            style: TextStyle(color: theme.textColorScheme.purple),
-          ),
-        );
-      } else {
-        spans.add(TextSpan(text: part));
-      }
-    }
-
-    return spans;
-  }
-
-  List<String> _splitPromptText(String text) {
-    final regex = RegExp(r'(\[[^\[\]]*?\])');
-
-    final result = <String>[];
-
-    text.splitMapJoin(
-      regex,
-      onMatch: (match) {
-        result.add(match.group(0)!);
-        return '';
-      },
-      onNonMatch: (nonMatch) {
-        result.add(nonMatch);
-        return '';
-      },
-    );
-
-    return result;
-  }
-}
-
-class _PromptExample extends StatelessWidget {
-  const _PromptExample({
-    required this.prompt,
-  });
-
-  final AiPrompt prompt;
-
-  @override
-  Widget build(BuildContext context) {
-    final theme = AppFlowyTheme.of(context);
-
-    return Container(
-      padding: EdgeInsets.all(theme.spacing.l),
-      decoration: BoxDecoration(
-        color: theme.fillColorScheme.quaternary,
-        borderRadius: BorderRadius.circular(theme.borderRadius.m),
-      ),
-      child: AIMarkdownText(
-        markdown: prompt.example,
-      ),
-    );
-  }
-}
@@ -1,341 +0,0 @@
-import 'dart:async';
-
-import 'package:appflowy/ai/ai.dart';
-import 'package:appflowy/generated/flowy_svgs.g.dart';
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy/util/theme_extension.dart';
-import 'package:appflowy_ui/appflowy_ui.dart';
-import 'package:diffutil_dart/diffutil.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:flowy_infra_ui/widget/spacing.dart';
-import 'package:flutter/material.dart';
-import 'package:flutter_bloc/flutter_bloc.dart';
-
-const Duration _listItemAnimationDuration = Duration(milliseconds: 150);
-
-class AiPromptVisibleList extends StatefulWidget {
-  const AiPromptVisibleList({
-    super.key,
-  });
-
-  @override
-  State<AiPromptVisibleList> createState() => _AiPromptVisibleListState();
-}
-
-class _AiPromptVisibleListState extends State<AiPromptVisibleList> {
-  final listKey = GlobalKey<AnimatedListState>();
-  final scrollController = ScrollController();
-  final List<AiPrompt> oldList = [];
-
-  late AiPromptSelectorCubit cubit;
-  late bool filterIsEmpty;
-
-  @override
-  void initState() {
-    super.initState();
-    cubit = context.read<AiPromptSelectorCubit>();
-    final textController = cubit.filterTextController;
-    final prompts = cubit.state.maybeMap(
-      ready: (value) => value.visiblePrompts,
-      orElse: () => <AiPrompt>[],
-    );
-    filterIsEmpty = textController.text.isEmpty;
-    textController.addListener(handleFilterTextChanged);
-    oldList.addAll(prompts);
-  }
-
-  @override
-  void dispose() {
-    cubit.filterTextController.removeListener(handleFilterTextChanged);
-    scrollController.dispose();
-    super.dispose();
-  }
-
-  @override
-  Widget build(BuildContext context) {
-    final spacing = AppFlowyTheme.of(context).spacing;
-    return BlocListener<AiPromptSelectorCubit, AiPromptSelectorState>(
-      listener: (context, state) {
-        final list = state.maybeMap(
-          ready: (state) => state.visiblePrompts,
-          orElse: () => <AiPrompt>[],
-        );
-        handleVisiblePromptListChanged(list);
-      },
-      child: Column(
-        children: [
-          VSpace(spacing.l),
-          Padding(
-            padding: EdgeInsets.symmetric(horizontal: spacing.l),
-            child: buildSearchField(context),
-          ),
-          VSpace(spacing.s),
-          Expanded(
-            child: AnimatedList(
-              controller: scrollController,
-              padding: EdgeInsets.all(spacing.l),
-              key: listKey,
-              initialItemCount: oldList.length,
-              itemBuilder: (context, index, animation) {
-                final cubit = context.read<AiPromptSelectorCubit>();
-
-                return cubit.state.maybeMap(
-                  ready: (state) {
-                    final prompt = state.visiblePrompts[index];
-                    return Padding(
-                      padding: EdgeInsets.only(
-                        top: index == 0 ? 0 : spacing.s,
-                        bottom: index == state.visiblePrompts.length - 1
-                            ? 0
-                            : spacing.s,
-                      ),
-                      child: _AiPromptListItem(
-                        animation: animation,
-                        prompt: prompt,
-                        isSelected: state.selectedPromptId == prompt.id,
-                      ),
-                    );
-                  },
-                  orElse: () => const SizedBox.shrink(),
-                );
-              },
-            ),
-          ),
-        ],
-      ),
-    );
-  }
-
-  Widget buildSearchField(BuildContext context) {
-    final theme = AppFlowyTheme.of(context);
-    final iconSize = 20.0;
-
-    return AFTextField(
-      groupId: "ai_prompt_category_list",
-      hintText: "Search",
-      controller: context.read<AiPromptSelectorCubit>().filterTextController,
-      autoFocus: true,
-      suffixIconConstraints: BoxConstraints.tightFor(
-        width: iconSize + theme.spacing.m,
-        height: iconSize,
-      ),
-      suffixIconBuilder: filterIsEmpty
-          ? null
-          : (context, isObscured) => TextFieldTapRegion(
-                groupId: "ai_prompt_category_list",
-                child: Padding(
-                  padding: EdgeInsets.only(right: theme.spacing.m),
-                  child: GestureDetector(
-                    onTap: () => context
-                        .read<AiPromptSelectorCubit>()
-                        .filterTextController
-                        .clear(),
-                    child: FlowySvg(
-                      FlowySvgs.toast_close_s,
-                      color: theme.textColorScheme.secondary,
-                      size: const Size.square(20),
-                    ),
-                  ),
-                ),
-              ),
-    );
-  }
-
-  void handleVisiblePromptListChanged(
-    List<AiPrompt> newList,
-  ) {
-    final updates = calculateListDiff(oldList, newList).getUpdatesWithData();
-
-    for (final update in updates) {
-      update.when(
-        insert: (pos, data) {
-          listKey.currentState?.insertItem(
-            pos,
-            duration: _listItemAnimationDuration,
-          );
-        },
-        remove: (pos, data) {
-          listKey.currentState?.removeItem(
-            pos,
-            (context, animation) {
-              final isSelected =
-                  context.read<AiPromptSelectorCubit>().state.maybeMap(
-                        ready: (state) => state.selectedPromptId == data.id,
-                        orElse: () => false,
-                      );
-              return _AiPromptListItem(
-                animation: animation,
-                prompt: data,
-                isSelected: isSelected,
-              );
-            },
-            duration: _listItemAnimationDuration,
-          );
-        },
-        change: (pos, oldData, newData) {},
-        move: (from, to, data) {},
-      );
-    }
-    oldList
-      ..clear()
-      ..addAll(newList);
-  }
-
-  void handleFilterTextChanged() {
-    setState(() {
-      filterIsEmpty = cubit.filterTextController.text.isEmpty;
-    });
-  }
-}
-
-class _AiPromptListItem extends StatefulWidget {
-  const _AiPromptListItem({
-    required this.animation,
-    required this.prompt,
-    required this.isSelected,
-  });
-
-  final Animation<double> animation;
-  final AiPrompt prompt;
-  final bool isSelected;
-
-  @override
-  State<_AiPromptListItem> createState() => _AiPromptListItemState();
-}
-
-class _AiPromptListItemState extends State<_AiPromptListItem> {
-  bool isHovering = false;
-  Timer? timer;
-
-  @override
-  void dispose() {
-    timer?.cancel();
-    timer = null;
-    super.dispose();
-  }
-
-  @override
-  Widget build(BuildContext context) {
-    final theme = AppFlowyTheme.of(context);
-    final cubit = context.read<AiPromptSelectorCubit>();
-
-    final curvedAnimation = CurvedAnimation(
-      parent: widget.animation,
-      curve: Curves.easeIn,
-    );
-
-    final surfacePrimaryHover =
-        Theme.of(context).isLightMode ? Color(0xFFF8FAFF) : Color(0xFF3C3F4E);
-
-    return FadeTransition(
-      opacity: curvedAnimation,
-      child: SizeTransition(
-        sizeFactor: curvedAnimation,
-        child: MouseRegion(
-          onEnter: (_) {
-            setState(() {
-              isHovering = true;
-              timer = Timer(const Duration(milliseconds: 300), () {
-                if (mounted) {
-                  cubit.selectPrompt(widget.prompt.id);
-                }
-              });
-            });
-          },
-          onExit: (_) {
-            setState(() {
-              isHovering = false;
-              timer?.cancel();
-            });
-          },
-          child: GestureDetector(
-            onTap: () {
-              cubit.selectPrompt(widget.prompt.id);
-            },
-            child: Stack(
-              children: [
-                Container(
-                  padding: EdgeInsets.all(theme.spacing.m),
-                  decoration: BoxDecoration(
-                    borderRadius: BorderRadius.circular(theme.borderRadius.m),
-                    border: Border.all(
-                      color: widget.isSelected
-                          ? isHovering
-                              ? theme.borderColorScheme.themeThickHover
-                              : theme.borderColorScheme.themeThick
-                          : isHovering
-                              ? theme.borderColorScheme.greyTertiaryHover
-                              : theme.borderColorScheme.greyTertiary,
-                    ),
-                    color: theme.surfaceColorScheme.primary,
-                  ),
-                  child: Column(
-                    crossAxisAlignment: CrossAxisAlignment.start,
-                    children: [
-                      Row(
-                        children: [
-                          Expanded(
-                            child: Text(
-                              widget.prompt.name,
-                              maxLines: 1,
-                              style: theme.textStyle.body.standard(
-                                color: theme.textColorScheme.primary,
-                              ),
-                              overflow: TextOverflow.ellipsis,
-                              softWrap: true,
-                            ),
-                          ),
-                        ],
-                      ),
-                      Text(
-                        widget.prompt.content,
-                        maxLines: 2,
-                        style: theme.textStyle.caption.standard(
-                          color: theme.textColorScheme.secondary,
-                        ),
-                        overflow: TextOverflow.ellipsis,
-                        softWrap: true,
-                      ),
-                    ],
-                  ),
-                ),
-                if (isHovering)
-                  Positioned(
-                    top: theme.spacing.s,
-                    right: theme.spacing.s,
-                    child: DecoratedBox(
-                      decoration: BoxDecoration(boxShadow: theme.shadow.small),
-                      child: AFBaseButton(
-                        onTap: () {
-                          Navigator.of(context).pop(widget.prompt);
-                        },
-                        builder: (context, isHovering, disabled) {
-                          return Text(
-                            LocaleKeys.ai_customPrompt_usePrompt.tr(),
-                            style: theme.textStyle.body.standard(
-                              color: theme.textColorScheme.primary,
-                            ),
-                          );
-                        },
-                        backgroundColor: (context, isHovering, disabled) {
-                          if (isHovering) {
-                            return surfacePrimaryHover;
-                          }
-                          return theme.surfaceColorScheme.primary;
-                        },
-                        padding: EdgeInsets.symmetric(
-                          vertical: theme.spacing.s,
-                          horizontal: theme.spacing.m,
-                        ),
-                        borderRadius: theme.borderRadius.m,
-                      ),
-                    ),
-                  ),
-              ],
-            ),
-          ),
-        ),
-      ),
-    );
-  }
-}
@@ -16,7 +16,8 @@ class AILoadingIndicator extends StatelessWidget {
  @override
  Widget build(BuildContext context) {
    final slice = Duration(milliseconds: duration.inMilliseconds ~/ 5);
-    return SelectionContainer.disabled(
+    return Padding(
+      padding: const EdgeInsets.only(top: 8.0),
      child: SizedBox(
        height: 20,
        child: SeparatedRow(
@@ -1,67 +0,0 @@
-import 'package:appflowy/ai/ai.dart';
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:flowy_infra_ui/flowy_infra_ui.dart';
-import 'package:flowy_infra_ui/style_widget/hover.dart';
-import 'package:flutter/material.dart';
-import 'package:flutter_bloc/flutter_bloc.dart';
-
-import '../ai_prompt_modal/ai_prompt_modal.dart';
-
-class BrowsePromptsButton extends StatelessWidget {
-  const BrowsePromptsButton({
-    super.key,
-    required this.onSelectPrompt,
-  });
-
-  final void Function(AiPrompt) onSelectPrompt;
-
-  @override
-  Widget build(BuildContext context) {
-    return FlowyTooltip(
-      message: LocaleKeys.ai_customPrompt_browsePrompts.tr(),
-      child: BlocProvider(
-        create: (context) => AiPromptSelectorCubit(),
-        child: Builder(
-          builder: (context) {
-            return GestureDetector(
-              onTap: () async {
-                final prompt = await showAiPromptModal(
-                  context,
-                  aiPromptSelectorCubit: context.read<AiPromptSelectorCubit>(),
-                );
-                if (context.mounted) {
-                  context.read<AiPromptSelectorCubit>().reset();
-                }
-                if (prompt != null && context.mounted) {
-                  onSelectPrompt(prompt);
-                }
-              },
-              behavior: HitTestBehavior.opaque,
-              child: SizedBox(
-                height: DesktopAIPromptSizes.actionBarButtonSize,
-                child: FlowyHover(
-                  style: const HoverStyle(
-                    borderRadius: BorderRadius.all(Radius.circular(8)),
-                  ),
-                  child: Padding(
-                    padding: const EdgeInsetsDirectional.all(4.0),
-                    child: Center(
-                      child: FlowyText(
-                        LocaleKeys.ai_customPrompt_browsePrompts.tr(),
-                        fontSize: 12,
-                        figmaLineHeight: 16,
-                        color: Theme.of(context).hintColor,
-                        overflow: TextOverflow.ellipsis,
-                      ),
-                    ),
-                  ),
-                ),
-              ),
-            );
-          },
-        ),
-      ),
-    );
-  }
-}
@@ -0,0 +1,53 @@
+import 'package:appflowy/plugins/ai_chat/application/chat_input_control_cubit.dart';
+import 'package:appflowy/plugins/ai_chat/presentation/layout_define.dart';
+import 'package:extended_text_field/extended_text_field.dart';
+import 'package:flutter/material.dart';
+
+import 'mentioned_page_text_span.dart';
+
+class PromptInputTextField extends StatelessWidget {
+  const PromptInputTextField({
+    super.key,
+    required this.cubit,
+    required this.textController,
+    required this.textFieldFocusNode,
+    required this.contentPadding,
+    this.hintText = "",
+  });
+
+  final ChatInputControlCubit cubit;
+  final TextEditingController textController;
+  final FocusNode textFieldFocusNode;
+  final EdgeInsetsGeometry contentPadding;
+  final String hintText;
+
+  @override
+  Widget build(BuildContext context) {
+    return ExtendedTextField(
+      controller: textController,
+      focusNode: textFieldFocusNode,
+      decoration: InputDecoration(
+        border: InputBorder.none,
+        enabledBorder: InputBorder.none,
+        focusedBorder: InputBorder.none,
+        contentPadding: contentPadding,
+        hintText: hintText,
+        hintStyle: AIChatUILayout.inputHintTextStyle(context),
+        isCollapsed: true,
+        isDense: true,
+      ),
+      keyboardType: TextInputType.multiline,
+      textCapitalization: TextCapitalization.sentences,
+      minLines: 1,
+      maxLines: null,
+      style: Theme.of(context).textTheme.bodyMedium,
+      specialTextSpanBuilder: PromptInputTextSpanBuilder(
+        inputControlCubit: cubit,
+        specialTextStyle: Theme.of(context).textTheme.bodyMedium?.copyWith(
+              color: Theme.of(context).colorScheme.primary,
+              fontWeight: FontWeight.w600,
+            ),
+      ),
+    );
+  }
+}
@@ -1,5 +1,5 @@
 import 'package:appflowy/generated/flowy_svgs.g.dart';
-import 'package:appflowy/ai/service/ai_prompt_input_bloc.dart';
+import 'package:appflowy/plugins/ai_chat/application/ai_prompt_input_bloc.dart';
 import 'package:appflowy/plugins/ai_chat/application/chat_entity.dart';
 import 'package:appflowy/plugins/ai_chat/application/chat_input_file_bloc.dart';
 import 'package:flowy_infra/theme_extension.dart';
@@ -13,9 +13,11 @@ import 'layout_define.dart';
 class PromptInputFile extends StatelessWidget {
  const PromptInputFile({
    super.key,
+    required this.chatId,
    required this.onDeleted,
  });

+  final String chatId;
  final void Function(ChatFile) onDeleted;

  @override
@@ -35,6 +37,7 @@ class PromptInputFile extends StatelessWidget {
          ),
          itemCount: files.length,
          itemBuilder: (context, index) => ChatFilePreview(
+            chatId: chatId,
            file: files[index],
            onDeleted: () => onDeleted(files[index]),
          ),
@@ -46,11 +49,13 @@ class PromptInputFile extends StatelessWidget {

 class ChatFilePreview extends StatefulWidget {
  const ChatFilePreview({
+    required this.chatId,
    required this.file,
    required this.onDeleted,
    super.key,
  });

+  final String chatId;
  final ChatFile file;
  final VoidCallback onDeleted;

@@ -49,9 +49,7 @@ class _PromptInputMentionPageMenuState
  void initState() {
    super.initState();
    Future.delayed(Duration.zero, () {
-      if (mounted) {
-        context.read<ChatInputControlCubit>().refreshViews();
-      }
+      context.read<ChatInputControlCubit>().refreshViews();
    });
  }

@@ -8,11 +8,11 @@ import 'package:flutter/material.dart';
 class PromptInputTextSpanBuilder extends SpecialTextSpanBuilder {
  PromptInputTextSpanBuilder({
    required this.inputControlCubit,
-    this.mentionedPageTextStyle,
+    this.specialTextStyle,
  });

  final ChatInputControlCubit inputControlCubit;
-  final TextStyle? mentionedPageTextStyle;
+  final TextStyle? specialTextStyle;

  @override
  SpecialText? createSpecialText(
@@ -25,22 +25,23 @@ class PromptInputTextSpanBuilder extends SpecialTextSpanBuilder {
      return null;
    }

-    if (isStart(flag, MentionedPageText.flag)) {
-      return MentionedPageText(
-        inputControlCubit,
-        mentionedPageTextStyle ?? textStyle,
-        onTap,
-        // scrubbing over text is kinda funky
-        start: index! - (MentionedPageText.flag.length - 1),
-      );
+    if (!isStart(flag, AtText.flag)) {
+      return null;
    }

-    return null;
+    // index is at the end of the start flag, so the start index should be index - (flag.length - 1)
+    return AtText(
+      inputControlCubit,
+      specialTextStyle ?? textStyle,
+      onTap,
+      // scrubbing over text is kinda funky
+      start: index! - (AtText.flag.length - 1),
+    );
  }
 }

-class MentionedPageText extends SpecialText {
-  MentionedPageText(
+class AtText extends SpecialText {
+  AtText(
    this.inputControlCubit,
    TextStyle? textStyle,
    SpecialTextGestureTapCallback? onTap, {
@@ -1,22 +1,24 @@
 import 'package:appflowy/generated/flowy_svgs.g.dart';
 import 'package:appflowy/generated/locale_keys.g.dart';
+import 'package:appflowy/plugins/ai_chat/application/chat_entity.dart';
 import 'package:easy_localization/easy_localization.dart';
 import 'package:flowy_infra_ui/flowy_infra_ui.dart';
 import 'package:flowy_infra_ui/style_widget/hover.dart';
 import 'package:flutter/material.dart';
 import 'package:universal_platform/universal_platform.dart';

-import '../../service/ai_entities.dart';
 import 'layout_define.dart';

 class PromptInputDesktopToggleFormatButton extends StatelessWidget {
  const PromptInputDesktopToggleFormatButton({
    super.key,
    required this.showFormatBar,
+    required this.predefinedFormat,
    required this.onTap,
  });

  final bool showFormatBar;
+  final PredefinedFormat predefinedFormat;
  final VoidCallback onTap;

  @override
@@ -48,31 +50,26 @@ class ChangeFormatBar extends StatelessWidget {
    required this.predefinedFormat,
    required this.spacing,
    required this.onSelectPredefinedFormat,
-    this.showImageFormats = true,
  });

  final PredefinedFormat? predefinedFormat;
  final double spacing;
  final void Function(PredefinedFormat) onSelectPredefinedFormat;
-  final bool showImageFormats;

  @override
  Widget build(BuildContext context) {
-    final showTextFormats = predefinedFormat?.imageFormat.hasText ?? true;
    return SizedBox(
      height: DesktopAIPromptSizes.predefinedFormatButtonHeight,
      child: SeparatedRow(
        mainAxisSize: MainAxisSize.min,
        separatorBuilder: () => HSpace(spacing),
        children: [
-          if (showImageFormats) ...[
-            _buildFormatButton(context, ImageFormat.text),
-            _buildFormatButton(context, ImageFormat.textAndImage),
-            _buildFormatButton(context, ImageFormat.image),
-          ],
-          if (showImageFormats && showTextFormats) _buildDivider(),
-          if (showTextFormats) ...[
-            _buildTextFormatButton(context, TextFormat.paragraph),
+          _buildFormatButton(context, ImageFormat.text),
+          _buildFormatButton(context, ImageFormat.textAndImage),
+          _buildFormatButton(context, ImageFormat.image),
+          if (predefinedFormat?.imageFormat.hasText ?? true) ...[
+            _buildDivider(),
+            _buildTextFormatButton(context, TextFormat.auto),
            _buildTextFormatButton(context, TextFormat.bulletList),
            _buildTextFormatButton(context, TextFormat.numberedList),
            _buildTextFormatButton(context, TextFormat.table),
@@ -91,8 +88,7 @@ class ChangeFormatBar extends StatelessWidget {
          return;
        }
        if (format.hasText) {
-          final textFormat =
-              predefinedFormat?.textFormat ?? TextFormat.paragraph;
+          final textFormat = predefinedFormat?.textFormat ?? TextFormat.auto;
          onSelectPredefinedFormat(
            PredefinedFormat(imageFormat: format, textFormat: textFormat),
          );
@@ -104,7 +100,6 @@ class ChangeFormatBar extends StatelessWidget {
      },
      child: FlowyTooltip(
        message: format.i18n,
-        preferBelow: false,
        child: SizedBox.square(
          dimension: _buttonSize,
          child: FlowyHover(
@@ -151,7 +146,6 @@ class ChangeFormatBar extends StatelessWidget {
      },
      child: FlowyTooltip(
        message: format.i18n,
-        preferBelow: false,
        child: SizedBox.square(
          dimension: _buttonSize,
          child: FlowyHover(
@@ -1,79 +0,0 @@
-import 'package:appflowy_ui/appflowy_ui.dart';
-import 'package:flutter/widgets.dart';
-
-final openingBracketReplacement = String.fromCharCode(0xFFFE);
-final closingBracketReplacement = String.fromCharCode(0xFFFD);
-
-class AiPromptInputTextEditingController extends TextEditingController {
-  AiPromptInputTextEditingController();
-
-  static String replace(String text) {
-    return text
-        .replaceAll('[', openingBracketReplacement)
-        .replaceAll(']', closingBracketReplacement);
-  }
-
-  static String restore(String text) {
-    return text
-        .replaceAll(openingBracketReplacement, '[')
-        .replaceAll(closingBracketReplacement, ']');
-  }
-
-  void usePrompt(String content) {
-    value = TextEditingValue(
-      text: content,
-      selection: TextSelection.collapsed(
-        offset: content.length,
-      ),
-    );
-  }
-
-  @override
-  TextSpan buildTextSpan({
-    required BuildContext context,
-    TextStyle? style,
-    required bool withComposing,
-  }) {
-    return TextSpan(
-      style: style,
-      children: <InlineSpan>[...getTextSpans(context)],
-    );
-  }
-
-  Iterable<TextSpan> getTextSpans(BuildContext context) {
-    final open = openingBracketReplacement;
-    final close = closingBracketReplacement;
-    final regex = RegExp('($open[^$open$close]*?$close)');
-    final theme = AppFlowyTheme.of(context);
-
-    final result = <TextSpan>[];
-
-    text.splitMapJoin(
-      regex,
-      onMatch: (match) {
-        final string = match.group(0)!;
-        result.add(
-          TextSpan(
-            text: restore(string),
-            style: theme.textStyle.body.standard().copyWith(
-                  color: theme.textColorScheme.purple,
-                  backgroundColor:
-                      theme.fillColorScheme.purpleThick.withAlpha(51),
-                ),
-          ),
-        );
-        return '';
-      },
-      onNonMatch: (nonMatch) {
-        result.add(
-          TextSpan(
-            text: restore(nonMatch),
-          ),
-        );
-        return '';
-      },
-    );
-
-    return result;
-  }
-}
@@ -1,262 +0,0 @@
-import 'package:appflowy/ai/ai.dart';
-import 'package:appflowy/generated/flowy_svgs.g.dart';
-import 'package:appflowy/generated/locale_keys.g.dart';
-import 'package:appflowy_backend/protobuf/flowy-ai/protobuf.dart';
-import 'package:easy_localization/easy_localization.dart';
-import 'package:flowy_infra_ui/flowy_infra_ui.dart';
-import 'package:flowy_infra_ui/style_widget/hover.dart';
-import 'package:flutter/material.dart';
-import 'package:flutter_bloc/flutter_bloc.dart';
-
-class SelectModelMenu extends StatefulWidget {
-  const SelectModelMenu({
-    super.key,
-    required this.aiModelStateNotifier,
-  });
-
-  final AIModelStateNotifier aiModelStateNotifier;
-
-  @override
-  State<SelectModelMenu> createState() => _SelectModelMenuState();
-}
-
-class _SelectModelMenuState extends State<SelectModelMenu> {
-  final popoverController = PopoverController();
-
-  @override
-  Widget build(BuildContext context) {
-    return BlocProvider(
-      create: (context) => SelectModelBloc(
-        aiModelStateNotifier: widget.aiModelStateNotifier,
-      ),
-      child: BlocBuilder<SelectModelBloc, SelectModelState>(
-        builder: (context, state) {
-          return AppFlowyPopover(
-            offset: Offset(-12.0, 0.0),
-            constraints: BoxConstraints(maxWidth: 250, maxHeight: 600),
-            direction: PopoverDirection.topWithLeftAligned,
-            margin: EdgeInsets.zero,
-            controller: popoverController,
-            popupBuilder: (popoverContext) {
-              return SelectModelPopoverContent(
-                models: state.models,
-                selectedModel: state.selectedModel,
-                onSelectModel: (model) {
-                  if (model != state.selectedModel) {
-                    context
-                        .read<SelectModelBloc>()
-                        .add(SelectModelEvent.selectModel(model));
-                  }
-                  popoverController.close();
-                },
-              );
-            },
-            child: _CurrentModelButton(
-              model: state.selectedModel,
-              onTap: () {
-                if (state.selectedModel != null) {
-                  popoverController.show();
-                }
-              },
-            ),
-          );
-        },
-      ),
-    );
-  }
-}
-
-class SelectModelPopoverContent extends StatelessWidget {
-  const SelectModelPopoverContent({
-    super.key,
-    required this.models,
-    required this.selectedModel,
-    this.onSelectModel,
-  });
-
-  final List<AIModelPB> models;
-  final AIModelPB? selectedModel;
-  final void Function(AIModelPB)? onSelectModel;
-
-  @override
-  Widget build(BuildContext context) {
-    if (models.isEmpty) {
-      return const SizedBox.shrink();
-    }
-
-    // separate models into local and cloud models
-    final localModels = models.where((model) => model.isLocal).toList();
-    final cloudModels = models.where((model) => !model.isLocal).toList();
-
-    return Padding(
-      padding: const EdgeInsets.all(8.0),
-      child: SingleChildScrollView(
-        child: Column(
-          mainAxisSize: MainAxisSize.min,
-          crossAxisAlignment: CrossAxisAlignment.start,
-          children: [
-            if (localModels.isNotEmpty) ...[
-              _ModelSectionHeader(
-                title: LocaleKeys.chat_switchModel_localModel.tr(),
-              ),
-              const VSpace(4.0),
-            ],
-            ...localModels.map(
-              (model) => _ModelItem(
-                model: model,
-                isSelected: model == selectedModel,
-                onTap: () => onSelectModel?.call(model),
-              ),
-            ),
-            if (cloudModels.isNotEmpty && localModels.isNotEmpty) ...[
-              const VSpace(8.0),
-              _ModelSectionHeader(
-                title: LocaleKeys.chat_switchModel_cloudModel.tr(),
-              ),
-              const VSpace(4.0),
-            ],
-            ...cloudModels.map(
-              (model) => _ModelItem(
-                model: model,
-                isSelected: model == selectedModel,
-                onTap: () => onSelectModel?.call(model),
-              ),
-            ),
-          ],
-        ),
-      ),
-    );
-  }
-}
-
-class _ModelSectionHeader extends StatelessWidget {
-  const _ModelSectionHeader({
-    required this.title,
-  });
-
-  final String title;
-
-  @override
-  Widget build(BuildContext context) {
-    return Padding(
-      padding: const EdgeInsets.only(top: 4, bottom: 2),
-      child: FlowyText(
-        title,
-        fontSize: 12,
-        figmaLineHeight: 16,
-        color: Theme.of(context).hintColor,
-        fontWeight: FontWeight.w500,
-      ),
-    );
-  }
-}
-
-class _ModelItem extends StatelessWidget {
-  const _ModelItem({
-    required this.model,
-    required this.isSelected,
-    required this.onTap,
-  });
-
-  final AIModelPB model;
-  final bool isSelected;
-  final VoidCallback onTap;
-
-  @override
-  Widget build(BuildContext context) {
-    return ConstrainedBox(
-      constraints: const BoxConstraints(minHeight: 32),
-      child: FlowyButton(
-        onTap: onTap,
-        margin: EdgeInsets.symmetric(horizontal: 8.0, vertical: 6.0),
-        text: Column(
-          crossAxisAlignment: CrossAxisAlignment.start,
-          children: [
-            FlowyText(
-              model.i18n,
-              figmaLineHeight: 20,
-              overflow: TextOverflow.ellipsis,
-            ),
-            if (model.desc.isNotEmpty)
-              FlowyText(
-                model.desc,
-                fontSize: 12,
-                figmaLineHeight: 16,
-                color: Theme.of(context).hintColor,
-                overflow: TextOverflow.ellipsis,
-              ),
-          ],
-        ),
-        rightIcon: isSelected
-            ? FlowySvg(
-                FlowySvgs.check_s,
-                size: const Size.square(20),
-                color: Theme.of(context).colorScheme.primary,
-              )
-            : null,
-      ),
-    );
-  }
-}
-
-class _CurrentModelButton extends StatelessWidget {
-  const _CurrentModelButton({
-    required this.model,
-    required this.onTap,
-  });
-
-  final AIModelPB? model;
-  final VoidCallback onTap;
-
-  @override
-  Widget build(BuildContext context) {
-    return FlowyTooltip(
-      message: LocaleKeys.chat_switchModel_label.tr(),
-      child: GestureDetector(
-        onTap: onTap,
-        behavior: HitTestBehavior.opaque,
-        child: SizedBox(
-          height: DesktopAIPromptSizes.actionBarButtonSize,
-          child: FlowyHover(
-            style: const HoverStyle(
-              borderRadius: BorderRadius.all(Radius.circular(8)),
-            ),
-            child: Padding(
-              padding: const EdgeInsetsDirectional.all(4.0),
-              child: Row(
-                mainAxisSize: MainAxisSize.min,
-                children: [
-                  Padding(
-                    // TODO: remove this after change icon to 20px
-                    padding: EdgeInsets.all(2),
-                    child: FlowySvg(
-                      FlowySvgs.ai_sparks_s,
-                      color: Theme.of(context).hintColor,
-                      size: Size.square(16),
-                    ),
-                  ),
-                  if (model != null && !model!.isDefault)
-                    Padding(
-                      padding: EdgeInsetsDirectional.only(end: 2.0),
-                      child: FlowyText(
-                        model!.i18n,
-                        fontSize: 12,
-                        figmaLineHeight: 16,
-                        color: Theme.of(context).hintColor,
-                        overflow: TextOverflow.ellipsis,
-                      ),
-                    ),
-                  FlowySvg(
-                    FlowySvgs.ai_source_drop_down_s,
-                    color: Theme.of(context).hintColor,
-                    size: const Size.square(8),
-                  ),
-                ],
-              ),
-            ),
-          ),
-        ),
-      ),
-    );
-  }
-}
@@ -2,6 +2,7 @@ import 'package:appflowy/generated/flowy_svgs.g.dart';
 import 'package:appflowy/generated/locale_keys.g.dart';
 import 'package:appflowy/mobile/presentation/base/flowy_search_text_field.dart';
 import 'package:appflowy/mobile/presentation/bottom_sheet/bottom_sheet.dart';
+import 'package:appflowy/plugins/ai_chat/application/chat_bloc.dart';
 import 'package:appflowy/plugins/ai_chat/application/chat_select_sources_cubit.dart';
 import 'package:appflowy/plugins/base/drag_handler.dart';
 import 'package:appflowy/workspace/application/sidebar/space/space_bloc.dart';
@@ -18,11 +19,9 @@ import 'select_sources_menu.dart';
 class PromptInputMobileSelectSourcesButton extends StatefulWidget {
  const PromptInputMobileSelectSourcesButton({
    super.key,
-    required this.selectedSourcesNotifier,
    required this.onUpdateSelectedSources,
  });

-  final ValueNotifier<List<String>> selectedSourcesNotifier;
  final void Function(List<String>) onUpdateSelectedSources;

  @override
@@ -37,15 +36,15 @@ class _PromptInputMobileSelectSourcesButtonState
  @override
  void initState() {
    super.initState();
-    widget.selectedSourcesNotifier.addListener(onSelectedSourcesChanged);
    WidgetsBinding.instance.addPostFrameCallback((_) {
-      onSelectedSourcesChanged();
+      cubit.updateSelectedSources(
+        context.read<ChatBloc>().state.selectedSourceIds,
+      );
    });
  }

  @override
  void dispose() {
-    widget.selectedSourcesNotifier.removeListener(onSelectedSourcesChanged);
    cubit.close();
    super.dispose();
  }
@@ -54,7 +53,7 @@ class _PromptInputMobileSelectSourcesButtonState
  Widget build(BuildContext context) {
    return BlocBuilder<UserWorkspaceBloc, UserWorkspaceState>(
      builder: (context, state) {
-        final userProfile = context.read<UserWorkspaceBloc>().state.userProfile;
+        final userProfile = context.read<UserWorkspaceBloc>().userProfile;
        final workspaceId = state.currentWorkspace?.workspaceId ?? '';
        return MultiBlocProvider(
          providers: [
@@ -71,49 +70,56 @@ class _PromptInputMobileSelectSourcesButtonState
          ],
          child: BlocBuilder<SpaceBloc, SpaceState>(
            builder: (context, state) {
-              return FlowyButton(
-                margin: const EdgeInsetsDirectional.fromSTEB(4, 6, 2, 6),
-                expandText: false,
-                text: Row(
-                  mainAxisSize: MainAxisSize.min,
-                  children: [
-                    FlowySvg(
-                      FlowySvgs.ai_page_s,
-                      color: Theme.of(context).iconTheme.color,
-                      size: const Size.square(20.0),
-                    ),
-                    FlowySvg(
-                      FlowySvgs.ai_source_drop_down_s,
-                      color: Theme.of(context).hintColor,
-                      size: const Size.square(10),
-                    ),
-                  ],
-                ),
-                onTap: () async {
-                  context
-                      .read<ChatSettingsCubit>()
-                      .refreshSources(state.spaces, state.currentSpace);
-                  await showMobileBottomSheet<void>(
-                    context,
-                    backgroundColor: Theme.of(context).colorScheme.surface,
-                    maxChildSize: 0.98,
-                    enableDraggableScrollable: true,
-                    scrollableWidgetBuilder: (_, scrollController) {
-                      return Expanded(
-                        child: BlocProvider.value(
-                          value: cubit,
-                          child: _MobileSelectSourcesSheetBody(
-                            scrollController: scrollController,
-                          ),
-                        ),
-                      );
-                    },
-                    builder: (context) => const SizedBox.shrink(),
-                  );
-                  if (context.mounted) {
-                    widget.onUpdateSelectedSources(cubit.selectedSourceIds);
-                  }
+              return BlocListener<ChatBloc, ChatState>(
+                listener: (context, state) {
+                  cubit
+                    ..updateSelectedSources(state.selectedSourceIds)
+                    ..updateSelectedStatus();
                },
+                child: FlowyButton(
+                  margin: const EdgeInsetsDirectional.fromSTEB(4, 6, 2, 6),
+                  expandText: false,
+                  text: Row(
+                    mainAxisSize: MainAxisSize.min,
+                    children: [
+                      FlowySvg(
+                        FlowySvgs.ai_page_s,
+                        color: Theme.of(context).iconTheme.color,
+                        size: const Size.square(20.0),
+                      ),
+                      FlowySvg(
+                        FlowySvgs.ai_source_drop_down_s,
+                        color: Theme.of(context).hintColor,
+                        size: const Size.square(10),
+                      ),
+                    ],
+                  ),
+                  onTap: () async {
+                    context
+                        .read<ChatSettingsCubit>()
+                        .refreshSources(state.spaces, state.currentSpace);
+                    await showMobileBottomSheet<void>(
+                      context,
+                      backgroundColor: Theme.of(context).colorScheme.surface,
+                      maxChildSize: 0.98,
+                      enableDraggableScrollable: true,
+                      scrollableWidgetBuilder: (_, scrollController) {
+                        return Expanded(
+                          child: BlocProvider.value(
+                            value: cubit,
+                            child: _MobileSelectSourcesSheetBody(
+                              scrollController: scrollController,
+                            ),
+                          ),
+                        );
+                      },
+                      builder: (context) => const SizedBox.shrink(),
+                    );
+                    if (context.mounted) {
+                      widget.onUpdateSelectedSources(cubit.selectedSourceIds);
+                    }
+                  },
+                ),
              );
            },
          ),
@@ -121,12 +127,6 @@ class _PromptInputMobileSelectSourcesButtonState
      },
    );
  }
-
-  void onSelectedSourcesChanged() {
-    cubit
-      ..updateSelectedSources(widget.selectedSourcesNotifier.value)
-      ..updateSelectedStatus();
-  }
 }

 class _MobileSelectSourcesSheetBody extends StatefulWidget {
@@ -2,8 +2,8 @@ import 'dart:math';

 import 'package:appflowy/generated/flowy_svgs.g.dart';
 import 'package:appflowy/generated/locale_keys.g.dart';
+import 'package:appflowy/plugins/ai_chat/application/chat_bloc.dart';
 import 'package:appflowy/plugins/ai_chat/application/chat_select_sources_cubit.dart';
-import 'package:appflowy/plugins/document/application/document_bloc.dart';
 import 'package:appflowy/workspace/application/sidebar/space/space_bloc.dart';
 import 'package:appflowy/workspace/application/user/user_workspace_bloc.dart';
 import 'package:appflowy/workspace/application/view/view_ext.dart';
@@ -24,11 +24,9 @@ import 'mention_page_menu.dart';
 class PromptInputDesktopSelectSourcesButton extends StatefulWidget {
  const PromptInputDesktopSelectSourcesButton({
    super.key,
-    required this.selectedSourcesNotifier,
    required this.onUpdateSelectedSources,
  });

-  final ValueNotifier<List<String>> selectedSourcesNotifier;
  final void Function(List<String>) onUpdateSelectedSources;

  @override
@@ -44,15 +42,15 @@ class _PromptInputDesktopSelectSourcesButtonState
  @override
  void initState() {
    super.initState();
-    widget.selectedSourcesNotifier.addListener(onSelectedSourcesChanged);
    WidgetsBinding.instance.addPostFrameCallback((_) {
-      onSelectedSourcesChanged();
+      cubit.updateSelectedSources(
+        context.read<ChatBloc>().state.selectedSourceIds,
+      );
    });
  }

  @override
  void dispose() {
-    widget.selectedSourcesNotifier.removeListener(onSelectedSourcesChanged);
    cubit.close();
    super.dispose();
  }
@@ -60,7 +58,7 @@ class _PromptInputDesktopSelectSourcesButtonState
  @override
  Widget build(BuildContext context) {
    final userWorkspaceBloc = context.read<UserWorkspaceBloc>();
-    final userProfile = userWorkspaceBloc.state.userProfile;
+    final userProfile = userWorkspaceBloc.userProfile;
    final workspaceId =
        userWorkspaceBloc.state.currentWorkspace?.workspaceId ?? '';

@@ -78,53 +76,51 @@ class _PromptInputDesktopSelectSourcesButtonState
      ],
      child: BlocBuilder<SpaceBloc, SpaceState>(
        builder: (context, state) {
-          return AppFlowyPopover(
-            constraints: BoxConstraints.loose(const Size(320, 380)),
-            offset: const Offset(0.0, -10.0),
-            direction: PopoverDirection.topWithCenterAligned,
-            margin: EdgeInsets.zero,
-            controller: popoverController,
-            onOpen: () {
-              context
-                  .read<ChatSettingsCubit>()
-                  .refreshSources(state.spaces, state.currentSpace);
+          return BlocListener<ChatBloc, ChatState>(
+            listener: (context, state) {
+              cubit
+                ..updateSelectedSources(state.selectedSourceIds)
+                ..updateSelectedStatus();
            },
-            onClose: () {
-              widget.onUpdateSelectedSources(cubit.selectedSourceIds);
-              context
-                  .read<ChatSettingsCubit>()
-                  .refreshSources(state.spaces, state.currentSpace);
-            },
-            popupBuilder: (_) {
-              return BlocProvider.value(
-                value: context.read<ChatSettingsCubit>(),
-                child: const _PopoverContent(),
-              );
-            },
-            child: _IndicatorButton(
-              selectedSourcesNotifier: widget.selectedSourcesNotifier,
-              onTap: () => popoverController.show(),
+            child: AppFlowyPopover(
+              constraints: BoxConstraints.loose(const Size(320, 380)),
+              offset: const Offset(0.0, -10.0),
+              direction: PopoverDirection.topWithCenterAligned,
+              margin: EdgeInsets.zero,
+              controller: popoverController,
+              onOpen: () {
+                context
+                    .read<ChatSettingsCubit>()
+                    .refreshSources(state.spaces, state.currentSpace);
+              },
+              onClose: () {
+                widget.onUpdateSelectedSources(cubit.selectedSourceIds);
+                context
+                    .read<ChatSettingsCubit>()
+                    .refreshSources(state.spaces, state.currentSpace);
+              },
+              popupBuilder: (_) {
+                return BlocProvider.value(
+                  value: context.read<ChatSettingsCubit>(),
+                  child: const _PopoverContent(),
+                );
+              },
+              child: _IndicatorButton(
+                onTap: () => popoverController.show(),
+              ),
            ),
          );
        },
      ),
    );
  }
-
-  void onSelectedSourcesChanged() {
-    cubit
-      ..updateSelectedSources(widget.selectedSourcesNotifier.value)
-      ..updateSelectedStatus();
-  }
 }

 class _IndicatorButton extends StatelessWidget {
  const _IndicatorButton({
-    required this.selectedSourcesNotifier,
    required this.onTap,
  });

-  final ValueNotifier<List<String>> selectedSourcesNotifier;
  final VoidCallback onTap;

  @override
@@ -145,22 +141,14 @@ class _IndicatorButton extends StatelessWidget {
              children: [
                FlowySvg(
                  FlowySvgs.ai_page_s,
-                  color: Theme.of(context).hintColor,
+                  color: Theme.of(context).iconTheme.color,
                ),
                const HSpace(2.0),
-                ValueListenableBuilder(
-                  valueListenable: selectedSourcesNotifier,
-                  builder: (context, selectedSourceIds, _) {
-                    final documentId =
-                        context.read<DocumentBloc?>()?.documentId;
-                    final label = documentId != null &&
-                            selectedSourceIds.length == 1 &&
-                            selectedSourceIds[0] == documentId
-                        ? LocaleKeys.chat_currentPage.tr()
-                        : selectedSourceIds.length.toString();
+                BlocBuilder<ChatBloc, ChatState>(
+                  builder: (context, state) {
                    return FlowyText(
-                      label,
-                      fontSize: 12,
+                      state.selectedSourceIds.length.toString(),
+                      fontSize: 14,
                      figmaLineHeight: 16,
                      color: Theme.of(context).hintColor,
                    );
@@ -170,7 +158,7 @@ class _IndicatorButton extends StatelessWidget {
                FlowySvg(
                  FlowySvgs.ai_source_drop_down_s,
                  color: Theme.of(context).hintColor,
-                  size: const Size.square(8),
+                  size: const Size.square(10),
                ),
              ],
            ),
Author	SHA1	Message	Date
Nathan.fooo	122fb89c87	chore: fix ai page user profile refresh (#7317 )	2025-02-04 20:40:06 +08:00
Lucas	e9201cce72	fix: editor stuck on image loading loop when uploading image in row document (#7313 ) * fix: editor stuck on image loading loop when uploading image in row document * test: editor stuck on image loading loop when uploading image in row document	2025-02-04 18:03:28 +08:00
Nathan.fooo	5e5693e249	chore: Support new error code (#7311 ) * chore: fetch model list * chore: suppor new error code	2025-02-03 20:52:31 +08:00
Nathan	5ac0eded3c	chore: fetch model list	2025-02-01 23:50:58 +08:00