mirror of
https://github.com/m4rcel-lol/m5rcode-ubuntu.git
synced 2025-12-06 19:13:57 +05:30
Uploading m5rcode Ubuntu Port to the repo
This commit is contained in:
113
commands/cmd_wdir.py
Normal file
113
commands/cmd_wdir.py
Normal file
@@ -0,0 +1,113 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user