// AI助手 - 前端应用
// 使用智谱 GLM-4.5-Air 模型（流式输出 + 多对话管理）

const CONFIG = {
  apiUrl: 'https://open.bigmodel.cn/api/paas/v4/chat/completions',
  apiKey: '2259e33a1357460abe17919aaf81e73d.K44a8LPQTmFM5PKm',
  model: 'glm-4.5-air',
  maxTokens: 2048
};

// 数据结构
let conversations = [];  // 对话列表
let currentConversation = null;  // 当前对话
let isLoading = false;

// DOM 元素
const appContainer = document.getElementById('app');
const messagesContainer = document.getElementById('messagesContainer');
const messagesDiv = document.getElementById('messages');
const userInput = document.getElementById('userInput');
const sendBtn = document.getElementById('sendBtn');
const welcome = document.getElementById('welcome');

// 初始化
document.addEventListener('DOMContentLoaded', () => {
  // 从本地存储加载对话列表
  const saved = localStorage.getItem('conversations');
  if (saved) {
    conversations = JSON.parse(saved);
  }
  
  // 兼容旧数据格式（chat_history）
  const oldHistory = localStorage.getItem('chat_history');
  if (oldHistory && conversations.length === 0) {
    const oldMessages = JSON.parse(oldHistory);
    if (oldMessages.length > 0) {
      // 转换旧数据为新格式
      const convertedConv = {
        id: Date.now().toString(),
        title: oldMessages[0].content.slice(0, 30) + (oldMessages[0].content.length > 30 ? '...' : ''),
        messages: oldMessages,
        createdAt: Date.now(),
        updatedAt: Date.now()
      };
      conversations.push(convertedConv);
      saveConversations();
      localStorage.removeItem('chat_history');  // 清理旧数据
    }
  }
  
  // 显示对话列表页面
  showConversationList();
});

// ==================== 对话列表页面 ====================

function showConversationList() {
  currentConversation = null;
  
  // 渲染对话列表
  const listHtml = `
    <div class="conversation-list-page">
      <header class="list-header">
        <div class="header-title">
          <span class="logo">🤖</span>
          <h1>AI助手</h1>
        </div>
      </header>
      
      <div class="list-content">
        <button class="new-chat-btn" onclick="createNewConversation()">
          <svg viewBox="0 0 24 24" width="20" height="20"><path fill="currentColor" d="M19 13h-6v6h-2v-6H5v-2h6V5h2v6h6v2z"/></svg>
          新建对话
        </button>
        
        <div class="conversation-list">
          ${conversations.length === 0 
            ? '<div class="empty-list">暂无对话记录</div>'
            : conversations.map(conv => `
              <div class="conversation-item" onclick="openConversation('${conv.id}')">
                <div class="conv-title">${escapeHtml(conv.title)}</div>
                <div class="conv-meta">${conv.messages.length} 条消息 · ${formatTime(conv.updatedAt)}</div>
                <button class="conv-delete-btn" onclick="event.stopPropagation(); deleteConversation('${conv.id}')" title="删除对话">
                  <svg viewBox="0 0 24 24" width="16" height="16"><path fill="currentColor" d="M6 19c0 1.1.9 2 2 2h8c1.1 0 2-.9 2-2V7H6v12zM19 4h-3.5l-1-1h-5l-1 1H5v2h14V4z"/></svg>
                </button>
              </div>
            `).join('')
          }
        </div>
      </div>
    </div>
  `;
  
  appContainer.innerHTML = listHtml;
}

// 创建新对话
function createNewConversation() {
  const newConv = {
    id: Date.now().toString(),
    title: '新对话',
    messages: [],
    createdAt: Date.now(),
    updatedAt: Date.now()
  };
  
  conversations.unshift(newConv);
  saveConversations();
  openConversation(newConv.id);
}

// 打开对话
function openConversation(id) {
  currentConversation = conversations.find(c => c.id === id);
  if (!currentConversation) {
    showConversationList();
    return;
  }
  
  // 渲染对话页面
  const chatHtml = `
    <div id="chatPage">
      <header class="header">
        <button class="back-btn" onclick="showConversationList()">
          <svg viewBox="0 0 24 24" width="24" height="24"><path fill="currentColor" d="M20 11H7.83l5.59-5.59L12 4l-8 8 8 8 1.41-1.41L7.83 13H20v-2z"/></svg>
        </button>
        <div class="header-title">
          <span class="logo">🤖</span>
          <h1>${escapeHtml(currentConversation.title)}</h1>
        </div>
        <button class="clear-btn" onclick="clearCurrentChat()" title="清空对话">
          <svg viewBox="0 0 24 24" width="20" height="20"><path fill="currentColor" d="M19 6.41L17.59 5 12 10.59 6.41 5 5 6.41 10.59 12 5 17.59 6.41 19 12 13.41 17.59 19 19 17.59 13.41 12 19 6.41z"/></svg>
        </button>
      </header>

      <div class="messages-container" id="messagesContainer">
        <div class="welcome" id="welcome" style="${currentConversation.messages.length > 0 ? 'display:none' : ''}">
          <div class="welcome-icon">👋</div>
          <h2>你好！我是AI助手</h2>
          <p>有什么可以帮助你的吗？</p>
          <div class="quick-actions">
            <button class="quick-btn" data-text="介绍一下你自己">介绍一下你自己</button>
            <button class="quick-btn" data-text="帮我写一段代码">帮我写代码</button>
            <button class="quick-btn" data-text="解释一个概念">解释概念</button>
          </div>
        </div>
        <div class="messages" id="messages"></div>
      </div>

      <div class="input-area">
        <textarea 
          id="userInput" 
          placeholder="输入消息..." 
          rows="1"
          onkeydown="handleKeyDown(event)"
          oninput="autoResize(this)"
        ></textarea>
        <button class="send-btn" onclick="sendMessage()" id="sendBtn">
          <svg viewBox="0 0 24 24" width="24" height="24"><path fill="currentColor" d="M2.01 21L23 12 2.01 3 2 10l15 2-15 2z"/></svg>
        </button>
      </div>
    </div>
  `;
  
  appContainer.innerHTML = chatHtml;
  
  // 重新获取 DOM 元素
  messagesContainer = document.getElementById('messagesContainer');
  messagesDiv = document.getElementById('messages');
  userInput = document.getElementById('userInput');
  sendBtn = document.getElementById('sendBtn');
  welcome = document.getElementById('welcome');
  
  // 渲染消息
  renderMessages();
  userInput.focus();
  
  // 绑定快捷按钮事件
  document.querySelectorAll('.quick-btn').forEach(btn => {
    btn.addEventListener('click', () => {
      const text = btn.getAttribute('data-text');
      userInput.value = text;
      sendMessage();
    });
  });
}

// 删除对话
function deleteConversation(id) {
  if (!confirm('确定要删除这个对话吗？')) return;
  
  conversations = conversations.filter(c => c.id !== id);
  saveConversations();
  showConversationList();
}

// ==================== 对话页面 ====================

// 自动调整输入框高度
function autoResize(textarea) {
  textarea.style.height = 'auto';
  textarea.style.height = Math.min(textarea.scrollHeight, 120) + 'px';
}

// 处理键盘事件
function handleKeyDown(event) {
  if (event.key === 'Enter' && !event.shiftKey) {
    event.preventDefault();
    sendMessage();
  }
}

// 发送快捷消息
function sendQuickMessage(text) {
  userInput.value = text;
  sendMessage();
}

// 发送消息（流式输出）
async function sendMessage() {
  if (!currentConversation) return;
  
  const text = userInput.value.trim();
  if (!text || isLoading) return;

  // 隐藏欢迎界面
  welcome.style.display = 'none';

  // 添加用户消息
  currentConversation.messages.push({ role: 'user', content: text });
  
  // 更新对话标题（第一条用户消息）
  if (currentConversation.title === '新对话') {
    currentConversation.title = text.slice(0, 30) + (text.length > 30 ? '...' : '');
    // 更新标题显示
    const titleEl = document.querySelector('.header h1');
    if (titleEl) {
      titleEl.textContent = currentConversation.title;
    }
  }
  
  currentConversation.updatedAt = Date.now();
  saveConversations();
  
  renderMessages();
  userInput.value = '';
  autoResize(userInput);

  // 调用流式生成
  await streamGenerate(currentConversation.messages.length - 1);
}

// 流式生成 AI 回复
async function streamGenerate(userMsgIndex) {
  isLoading = true;
  sendBtn.disabled = true;

  const aiMessageIndex = currentConversation.messages.length;
  currentConversation.messages.push({ role: 'assistant', content: '' });
  renderMessages();
  
  const lastMessageEl = messagesDiv.lastElementChild;
  const contentEl = lastMessageEl.querySelector('.message-content');
  contentEl.innerHTML = '<span class="streaming-cursor">▌</span>';

  try {
    const response = await fetch(CONFIG.apiUrl, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
        'Authorization': `Bearer ${CONFIG.apiKey}`
      },
      body: JSON.stringify({
        model: CONFIG.model,
        messages: currentConversation.messages.slice(0, aiMessageIndex).map(m => ({
          role: m.role,
          content: m.content
        })),
        max_tokens: CONFIG.maxTokens,
        stream: true
      })
    });

    if (!response.ok) {
      throw new Error(`API 错误: ${response.status}`);
    }

    const reader = response.body.getReader();
    const decoder = new TextDecoder();
    let buffer = '';

    while (true) {
      const { done, value } = await reader.read();
      if (done) break;

      buffer += decoder.decode(value, { stream: true });
      const lines = buffer.split('\n');
      buffer = lines.pop() || '';

      for (const line of lines) {
        if (line.startsWith('data: ')) {
          const jsonStr = line.slice(6).trim();
          if (jsonStr === '[DONE]') continue;
          
          try {
            const data = JSON.parse(jsonStr);
            if (data.choices && data.choices[0]?.delta?.content) {
              currentConversation.messages[aiMessageIndex].content += data.choices[0].delta.content;
              contentEl.innerHTML = renderMarkdown(currentConversation.messages[aiMessageIndex].content) + '<span class="streaming-cursor">▌</span>';
              scrollToBottom();
            }
          } catch (e) {}
        }
      }
    }

    contentEl.innerHTML = renderMarkdown(currentConversation.messages[aiMessageIndex].content);

  } catch (error) {
    console.error('Error:', error);
    currentConversation.messages[aiMessageIndex].content = `抱歉，出现了错误：${error.message}\n\n请检查网络连接后重试。`;
    contentEl.innerHTML = renderMarkdown(currentConversation.messages[aiMessageIndex].content);
  } finally {
    isLoading = false;
    sendBtn.disabled = false;
    currentConversation.updatedAt = Date.now();
    saveConversations();
    renderMessages();
  }
}

// 重新生成 AI 回复
async function regenerate(index) {
  if (!currentConversation || isLoading || index < 1) return;
  
  const userMsgIndex = index - 1;
  if (currentConversation.messages[userMsgIndex].role !== 'user') return;
  
  currentConversation.messages.splice(index, 1);
  currentConversation.updatedAt = Date.now();
  saveConversations();
  
  await streamGenerate(userMsgIndex);
}

// 删除消息
function deleteMessage(index) {
  if (!currentConversation || isLoading) return;
  
  const msg = currentConversation.messages[index];
  
  if (msg.role === 'assistant') {
    if (index > 0 && currentConversation.messages[index - 1].role === 'user') {
      currentConversation.messages.splice(index - 1, 2);
    } else {
      currentConversation.messages.splice(index, 1);
    }
  } else {
    if (index < currentConversation.messages.length - 1 && currentConversation.messages[index + 1].role === 'assistant') {
      currentConversation.messages.splice(index, 2);
    } else {
      currentConversation.messages.splice(index, 1);
    }
  }
  
  currentConversation.updatedAt = Date.now();
  saveConversations();
  renderMessages();
  
  if (currentConversation.messages.length === 0) {
    welcome.style.display = 'block';
  }
}

// 复制消息（复制原文）
function copyMessage(index) {
  if (!currentConversation) return;
  
  const content = currentConversation.messages[index].content;
  
  navigator.clipboard.writeText(content).then(() => {
    showToast('已复制到剪贴板');
  }).catch(err => {
    const textarea = document.createElement('textarea');
    textarea.value = content;
    textarea.style.position = 'fixed';
    textarea.style.opacity = '0';
    document.body.appendChild(textarea);
    textarea.select();
    document.execCommand('copy');
    document.body.removeChild(textarea);
    showToast('已复制到剪贴板');
  });
}

// 清空当前对话
function clearCurrentChat() {
  if (!currentConversation) return;
  
  if (confirm('确定要清空当前对话吗？')) {
    currentConversation.messages = [];
    currentConversation.updatedAt = Date.now();
    saveConversations();
    renderMessages();
    welcome.style.display = 'block';
  }
}

// 渲染消息
function renderMessages() {
  if (!currentConversation) return;
  
  // 根据消息数量显示/隐藏欢迎界面
  if (welcome) {
    welcome.style.display = currentConversation.messages.length > 0 ? 'none' : 'block';
  }
  
  messagesDiv.innerHTML = currentConversation.messages.map((msg, index) => {
    const isUser = msg.role === 'user';
    const avatar = isUser ? '👤' : '🤖';
    const content = renderMarkdown(msg.content);
    
    const copyIcon = `<svg viewBox="0 0 24 24" width="16" height="16"><path fill="currentColor" d="M16 1H4c-1.1 0-2 .9-2 2v14h2V3h12V1zm3 4H8c-1.1 0-2 .9-2 2v14c0 1.1.9 2 2 2h11c1.1 0 2-.9 2-2V7c0-1.1-.9-2-2-2zm0 16H8V7h11v14z"/></svg>`;
    
    const actions = isUser 
      ? `<div class="message-actions">
          <button class="action-btn copy-btn" onclick="copyMessage(${index})" title="复制">${copyIcon}</button>
          <button class="action-btn delete-btn" onclick="deleteMessage(${index})" title="删除">
            <svg viewBox="0 0 24 24" width="16" height="16"><path fill="currentColor" d="M6 19c0 1.1.9 2 2 2h8c1.1 0 2-.9 2-2V7H6v12zM19 4h-3.5l-1-1h-5l-1 1H5v2h14V4z"/></svg>
          </button>
        </div>`
      : `<div class="message-actions">
          <button class="action-btn copy-btn" onclick="copyMessage(${index})" title="复制">${copyIcon}</button>
          <button class="action-btn regenerate-btn" onclick="regenerate(${index})" title="重新生成">
            <svg viewBox="0 0 24 24" width="16" height="16"><path fill="currentColor" d="M17.65 6.35C16.2 4.9 14.21 4 12 4c-4.42 0-7.99 3.58-7.99 8s3.57 8 7.99 8c3.73 0 6.84-2.55 7.73-6h-2.08c-.82 2.33-3.04 4-5.65 4-3.31 0-6-2.69-6-6s2.69-6 6-6c1.66 0 3.14.69 4.22 1.78L13 11h7V4l-2.35 2.35z"/></svg>
          </button>
          <button class="action-btn delete-btn" onclick="deleteMessage(${index})" title="删除">
            <svg viewBox="0 0 24 24" width="16" height="16"><path fill="currentColor" d="M6 19c0 1.1.9 2 2 2h8c1.1 0 2-.9 2-2V7H6v12zM19 4h-3.5l-1-1h-5l-1 1H5v2h14V4z"/></svg>
          </button>
        </div>`;
    
    return `
      <div class="message ${msg.role}" data-index="${index}">
        <div class="message-avatar">${avatar}</div>
        <div class="message-body">
          <div class="message-content">${content}</div>
          ${actions}
        </div>
      </div>
    `;
  }).join('');

  scrollToBottom();
}

// ==================== 工具函数 ====================

// 渲染 Markdown
function renderMarkdown(text) {
  if (!text) return '';
  
  marked.setOptions({
    breaks: true,
    gfm: true
  });
  
  return marked.parse(text);
}

// 滚动到底部
function scrollToBottom() {
  if (messagesContainer) {
    messagesContainer.scrollTop = messagesContainer.scrollHeight;
  }
}

// 保存对话列表
function saveConversations() {
  localStorage.setItem('conversations', JSON.stringify(conversations));
}

// 显示提示
function showToast(message) {
  const toast = document.createElement('div');
  toast.className = 'toast';
  toast.textContent = message;
  document.body.appendChild(toast);
  
  setTimeout(() => toast.classList.add('show'), 10);
  setTimeout(() => {
    toast.classList.remove('show');
    setTimeout(() => document.body.removeChild(toast), 300);
  }, 2000);
}

// HTML转义
function escapeHtml(text) {
  const div = document.createElement('div');
  div.textContent = text;
  return div.innerHTML;
}

// 格式化时间
function formatTime(timestamp) {
  const date = new Date(timestamp);
  const now = new Date();
  const diff = now - date;
  
  if (diff < 60000) return '刚刚';
  if (diff < 3600000) return Math.floor(diff / 60000) + '分钟前';
  if (diff < 86400000) return Math.floor(diff / 3600000) + '小时前';
  if (diff < 604800000) return Math.floor(diff / 86400000) + '天前';
  
  return date.toLocaleDateString('zh-CN');
}

// PWA 注册
if ('serviceWorker' in navigator) {
  navigator.serviceWorker.register('sw.js').catch(() => {});
}