#!/usr/bin/env python3
import argparse
import json
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
from urllib.parse import parse_qs, urlparse


def build_results(engine: str, query: str):
    lowered = (engine or "").lower()
    if "google" in lowered and "video" in lowered:
        return [
            {
                "title": f"{query.title()} cinematic b-roll",
                "url": "https://www.youtube.com/watch?v=abcdefghijk",
                "content": f"{query} stock footage edit reference",
                "thumbnail": "https://i.ytimg.com/vi/abcdefghijk/hqdefault.jpg",
                "engine": engine,
            }
        ]

    return [
        {
            "title": f"{query.title()} envato clip",
            "url": "https://elements.envato.com/city-rain-b-roll-AB12CD",
            "content": f"{query} stock footage cinematic scene",
            "thumbnail": "https://images.example.com/envato-city-rain.jpg",
            "thumbnail_src": "https://images.example.com/envato-city-rain.jpg",
            "engine": engine,
        },
        {
            "title": f"{query.title()} artgrid clip",
            "url": "https://artgrid.io/clip/123456/city-rain-night",
            "content": f"{query} editorial footage establishing shot",
            "thumbnail": "https://images.example.com/artgrid-city-rain.jpg",
            "thumbnail_src": "https://images.example.com/artgrid-city-rain.jpg",
            "img_src": "https://images.example.com/artgrid-city-rain.jpg",
            "engine": engine,
        },
    ]


class Handler(BaseHTTPRequestHandler):
    def do_GET(self):
        parsed = urlparse(self.path)
        if parsed.path != "/search":
            self.send_response(404)
            self.end_headers()
            return

        params = parse_qs(parsed.query)
        query = params.get("q", ["city rain"])[0]
        engine = params.get("engines", [""])[0]
        payload = {"results": build_results(engine, query)}

        body = json.dumps(payload).encode("utf-8")
        self.send_response(200)
        self.send_header("Content-Type", "application/json")
        self.send_header("Content-Length", str(len(body)))
        self.end_headers()
        self.wfile.write(body)

    def log_message(self, format, *args):
        return


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--port", type=int, default=18080)
    args = parser.parse_args()

    server = ThreadingHTTPServer(("127.0.0.1", args.port), Handler)
    try:
        server.serve_forever()
    except KeyboardInterrupt:
        pass
    finally:
        server.server_close()


if __name__ == "__main__":
    main()