import os
import sys
from socketserver import ThreadingTCPServer, StreamRequestHandler
class GopherHandler(StreamRequestHandler):
def handle(self):
request = self.rfile.readline().strip().decode('utf-8')
path = self.server.root + '/' + request.lstrip('/')
print('{}: {} => {}'.format(self.client_address[0], request, path))
if os.path.isfile(path):
with open(path, 'rb') as resource:
self.wfile.write(resource.read())
elif os.path.isdir(path):
if os.path.isfile(path + '/index.gph'):
with open(path + '/index.gph', 'rb') as resource:
self.wfile.write(resource.read())
else:
response = ''
for item in os.listdir(path):
if os.path.isdir(os.path.join(path, item)):
response += '1{1}\t{0}/{1}\t{2}\t{3}\r\n'.format(request, item, self.server.hostname, self.server.server_address[1])
else:
response += '0{1}\t{0}/{1}\t{2}\t{3}\r\n'.format(request, item, self.server.hostname, self.server.server_address[1])
self.wfile.write(response.encode('utf-8'))
self.wfile.write(b'.\r\n')
else:
self.wfile.write(b'3Not found\t\tlocalhost\t7000\r\n.\r\n')
class GopherServer(ThreadingTCPServer):
allow_reuse_address = True
def __init__(self, host, port, root, hostname=None):
super().__init__((host, port), GopherHandler)
self.root = root
self.hostname = host if hostname is None else hostname
if __name__ == '__main__':
args = sys.argv[1:]
host = '127.0.0.1'
name = None
port = 7000
root = '.'
while len(args) > 0:
option = args.pop(0)
if option == '-h':
host = args.pop(0)
elif option == '-n':
name = args.pop(0)
elif option == '-p':
port = int(args.pop(0))
elif option == '-d':
root = args.pop(0)
with GopherServer(host, port, root, name) as server:
server.serve_forever()
