#!/usr/bin/env python3
"""Simple file server with download headers"""
from http.server import HTTPServer, SimpleHTTPRequestHandler
import os

class DownloadHandler(SimpleHTTPRequestHandler):
    def end_headers(self):
        path = self.translate_path(self.path)
        if os.path.isfile(path):
            self.send_header('Content-Disposition', 'attachment')
            self.send_header('Access-Control-Allow-Origin', '*')
        SimpleHTTPRequestHandler.end_headers(self)
    
    def do_OPTIONS(self):
        self.send_response(200)
        self.send_header('Access-Control-Allow-Origin', '*')
        self.send_header('Access-Control-Allow-Methods', 'GET, OPTIONS')
        self.send_header('Access-Control-Allow-Headers', '*')
        self.end_headers()

os.chdir('/root')
print(f'Server on port 9000, serving from {os.getcwd()}')
HTTPServer(('0.0.0.0', 9000), DownloadHandler).serve_forever()