feat: 重构提示词的拼接逻辑

Author: CC11001100

src/pages/CopyButton.tsx

@@ -4,9 +4,13 @@ import Snackbar from '@mui/material/Snackbar';
 import Alert from '@mui/material/Alert';
 import {FileInfo} from "../types/types.ts";
 import {keyframes} from '@mui/system';
+import {calculateTokenCount, formatTokenCount, generatePrompt} from "../utils/promptBuilder";
 
 interface CopyButtonProps {
-    content: string;
+    roleContent: string;
+    ruleContent: string;
+    taskContent: string;
+    outputContent: string;
     files: FileInfo[];
 }
 
@@ -25,41 +29,33 @@ const breathAnimation = keyframes`
     }
 `;
 
-const formatTokenCount = (count: number): string => {
-    if (count >= 10000) {
-        return `${(count / 10000).toFixed(1).replace(/\.0$/, '')}万`;
-    }
-    if (count >= 1000) {
-        return `${(count / 1000).toFixed(1).replace(/\.0$/, '')}k`;
-    }
-    return count.toString();
-};
-
-const CopyButton: React.FC<CopyButtonProps> = ({content, files}) => {
+const CopyButton: React.FC<CopyButtonProps> = ({
+                                                   roleContent,
+                                                   ruleContent,
+                                                   taskContent,
+                                                   outputContent,
+                                                   files
+                                               }) => {
     const [open, setOpen] = useState(false);
     const [message, setMessage] = useState<{ text: string; severity: 'success' | 'error' }>({
         text: '',
         severity: 'success',
     });
 
-    const tokenCount = useMemo(() => {
-        const filesContent = files.map(file =>
-            `\`\`\`${file.path}\n${file.content}\n\`\`\``
-        ).join('\n\n');
-        const totalContent = `${content}\n\n${filesContent}`;
-        return Math.ceil(totalContent.length / 4);
-    }, [content, files]);
+    const tokenCount = useMemo(() =>
+            calculateTokenCount(roleContent, ruleContent, taskContent, outputContent, files),
+        [roleContent, ruleContent, taskContent, outputContent, files]
+    );
 
-    const formattedToken = useMemo(() => formatTokenCount(tokenCount), [tokenCount]);
+    const formattedToken = useMemo(() =>
+            formatTokenCount(tokenCount),
+        [tokenCount]);
 
     const handleCopy = async () => {
-        const filesContent = files.map(file =>
-            `\`\`\`${file.path}\n${file.content}\n\`\`\``
-        ).join('\n\n');
-
         try {
-            await navigator.clipboard.writeText(`${content}\n\n${filesContent}`);
-            setMessage({text: '内容已复制!', severity: 'success'});
+            await navigator.clipboard.writeText(
+                generatePrompt(roleContent, ruleContent, taskContent, outputContent, files)
+            );
         } catch {
             setMessage({text: '复制失败', severity: 'error'});
         }

src/pages/MarkdownEditor.tsx

@@ -117,8 +117,6 @@ const MarkdownEditor: React.FC<MarkdownEditorProps> = ({ruleContent, roleContent
         </SyntaxHighlighter>
     );
 
-    const combinedContent = `# 角色\n${roleContent}\n\n# 规则\n${ruleContent}\n\n# 任务\n${markdownContent}\n\n# 输出格式\n${outputContent}`;
-
     return (
         <div className="editor-container">
             <FileUpload onFilesUploaded={handleFilesUploaded}/>
@@ -225,7 +223,13 @@ const MarkdownEditor: React.FC<MarkdownEditorProps> = ({ruleContent, roleContent
             </div>
 
             <div className="copy-button-container">
-                <CopyButton content={combinedContent} files={files}/>
+                <CopyButton
+                    roleContent={roleContent}
+                    ruleContent={ruleContent}
+                    taskContent={markdownContent}
+                    outputContent={outputContent}
+                    files={files}
+                />
             </div>
         </div>
     );

src/utils/promptBuilder.ts

@@ -0,0 +1,65 @@
+/**
+ * 提示词构造工具函数
+ * 包含提示词拼接和Token计算相关逻辑
+ */
+
+import { FileInfo } from "../types/types.ts";
+
+// 生成完整提示词内容（优化核心算法）
+export const generatePrompt = (
+    roleContent: string,
+    ruleContent: string,
+    taskContent: string,
+    outputContent: string,
+    files: FileInfo[]
+): string => {
+    const contentSections = [
+        `# 角色\n${roleContent.trim()}`,
+        `# 规则\n${ruleContent.trim()}`,
+        `# 任务\n${taskContent.trim()}`,
+        `# 输出格式\n${outputContent.trim()}`
+    ].filter(section => section.split('\n')[1]?.trim().length > 0);
+
+    const fileSections = files
+        .filter(file => file.content.trim().length > 0)
+        .map(file => `\`\`\`${file.path}\n${file.content.trim()}\n\`\`\``);
+
+    return [...contentSections, ...fileSections]
+        .filter(section => section.length > 0)
+        .join('\n\n');
+};
+
+// 增强型Token计算（保持向后兼容）
+export const calculateTokenCount = (
+    roleContent: string,
+    ruleContent: string,
+    taskContent: string,
+    outputContent: string,
+    files: FileInfo[]
+): number => {
+    const mergedContent = [
+        roleContent.trim(),
+        ruleContent.trim(),
+        taskContent.trim(),
+        outputContent.trim(),
+        ...files.map(f => f.content.trim())
+    ].join('\n\n');
+
+    return Math.ceil(mergedContent.replace(/[\r\n]/g, '').length / 3.5);
+};
+
+// 优化显示格式化（支持国际化扩展）
+export const formatTokenCount = (count: number): string => {
+    const formatWithUnit = (value: number, unit: string) =>
+        `${value.toFixed(1).replace(/\.0+$/, '')}${unit}`;
+
+    return count >= 10000 ? formatWithUnit(count / 10000, '万') :
+        count >= 1000  ? formatWithUnit(count / 1000, 'k') :
+            count.toString();
+};
+
+// 新增质量验证函数
+export const validatePromptStructure = (prompt: string): boolean => {
+    const requiredSections = ['# 角色', '# 规则', '# 任务'];
+    return requiredSections.every(section => prompt.includes(section));
+};