{
"tool_response": {
"filePath": "\/var\/www\/tools\/ki-protokoll\/claude-hook\/log_to_db.py",
"oldString": " elif event_name in ['Stop', 'SubagentStop']:\n request_data = {\n 'event': event_name,\n 'stop_hook_active': data.get('stop_hook_active', False)\n }\n response_data = {'status': 'completed'}",
"newString": " elif event_name in ['Stop', 'SubagentStop']:\n # Bei Stop: Hole Assistant-Response aus Session und schließe pending Prompts\n session_id = data.get('session_id', '')\n if session_id:\n assistant_response = get_assistant_response_from_session(session_id)\n if assistant_response:\n close_pending_user_prompts(session_id, assistant_response)\n\n request_data = {\n 'event': event_name,\n 'stop_hook_active': data.get('stop_hook_active', False)\n }\n response_data = {'status': 'completed'}",
"originalFile": "#!\/usr\/bin\/env python3\n\"\"\"\nClaude Code Hook Logger v2 für KI-Protokoll-Datenbank\nVerbesserte Version mit Request\/Response-Pairing und Duration-Tracking\n\"\"\"\n\nimport json\nimport os\nimport sys\nimport re\nimport datetime\nimport hashlib\nimport pymysql\nimport tempfile\nfrom pathlib import Path\nfrom typing import Dict, Any, Optional\n\n# .env aus Hook-Verzeichnis laden\nfrom dotenv import load_dotenv\nload_dotenv(Path(__file__).parent \/ '.env')\n\n# Konfiguration aus Environment-Variablen\nDB_CONFIG = {\n 'host': os.environ.get('CLAUDE_DB_HOST', 'localhost'),\n 'port': int(os.environ.get('CLAUDE_DB_PORT', '3306')),\n 'user': os.environ.get('CLAUDE_DB_USER', 'claude_code'),\n 'password': os.environ.get('CLAUDE_DB_PASSWORD', ''),\n 'database': os.environ.get('CLAUDE_DB_NAME', 'ki_protokoll'),\n 'charset': 'utf8mb4'\n}\n\n# Session-Tracking im temporären Verzeichnis\nTEMP_DIR = Path(tempfile.gettempdir()) \/ \"claude_hooks\"\nTEMP_DIR.mkdir(exist_ok=True)\n\n# Sicherheitseinstellungen\nMAX_FIELD_LENGTH = 10000\nSENSITIVE_KEY_PATTERNS = re.compile(r\"(?i)(password|pass|secret|token|apikey|api_key|authorization|auth|bearer|credential)\")\nSENSITIVE_VALUE_PATTERNS = [\n re.compile(r\"(?i)\\bAKIA[0-9A-Z]{16}\\b\"),\n re.compile(r\"(?i)\\b(?:sk|rk|pk)[0-9A-Za-z]{20,}\\b\"),\n re.compile(r\"(?i)\\beyJ[a-zA-Z0-9-]{10,}\\.[a-zA-Z0-9_-]{10,}\\.[a-zA-Z0-9_-]{10,}\\b\")\n]\n\ndef get_client_ip() -> str:\n \"\"\"Ermittelt die Client-IP-Adresse\"\"\"\n ssh_client = os.environ.get('SSH_CLIENT', '')\n if ssh_client:\n return ssh_client.split()[0]\n ssh_connection = os.environ.get('SSH_CONNECTION', '')\n if ssh_connection:\n return ssh_connection.split()[0]\n return '127.0.0.1'\n\ndef sanitize_data(obj: Any) -> Any:\n \"\"\"Entfernt oder maskiert sensible Daten\"\"\"\n if isinstance(obj, dict):\n result = {}\n for key, value in obj.items():\n if SENSITIVE_KEY_PATTERNS.search(str(key)):\n result[key] = '[REDACTED]'\n else:\n result[key] = sanitize_data(value)\n return result\n elif isinstance(obj, list):\n return [sanitize_data(item) for item in obj]\n elif isinstance(obj, str):\n for pattern in SENSITIVE_VALUE_PATTERNS:\n if pattern.search(obj):\n return '[REDACTED]'\n if len(obj) > MAX_FIELD_LENGTH:\n hash_value = hashlib.sha256(obj.encode('utf-8', errors='ignore')).hexdigest()[:16]\n return obj[:MAX_FIELD_LENGTH] + f'... [TRUNCATED-{hash_value}]'\n return obj\n return obj\n\ndef estimate_tokens(text: str) -> int:\n \"\"\"Grobe Token-Schätzung (4 Zeichen = 1 Token)\"\"\"\n if not text:\n return 0\n return max(1, len(text) \/\/ 4)\n\ndef get_session_tracking_key(data: Dict[str, Any]) -> str:\n \"\"\"Erstellt einen eindeutigen Key für Session-Tracking\"\"\"\n session_id = data.get('session_id', '')\n event_name = data.get('hook_event_name', '')\n tool_name = data.get('tool_name', '')\n\n # Für Tool-Events: tool_name-spezifischer Key\n if event_name in ['PreToolUse', 'PostToolUse'] and tool_name:\n return f\"{session_id}_{tool_name}_{event_name}\"\n\n # Für andere Events: event-spezifischer Key\n return f\"{session_id}_{event_name}\"\n\ndef save_pending_request(data: Dict[str, Any], db_id: int) -> None:\n \"\"\"Speichert pending Request für spätere Response-Zuordnung\"\"\"\n try:\n key = get_session_tracking_key(data)\n tracking_file = TEMP_DIR \/ f\"{key}.json\"\n\n tracking_data = {\n 'db_id': db_id,\n 'timestamp': datetime.datetime.now().isoformat(),\n 'event': data.get('hook_event_name'),\n 'tool_name': data.get('tool_name', ''),\n 'session_id': data.get('session_id', '')\n }\n\n with open(tracking_file, 'w') as f:\n json.dump(tracking_data, f)\n\n except Exception as e:\n print(f\"Session tracking save error: {e}\", file=sys.stderr)\n\ndef find_matching_request(data: Dict[str, Any]) -> Optional[int]:\n \"\"\"Findet matching Request für Response-Event\"\"\"\n try:\n event_name = data.get('hook_event_name', '')\n\n # PostToolUse sucht nach PreToolUse\n if event_name == 'PostToolUse':\n search_data = dict(data)\n search_data['hook_event_name'] = 'PreToolUse'\n key = get_session_tracking_key(search_data)\n else:\n return None\n\n tracking_file = TEMP_DIR \/ f\"{key}.json\"\n\n if tracking_file.exists():\n with open(tracking_file, 'r') as f:\n tracking_data = json.load(f)\n\n # Cleanup\n tracking_file.unlink()\n return tracking_data['db_id']\n\n except Exception as e:\n print(f\"Session tracking find error: {e}\", file=sys.stderr)\n\n return None\n\ndef update_request_with_response(db_id: int, response_data: str) -> None:\n \"\"\"Updated existing Request mit Response-Daten und berechnet Duration\"\"\"\n try:\n connection = pymysql.connect(**DB_CONFIG)\n\n with connection.cursor() as cursor:\n current_time = datetime.datetime.now()\n\n # Erst Response und Timestamp setzen\n tokens_output = estimate_tokens(response_data)\n\n sql = \"\"\"\n UPDATE protokoll\n SET response = %s,\n response_timestamp = %s,\n tokens_output = %s,\n tokens_total = tokens_input + %s,\n status = 'completed'\n WHERE id = %s\n \"\"\"\n\n cursor.execute(sql, (\n response_data,\n current_time,\n tokens_output,\n tokens_output,\n db_id\n ))\n\n # Dann Duration aus Timestamps berechnen und setzen (mit Microsekunden-Präzision)\n duration_sql = \"\"\"\n UPDATE protokoll\n SET duration_ms = ROUND(TIMESTAMPDIFF(MICROSECOND, request_timestamp, response_timestamp) \/ 1000, 3)\n WHERE id = %s\n \"\"\"\n\n cursor.execute(duration_sql, (db_id,))\n\n connection.commit()\n\n except Exception as e:\n print(f\"Database update error: {e}\", file=sys.stderr)\n finally:\n if 'connection' in locals():\n connection.close()\n\n\ndef get_assistant_response_from_session(session_id: str) -> Optional[str]:\n \"\"\"Liest die letzte Assistant-Response aus der Session-JSONL\"\"\"\n try:\n # Claude Code speichert Sessions unter ~\/.claude\/projects\/\n claude_dir = Path.home() \/ '.claude' \/ 'projects'\n\n # Finde das richtige Projektverzeichnis (basierend auf cwd)\n for project_dir in claude_dir.iterdir():\n if not project_dir.is_dir():\n continue\n\n session_file = project_dir \/ f\"{session_id}.jsonl\"\n if session_file.exists():\n # Letzte Zeilen der JSONL lesen (rückwärts für Performance)\n lines = session_file.read_text().strip().split('\\n')\n\n # Suche letzte Assistant-Message (von hinten)\n for line in reversed(lines[-50:]): # Nur letzte 50 Zeilen\n try:\n entry = json.loads(line)\n if entry.get('type') == 'assistant':\n message = entry.get('message', {})\n # Extrahiere Text-Content aus der Message\n content_parts = message.get('content', [])\n text_parts = []\n for part in content_parts:\n if isinstance(part, dict) and part.get('type') == 'text':\n text_parts.append(part.get('text', ''))\n elif isinstance(part, str):\n text_parts.append(part)\n\n if text_parts:\n return sanitize_data('\\n'.join(text_parts))\n except json.JSONDecodeError:\n continue\n\n return None\n\n except Exception as e:\n print(f\"Session read error: {e}\", file=sys.stderr)\n return None\n\n\ndef close_pending_user_prompts(session_id: str, assistant_response: str) -> None:\n \"\"\"Schließt offene UserPromptSubmit-Einträge mit der Assistant-Response\"\"\"\n try:\n connection = pymysql.connect(**DB_CONFIG)\n\n with connection.cursor() as cursor:\n current_time = datetime.datetime.now()\n tokens_output = estimate_tokens(assistant_response)\n\n response_json = json.dumps({\n 'assistant_response': assistant_response\n }, ensure_ascii=False)\n\n # Finde und update alle pending UserPromptSubmit-Einträge dieser Session\n # (basierend auf client_name und timestamp der letzten Stunde)\n sql = \"\"\"\n UPDATE protokoll\n SET response = %s,\n response_timestamp = %s,\n tokens_output = %s,\n tokens_total = tokens_input + %s,\n duration_ms = ROUND(TIMESTAMPDIFF(MICROSECOND, request_timestamp, %s) \/ 1000, 3),\n status = 'completed'\n WHERE status = 'pending'\n AND JSON_EXTRACT(request, '$.event') = 'UserPromptSubmit'\n AND timestamp > DATE_SUB(%s, INTERVAL 1 HOUR)\n \"\"\"\n\n cursor.execute(sql, (\n response_json,\n current_time,\n tokens_output,\n tokens_output,\n current_time,\n current_time\n ))\n\n connection.commit()\n\n except Exception as e:\n print(f\"Close pending prompts error: {e}\", file=sys.stderr)\n finally:\n if 'connection' in locals():\n connection.close()\n\n# Entfernt - Duration wird jetzt direkt in SQL berechnet\n\ndef log_to_database(data: Dict[str, Any]) -> Optional[int]:\n \"\"\"Schreibt Protokoll-Eintra... [TRUNCATED-441e360b5f610f12]",
"structuredPatch": [
{
"oldStart": 323,
"oldLines": 6,
"newStart": 323,
"newLines": 13,
"lines": [
" }",
" ",
" elif event_name in ['Stop', 'SubagentStop']:",
"+ # Bei Stop: Hole Assistant-Response aus Session und schließe pending Prompts",
"+ session_id = data.get('session_id', '')",
"+ if session_id:",
"+ assistant_response = get_assistant_response_from_session(session_id)",
"+ if assistant_response:",
"+ close_pending_user_prompts(session_id, assistant_response)",
"+",
" request_data = {",
" 'event': event_name,",
" 'stop_hook_active': data.get('stop_hook_active', False)"
]
}
],
"userModified": false,
"replaceAll": false
}
}