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")
async def mirror_work(self, work_url: str, format: str = "html") -> Dict: """Mirror a single work from AO3""" work_id = self._extract_work_id(work_url) # Check if already mirrored if self._is_mirrored(work_id): return {"status": "exists", "work_id": work_id} # Fetch work data work_data = await self._fetch_work(work_url) # Save metadata self._save_metadata(work_id, work_data['metadata']) # Save content self._save_content(work_id, work_data['content'], format) return {"status": "success", "work_id": work_id}
class AO3Mirror: def (self, cache_dir: str = "ao3_cache"): self.cache_dir = Path(cache_dir) self.cache_dir.mkdir(exist_ok=True) self.work_dir = self.cache_dir / "works" self.work_dir.mkdir(exist_ok=True)
def _is_mirrored(self, work_id: str) -> bool: """Check if work is already mirrored""" return (self.work_dir / work_id / "metadata.json").exists() 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}
def _save_metadata(self, work_id: str, metadata: WorkMetadata): """Save work metadata as JSON""" work_path = self.work_dir / work_id work_path.mkdir(exist_ok=True) metadata_file = work_path / "metadata.json" with open(metadata_file, 'w', encoding='utf-8') as f: json.dump(asdict(metadata), f, indent=2, ensure_ascii=False)
return jsonify(result) @app.route('/api/library', methods=['GET']) def get_library(): works = [] work_dir = mirror.work_dir url: str) ->
<div id="readerModal" class="modal"> <div class="modal-content"> <div id="readerContent"></div> <button onclick="closeModal()" style="margin-top: 20px;">Close</button> </div> </div>
@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')
<div class="input-group"> <label>AO3 URL</label> <input type="text" id="urlInput" placeholder="https://archiveofourown.org/works/12345678"> </div> <div class="input-group"> <label>Format</label> <select id="formatSelect"> <option value="html">HTML (Original)</option> <option value="txt">Plain Text</option> <option value="epub">EPUB</option> </select> </div> <button onclick="mirrorWork()">Mirror Work</button> <button onclick="mirrorSeries()" style="margin-left: 10px;">Mirror Series</button> </div> <div class="card"> <h2>📥 Download Queue</h2> <div id="queue"></div> </div> <div class="card"> <h2>📖 Mirrored Library</h2> <div id="library" class="library-grid"></div> </div> </div> format: str = "html") ->
html_path = work_path / 'work.html' if html_path.exists(): with open(html_path, 'r', encoding='utf-8') as f: content = f.read() else: content = "<p>Content not available</p>"
async def mirror_bookmarks(self, user: str, page_limit: int = None) -> Dict: """Mirror all bookmarked works of a user""" # Respect rate limits await self._rate_limit() # Implementation continues... pass
def _save_content(self, work_id: str, content: str, format: str): """Save work content in specified format""" work_path = self.work_dir / work_id if format == "html": output_file = work_path / "work.html" elif format == "txt": output_file = work_path / "work.txt" elif format == "epub": output_file = work_path / "work.epub" else: raise ValueError(f"Unsupported format: {format}") with open(output_file, 'w', encoding='utf-8') as f: f.write(content) <!-- templates/index.html --> <!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <title>AO3 Mirror Tool</title> <style> * { margin: 0; padding: 0; box-sizing: border-box; } body { font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, sans-serif; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); min-height: 100vh; padding: 20px; } .container { max-width: 1200px; margin: 0 auto; } .card { background: white; border-radius: 15px; padding: 30px; margin-bottom: 20px; box-shadow: 0 10px 30px rgba(0,0,0,0.2); } h1 { color: #333; margin-bottom: 10px; } .subtitle { color: #666; margin-bottom: 30px; } .input-group { margin-bottom: 20px; } label { display: block; margin-bottom: 8px; color: #555; font-weight: 500; } input, select { width: 100%; padding: 12px; border: 2px solid #e0e0e0; border-radius: 8px; font-size: 16px; transition: border-color 0.3s; } input:focus { outline: none; border-color: #667eea; } button { background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; border: none; padding: 12px 24px; border-radius: 8px; font-size: 16px; cursor: pointer; transition: transform 0.2s; } button:hover { transform: translateY(-2px); } .queue-item { background: #f8f9fa; border-left: 4px solid #667eea; padding: 15px; margin-bottom: 10px; border-radius: 8px; } .status { display: inline-block; padding: 4px 12px; border-radius: 20px; font-size: 12px; font-weight: 500; } .status.pending { background: #fff3cd; color: #856404; } .status.completed { background: #d4edda; color: #155724; } .status.failed { background: #f8d7da; color: #721c24; } .progress-bar { width: 100%; height: 6px; background: #e0e0e0; border-radius: 3px; overflow: hidden; margin-top: 10px; } .progress-fill { height: 100%; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); transition: width 0.3s; } .library-grid { display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px; margin-top: 20px; } .work-card { background: #f8f9fa; border-radius: 10px; padding: 15px; cursor: pointer; transition: transform 0.2s, box-shadow 0.2s; } .work-card:hover { transform: translateY(-3px); box-shadow: 0 5px 15px rgba(0,0,0,0.1); } .work-title { font-weight: 600; color: #333; margin-bottom: 8px; } .work-author { color: #666; font-size: 14px; margin-bottom: 8px; } .work-stats { display: flex; gap: 15px; font-size: 12px; color: #888; } .modal { display: none; position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(0,0,0,0.5); justify-content: center; align-items: center; z-index: 1000; } .modal-content { background: white; border-radius: 15px; padding: 30px; max-width: 800px; max-height: 80vh; overflow-y: auto; } @keyframes spin { 0% { transform: rotate(0deg); } 100% { transform: rotate(360deg); } } .spinner { border: 3px solid #f3f3f3; border-top: 3px solid #667eea; border-radius: 50%; width: 40px; height: 40px; animation: spin 1s linear infinite; margin: 20px auto; } </style> </head> <body> <div class="container"> <div class="card"> <h1>📚 AO3 Mirror Tool</h1> <p class="subtitle">Archive works for offline reading with full metadata preservation</p>
with open(work_path / 'metadata.json', 'r', encoding='utf-8') as f: metadata = json.load(f)
mirror = AO3Mirror()