feat: improve ui logic and support more email operations

1 files changed, 13 insertions, 2 deletions

diff --git a/src/ui/UIModuleManager.cpp b/src/ui/UIModuleManager.cpp
index ed18f473..ba6172a4 100644
--- a/src/ui/UIModuleManager.cpp
+++ b/src/ui/UIModuleManager.cpp
@@ -61,7 +61,7 @@ auto UIModuleManager::DeclareMountPoint(
 
 auto UIModuleManager::MountEntry(const QString& id,
                                  QMap<QString, QString> meta_data,
-                                 EntryFactory factory) -> bool {
+                                 QObjectFactory factory) -> bool {
   if (id.isEmpty() || !mount_points_.contains(id)) return false;
 
   if (factory == nullptr) return false;
@@ -82,7 +82,7 @@ auto UIModuleManager::QueryMountedEntries(QString id) -> QList<MountedUIEntry> {
 }
 
 auto MountedUIEntry::GetWidget() const -> QWidget* {
-  return qobject_cast<QWidget*>(static_cast<QObject*>(factory_(id_.toUtf8())));
+  return qobject_cast<QWidget*>(static_cast<QObject*>(factory_(nullptr)));
 }
 
 auto MountedUIEntry::GetMetaDataByDefault(
@@ -176,4 +176,15 @@ void UIModuleManager::TranslateAllModulesParams() {
 #endif
 }
 
+auto UIModuleManager::RegisterQObject(const QString& id, QObject* p) -> bool {
+  if (id.isEmpty() || registered_qobjects_.contains(id)) return false;
+
+  registered_qobjects_[id] = p;
+  return true;
+}
+
+auto UIModuleManager::GetQObject(const QString& id) -> QObject* {
+  return registered_qobjects_.value(id, nullptr);
+}
+
 }  // namespace GpgFrontend::UI
\ No newline at end of file