diff --git a/main.py b/main.py index e8e5612..0a6e740 100644 --- a/main.py +++ b/main.py @@ -44,16 +44,26 @@ class WebServer: async def handle(self, reader: asyncio.StreamReader, writer: asyncio.StreamWriter) -> None: addr = writer.get_extra_info("peername") await self.log(self.conn_msg, addr) - data = await reader.read(self._read_buffer) - data = unquote(data.decode()) + rdata = await reader.read(self._read_buffer) + data = rdata.decode() if not data: return - real_addr = "".join([l.replace("X-Real-IP: ", "") for l in data.split("\n") if l.startswith("X-Real-IP: ")][0]).strip() # ТАКОЙ ГАВНАКОД XD + real_addr = None + for line in data.split("\n"): + if line.startswith("X-Real-IP: "): + real_addr = line[len("X-Real-IP: "):].strip() + break + if real_addr and self.proxied: addr = (real_addr, addr[1]) + request = data.split("\n")[0] - file_name = request.split()[1][1:] - file_path = self.directory + file_name + parts = request.split() + if len(parts) < 2: return + + path = unquote(parts[1]) + file_name = path[1:] if path.startswith('/') else path + file_path = os.path.abspath(os.path.join(self.directory, file_name)) if os.path.isfile(file_path): mime, _ = mimetypes.guess_type(file_path) @@ -90,20 +100,27 @@ class WebServer: resp = await f.read() files = "" - for file_name in sorted(os.listdir(file_path)): # TODO: добавить настройки отображения файла - file = "/".join(file_path.split("/")[1:]) +"/" + file_name - - if os.path.isdir(file_path + "/" + file_name): - file_name = "/" + file_name - - modify_time = (os.path.getmtime(file_path + "/" + file_name)) + norm_path = file_path.replace('\\', '/') + base_path = norm_path.replace(self.directory.replace('\\', '/'), '').strip('/') + + for item in sorted(os.listdir(file_path)): + item_path = os.path.join(file_path, item) + is_dir = os.path.isdir(item_path) + + rel_path = f"{base_path}/{item}" if base_path else item + rel_path = rel_path.replace('\\', '/') + + if is_dir: + rel_path += "/" + item += "/" + + modify_time = os.path.getmtime(item_path) modify_datetime = datetime.datetime.fromtimestamp(modify_time) formatted_time = modify_datetime.strftime("%d.%m.%Y %H:%M:%S") - - files += f'{file_name} | {formatted_time}
' + + files += f'{item} | {formatted_time}
\n' resp = resp.replace("", files) - resp = resp.replace("//", "/") resp = resp.encode() headers = ( "HTTP/1.1 200 OK\r\n" @@ -113,15 +130,17 @@ class WebServer: "\r\n" ) - writer.write("".join(headers).encode() + resp) + writer.write(headers.encode() + resp) else: await self.log(self.e404_msg, addr, file_path) + async with aiofiles.open(self._e404_file, "r", encoding="utf-8") as f: + content = await f.read() response = ( "HTTP/1.1 404 Not Found\r\n" "Content-Type: text/html; charset=utf-8\r\n" f"Server: {self.name}\r\n" "\r\n" - f"{open(self._e404_file, encoding="utf-8").read()}" + f"{content}" ) writer.write(response.encode()) await writer.drain()