diff --git a/python/bin/pmtiles-serve b/python/bin/pmtiles-serve index edd35ba..5e03059 100755 --- a/python/bin/pmtiles-serve +++ b/python/bin/pmtiles-serve @@ -1,26 +1,34 @@ #!/usr/bin/env python3 -import sys -import re -import json +import argparse import http.server +import json +import re from socketserver import ThreadingMixIn +import sys from pmtiles.reader import read + +# https://docs.python.org/3/library/http.server.html class ThreadingSimpleServer(ThreadingMixIn, http.server.HTTPServer): pass -if len(sys.argv) <= 1: - print("Usage: pmtiles-serve PMTILES_FILE") - exit(1) +parser = argparse.ArgumentParser(description='Convert between PMTiles and other archive formats.') +parser.add_argument('pmtiles_file',help='PMTiles archive to serve') +parser.add_argument('port',help='Port to bind to') +parser.add_argument('--bind', help='Address to bind server to: default localhost') +parser.add_argument('--cors-allow-all', help='Return Access-Control-Allow-Origin:* header',action='store_true') +args = parser.parse_args() -with read(sys.argv[1]) as reader: +with read(args.pmtiles_file) as reader: fmt = reader.metadata['format'] class Handler(http.server.SimpleHTTPRequestHandler): def do_GET(self): if self.path == "/metadata": self.send_response(200) + if args.cors_allow_all: + self.send_header('Access-Control-Allow-Origin','*') self.end_headers() self.wfile.write(json.dumps(reader.metadata).encode('utf-8')) return @@ -40,11 +48,16 @@ with read(sys.argv[1]) as reader: self.wfile.write("tile not found".encode('utf-8')) return self.send_response(200) - self.end_headers() - self.send_header('Content-Type','image/png') + if args.cors_allow_all: + self.send_header('Access-Control-Allow-Origin','*') + if fmt == 'pbf': + self.send_header('Content-Type','application/x-protobuf') + else: + self.send_header('Content-Type','image/' + fmt) self.end_headers() self.wfile.write(data) - print('serving localhost:8000/{z}/{x}/{y}.' + fmt) - httpd = ThreadingSimpleServer(('', 8000), Handler) + bind = args.bind or 'localhost' + print(f'serving {bind}:{args.port}/{{z}}/{{x}}/{{y}}.{fmt}, for development only') + httpd = ThreadingSimpleServer((args.bind or '', int(args.port)), Handler) httpd.serve_forever()