python: Use importlib for python3.
[ashd.git] / python / htredir
1 #!/usr/bin/python
2
3 import sys, os, getopt
4
5 def destructurl(url):
6     if "://" in url:
7         p = url.index("://")
8         scheme, url = url[:p], url[p + 3]
9         if "/" in url:
10             p = url.index("/")
11             host, url = url[:p], url[p + 1:]
12         else:
13             host, url = url, ""
14     else:
15         scheme = None
16         host = None
17     return scheme, host, url
18
19 def usage(out):
20     out.write("usage: htredir [-hp] TARGET METHOD URL REST\n")
21
22 status = "302 Found"
23 opts, args = getopt.getopt(sys.argv[1:], "hp")
24 for o, a in opts:
25     if o == "-h":
26         usage(sys.stdout)
27         sys.exit(0)
28     elif o == "-p":
29         status = "301 Moved Permanently"
30 if len(args) != 4:
31     usage(sys.stderr)
32     sys.exit(1)
33 target, method, url, rest = args
34 scheme = os.getenv("REQ_X_ASH_PROTOCOL")
35 host = os.getenv("REQ_HOST")
36 me = url
37 if me[-len(rest):] == rest:
38     me = me[:-len(rest)]
39 tscheme, thost, target = destructurl(target)
40 if tscheme: scheme = tscheme
41 if thost: host = thost
42 if len(target) > 0 and target[0] == "/":
43     pass
44 else:
45     if "/" in me:
46         p = me.rindex("/")
47         target = me[:p + 1] + target
48 if len(target) > 0 and target[0] == "/":
49     target = target[1:]
50 if scheme and host:
51     target = "%s://%s/%s" % (scheme, host, target)
52 else:
53     # Illegal, but the only option (the premises are illegal anyway)
54     pass
55
56 try:
57     sys.stdout.write("HTTP/1.1 %s\n" % status)
58     sys.stdout.write("Location: %s\n" % target)
59     sys.stdout.write("Content-Length: 0\n")
60     sys.stdout.write("\n")
61 except IOError:
62     sys.exit(1)