feat: add knowledge serve command with HTTP server

Author: mdesmet

src/datapilot/core/knowledge/cli.py

@@ -1,7 +1,101 @@
 import click
+import json
+from http.server import HTTPServer, BaseHTTPRequestHandler
+from urllib.parse import urlparse
+from urllib.request import Request, urlopen
+from urllib.error import URLError, HTTPError
+import re
 
 
 @click.group()
 def cli():
     """knowledge specific commands."""
 
+
+@cli.command()
+@click.option('--port', default=3000, help='Port to run the server on')
+@click.pass_context
+def serve(ctx, port):
+    """Serve knowledge bases via HTTP server."""
+    # Get configuration from parent context
+    token = ctx.parent.obj.get('token')
+    instance_name = ctx.parent.obj.get('instance_name')
+    backend_url = ctx.parent.obj.get('backend_url')
+    
+    if not token or not instance_name:
+        click.echo("Error: API token and instance name are required. Use --token and --instance-name options or set them in config.", err=True)
+        ctx.exit(1)
+    
+    class KnowledgeBaseHandler(BaseHTTPRequestHandler):
+        def do_GET(self):
+            """Handle GET requests."""
+            path = urlparse(self.path).path
+            
+            # Match /knowledge_bases/{uuid} pattern
+            match = re.match(r'^/knowledge_bases/([a-fA-F0-9]{8}-[a-fA-F0-9]{4}-[a-fA-F0-9]{4}-[a-fA-F0-9]{4}-[a-fA-F0-9]{12})$', path)
+            
+            if match:
+                public_id = match.group(1)
+                self.handle_knowledge_base(public_id)
+            elif path == '/health':
+                self.handle_health()
+            else:
+                self.send_error(404, "Not Found")
+        
+        def handle_knowledge_base(self, public_id):
+            """Fetch and return knowledge base data."""
+            url = f"{backend_url}/knowledge_bases/public/{public_id}"
+            
+            headers = {
+                'Authorization': f'Bearer {token}',
+                'X-Tenant': instance_name,
+                'Content-Type': 'application/json'
+            }
+            
+            req = Request(url, headers=headers)
+            
+            try:
+                with urlopen(req) as response:
+                    data = response.read()
+                    self.send_response(200)
+                    self.send_header('Content-Type', 'application/json')
+                    self.end_headers()
+                    self.wfile.write(data)
+            except HTTPError as e:
+                error_data = e.read().decode('utf-8') if e.read() else '{"error": "HTTP Error"}'
+                self.send_response(e.code)
+                self.send_header('Content-Type', 'application/json')
+                self.end_headers()
+                self.wfile.write(error_data.encode('utf-8'))
+            except URLError as e:
+                self.send_response(500)
+                self.send_header('Content-Type', 'application/json')
+                self.end_headers()
+                error_msg = json.dumps({'error': str(e)})
+                self.wfile.write(error_msg.encode('utf-8'))
+        
+        def handle_health(self):
+            """Handle health check endpoint."""
+            self.send_response(200)
+            self.send_header('Content-Type', 'application/json')
+            self.end_headers()
+            self.wfile.write(json.dumps({'status': 'ok'}).encode('utf-8'))
+        
+        def log_message(self, format, *args):
+            """Override to use click.echo for logging."""
+            click.echo(f"{self.address_string()} - {format % args}")
+    
+    server_address = ('', port)
+    httpd = HTTPServer(server_address, KnowledgeBaseHandler)
+    
+    click.echo(f"Starting knowledge base server on port {port}...")
+    click.echo(f"Backend URL: {backend_url}")
+    click.echo(f"Instance: {instance_name}")
+    click.echo(f"Server running at http://localhost:{port}")
+    
+    try:
+        httpd.serve_forever()
+    except KeyboardInterrupt:
+        click.echo("\nShutting down server...")
+        httpd.shutdown()
+