if mirror_type == 'work': result = asyncio.run(mirror.mirror_work(url, format)) elif mirror_type == 'series': result = asyncio.run(mirror.mirror_series(url)) else: return jsonify({'error': 'Invalid type'}), 400
def _extract_work_id(self, url: str) -> str: """Extract work ID from AO3 URL""" import re match = re.search(r'/works/(\d+)', url) if match: return match.group(1) raise ValueError("Invalid AO3 work URL")
<script> let queue = []; async function mirrorWork() { const url = document.getElementById('urlInput').value; const format = document.getElementById('formatSelect').value; if (!url) { alert('Please enter an AO3 URL'); return; } addToQueue(url, 'work', format); await processQueue(); } async function mirrorSeries() { const url = document.getElementById('urlInput').value; const format = document.getElementById('formatSelect').value; if (!url) { alert('Please enter an AO3 series URL'); return; } addToQueue(url, 'series', format); await processQueue(); } function addToQueue(url, type, format) { queue.push({ id: Date.now(), url: url, type: type, format: format, status: 'pending' }); updateQueueDisplay(); } async function processQueue() { while (queue.length > 0) { const item = queue[0]; if (item.status === 'processing') break; item.status = 'processing'; updateQueueDisplay(); try { const response = await fetch('/api/mirror', { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify(item) }); const result = await response.json(); if (result.status === 'success') { item.status = 'completed'; loadLibrary(); } else { item.status = 'failed'; item.error = result.error; } } catch (error) { item.status = 'failed'; item.error = error.message; } updateQueueDisplay(); queue.shift(); await new Promise(resolve => setTimeout(resolve, 1000)); } } function updateQueueDisplay() { const queueDiv = document.getElementById('queue'); if (queue.length === 0) { queueDiv.innerHTML = '<p style="color: #888;">No active downloads</p>'; return; } queueDiv.innerHTML = queue.map(item => ` <div class="queue-item"> <strong>${item.url}</strong> <span class="status ${item.status}">${item.status}</span> ${item.error ? `<div style="color: red; font-size: 12px; margin-top: 5px;">${item.error}</div>` : ''} </div> `).join(''); } async function loadLibrary() { const response = await fetch('/api/library'); const works = await response.json(); const libraryDiv = document.getElementById('library'); if (works.length === 0) { libraryDiv.innerHTML = '<p style="color: #888;">No mirrored works yet</p>'; return; } libraryDiv.innerHTML = works.map(work => ` <div class="work-card" onclick="readWork('${work.work_id}')"> <div class="work-title">${escapeHtml(work.title)}</div> <div class="work-author">by ${escapeHtml(work.author)}</div> <div class="work-stats"> <span>📄 ${work.word_count.toLocaleString()} words</span> <span>📖 ${work.chapters} chapters</span> <span>❤️ ${work.kudos}</span> </div> </div> `).join(''); } async function readWork(workId) { const response = await fetch(`/api/read/${workId}`); const data = await response.json(); const modal = document.getElementById('readerModal'); const content = document.getElementById('readerContent'); content.innerHTML = ` <h2>${escapeHtml(data.metadata.title)}</h2> <p><strong>by ${escapeHtml(data.metadata.author)}</strong></p> <div style="margin: 20px 0;">${data.content}</div> `; modal.style.display = 'flex'; } function closeModal() { document.getElementById('readerModal').style.display = 'none'; } function escapeHtml(text) { const div = document.createElement('div'); div.textContent = text; return div.innerHTML; } // Load library on page load loadLibrary(); </script> </body> </html> # api.py from flask import Flask, request, jsonify, send_file from flask_cors import CORS import asyncio import json from pathlib import Path app = Flask( name ) CORS(app) ao3 mirror
async def mirror_series(self, series_url: str) -> Dict: """Mirror an entire series""" series_id = self._extract_series_id(series_url) works = await self._get_series_works(series_url) mirrored = [] for work_url in works: result = await self.mirror_work(work_url) mirrored.append(result) return {"series_id": series_id, "works": mirrored}
with open(work_path / 'metadata.json', 'r', encoding='utf-8') as f: metadata = json.load(f) if mirror_type == 'work': result = asyncio
mirror = AO3Mirror()
@app.route('/api/mirror', methods=['POST']) def mirror_endpoint(): data = request.json url = data.get('url') mirror_type = data.get('type', 'work') format = data.get('format', 'html') 400 def _extract_work_id(self
for work_path in work_dir.iterdir(): if work_path.is_dir(): metadata_file = work_path / 'metadata.json' if metadata_file.exists(): with open(metadata_file, 'r', encoding='utf-8') as f: metadata = json.load(f) works.append({ 'work_id': metadata['work_id'], 'title': metadata['title'], 'author': metadata['author'], 'word_count': metadata['word_count'], 'chapters': metadata['chapters'], 'kudos': metadata['kudos'] })