#!/usr/bin/env python3
"""Serve POC files on http://127.0.0.1:8080/."""
import http.server
import socketserver
import os
import sys

PORT = 8080
os.chdir(os.path.dirname(os.path.abspath(__file__)))

class Handler(http.server.SimpleHTTPRequestHandler):
    def end_headers(self):
        if self.path.endswith('.rar'):
            self.send_header('Content-Type', 'application/x-rar-compressed')
            self.send_header('Content-Disposition',
                             f'attachment; filename="{os.path.basename(self.path)}"')
        super().end_headers()

    def log_message(self, fmt, *args):
        sys.stderr.write("[server] " + (fmt % args) + "\n")

socketserver.TCPServer.allow_reuse_address = True
with socketserver.TCPServer(("127.0.0.1", PORT), Handler) as httpd:
    sys.stderr.write(f"[server] listening on http://127.0.0.1:{PORT}/\n")
    httpd.serve_forever()