feat(ai-scribe): Improve data & domain layer for ai scribe

This commit is contained in:
dab246
2025-12-18 04:31:18 +07:00
committed by Dat H. Pham
parent 1a1b8f489f
commit 860393fc90
16 changed files with 94 additions and 117 deletions
+26 -4
View File
@@ -1,13 +1,35 @@
import 'package:core/utils/app_logger.dart';
import 'package:core/utils/platform_info.dart';
import 'package:jmap_dart_client/jmap/account_id.dart';
import 'package:jmap_dart_client/jmap/core/session/session.dart';
import 'package:scribe/scribe.dart';
import 'package:scribe/scribe/ai/presentation/bindings/ai_scribe_bindings.dart';
import 'package:tmail_ui_user/features/home/domain/extensions/session_extensions.dart';
mixin AiScribeMixin {
bool isAICapabilitySupported({Session? session, AccountId? accountId}) {
AICapability? getAICapability({Session? session, AccountId? accountId}) {
if (PlatformInfo.isMobile) return null;
if (accountId == null || session == null) {
return false;
return null;
}
return session.getAICapability(accountId);
}
void injectAIScribeBindings(Session? session, AccountId? accountId) {
try {
final aiCapability = getAICapability(
session: session,
accountId: accountId,
);
final scribeEndpoint = aiCapability?.scribeEndpoint;
if (scribeEndpoint == null || scribeEndpoint.isEmpty) return;
AIScribeBindings(scribeEndpoint).dependencies();
} catch (e) {
logError('AiScribeMixin::injectAIScribeBindings(): $e');
}
final aiCapability = session.getAICapability(accountId);
return aiCapability != null;
}
}
@@ -263,8 +263,6 @@ class ComposerController extends BaseController
TransformHtmlEmailContentInteractor get transformHtmlEmailContentInteractor => _transformHtmlEmailContentInteractor;
GenerateAITextInteractor get generateAITextInteractor => Get.find<GenerateAITextInteractor>();
String get ownEmailAddress =>
mailboxDashBoardController.ownEmailAddress.value;
@@ -10,10 +10,11 @@ import 'package:tmail_ui_user/features/composer/presentation/mixin/text_selectio
extension HandleAiScribeInComposerExtension on ComposerController {
bool get isAIScribeAvailable {
return isAICapabilitySupported(
final aiCapability = getAICapability(
session: mailboxDashBoardController.sessionCurrent,
accountId: mailboxDashBoardController.accountId.value,
);
return aiCapability?.isScribeEndpointAvailable == true;
}
Future<String> _getTextOnlyContentInEditor() async {
@@ -31,6 +31,7 @@ import 'package:receive_sharing_intent/receive_sharing_intent.dart';
import 'package:rxdart/transformers.dart';
import 'package:server_settings/server_settings/tmail_server_settings_extension.dart';
import 'package:tmail_ui_user/features/base/action/ui_action.dart';
import 'package:tmail_ui_user/features/base/mixin/ai_scribe_mixin.dart';
import 'package:tmail_ui_user/features/base/mixin/contact_support_mixin.dart';
import 'package:tmail_ui_user/features/base/mixin/message_dialog_action_manager.dart';
import 'package:tmail_ui_user/features/base/mixin/own_email_address_mixin.dart';
@@ -78,15 +79,8 @@ import 'package:tmail_ui_user/features/email/presentation/model/composer_argumen
import 'package:tmail_ui_user/features/email/presentation/utils/email_utils.dart';
import 'package:tmail_ui_user/features/email_recovery/presentation/model/email_recovery_arguments.dart';
import 'package:tmail_ui_user/features/home/data/exceptions/session_exceptions.dart';
import 'package:tmail_ui_user/features/home/domain/extensions/session_extensions.dart';
import 'package:tmail_ui_user/features/home/domain/state/auto_sign_in_via_deep_link_state.dart';
import 'package:tmail_ui_user/features/home/domain/usecases/store_session_interactor.dart';
import 'package:scribe/scribe/ai/data/datasource/ai_datasource.dart';
import 'package:scribe/scribe/ai/data/datasource_impl/ai_datasource_impl.dart';
import 'package:scribe/scribe/ai/data/repository/ai_repository_impl.dart';
import 'package:scribe/scribe/ai/domain/repository/ai_scribe_repository.dart';
import 'package:scribe/scribe/ai/domain/usecases/generate_ai_text_interactor.dart';
import 'package:scribe/scribe/ai/presentation/bindings/ai_scribe_bindings.dart';
import 'package:tmail_ui_user/features/identity_creator/domain/state/get_identity_cache_on_web_state.dart';
import 'package:tmail_ui_user/features/identity_creator/domain/usecase/get_identity_cache_on_web_interactor.dart';
import 'package:tmail_ui_user/features/login/domain/exceptions/logout_exception.dart';
@@ -223,7 +217,8 @@ import 'package:uuid/uuid.dart';
class MailboxDashBoardController extends ReloadableController
with ContactSupportMixin,
OwnEmailAddressMixin,
SaaSPremiumMixin {
SaaSPremiumMixin,
AiScribeMixin {
final RemoveEmailDraftsInteractor _removeEmailDraftsInteractor = Get.find<RemoveEmailDraftsInteractor>();
final EmailReceiveManager _emailReceiveManager = Get.find<EmailReceiveManager>();
@@ -825,41 +820,6 @@ class MailboxDashBoardController extends ReloadableController
}
}
void injectAIScribeBindings(Session? session, AccountId? accountId) {
try {
if (!PlatformInfo.isWeb) return;
if (session == null || accountId == null) return;
final aiCapability = session.getAICapability(accountId);
final scribeEndpoint = aiCapability?.scribeEndpoint;
if (scribeEndpoint == null || scribeEndpoint.isEmpty) return;
// Delete existing AI bindings if they exist
if (Get.isRegistered<AIDataSourceImpl>()) {
Get.delete<AIDataSourceImpl>(force: true);
}
if (Get.isRegistered<AIDataSource>()) {
Get.delete<AIDataSource>(force: true);
}
if (Get.isRegistered<AIScribeRepositoryImpl>()) {
Get.delete<AIScribeRepositoryImpl>(force: true);
}
if (Get.isRegistered<AIScribeRepository>()) {
Get.delete<AIScribeRepository>(force: true);
}
if (Get.isRegistered<GenerateAITextInteractor>()) {
Get.delete<GenerateAITextInteractor>(force: true);
}
// Reinitialize with the correct endpoint
AIScribeBindings(scribeEndpoint: scribeEndpoint).dependencies();
} catch (e) {
logError('MailboxDashBoardController::injectAIScribeBindings(): $e');
}
}
@override
Future<void> injectFCMBindings(Session? session, AccountId? accountId) async {
try {
@@ -14,6 +14,7 @@ import 'package:model/model.dart';
import 'package:rule_filter/rule_filter/capability_rule_filter.dart';
import 'package:server_settings/server_settings/capability_server_settings.dart';
import 'package:tmail_ui_user/features/base/action/ui_action.dart';
import 'package:tmail_ui_user/features/base/mixin/ai_scribe_mixin.dart';
import 'package:tmail_ui_user/features/base/mixin/own_email_address_mixin.dart';
import 'package:tmail_ui_user/features/base/reloadable/reloadable_controller.dart';
import 'package:tmail_ui_user/features/base/widget/dialog_picker/color_dialog_picker.dart';
@@ -55,7 +56,7 @@ import 'package:tmail_ui_user/main/routes/route_utils.dart';
import 'package:tmail_ui_user/main/utils/app_config.dart';
class ManageAccountDashBoardController extends ReloadableController
with OwnEmailAddressMixin {
with OwnEmailAddressMixin, AiScribeMixin {
GetAllVacationInteractor? _getAllVacationInteractor;
UpdateVacationInteractor? _updateVacationInteractor;
@@ -152,6 +153,7 @@ class ManageAccountDashBoardController extends ReloadableController
_setUpMinInputLengthAutocomplete();
_bindingInteractorForMenuItemView(sessionCurrent, accountId.value);
_getVacationResponse();
injectAIScribeBindings(sessionCurrent, accountId.value);
paywallController = PaywallController(
ownEmailAddress: ownEmailAddress.value,
);