[feature]: Enable custom rules support in the rule knowledge

src/api/common.d.ts

@@ -177,6 +177,8 @@ export interface IAuditResDataV1 {
 }
 
 export interface IAuditResult {
+  db_type?: string;
+
   level?: string;
 
   message?: string;

src/components/AuditResultErrorMessage/index.test.tsx

@@ -6,17 +6,24 @@ import { renderWithRouter } from '../../testUtils/customRender';
 
 describe('AuditResultErrorMessage', () => {
   const auditResult: IAuditResult[] = [
-    { level: 'error', message: 'schema 不存在', rule_name: '' },
+    {
+      level: 'error',
+      message: 'schema 不存在',
+      rule_name: '',
+      db_type: 'MySQL',
+    },
     {
       level: 'error',
       message:
         '禁止使用没有where条件的sql语句或者使用where 1=1等变相没有条件的sql',
       rule_name: 'all_check_where_is_invalid',
+      db_type: 'MySQL',
     },
     {
       level: 'warn',
       message: 'select 语句必须带limit,且限制数不得超过1000',
       rule_name: 'dml_check_select_limit',
+      db_type: 'MySQL',
     },
   ];
 
@@ -55,6 +62,7 @@ describe('AuditResultErrorMessage', () => {
     expect(getRulesSpy).toBeCalledTimes(1);
     expect(getRulesSpy).nthCalledWith(1, {
       filter_rule_names: 'all_check_where_is_invalid,dml_check_select_limit',
+      filter_db_type: 'MySQL',
     });
 
     fireEvent.mouseEnter(

src/components/AuditResultErrorMessage/index.tsx

@@ -22,14 +22,22 @@ const AuditResultErrorMessage: React.FC<AuditResultErrorMessageProps> = (
     [props.auditResult]
   );
 
+  const filterDbType = useMemo(() => {
+    return props.auditResult?.find((v) => !!v.db_type)?.db_type;
+  }, [props.auditResult]);
+
   const { data: ruleInfo } = useRequest(
     () =>
       rule_template
         .getRuleListV1({
           filter_rule_names: filterRuleNames.join(','),
+          filter_db_type: filterDbType,
         })
         .then((res) => res.data.data),
-    { ready: !!filterRuleNames.length, refreshDeps: [filterRuleNames] }
+    {
+      ready: !!filterRuleNames.length,
+      refreshDeps: [filterRuleNames, filterDbType],
+    }
   );
 
   return (
@@ -58,8 +66,7 @@ const AuditResultErrorMessage: React.FC<AuditResultErrorMessageProps> = (
 
                         {/* IFTRUE_isEE */}
                         {'  '}
-                        {/* 暂时不支持自定义规则 */}
-                        <EmptyBox if={!rule?.is_custom_rule && !!rule?.db_type}>
+                        <EmptyBox if={!!rule?.db_type}>
                           <Link
                             target="_blank"
                             to={`rule/knowledge/${v.rule_name}?db_type=${rule?.db_type}`}

src/components/RuleList/index.tsx

@@ -88,8 +88,7 @@ const RuleList: React.FC<RuleListProps> = (props) => {
                         {item.annotation}
                         {/* IFTRUE_isEE */}
                         {'  '}
-                        {/* 暂时不支持自定义规则 */}
-                        <EmptyBox if={!item?.is_custom_rule && !!item?.db_type}>
+                        <EmptyBox if={!!item?.db_type}>
                           <Link
                             target="_blank"
                             to={`rule/knowledge/${item.rule_name}?db_type=${item?.db_type}`}

src/page/Order/AuditResult/__test__/index.test.tsx

@@ -225,6 +225,7 @@ describe('Order/Detail/AuditResult', () => {
     expect(getAllRulesSpy).toBeCalledTimes(1);
     expect(getAllRulesSpy).nthCalledWith(1, {
       filter_rule_names: 'all_check_where_is_invalid',
+      filter_db_type: 'MySQL',
     });
   });
 });

src/page/Order/Detail/__testData__/index.tsx

@@ -524,6 +524,7 @@ export const taskSqls = [
         message:
           '禁止使用没有where条件的sql语句或者使用where 1=1等变相没有条件的sql',
         rule_name: 'all_check_where_is_invalid',
+        db_type: 'MySQL',
       },
     ],
     audit_level: 'error',

src/page/RuleKnowledge/RuleUnderstand/index.tsx

@@ -18,6 +18,7 @@ const RuleUnderstand: React.FC<RuleUnderstandProps> = ({
   dbType,
   loading,
   isAdmin,
+  isCustomRule,
 }) => {
   const { t } = useTranslation();
   const [modifyFlag, { setTrue: startModify, setFalse: modifyFinish }] =
@@ -45,12 +46,19 @@ const RuleUnderstand: React.FC<RuleUnderstandProps> = ({
   };
   const submit = () => {
     startSubmit();
-    rule_template
-      .updateRuleKnowledge({
-        rule_name: ruleName,
-        knowledge_content: editValue,
-        db_type: dbType,
-      })
+    const request = isCustomRule
+      ? rule_template.updateCustomRuleKnowledge({
+          rule_name: ruleName,
+          knowledge_content: editValue,
+          db_type: dbType,
+        })
+      : rule_template.updateRuleKnowledge({
+          rule_name: ruleName,
+          knowledge_content: editValue,
+          db_type: dbType,
+        });
+
+    request
       .then((res) => {
         if (res.data.code === ResponseCode.SUCCESS) {
           message.success(t('ruleKnowledge.successTips'));

src/page/RuleKnowledge/RuleUnderstand/index.type.ts

@@ -7,6 +7,7 @@ export type RuleUnderstandProps = {
   dbType: string;
   loading: boolean;
   isAdmin: boolean;
+  isCustomRule: boolean;
 };
 
 export type EditKnowledgeContentProps = {

src/page/RuleKnowledge/index.tsx

@@ -8,32 +8,65 @@ import RuleUnderstand from './RuleUnderstand';
 import { useEffect, useState } from 'react';
 import EmptyBox from '../../components/EmptyBox';
 import useCurrentUser from '../../hooks/useCurrentUser';
+import { ResponseCode } from '../../data/common';
 
 const RuleKnowledge: React.FC = () => {
   const { t } = useTranslation();
   const location = useLocation();
   const theme = useTheme();
   const { ruleName = '' } = useParams<{ ruleName: string }>();
   const [dbType, setDbType] = useState<string>();
+  const [isCustomRule, setIsCustomRule] = useState(false);
   const { isAdmin } = useCurrentUser();
 
   const {
     data: ruleKnowledgeInfo,
     loading,
     refresh,
+    run: getRuleKnowledge,
   } = useRequest(
-    () =>
-      rule_template
-        .getRuleKnowledgeV1({ rule_name: ruleName, db_type: dbType! })
+    (isCustomType: boolean, dbType: string) => {
+      if (isCustomType) {
+        return rule_template
+          .getCustomRuleKnowledgeV1({ rule_name: ruleName, db_type: dbType })
+          .then((res) => {
+            return res.data.data;
+          });
+      }
+      return rule_template
+        .getRuleKnowledgeV1({ rule_name: ruleName, db_type: dbType })
         .then((res) => {
           return res.data.data;
-        }),
+        });
+    },
     {
-      ready: !!ruleName && !!dbType,
-      refreshDeps: [ruleName, dbType],
+      manual: true,
     }
   );
 
+  useEffect(() => {
+    if (!!ruleName && !!dbType) {
+      rule_template
+        .getRuleListV1({
+          filter_rule_names: ruleName,
+          filter_db_type: dbType,
+        })
+        .then((res) => {
+          if (
+            res.data.code === ResponseCode.SUCCESS &&
+            res.data.data?.length === 1
+          ) {
+            const ruleInfo = res.data.data[0];
+            if (!!ruleInfo.db_type) {
+              setIsCustomRule(!!ruleInfo.is_custom_rule);
+              getRuleKnowledge(!!ruleInfo.is_custom_rule, ruleInfo.db_type);
+            }
+          }
+        });
+    }
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [dbType]);
+
   useEffect(() => {
     const searchParams = new URLSearchParams(location.search);
 
@@ -73,6 +106,7 @@ const RuleKnowledge: React.FC = () => {
               refresh={refresh}
               dbType={dbType!}
               isAdmin={isAdmin}
+              isCustomRule={isCustomRule}
             />
           </EmptyBox>
         </Space>