mirror of
https://github.com/Viren070/mediaflow-proxy.git
synced 2025-12-01 23:22:12 +01:00
mhdzumair
65 lines
3.1 KiB
Python
65 lines
3.1 KiB
Python
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field, IPvAnyAddress, ConfigDict
|
|
|
|
|
|
class GenerateUrlRequest(BaseModel):
|
|
mediaflow_proxy_url: str = Field(..., description="The base URL for the mediaflow proxy.")
|
|
endpoint: str | None = Field(None, description="The specific endpoint to be appended to the base URL.")
|
|
destination_url: str | None = Field(None, description="The destination URL to which the request will be proxied.")
|
|
query_params: dict | None = Field(
|
|
default_factory=dict, description="Query parameters to be included in the request."
|
|
)
|
|
request_headers: dict | None = Field(default_factory=dict, description="Headers to be included in the request.")
|
|
response_headers: dict | None = Field(default_factory=dict, description="Headers to be included in the response.")
|
|
expiration: int | None = Field(
|
|
None, description="Expiration time for the URL in seconds. If not provided, the URL will not expire."
|
|
)
|
|
api_password: str | None = Field(
|
|
None, description="API password for encryption. If not provided, the URL will only be encoded."
|
|
)
|
|
ip: IPvAnyAddress | None = Field(None, description="The IP address to restrict the URL to.")
|
|
|
|
|
|
class GenericParams(BaseModel):
|
|
model_config = ConfigDict(populate_by_name=True)
|
|
|
|
|
|
class HLSManifestParams(GenericParams):
|
|
destination: str = Field(..., description="The URL of the HLS manifest.", alias="d")
|
|
key_url: str | None = Field(
|
|
None,
|
|
description="The HLS Key URL to replace the original key URL. Defaults to None. (Useful for bypassing some sneaky protection)",
|
|
)
|
|
|
|
|
|
class ProxyStreamParams(GenericParams):
|
|
destination: str = Field(..., description="The URL of the stream.", alias="d")
|
|
|
|
|
|
class MPDManifestParams(GenericParams):
|
|
destination: str = Field(..., description="The URL of the MPD manifest.", alias="d")
|
|
key_id: str | None = Field(None, description="The DRM key ID (optional).")
|
|
key: str | None = Field(None, description="The DRM key (optional).")
|
|
|
|
|
|
class MPDPlaylistParams(GenericParams):
|
|
destination: str = Field(..., description="The URL of the MPD manifest.", alias="d")
|
|
profile_id: str = Field(..., description="The profile ID to generate the playlist for.")
|
|
key_id: str | None = Field(None, description="The DRM key ID (optional).")
|
|
key: str | None = Field(None, description="The DRM key (optional).")
|
|
|
|
|
|
class MPDSegmentParams(GenericParams):
|
|
init_url: str = Field(..., description="The URL of the initialization segment.")
|
|
segment_url: str = Field(..., description="The URL of the media segment.")
|
|
mime_type: str = Field(..., description="The MIME type of the segment.")
|
|
key_id: str | None = Field(None, description="The DRM key ID (optional).")
|
|
key: str | None = Field(None, description="The DRM key (optional).")
|
|
|
|
|
|
class ExtractorURLParams(GenericParams):
|
|
host: Literal["Doodstream", "Mixdrop", "Uqload"] = Field(..., description="The host to extract the URL from.")
|
|
destination: str = Field(..., description="The URL of the stream.", alias="d")
|
|
redirect_stream: bool = Field(False, description="Whether to redirect to the stream endpoint automatically.")
|