From ace03e5b5a6f3d62ce4932826774aeed03ae9b1e Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Wed, 11 Jun 2025 20:07:33 +0800
Subject: [PATCH] fix: enhance knowledge retrieval logic with dynamic SQL
 selection based on workspace management

---
 apps/knowledge/sql/list_knowledge_user.sql    | 49 ++++++++++++++++
 apps/knowledge/sql/list_knowledge_user_ee.sql | 58 +++++++++++++++++++
 2 files changed, 107 insertions(+)
 create mode 100644 apps/knowledge/sql/list_knowledge_user.sql
 create mode 100644 apps/knowledge/sql/list_knowledge_user_ee.sql

diff --git a/apps/knowledge/sql/list_knowledge_user.sql b/apps/knowledge/sql/list_knowledge_user.sql
new file mode 100644
index 000000000..a780a6cbb
--- /dev/null
+++ b/apps/knowledge/sql/list_knowledge_user.sql
@@ -0,0 +1,49 @@
+SELECT *
+FROM (SELECT "temp_knowledge".id::text, "temp_knowledge".name,
+             "temp_knowledge".desc,
+             "temp_knowledge".type,
+             'knowledge'                               as resource_type,
+             "temp_knowledge".workspace_id,
+             "temp_knowledge".folder_id,
+             "temp_knowledge".user_id,
+             "temp_knowledge".create_time,
+             "temp_knowledge".update_time,
+             "temp_knowledge".file_size_limit,
+             "temp_knowledge".file_count_limit,
+             "document_temp"."char_length",
+             CASE
+                 WHEN
+                     "app_knowledge_temp"."count" IS NULL THEN 0
+                 ELSE "app_knowledge_temp"."count" END AS application_mapping_count,
+             "document_temp".document_count
+      FROM (SELECT knowledge.*
+            FROM knowledge knowledge ${knowledge_custom_sql}
+            AND id in (select target
+                   from workspace_user_resource_permission
+                   where auth_target_type = 'KNOWLEDGE'
+                     and 'VIEW' = any (permission_list))) temp_knowledge
+               LEFT JOIN (SELECT "count"("id") AS document_count, "sum"("char_length") "char_length", knowledge_id
+                          FROM "document"
+                          GROUP BY knowledge_id) "document_temp" ON temp_knowledge."id" = "document_temp".knowledge_id
+               LEFT JOIN (SELECT "count"("id"), knowledge_id
+                          FROM application_knowledge_mapping
+                          GROUP BY knowledge_id) app_knowledge_temp
+                         ON temp_knowledge."id" = "app_knowledge_temp".knowledge_id
+      UNION
+      SELECT "id",
+             "name",
+             "desc",
+             0           as "type",
+             'folder'    as "resource_type",
+             "workspace_id",
+             "parent_id" as "folder_id",
+             "user_id",
+             "create_time",
+             "update_time",
+             0           as file_size_limit,
+             0           as file_count_limit,
+             0 as char_length,
+                0 as application_mapping_count,
+                0 as document_count
+      from knowledge_folder ${folder_query_set}) temp
+    ${default_sql}
\ No newline at end of file
diff --git a/apps/knowledge/sql/list_knowledge_user_ee.sql b/apps/knowledge/sql/list_knowledge_user_ee.sql
new file mode 100644
index 000000000..4813966e9
--- /dev/null
+++ b/apps/knowledge/sql/list_knowledge_user_ee.sql
@@ -0,0 +1,58 @@
+SELECT *
+FROM (SELECT "temp_knowledge".id::text, "temp_knowledge".name,
+             "temp_knowledge".desc,
+             "temp_knowledge".type,
+             'knowledge'                               as resource_type,
+             "temp_knowledge".workspace_id,
+             "temp_knowledge".folder_id,
+             "temp_knowledge".user_id,
+             "temp_knowledge".create_time,
+             "temp_knowledge".update_time,
+             "temp_knowledge".file_size_limit,
+             "temp_knowledge".file_count_limit,
+             "document_temp"."char_length",
+             CASE
+                 WHEN
+                     "app_knowledge_temp"."count" IS NULL THEN 0
+                 ELSE "app_knowledge_temp"."count" END AS application_mapping_count,
+             "document_temp".document_count
+      FROM (SELECT knowledge.*
+            FROM knowledge knowledge ${knowledge_custom_sql}
+            AND id in (select target
+                   from workspace_user_resource_permission
+                   where auth_target_type = 'KNOWLEDGE'
+                     and case
+                             when auth_type = 'ROLE' then
+                                 'KNOWLEDGE_READ' in (select permission_id
+                                                        from role_permission
+                                                        where role_id in (select role_id
+                                                                          from user_role_relation))
+                             else
+                                 'VIEW' = any (permission_list)
+                       end
+            )) temp_knowledge
+               LEFT JOIN (SELECT "count"("id") AS document_count, "sum"("char_length") "char_length", knowledge_id
+                          FROM "document"
+                          GROUP BY knowledge_id) "document_temp" ON temp_knowledge."id" = "document_temp".knowledge_id
+               LEFT JOIN (SELECT "count"("id"), knowledge_id
+                          FROM application_knowledge_mapping
+                          GROUP BY knowledge_id) app_knowledge_temp
+                         ON temp_knowledge."id" = "app_knowledge_temp".knowledge_id
+      UNION
+      SELECT "id",
+             "name",
+             "desc",
+             0           as "type",
+             'folder'    as "resource_type",
+             "workspace_id",
+             "parent_id" as "folder_id",
+             "user_id",
+             "create_time",
+             "update_time",
+             0           as file_size_limit,
+             0           as file_count_limit,
+             0 as char_length,
+                0 as application_mapping_count,
+                0 as document_count
+      from knowledge_folder ${folder_query_set}) temp
+    ${default_sql}
\ No newline at end of file