import re

with open('src/components/SandboxSimulation.tsx', 'r') as f:
    content = f.read()

# 1. remove interceptCount from initial states
content = re.sub(r"    alerts: \[\],\n    interceptCount: 0\n  \}\);", "    alerts: []\n  });", content)
content = re.sub(r"      alerts: \[\],\n      interceptCount: 0\n    \}\);", "      alerts: []\n    });", content)

# 2. handleSendLine
old_handle = """      // Perform rule matching
      let currentAlerts = [...prev.alerts];
      let silentCount = prev.interceptCount;
      const activeRules = rules.filter(r => r.enabled);

      activeRules.forEach(rule => {
        // Check if any keyword matches
        const matchedKeyword = rule.keywords.find(kw => 
          text.toLowerCase().includes(kw.toLowerCase())
        );

        if (matchedKeyword) {
          // Check if this rule was already triggered in the current session
          const alreadyTriggered = currentAlerts.some(a => a.ruleId === rule.id);

          if (alreadyTriggered) {
            // Duplicate hit: silent interception
            silentCount += 1;
            // Silent notify
            addToast(`规则「${rule.name}」在 seq: ${nextSeq} 中再次触发，已被系统去重静默拦截！`, 'info');
          } else {
            // Fresh alert!
            const alertId = `alert-${rule.id}-${Date.now()}`;
            const newAlert: Alert = {
              id: alertId,
              ruleId: rule.id,
              ruleName: rule.name,
              keyword: matchedKeyword,
              text: text.trim(),
              seq: nextSeq,
              timestamp,
            };
            currentAlerts = [...currentAlerts, newAlert];
            addToast(`⚠️ 触发预警: 「${rule.name}」（命中词: ${matchedKeyword}）`, 'warning');
          }
        }
      });

      return {
        ...prev,
        lines: updatedLines,
        alerts: currentAlerts,
        interceptCount: silentCount
      };"""

new_handle = """      // Perform rule matching
      let currentAlerts = [...prev.alerts];
      const activeRules = rules.filter(r => r.enabled);

      activeRules.forEach(rule => {
        // Check if any keyword matches
        const matchedKeyword = rule.keywords.find(kw => 
          text.toLowerCase().includes(kw.toLowerCase())
        );

        if (matchedKeyword) {
          // Fresh alert!
          const alertId = `alert-${rule.id}-${Date.now()}`;
          const newAlert: Alert = {
            id: alertId,
            ruleId: rule.id,
            ruleName: rule.name,
            keyword: matchedKeyword,
            text: text.trim(),
            seq: nextSeq,
            timestamp,
          };
          currentAlerts = [...currentAlerts, newAlert];
          addToast(`⚠️ 触发预警: 「${rule.name}」（命中词: ${matchedKeyword}）`, 'warning');
        }
      });

      return {
        ...prev,
        lines: updatedLines,
        alerts: currentAlerts
      };"""

content = content.replace(old_handle, new_handle)

# 3. UI
ui_to_remove = """            <div className="flex items-center space-x-1 font-mono text-[10px] font-bold">
              <span className="text-slate-400">去重静默拦截:</span>
              <span className={`px-1.5 py-0.2 rounded ${
                session.interceptCount > 0 
                  ? 'bg-amber-100 text-amber-800' 
                  : 'bg-slate-200 text-slate-600'
              }`}>
                {session.interceptCount} 次
              </span>
            </div>"""

content = content.replace(ui_to_remove, "")

with open('src/components/SandboxSimulation.tsx', 'w') as f:
    f.write(content)