289 lines
7.2 KiB
JavaScript
289 lines
7.2 KiB
JavaScript
/**
|
||
* 工具函数集合
|
||
*/
|
||
|
||
// HTML转义，防止XSS攻击
|
||
function escapeHtml(text) {
|
||
const div = document.createElement('div');
|
||
div.textContent = text;
|
||
return div.innerHTML;
|
||
}
|
||
|
||
// 滚动到指定元素的底部
|
||
function scrollToBottom(element) {
|
||
if (element) {
|
||
element.scrollTop = element.scrollHeight;
|
||
}
|
||
}
|
||
|
||
// 生成唯一ID
|
||
function generateId(prefix = 'id') {
|
||
return `${prefix}-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
|
||
}
|
||
|
||
// 解析SVG响应，提取SVG内容和前后文本
|
||
function parseSVGResponse(response) {
|
||
const svgRegex = /```svg\s*([\s\S]*?)```/i;
|
||
const match = response.match(svgRegex);
|
||
|
||
if (match) {
|
||
const svgContent = match[1].trim();
|
||
const beforeText = response.substring(0, match.index).trim();
|
||
const afterText = response.substring(match.index + match[0].length).trim();
|
||
|
||
return {
|
||
svgContent,
|
||
beforeText,
|
||
afterText
|
||
};
|
||
}
|
||
|
||
return {
|
||
svgContent: null,
|
||
beforeText: response,
|
||
afterText: ''
|
||
};
|
||
}
|
||
|
||
// 下载文件
|
||
function downloadFile(content, filename, mimeType = 'text/plain') {
|
||
const blob = new Blob([content], { type: mimeType });
|
||
const url = URL.createObjectURL(blob);
|
||
const a = document.createElement('a');
|
||
a.href = url;
|
||
a.download = filename;
|
||
document.body.appendChild(a);
|
||
a.click();
|
||
document.body.removeChild(a);
|
||
URL.revokeObjectURL(url);
|
||
}
|
||
|
||
// 显示状态信息
|
||
function showStatus(element, message, type = 'info') {
|
||
if (!element) return;
|
||
|
||
element.classList.remove('hidden');
|
||
element.textContent = message;
|
||
|
||
// 移除所有状态类
|
||
element.classList.remove('border-gray-300', 'bg-gray-50', 'text-gray-600');
|
||
element.classList.remove('border-green-500', 'bg-green-50', 'text-green-700');
|
||
element.classList.remove('border-red-500', 'bg-red-50', 'text-red-700');
|
||
element.classList.remove('border-blue-500', 'bg-blue-50', 'text-blue-700');
|
||
|
||
// 根据类型添加相应的样式类
|
||
switch (type) {
|
||
case 'success':
|
||
element.classList.add('border-green-500', 'bg-green-50', 'text-green-700');
|
||
break;
|
||
case 'error':
|
||
element.classList.add('border-red-500', 'bg-red-50', 'text-red-700');
|
||
break;
|
||
case 'loading':
|
||
element.classList.add('border-blue-500', 'bg-blue-50', 'text-blue-700');
|
||
break;
|
||
default:
|
||
element.classList.add('border-gray-300', 'bg-gray-50', 'text-gray-600');
|
||
}
|
||
}
|
||
|
||
// 本地存储操作
|
||
const storage = {
|
||
// 保存数据到本地存储
|
||
set(key, value) {
|
||
try {
|
||
localStorage.setItem(key, JSON.stringify(value));
|
||
return true;
|
||
} catch (error) {
|
||
console.error('保存到本地存储失败:', error);
|
||
return false;
|
||
}
|
||
},
|
||
|
||
// 从本地存储获取数据
|
||
get(key, defaultValue = null) {
|
||
try {
|
||
const item = localStorage.getItem(key);
|
||
return item ? JSON.parse(item) : defaultValue;
|
||
} catch (error) {
|
||
console.error('从本地存储获取数据失败:', error);
|
||
return defaultValue;
|
||
}
|
||
},
|
||
|
||
// 删除本地存储中的数据
|
||
remove(key) {
|
||
try {
|
||
localStorage.removeItem(key);
|
||
return true;
|
||
} catch (error) {
|
||
console.error('删除本地存储数据失败:', error);
|
||
return false;
|
||
}
|
||
}
|
||
};
|
||
|
||
// 防抖函数
|
||
function debounce(func, wait) {
|
||
let timeout;
|
||
return function executedFunction(...args) {
|
||
const later = () => {
|
||
clearTimeout(timeout);
|
||
func(...args);
|
||
};
|
||
clearTimeout(timeout);
|
||
timeout = setTimeout(later, wait);
|
||
};
|
||
}
|
||
|
||
// 节流函数
|
||
function throttle(func, limit) {
|
||
let inThrottle;
|
||
return function() {
|
||
const args = arguments;
|
||
const context = this;
|
||
if (!inThrottle) {
|
||
func.apply(context, args);
|
||
inThrottle = true;
|
||
setTimeout(() => inThrottle = false, limit);
|
||
}
|
||
};
|
||
}
|
||
|
||
// 格式化日期时间
|
||
function formatDateTime(date = new Date()) {
|
||
return date.toLocaleString('zh-CN', {
|
||
year: 'numeric',
|
||
month: '2-digit',
|
||
day: '2-digit',
|
||
hour: '2-digit',
|
||
minute: '2-digit',
|
||
second: '2-digit'
|
||
});
|
||
}
|
||
|
||
// 深拷贝对象
|
||
function deepClone(obj) {
|
||
if (obj === null || typeof obj !== 'object') return obj;
|
||
if (obj instanceof Date) return new Date(obj.getTime());
|
||
if (obj instanceof Array) return obj.map(item => deepClone(item));
|
||
if (typeof obj === 'object') {
|
||
const clonedObj = {};
|
||
for (const key in obj) {
|
||
if (obj.hasOwnProperty(key)) {
|
||
clonedObj[key] = deepClone(obj[key]);
|
||
}
|
||
}
|
||
return clonedObj;
|
||
}
|
||
}
|
||
|
||
// 检查对象是否为空
|
||
function isEmpty(obj) {
|
||
if (obj == null) return true;
|
||
if (Array.isArray(obj) || typeof obj === 'string') return obj.length === 0;
|
||
if (typeof obj === 'object') return Object.keys(obj).length === 0;
|
||
return false;
|
||
}
|
||
|
||
// 自动调整文本域高度
|
||
function autoResizeTextarea(textarea) {
|
||
if (!textarea) return;
|
||
|
||
// 重置高度以获取正确的scrollHeight
|
||
textarea.style.height = 'auto';
|
||
|
||
// 计算新高度，限制最大高度
|
||
const newHeight = Math.min(textarea.scrollHeight, 120); // 最大120px（约5行）
|
||
textarea.style.height = newHeight + 'px';
|
||
}
|
||
|
||
// 流式文本处理
|
||
class StreamProcessor {
|
||
constructor(onChunk, onComplete) {
|
||
this.onChunk = onChunk;
|
||
this.onComplete = onComplete;
|
||
this.buffer = '';
|
||
}
|
||
|
||
// 处理数据块
|
||
processChunk(chunk) {
|
||
this.buffer += chunk;
|
||
|
||
// 尝试解析完整的JSON行
|
||
const lines = this.buffer.split('\n');
|
||
this.buffer = lines.pop() || ''; // 保留不完整的行
|
||
|
||
for (const line of lines) {
|
||
if (line.trim()) {
|
||
try {
|
||
// 处理SSE格式
|
||
if (line.startsWith('data: ')) {
|
||
const data = line.slice(6);
|
||
if (data === '[DONE]') {
|
||
this.onComplete();
|
||
return;
|
||
}
|
||
|
||
const parsed = JSON.parse(data);
|
||
this.onChunk(parsed);
|
||
}
|
||
} catch (error) {
|
||
console.warn('解析流数据失败:', error, line);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// 创建流式请求
|
||
async function createStreamRequest(url, options, onChunk, onComplete) {
|
||
const processor = new StreamProcessor(onChunk, onComplete);
|
||
|
||
try {
|
||
const response = await fetch(url, {
|
||
...options,
|
||
headers: {
|
||
...options.headers,
|
||
'Accept': 'text/event-stream',
|
||
'Cache-Control': 'no-cache'
|
||
}
|
||
});
|
||
|
||
if (!response.ok) {
|
||
throw new Error(`HTTP ${response.status}: ${response.statusText}`);
|
||
}
|
||
|
||
const reader = response.body.getReader();
|
||
const decoder = new TextDecoder();
|
||
|
||
while (true) {
|
||
const { done, value } = await reader.read();
|
||
if (done) break;
|
||
|
||
const chunk = decoder.decode(value, { stream: true });
|
||
processor.processChunk(chunk);
|
||
}
|
||
} catch (error) {
|
||
throw error;
|
||
}
|
||
}
|
||
|
||
// 导出工具函数
|
||
window.Utils = {
|
||
escapeHtml,
|
||
scrollToBottom,
|
||
generateId,
|
||
parseSVGResponse,
|
||
downloadFile,
|
||
showStatus,
|
||
storage,
|
||
debounce,
|
||
throttle,
|
||
formatDateTime,
|
||
deepClone,
|
||
isEmpty,
|
||
autoResizeTextarea,
|
||
StreamProcessor,
|
||
createStreamRequest
|
||
}; |