From 69819eba7d7d0682d16717cb9c727d5a4530e676 Mon Sep 17 00:00:00 2001
From: overcuriousity <overcuriousity@posteo.org>
Date: Fri, 18 Jul 2025 14:30:58 +0200
Subject: [PATCH] dynamic yaml loading part 1

---
 src/components/AIQueryInterface.astro | 17 +++++++++--------
 src/data/tools.yaml                   |  7 ++++++-
 src/pages/api/ai/query.ts             | 12 +++++++-----
 3 files changed, 22 insertions(+), 14 deletions(-)

diff --git a/src/components/AIQueryInterface.astro b/src/components/AIQueryInterface.astro
index fb1eeb7..e58399a 100644
--- a/src/components/AIQueryInterface.astro
+++ b/src/components/AIQueryInterface.astro
@@ -8,6 +8,7 @@ const yamlPath = path.join(process.cwd(), 'src/data/tools.yaml');
 const yamlContent = await fs.readFile(yamlPath, 'utf8');
 const data = load(yamlContent) as any;
 const tools = data.tools;
+const phases = data.phases;
 ---
 
 <!-- AI Query Interface -->
@@ -87,7 +88,7 @@ const tools = data.tools;
   </div>
 </section>
 
-<script define:vars={{ tools }}>
+<script define:vars={{ tools, phases }}>
 document.addEventListener('DOMContentLoaded', () => {
   const aiInterface = document.getElementById('ai-interface');
   const aiInput = document.getElementById('ai-query-input');
@@ -278,13 +279,13 @@ document.addEventListener('DOMContentLoaded', () => {
   function displayResults(recommendation, originalQuery) {
     // Group tools by phase
     const toolsByPhase = {};
-    const phaseOrder = ['data-collection', 'examination', 'analysis', 'reporting'];
-    const phaseNames = {
-      'data-collection': 'Datensammlung',
-      'examination': 'Auswertung', 
-      'analysis': 'Analyse',
-      'reporting': 'Bericht & Präsentation'
-    };
+    
+    // Replace hardcoded values with dynamic data from YAML
+    const phaseOrder = phases.filter(phase => phase.id !== 'collaboration-general').map(phase => phase.id);
+    const phaseNames = phases.reduce((acc, phase) => {
+      acc[phase.id] = phase.name;
+      return acc;
+    }, {});
 
     // Initialize phases
     phaseOrder.forEach(phase => {
diff --git a/src/data/tools.yaml b/src/data/tools.yaml
index 2e958f7..7623335 100644
--- a/src/data/tools.yaml
+++ b/src/data/tools.yaml
@@ -825,11 +825,16 @@ domains:
 phases:
   - id: data-collection
     name: Datensammlung
+    description: "Imaging, Acquisition, Remote Collection Tools"
   - id: examination
     name: Auswertung
+    description: "Parsing, Extraction, Initial Analysis Tools"
   - id: analysis
     name: Analyse
+    description: "Deep Analysis, Correlation, Visualization Tools"
   - id: reporting
     name: Bericht & Präsentation
+    description: "Documentation, Visualization, Presentation Tools (z.B. QGIS für Geodaten, Timeline-Tools)"
   - id: collaboration-general
-    name: Übergreifend & Kollaboration
\ No newline at end of file
+    name: Übergreifend & Kollaboration
+    description: "Cross-cutting tools and collaboration platforms"
\ No newline at end of file
diff --git a/src/pages/api/ai/query.ts b/src/pages/api/ai/query.ts
index 6d2a6e8..a562e04 100644
--- a/src/pages/api/ai/query.ts
+++ b/src/pages/api/ai/query.ts
@@ -107,6 +107,11 @@ function createSystemPrompt(toolsData: any): string {
     `- ${domain.id}: ${domain.name}`
   ).join('\n');
 
+  const phaseDescriptions = toolsData.phases
+    .filter((phase: any) => phase.id !== 'collaboration-general')
+    .map((phase: any) => `- ${phase.name}: ${phase.description || 'Tools for this phase'}`)
+    .join('\n');
+
   return `Du bist ein DFIR (Digital Forensics and Incident Response) Experte, der Ermittlern bei der Toolauswahl hilft.
 
 VERFÜGBARE TOOLS DATABASE:
@@ -128,10 +133,7 @@ WICHTIGE REGELN:
 7. Bewerbe NIEMALS Proprietäre Software fälschlicherweise als Open-Source-Tools, erkenne aber an, falls diese besser geeignet sein könnte.
 
 TOOL-AUSWAHL NACH PHASE:
-- Datensammlung: Imaging, Acquisition, Remote Collection Tools
-- Auswertung: Parsing, Extraction, Initial Analysis Tools  
-- Analyse: Deep Analysis, Correlation, Visualization Tools
-- Berichterstattung: Documentation, Visualization, Presentation Tools (z.B. QGIS für Geodaten, Timeline-Tools)
+${phaseDescriptions}
 
 ANTWORT-FORMAT (strict JSON):
 {
@@ -140,7 +142,7 @@ ANTWORT-FORMAT (strict JSON):
     {
       "name": "EXAKTER Name aus der Database",
       "priority": "high|medium|low", 
-      "phase": "data-collection|examination|analysis|reporting",
+      "phase": "${toolsData.phases.filter((p: any) => p.id !== 'collaboration-general').map((p: any) => p.id).join('|')}",
       "justification": "Warum dieses Tool für diese Phase und Szenario geeignet ist"
     }
   ],