feat: ai billing

frontend/appflowy_flutter/lib/plugins/document/application/document_service.dart

@@ -136,7 +136,7 @@ class DocumentService {
   }) async {
     final workspace = await FolderEventReadCurrentWorkspace().send();
     return workspace.fold((l) async {
-      final payload = UploadedFilePB(
+      final payload = DownloadFilePB(
         url: url,
       );
       final result = await DocumentEventDownloadFile(payload).send();

frontend/appflowy_flutter/lib/plugins/document/presentation/editor_plugins/image/image_util.dart

@@ -7,6 +7,7 @@ import 'package:appflowy/startup/startup.dart';
 import 'package:appflowy/util/file_extension.dart';
 import 'package:appflowy/workspace/application/settings/application_data_storage.dart';
 import 'package:appflowy_backend/log.dart';
+import 'package:appflowy_backend/protobuf/flowy-error/code.pb.dart';
 import 'package:easy_localization/easy_localization.dart';
 import 'package:flowy_infra/uuid.dart';
 import 'package:path/path.dart' as p;
@@ -63,6 +64,12 @@ Future<(String? path, String? errorMessage)> saveImageToCloudStorage(
       );
       return (s.url, null);
     },
-    (e) => (null, e.msg),
+    (err) {
+      if (err.code == ErrorCode.FileStorageLimitExceeded) {
+        return (null, LocaleKeys.sideBar_storageLimitDialogTitle.tr());
+      } else {
+        return (null, err.msg);
+      }
+    },
   );
 }

frontend/appflowy_flutter/lib/plugins/document/presentation/editor_plugins/openai/service/error.dart

@@ -1,14 +1,26 @@
 import 'package:freezed_annotation/freezed_annotation.dart';
+
 part 'error.freezed.dart';
 part 'error.g.dart';
 
 @freezed
 class AIError with _$AIError {
   const factory AIError({
-    String? code,
     required String message,
+    @Default(AIErrorCode.other) AIErrorCode code,
   }) = _AIError;
 
   factory AIError.fromJson(Map<String, Object?> json) =>
       _$AIErrorFromJson(json);
 }
+
+enum AIErrorCode {
+  @JsonValue('AIResponseLimitExceeded')
+  aiResponseLimitExceeded,
+  @JsonValue('Other')
+  other,
+}
+
+extension AIErrorExtension on AIError {
+  bool get isLimitExceeded => code == AIErrorCode.aiResponseLimitExceeded;
+}

frontend/appflowy_flutter/lib/plugins/document/presentation/editor_plugins/openai/widgets/ai_limit_dialog.dart

@@ -0,0 +1,35 @@
+import 'package:appflowy/generated/locale_keys.g.dart';
+import 'package:appflowy/workspace/presentation/widgets/dialogs.dart';
+import 'package:easy_localization/easy_localization.dart';
+import 'package:flutter/material.dart';
+
+class _AILimitDialog extends StatelessWidget {
+  const _AILimitDialog({
+    required this.message,
+    required this.onOkPressed,
+  });
+  final VoidCallback onOkPressed;
+  final String message;
+
+  @override
+  Widget build(BuildContext context) {
+    return NavigatorOkCancelDialog(
+      message: message,
+      okTitle: LocaleKeys.button_ok.tr(),
+      onOkPressed: onOkPressed,
+      titleUpperCase: false,
+    );
+  }
+}
+
+void showAILimitDialog(BuildContext context, String message) {
+  showDialog(
+    context: context,
+    barrierDismissible: false,
+    useRootNavigator: false,
+    builder: (dialogContext) => _AILimitDialog(
+      message: message,
+      onOkPressed: () {},
+    ),
+  );
+}

frontend/appflowy_flutter/lib/plugins/document/presentation/editor_plugins/openai/widgets/auto_completion_node_widget.dart

@@ -1,6 +1,7 @@
 import 'package:appflowy/generated/locale_keys.g.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/base/build_context_extension.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/base/text_robot.dart';
+import 'package:appflowy/plugins/document/presentation/editor_plugins/openai/service/error.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/openai/widgets/discard_dialog.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/openai/widgets/loading.dart';
 import 'package:appflowy/user/application/ai_service.dart';
@@ -17,6 +18,8 @@ import 'package:flowy_infra_ui/widget/spacing.dart';
 import 'package:flutter/material.dart';
 import 'package:provider/provider.dart';
 
+import 'ai_limit_dialog.dart';
+
 class AutoCompletionBlockKeys {
   const AutoCompletionBlockKeys._();
 
@@ -225,11 +228,15 @@ class _AutoCompletionBlockComponentState
       onError: (error) async {
         barrierDialog?.dismiss();
         if (mounted) {
-          showSnackBarMessage(
-            context,
-            error.message,
-            showCancel: true,
-          );
+          if (error.isLimitExceeded) {
+            showAILimitDialog(context, error.message);
+          } else {
+            showSnackBarMessage(
+              context,
+              error.message,
+              showCancel: true,
+            );
+          }
         }
       },
     );
@@ -304,11 +311,15 @@ class _AutoCompletionBlockComponentState
       onEnd: () async {},
       onError: (error) async {
         if (mounted) {
-          showSnackBarMessage(
-            context,
-            error.message,
-            showCancel: true,
-          );
+          if (error.isLimitExceeded) {
+            showAILimitDialog(context, error.message);
+          } else {
+            showSnackBarMessage(
+              context,
+              error.message,
+              showCancel: true,
+            );
+          }
         }
       },
     );

frontend/appflowy_flutter/lib/plugins/document/presentation/editor_plugins/openai/widgets/smart_edit_node_widget.dart

@@ -2,6 +2,7 @@ import 'dart:async';
 
 import 'package:appflowy/generated/locale_keys.g.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/openai/service/ai_client.dart';
+import 'package:appflowy/plugins/document/presentation/editor_plugins/openai/service/error.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/openai/widgets/discard_dialog.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/openai/widgets/smart_edit_action.dart';
 import 'package:appflowy/startup/startup.dart';
@@ -16,6 +17,8 @@ import 'package:flutter/material.dart';
 import 'package:http/http.dart' as http;
 import 'package:provider/provider.dart';
 
+import 'ai_limit_dialog.dart';
+
 class SmartEditBlockKeys {
   const SmartEditBlockKeys._();
 
@@ -426,11 +429,15 @@ class _SmartEditInputWidgetState extends State<SmartEditInputWidget> {
         });
       },
       onError: (error) async {
-        showSnackBarMessage(
-          context,
-          error.message,
-          showCancel: true,
-        );
+        if (error.isLimitExceeded) {
+          showAILimitDialog(context, error.message);
+        } else {
+          showSnackBarMessage(
+            context,
+            error.message,
+            showCancel: true,
+          );
+        }
         await _onExit();
       },
     );

frontend/appflowy_flutter/lib/shared/feature_flags.dart

@@ -91,6 +91,7 @@ enum FeatureFlag {
 
   bool get isOn {
     if ([
+      FeatureFlag.planBilling,
       // release this feature in version 0.6.1
       FeatureFlag.spaceDesign,
       // release this feature in version 0.5.9
@@ -110,14 +111,14 @@ enum FeatureFlag {
     }
 
     switch (this) {
+      case FeatureFlag.planBilling:
       case FeatureFlag.search:
       case FeatureFlag.syncDocument:
       case FeatureFlag.syncDatabase:
       case FeatureFlag.spaceDesign:
         return true;
       case FeatureFlag.collaborativeWorkspace:
       case FeatureFlag.membersSettings:
-      case FeatureFlag.planBilling:
       case FeatureFlag.unknown:
         return false;
     }

frontend/appflowy_flutter/lib/startup/tasks/appflowy_cloud_task.dart

@@ -95,7 +95,9 @@ class AppFlowyCloudDeepLink {
     }
 
     if (_isPaymentSuccessUri(uri)) {
-      return getIt<SubscriptionSuccessListenable>().onPaymentSuccess();
+      Log.debug("Payment success deep link: ${uri.toString()}");
+      final plan = uri.queryParameters['plan'];
+      return getIt<SubscriptionSuccessListenable>().onPaymentSuccess(plan);
     }
 
     return _isAuthCallbackDeepLink(uri).fold(

frontend/appflowy_flutter/lib/user/application/ai_service.dart

@@ -2,13 +2,15 @@ import 'dart:async';
 import 'dart:ffi';
 import 'dart:isolate';
 
+import 'package:appflowy/generated/locale_keys.g.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/openai/service/ai_client.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/openai/service/error.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/openai/service/text_completion.dart';
 import 'package:appflowy/plugins/document/presentation/editor_plugins/openai/widgets/smart_edit_action.dart';
 import 'package:appflowy_backend/dispatch/dispatch.dart';
 import 'package:appflowy_backend/protobuf/flowy-chat/entities.pb.dart';
 import 'package:appflowy_result/appflowy_result.dart';
+import 'package:easy_localization/easy_localization.dart';
 import 'package:fixnum/fixnum.dart' as fixnum;
 
 class AppFlowyAIService implements AIRepository {
@@ -85,6 +87,15 @@ class CompletionStream {
     _port.handler = _controller.add;
     _subscription = _controller.stream.listen(
       (event) async {
+        if (event == "AI_RESPONSE_LIMIT") {
+          onError(
+            AIError(
+              message: LocaleKeys.sideBar_aiResponseLitmit.tr(),
+              code: AIErrorCode.aiResponseLimitExceeded,
+            ),
+          );
+        }
+
         if (event.startsWith("start:")) {
           await onStart();
         }
@@ -96,6 +107,7 @@ class CompletionStream {
         if (event.startsWith("finish:")) {
           await onEnd();
         }
+
         if (event.startsWith("error:")) {
           onError(AIError(message: event.substring(6)));
         }

frontend/appflowy_flutter/lib/user/application/user_service.dart

@@ -2,6 +2,7 @@ import 'dart:async';
 
 import 'package:appflowy/env/cloud_env.dart';
 import 'package:appflowy/startup/startup.dart';
+import 'package:appflowy/workspace/application/settings/plan/workspace_subscription_ext.dart';
 import 'package:appflowy_backend/dispatch/dispatch.dart';
 import 'package:appflowy_backend/protobuf/flowy-error/errors.pb.dart';
 import 'package:appflowy_backend/protobuf/flowy-folder/workspace.pb.dart';
@@ -10,7 +11,10 @@ import 'package:appflowy_result/appflowy_result.dart';
 import 'package:fixnum/fixnum.dart';
 
 abstract class IUserBackendService {
-  Future<FlowyResult<void, FlowyError>> cancelSubscription(String workspaceId);
+  Future<FlowyResult<void, FlowyError>> cancelSubscription(
+    String workspaceId,
+    SubscriptionPlanPB plan,
+  );
   Future<FlowyResult<PaymentLinkPB, FlowyError>> createSubscription(
     String workspaceId,
     SubscriptionPlanPB plan,
@@ -228,9 +232,10 @@ class UserBackendService implements IUserBackendService {
     return UserEventLeaveWorkspace(data).send();
   }
 
-  static Future<FlowyResult<RepeatedWorkspaceSubscriptionPB, FlowyError>>
-      getWorkspaceSubscriptions() {
-    return UserEventGetWorkspaceSubscriptions().send();
+  static Future<FlowyResult<WorkspaceSubscriptionInfoPB, FlowyError>>
+      getWorkspaceSubscriptionInfo(String workspaceId) {
+    final params = UserWorkspaceIdPB.create()..workspaceId = workspaceId;
+    return UserEventGetWorkspaceSubscriptionInfo(params).send();
   }
 
   Future<FlowyResult<WorkspaceMemberPB, FlowyError>>
@@ -250,15 +255,32 @@ class UserBackendService implements IUserBackendService {
       ..recurringInterval = RecurringIntervalPB.Year
       ..workspaceSubscriptionPlan = plan
       ..successUrl =
-          '${getIt<AppFlowyCloudSharedEnv>().appflowyCloudConfig.base_url}/web/payment-success';
+          '${getIt<AppFlowyCloudSharedEnv>().appflowyCloudConfig.base_url}/web/payment-success?plan=${plan.toRecognizable()}';
     return UserEventSubscribeWorkspace(request).send();
   }
 
   @override
   Future<FlowyResult<void, FlowyError>> cancelSubscription(
     String workspaceId,
+    SubscriptionPlanPB plan,
   ) {
-    final request = UserWorkspaceIdPB()..workspaceId = workspaceId;
+    final request = CancelWorkspaceSubscriptionPB()
+      ..workspaceId = workspaceId
+      ..plan = plan;
+
     return UserEventCancelWorkspaceSubscription(request).send();
   }
+
+  Future<FlowyResult<void, FlowyError>> updateSubscriptionPeriod(
+    String workspaceId,
+    SubscriptionPlanPB plan,
+    RecurringIntervalPB interval,
+  ) {
+    final request = UpdateWorkspaceSubscriptionPaymentPeriodPB()
+      ..workspaceId = workspaceId
+      ..plan = plan
+      ..recurringInterval = interval;
+
+    return UserEventUpdateWorkspaceSubscriptionPaymentPeriod(request).send();
+  }
 }