#!/usr/bin/python

from os import environ as env
import urllib.request
import json
import sys

def error():
  print("Content-type: text/html")
  print()
  print("HELLO!")

def proxy(url,ext,callback):
  req = urllib.request.urlopen(url)
  print("Content-type: image/" + ext)
  print()
  sys.stdout.flush()
  if callback:
  	sys.stdout.buffer.write( callback + "('" )
		sys.stdout.buffer.write( json.dump({ data: req.read() })
  	sys.stdout.buffer.write( callback + "')" )
  else:
		sys.stdout.buffer.write( req.read() )

qs = env['QUERY_STRING'].split('&callback=')
path = qs[0]
callback = qs[1]
ext = path[-3:].lower()

if path[0:4] == "http" and ext in ("gif","jpg","png","peg"):
  proxy(path,ext,callback)
else:
  error()