from http.server import BaseHTTPRequestHandler, HTTPServer
from urllib.parse import parse_qs
import json
import logging
from datetime import datetime
import requests  # For making HTTP requests to DeepSeek API

# Configure logging
logging.basicConfig(
    filename="duplicati_logs.log",
    level=logging.INFO,
    format="%(asctime)s - %(message)s",
)

# DeepSeek API configuration
DEEPSEEK_API_URL = "https://api.deepseek.com/v1/chat/completions"  # DeepSeek Chat API endpoint
DEEPSEEK_API_KEY = "sk-175159668af9430ea6208f5636b24198"  # Replace with your DeepSeek API key

class DuplicatiLogHandler(BaseHTTPRequestHandler):
    def do_POST(self):
        # Get the length of the data
        content_length = int(self.headers["Content-Length"])
        # Read the POST data
        post_data = self.rfile.read(content_length).decode("utf-8")
        # Parse the form-urlencoded data
        data = parse_qs(post_data)

        # Log the data
        self.log_data(data)

        # Send the log message to DeepSeek Chat API for analysis
        analysis_result, diagnostic_message = self.send_to_deepseek_chat(data)
        if analysis_result:
            logging.info(f"DeepSeek Analysis Result: {analysis_result}")
            if diagnostic_message:
                logging.info(f"Diagnostic Message: {diagnostic_message}")

        # Send a response
        self.send_response(200)
        self.send_header("Content-type", "application/json")
        self.end_headers()
        response = {
            "status": "success",
            "message": "Log received",
            "analysis_result": analysis_result,
            "diagnostic_message": diagnostic_message,
        }
        self.wfile.write(json.dumps(response).encode("utf-8"))

    def log_data(self, data):
        """Log the received data to a file."""
        log_entry = {
            "timestamp": datetime.now().isoformat(),
            "data": data,
        }
        logging.info(json.dumps(log_entry))

    def send_to_deepseek_chat(self, data):
        """Send the log message to DeepSeek Chat API for analysis."""
        try:
            # Prepare the payload for DeepSeek Chat API
            payload = {
                "model": "deepseek-chat",  # Specify the model to use
                "messages": [
                    {
                        "role": "system",
                        "content": "You are an operator analyzing backup logs. "
                                  "Check the log message for abnormalities and respond with 'normal' or 'abnormal'. "
                                  "If the backup is abnormal, provide a detailed diagnostic message explaining the issue and suggesting possible fixes.",
                    },
                    {
                        "role": "user",
			"content": json.dumps(data),  # Pass the raw data as a JSON string
                    },
                ],
            }

            # Make a POST request to DeepSeek Chat API
            headers = {
                "Authorization": f"Bearer {DEEPSEEK_API_KEY}",
                "Content-Type": "application/json",
            }
            response = requests.post(DEEPSEEK_API_URL, headers=headers, json=payload)
            response.raise_for_status()  # Raise an error for bad status codes

            # Parse the response
            result = response.json()
            chat_response = result["choices"][0]["message"]["content"]

            # Extract analysis result and diagnostic message
            if "abnormal" in chat_response.lower():
                analysis_result = "abnormal"
                diagnostic_message = chat_response  # Use the full response as the diagnostic message
            else:
                analysis_result = "normal"
                diagnostic_message = None

            return analysis_result, diagnostic_message
        except requests.exceptions.RequestException as e:
            logging.error(f"Failed to send data to DeepSeek Chat API: {e}")
            return None, None

def run(server_class=HTTPServer, handler_class=DuplicatiLogHandler, port=8680):
    server_address = ("", port)
    httpd = server_class(server_address, handler_class)
    print(f"Starting Duplicati log server on port {port}...")
    httpd.serve_forever()

if __name__ == "__main__":
    run()