111 lines
3.3 KiB
Python
111 lines
3.3 KiB
Python
import os
|
|
|
|
from http.server import BaseHTTPRequestHandler
|
|
import socketserver
|
|
import threading
|
|
import pyqrcode
|
|
import socket
|
|
import struct
|
|
import base64
|
|
|
|
|
|
from SuperBeam.RequestHandlers import (
|
|
ChunkTemplateHandler, ThumbHandler, ApkRequestHandler, LegacyListHandler,
|
|
SingleFileHandler, ZipFileHandler, SuperStreamHandler, AssetHandler,
|
|
JsonListHandler
|
|
)
|
|
|
|
|
|
class ThreadedTCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer):
|
|
pass
|
|
|
|
|
|
class SuperBeamServer(BaseHTTPRequestHandler):
|
|
|
|
def do_GET(self):
|
|
# TODO Add globbing
|
|
print(threading.current_thread())
|
|
paths = {
|
|
'index': ChunkTemplateHandler,
|
|
'/': ChunkTemplateHandler,
|
|
'/index.htm': ChunkTemplateHandler, # TODO handler2
|
|
'/getthumb': ThumbHandler,
|
|
'/light': ChunkTemplateHandler, # TODO handler2
|
|
'/superlist': LegacyListHandler,
|
|
'/jsonlist': JsonListHandler,
|
|
'/get/': SingleFileHandler,
|
|
'/getapk': ApkRequestHandler,
|
|
'/getzip': ZipFileHandler,
|
|
'/getstream': SuperStreamHandler,
|
|
'*': AssetHandler,
|
|
}
|
|
|
|
if self.path in paths: # TODO startswith key check
|
|
self.respond(paths[self.path])
|
|
|
|
def respond(self, handler):
|
|
print('='*30 + ' DEBUG ' + '='*30)
|
|
print('CONNECTION:', self.connection)
|
|
print('HEADERS :', self.headers)
|
|
print('REQUEST :', self.request)
|
|
print('REQUEST* :', self.requestline)
|
|
print('PATH :', self.path)
|
|
print('COMMAND :', self.command)
|
|
print('HANDLER :', handler)
|
|
print('='*30 + ' DEBUG ' + '='*30)
|
|
handler.handle(self)
|
|
|
|
|
|
def get_primary_ip():
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
try:
|
|
# doesn't even have to be reachable
|
|
s.connect(('10.255.255.255', 1))
|
|
IP = s.getsockname()[0]
|
|
except Exception as exc:
|
|
print(exc)
|
|
IP = '127.0.0.1'
|
|
finally:
|
|
s.close()
|
|
return IP
|
|
|
|
|
|
def show_qrcode():
|
|
primary_ip = get_primary_ip()
|
|
octets = [1]
|
|
octets.extend([int(_) for _ in primary_ip.split('.')[::-1]])
|
|
url = 'http://superbe.am/q?' + base64.b64encode(
|
|
struct.pack('BBBBB', *octets)
|
|
).decode('utf-8')
|
|
qrcode = pyqrcode.create(url)
|
|
print('\033[1;37;37m████████' + '██'*(len(qrcode.code[0])))
|
|
for row in qrcode.code:
|
|
print('\033[1;37;37m████', end='')
|
|
for block in row:
|
|
if block == 1:
|
|
print('\033[1;30;30m ', end='')
|
|
else:
|
|
print('\033[1;37;37m██', end='')
|
|
print('\033[1;37;37m████')
|
|
print('\033[1;37;37m████████' + '██'*(len(qrcode.code[0])))
|
|
|
|
|
|
def build_filelist(paths, recursive=True):
|
|
filelist = []
|
|
for path in paths:
|
|
if os.path.isdir(path) and recursive:
|
|
for root, dirs, files in os.walk(path):
|
|
for file_ in files:
|
|
filelist.append(os.path.join(root, file_))
|
|
else:
|
|
filelist.append(path)
|
|
return filelist
|
|
|
|
|
|
def serve_forever(files):
|
|
Handler = SuperBeamServer
|
|
Handler.files = files
|
|
with socketserver.ThreadingTCPServer(('', 8080), Handler) as httpd:
|
|
show_qrcode()
|
|
httpd.serve_forever()
|