mirror of
https://github.com/m4rcel-lol/m5rcode.git
synced 2025-12-06 19:13:57 +05:30
114 lines
4.4 KiB
Python
114 lines
4.4 KiB
Python
import requests
|
|
from bs4 import BeautifulSoup
|
|
from urllib.parse import urljoin
|
|
from colorama import Fore, Style
|
|
import re
|
|
|
|
def strip_ansi(text):
|
|
return re.sub(r'\x1b\[[0-9;]*m', '', text)
|
|
|
|
class WdirCommand:
|
|
def __init__(self, url):
|
|
self.url = url.strip()
|
|
|
|
def run(self):
|
|
box_min = 72
|
|
if not self.url:
|
|
print(Fore.RED + "Usage: wdir <url>" + Style.RESET_ALL)
|
|
return
|
|
|
|
# Ensure scheme
|
|
if not self.url.startswith("http://") and not self.url.startswith("https://"):
|
|
self.url = "http://" + self.url
|
|
|
|
try:
|
|
print(Fore.CYAN + f"[FETCH] Scanning directory at {self.url}..." + Style.RESET_ALL)
|
|
resp = requests.get(self.url, timeout=5)
|
|
resp.raise_for_status()
|
|
except Exception as e:
|
|
print(Fore.RED + f"[ERR] Failed to fetch {self.url}: {e}" + Style.RESET_ALL)
|
|
return
|
|
|
|
soup = BeautifulSoup(resp.text, "html.parser")
|
|
links = soup.find_all("a")
|
|
|
|
files = []
|
|
for link in links:
|
|
href = link.get("href")
|
|
if not href:
|
|
continue
|
|
if href.startswith("?") or href.startswith("#") or href.startswith("../"):
|
|
continue
|
|
is_dir = href.endswith("/")
|
|
filename = href.rstrip("/").split("/")[-1]
|
|
if not is_dir and re.search(r"\.(php|html?|asp|aspx|jsp)$", filename, re.I):
|
|
continue
|
|
if is_dir:
|
|
ftype = "Directory"
|
|
elif "." in filename:
|
|
ftype = f".{filename.split('.')[-1]} file"
|
|
else:
|
|
ftype = "File"
|
|
row_text = link.parent.get_text(" ", strip=True)
|
|
size_match = re.search(r"(\d+(?:\.\d+)?\s*(?:KB|MB|GB|B))", row_text, re.I)
|
|
date_match = re.search(r"(\d{4}-\d{2}-\d{2} \d{2}:\d{2})", row_text)
|
|
size = size_match.group(1) if size_match else "-"
|
|
modified = date_match.group(1) if date_match else "-"
|
|
files.append((filename, ftype, size, modified))
|
|
|
|
# Calculate column widths for nice boxed output
|
|
col_names = ["Name", "Type", "Size", "Modified"]
|
|
pad = [30, 15, 12, 20]
|
|
table_width = sum(pad) + len(pad) + 1 # columns + spaces + box
|
|
width = max(box_min, table_width + 2)
|
|
|
|
# If no files, pretty box saying so
|
|
if not files:
|
|
print(Fore.MAGENTA + "╔" + "═" * (width - 2) + "╗")
|
|
out = "No files or directories found (maybe directory listing is disabled)."
|
|
out = out.center(width - 2)
|
|
print(Fore.MAGENTA + "║" + Fore.YELLOW + out + Fore.MAGENTA + "║")
|
|
print(Fore.MAGENTA + "╚" + "═" * (width - 2) + "╝" + Style.RESET_ALL)
|
|
return
|
|
|
|
# Pretty header
|
|
print(Fore.MAGENTA + "╔" + "═" * (width - 2) + "╗")
|
|
title = "Web Directory Listing"
|
|
print(Fore.MAGENTA + "║" + Fore.CYAN + Style.BRIGHT + title.center(width - 2) + Fore.MAGENTA + "║")
|
|
print(Fore.MAGENTA + "╟" + "─" * (width - 2) + "╢")
|
|
# Table header
|
|
header = (
|
|
Fore.LIGHTMAGENTA_EX
|
|
+ f"{col_names[0]:<{pad[0]}} {col_names[1]:<{pad[1]}} {col_names[2]:<{pad[2]}} {col_names[3]:<{pad[3]}}"
|
|
+ Style.RESET_ALL
|
|
)
|
|
print(
|
|
Fore.MAGENTA + "║"
|
|
+ header
|
|
+ " " * (width - 2 - len(strip_ansi(header)))
|
|
+ Fore.MAGENTA + "║"
|
|
)
|
|
print(Fore.MAGENTA + "╟" + "─" * (width - 2) + "╢")
|
|
|
|
# Table rows
|
|
for fname, ftype, size, modified in files:
|
|
if ftype == "Directory":
|
|
color = Fore.BLUE + Style.BRIGHT
|
|
elif ftype.endswith("file"):
|
|
color = Fore.CYAN
|
|
else:
|
|
color = Fore.WHITE
|
|
filecol = f"{color}{fname:<{pad[0]}}{Style.RESET_ALL}"
|
|
typecol = f"{Style.DIM}{Fore.WHITE}{ftype:<{pad[1]}}{Style.RESET_ALL}"
|
|
sizecol = f"{Style.DIM}{Fore.LIGHTWHITE_EX}{size:<{pad[2]}}{Style.RESET_ALL}"
|
|
modcol = f"{Style.DIM}{Fore.LIGHTWHITE_EX}{modified:<{pad[3]}}{Style.RESET_ALL}"
|
|
row = f"{filecol} {typecol} {sizecol} {modcol}"
|
|
print(
|
|
Fore.MAGENTA + "║"
|
|
+ row
|
|
+ " " * (width - 2 - len(strip_ansi(row)))
|
|
+ Fore.MAGENTA + "║"
|
|
)
|
|
# Footer
|
|
print(Fore.MAGENTA + "╚" + "═" * (width - 2) + "╝" + Style.RESET_ALL)
|