Files
ohm_streaming/app/downloaders/__init__.py
T
root 1fe7392063 feat: Complete Sonarr integration with security enhancements
This commit adds comprehensive Sonarr webhook integration and implements
critical security improvements identified in code review.

## Sonarr Integration
- Full webhook support for Grab, Download, Rename, Delete, and Test events
- HMAC SHA256 signature verification for webhook authentication
- Series mapping system (Sonarr TVDB ID → Anime Provider URL)
- 11 new API endpoints for configuration, mappings, search, and downloads
- Comprehensive test suite (31 tests, all passing)
- Complete documentation in docs/SONARR_INTEGRATION.md

## Security Enhancements
- CORS restricted to specific origins (user's IP: 192.168.1.204:3000)
- Path traversal prevention via sanitize_filename() and is_safe_filename()
- Structured logging infrastructure (replaced all print() statements)
- Environment-based configuration with .env support
- Filename sanitization prevents malicious path attacks

## New Features
- Lpayer and Sibnet downloader support
- Kitsu API integration for anime metadata
- Recommendation engine based on download history
- Latest releases endpoint for new anime
- Modular web interface with component-based templates

## Configuration
- Centralized settings via app/config.py with pydantic-settings
- Sonarr config auto-created in config/ directory
- Example configurations provided for easy setup

## Tests
- 31 Sonarr integration tests (23 functionality + 9 security)
- 100+ tests passing in core test files
- Security utilities fully tested

## Documentation
- Updated CLAUDE.md with Sonarr and testing info
- Added IMPROVEMENTS_2024-01-24.md analysis
- Added SONARR_IMPLEMENTATION.md technical summary

Generated with [Claude Code](https://claude.ai/code)
via [Happy](https://happy.engineering)

Co-Authored-By: Claude <noreply@anthropic.com>
Co-Authored-By: Happy <yesreply@happy.engineering>
2026-01-24 21:25:47 +00:00

53 lines
1.6 KiB
Python

from .base import BaseDownloader
from .unfichier import UnFichierDownloader
from .doodstream import DoodStreamDownloader
from .rapidfile import RapidFileDownloader
from .uptobox import UptoboxDownloader
from .animesama import AnimeSamaDownloader
from .animeultime import AnimeUltimeDownloader
from .nekosama import NekoSamaDownloader
from .vostfree import VostfreeDownloader
from .vidmoly import VidMolyDownloader
from .sendvid import SendVidDownloader
from .sibnet import SibnetDownloader
from .lpayer import LpayerDownloader
def get_downloader(url: str) -> BaseDownloader:
"""Factory function to get the appropriate downloader for a URL"""
downloaders = [
# Anime sites
AnimeSamaDownloader(),
AnimeUltimeDownloader(),
NekoSamaDownloader(),
VostfreeDownloader(),
# File hosts
UnFichierDownloader(),
UptoboxDownloader(),
DoodStreamDownloader(),
RapidFileDownloader(),
VidMolyDownloader(),
SendVidDownloader(),
SibnetDownloader(),
LpayerDownloader(),
]
for downloader in downloaders:
if downloader.can_handle(url):
return downloader
# Return generic downloader if no match
return GenericDownloader()
class GenericDownloader(BaseDownloader):
"""Generic downloader for unhandled hosts"""
def can_handle(self, url: str) -> bool:
return True
async def get_download_link(self, url: str) -> tuple[str, str]:
# Just return the URL as-is
filename = url.split('/')[-1] or "download"
return url, filename