#!/usr/bin/env python3
"""Serves poc.html and the test media files on http://127.0.0.1:8080/."""
import http.server
import os
import socketserver
import sys

ROOT = os.path.dirname(os.path.abspath(__file__))
# Adjust to your checkout location if different.
MEDIA_DIR = '/chromium/chromium/src/media/test/data'

class Handler(http.server.SimpleHTTPRequestHandler):
    def translate_path(self, path):
        if path.startswith('/media/'):
            return os.path.join(MEDIA_DIR, path[len('/media/'):])
        return os.path.join(ROOT, path.lstrip('/'))

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

if __name__ == '__main__':
    port = int(sys.argv[1]) if len(sys.argv) > 1 else 8080
    with socketserver.TCPServer(('127.0.0.1', port), Handler) as httpd:
        httpd.allow_reuse_address = True
        sys.stderr.write(f'[server] listening on http://127.0.0.1:{port}/\n')
        httpd.serve_forever()