import os

import glob

import re

import sys

import urllib.request

import urllib.parse

import html # Add this new import!

 

def get_channel_id(filename):

match = re.search(r'\[(\d+)\]\.html$', filename)

if match:

return int(match.group(1))

return float('inf')

 

 

def download_and_localize_images(target_dir, log_files):

img_dir_name = "images"

img_dir_path = os.path.join(target_dir, img_dir_name)

os.makedirs(img_dir_path, exist_ok=True)

 

downloaded_urls = {}

img_counter = 0

 

print("Scanning log files for external images to download locally...")

src_pattern = re.compile(r'src=["\'](https?://[^"\']+)["\']', re.IGNORECASE)

 

for file_path in log_files:

filename = os.path.basename(file_path)

print(f" Processing image links in: {filename}")

 

with open(file_path, "r", encoding="utf-8") as f:

html_content = f.read()

 

urls = src_pattern.findall(html_content)

if not urls:

continue

 

modified = False

for url in urls:

if url.startswith(('data:', 'file:', './')) or url in downloaded_urls:

if url in downloaded_urls:

html_content = html_content.replace(url, downloaded_urls[url])

modified = True

continue

 

try:

parsed_url = urllib.parse.urlparse(url)

clean_path = parsed_url.path

ext = os.path.splitext(clean_path)[1]

 

if '?' in ext:

ext = ext.split('?')[0]

if not ext or len(ext) > 5:

ext = ".png"

 

img_counter += 1

local_img_name = f"img_{img_counter:04d}{ext}"

local_img_path = os.path.join(img_dir_path, local_img_name)

 

req = urllib.request.Request(

html.unescape(url),

headers={'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64)'}

)

 

with urllib.request.urlopen(req, timeout=10) as response, open(local_img_path, "wb") as out_file:

out_file.write(response.read())

 

relative_local_path = f"{img_dir_name}/{local_img_name}"

downloaded_urls[url] = relative_local_path

html_content = html_content.replace(url, relative_local_path)

modified = True

 

except Exception as e:

print(f" Warning: Could not download asset {url}... Error: {e}")

 

if modified:

with open(file_path, "w", encoding="utf-8") as f:

f.write(html_content)

 

print(f"Successfully cached {img_counter} image assets locally.")

 

 

def build_discord_archive(target_dir, game_name):

target_dir = os.path.abspath(target_dir)

 

if not os.path.isdir(target_dir):

print(f"Error: The directory '{target_dir}' does not exist.")

return

 

all_html_files = glob.glob(os.path.join(target_dir, "*.html"))

exclude_files = {"index.html", "sidebar.html", "welcome.html", "public_index.html"}

log_files = [f for f in all_html_files if os.path.basename(f) not in exclude_files]

 

if not log_files:

print(f"No Discord chat log HTML files found in '{target_dir}'.")

return

 

# Download image resources locally

download_and_localize_images(target_dir, log_files)

 

# Sort files chronologically by channel ID

log_files.sort(key=lambda x: get_channel_id(os.path.basename(x)))

 

print(f"Compiling Single Page Application layout for '{game_name}'...")

 

sidebar_buttons = []

chat_panels = []

 

for idx, file_path in enumerate(log_files):

filename = os.path.basename(file_path)

 

clean_name = filename.split(" - ")[-1].split(" [")[0]

if not clean_name or clean_name.endswith(".html"):

clean_name = os.path.splitext(filename)[0]

clean_name = clean_name.replace('"', '"')

 

with open(file_path, "r", encoding="utf-8") as f:

raw_html = f.read()

 

body_match = re.search(r"<body[^>]*>(.*)</body>", raw_html, re.DOTALL | re.IGNORECASE)

body_content = body_match.group(1) if body_match else raw_html

 

panel_id = f"chan_panel_{idx}"

 

btn_html = f' <button class="channel-link" onclick="switchChannel(\'{panel_id}\', this)">{clean_name}</button>'

sidebar_buttons.append(btn_html)

 

panel_html = f' <div id="{panel_id}" class="chat-container-view" style="display: none;">\n{body_content}\n </div>'

chat_panels.append(panel_html)

 

sidebar_content_block = "\n".join(sidebar_buttons)

chat_panels_block = "\n".join(chat_panels)

 

master_spa_tmpl = f"""<!DOCTYPE html>

<html lang="en">

<head>

<meta charset="UTF-8">

<title>{game_name} - Chat Logs</title>

<style>

* {{ box-sizing: border-box; margin: 0; padding: 0; }}

body, html {{ height: 100%; background-color: #202225; overflow: hidden; display: flex; }}

 

/* Sidebar layout styles */

.sidebar {{ width: 260px; background-color: #2f3136; height: 100%; display: flex; flex-direction: column; border-right: 1px solid rgba(255, 255, 255, 0.05); flex-shrink: 0; }}

.sidebar-header {{ padding: 1rem; color: #ffffff; font-weight: 600; font-size: 1.1rem; border-bottom: 1px solid rgba(0, 0, 0, 0.2); margin-bottom: 10px; }}

.channel-list {{ padding: 0 8px; overflow-y: auto; flex: 1; display: flex; flex-direction: column; gap: 2px; }}

 

.channel-link {{ display: flex; align-items: center; padding: 8px 12px; color: #8e9297; background: none; border: none; border-radius: 4px; font-size: 0.95rem; font-weight: 500; text-align: left; width: 100%; cursor: pointer; transition: background-color 0.1s, color 0.1s; }}

.channel-link::before {{ content: "# "; font-size: 1.1rem; color: #4f545c; margin-right: 4px; }}

.channel-link:hover {{ background-color: #34363c; color: #dcddde; }}

.channel-link.active {{ background-color: #393c43; color: #ffffff; }}

.channel-link.active::before {{ color: #b9bbbe; }}

 

/* Expansive content layout layers */

.main-viewer {{ flex: 1; height: 100%; background-color: #36393f; position: relative; }}

.chat-container-view {{ width: 100%; height: 100%; overflow-y: auto; overflow-x: hidden; }}

 

/* Retain standard Discord layout width rules cleanly across screen space */

.preamble, .chatlog, .postamble {{ max-width: 100% !important; width: 100% !important; }}

 

/* Fixed Welcome Screen Layout centering elements safely */

.welcome-screen {{ display: flex; align-items: center; justify-content: center; height: 100%; width: 100%; color: #72767d; font-family: sans-serif; padding: 20px; }}

.welcome-container {{ max-width: 500px; width: 100%; display: flex; flex-direction: column; align-items: center; justify-content: center; }}

.welcome-container h2 {{ color: #b9bbbe; margin-bottom: 12px; font-size: 1.6rem; text-align: center !important; width: 100%; }}

.welcome-container p {{ font-size: 0.95rem; line-height: 1.5; text-align: center !important; width: 100%; }}

</style>

</head>

<body>

 

<div class="sidebar">

<div class="sidebar-header">Text Channels</div>

<div class="channel-list">

{sidebar_content_block}

</div>

</div>

 

<div class="main-viewer">

<div id="welcome-layer" class="welcome-screen">

<div class="welcome-container">

<h2>Welcome to the {game_name} Archive</h2>

<p>Select a channel from the sidebar list on the left to view specific round logs. Your scroll positions will remain locked automatically as you switch.</p>

</div>

</div>

 

{chat_panels_block}

</div>

 

<script>

function switchChannel(targetPanelId, activeBtn) {{

// Hide the welcome block layer

document.getElementById('welcome-layer').style.display = 'none';

 

// Hide all other viewport logs

document.querySelectorAll('.chat-container-view').forEach(panel => {{

panel.style.display = 'none';

}});

 

// Clear existing navbar highlight tracks

document.querySelectorAll('.channel-link').forEach(btn => {{

btn.classList.remove('active');

}});

 

// Activate target visibility layers

document.getElementById(targetPanelId).style.display = 'block';

activeBtn.classList.add('active');

}}

</script>

</body>

</html>"""

 

output_path = os.path.join(target_dir, "index.html")

with open(output_path, "w", encoding="utf-8") as f:

f.write(master_spa_tmpl)

 

print(f"\nSuccess! Built your instant-switch, scroll-preserving archive sheet at:\n{output_path}")

 

 

if __name__ == "__main__":

if len(sys.argv) > 2:

path_input = sys.argv[1]

game_input = sys.argv[2]

else:

path_input = input("Enter the path to your Discord logs directory: ").strip()

game_input = input("Enter the name of the game: ").strip()

 

build_discord_archive(path_input, game_input)