run.py

from fastapi import FastAPI, HTTPException,Request,Query
from fastapi.responses import JSONResponse
from filmpertutti import filmpertutti
from streamingcommunity import streaming_community
from tantifilm import tantifilm
from lordchannel import lordchannel
from streamingwatch import streamingwatch
import json
import config
import logging
from okru import okru_get_url
from animeworld import animeworld
from dictionaries import okru,STREAM,extra_sources,webru_vary,webru_dlhd
from epg import tivu, tivu_get,epg_guide,convert_bho_1,convert_bho_2
from webru import webru
import httpx
# Configure logging
FILMPERTUTTI = config.FILMPERTUTTI
STREAMINGCOMMUNITY = config.STREAMINGCOMMUNITY
MYSTERIUS = config.MYSTERIUS
TUTTIFILM = config.TUTTIFILM
TF_DOMAIN = config.TF_DOMAIN
LORDCHANNEL = config.LORDCHANNEL
STREAMINGWATCH= config.STREAMINGWATCH
ANIMEWORLD = config.ANIMEWORLD
HOST = config.HOST
PORT = int(config.PORT)
HF = config.HF
if HF == "1":
    HF = "🤗️"
    #Cool code to set the hugging face if the service is hosted there.
else:
    HF = ""
if MYSTERIUS == "1":
    from cool import cool

app = FastAPI()
MANIFEST = {
    "id": "org.stremio.mammamia",
    "version": "1.0.5",
    "catalogs": [
        {
            "type": "tv",
            "id": "tv_channels",
            "name": "TV Channels",
            "extra": [
                {
                    "name": "genre",
                    "isRequired": False,
                    "options": ["Rai", "Mediaset", "Sky", "Euronews", "La7", "Warner Bros", "FIT", "Sportitalia","RSI","DAZN", "Rakuten", "Pluto", "A+E", "Paramount", "Chill"]
                }
            ]
        }
    ],
    "resources": ["stream", "catalog", "meta"],
    "types": ["movie", "series", "tv"],
    "name": "Mamma Mia",
    "description": "Addon providing HTTPS Streams for Italian Movies, Series, and Live TV! Note that you need to have Kitsu Addon installed in order to watch Anime",
    "logo": "https://creazilla-store.fra1.digitaloceanspaces.com/emojis/49647/pizza-emoji-clipart-md.png"
}



def respond_with(data):
    resp = JSONResponse(data)
    resp.headers['Access-Control-Allow-Origin'] = '*'
    resp.headers['Access-Control-Allow-Headers'] = '*'
    return resp

@app.get('/manifest.json')
def addon_manifest():
    return respond_with(MANIFEST)

@app.get('/')
def root():
    return "Hello, this is a Stremio Addon providing HTTPS Stream for Italian Movies/Series, to install it add /manifest.json to the url and then add it into the Stremio search bar"

async def addon_catalog(type: str, id: str, genre: str = None):
    print(f"Received genre parameter: {genre}")
    if type != "tv":
        raise HTTPException(status_code=404)
    
    catalogs = {"metas": []}
    
    for channel in STREAM["channels"]:
        if genre and genre not in channel.get("genres", []):
            continue  # Skip channels that don't match the selected genre
        
        description = f'Watch {channel["title"]}'
        catalogs["metas"].append({
            "id": channel["id"],
            "type": "tv",
            "name": channel["title"],
            "poster": channel["poster"],  # Add poster URL if available
            "description": description,
            "genres": channel.get("genres", [])
        })

    return catalogs
@app.get('/catalog/{type}/{id}.json')
async def first_catalog(type: str, id: str, genre: str = None):
    catalogs = await addon_catalog(type, id,genre)
    return respond_with(catalogs)

@app.get('/catalog/{type}/{id}/genre={genre}.json')
async def first_catalog(type: str, id: str, genre: str = None):
    catalogs = await addon_catalog(type, id,genre)
    return respond_with(catalogs)

@app.get('/meta/tv/{id}.json')
async def addon_meta(id: str):
    # Find the channel by ID
    channel = next((ch for ch in STREAM['channels'] if ch['id'] == id), None)
    
    if not channel:
        raise HTTPException(status_code=404, detail="Channel not found")
    if channel["id"] in convert_bho_1 or channel["id"] in convert_bho_2:
            description,title =  await epg_guide(channel["id"])
    else:
        description = f'Watch {channel["title"]}'
        title = ""
    meta = {
        'meta': {
            'id': channel['id'],
            'type': 'tv',
            'name': channel['name'],
            'poster': channel['poster'],
            'posterShape': 'landscape',
            'description': title + "\n" + description,
            # Additional fields can be added here
            'background': channel['poster'],  # Example of using the same poster as background
            'logo': channel['poster'],
            'genres': channel.get('genres', []),  # Example of using the same poster as logo
        }
    }
    if 'url' in channel:
        meta['meta']['url'] = channel['url']  # Using the stream URL as a website link
    return respond_with(meta)


@app.get('/stream/{type}/{id}.json')
async def addon_stream(type, id):
    if type not in MANIFEST['types']:
        raise HTTPException(status_code=404)
    streams = {'streams': []}
    async with httpx.AsyncClient() as client:
        if type == "tv":
            for channel in STREAM["channels"]:
                if channel["id"] == id:
                    i = 0
                    if 'url' in channel:
                        i = i+1
                        streams['streams'].append({
                            'title': f"Server {i} " + f" "+ channel['name'] + " " + channel['title'] ,
                            'url': channel['url']
                            })
                    if id in okru:
                        i = i+1
                        channel_url = await okru_get_url(id,client)
                        streams['streams'].append({
                            'title':  f"Server {i} " +  channel['title'] + " OKRU",
                            'url': channel_url
                        })
                    if id in extra_sources:
                        list_sources = extra_sources[id]
                        for item in list_sources:
                            i = i+1
                            streams['streams'].append({'title':f"Server {i} " + channel['title'],'url': item})
                    if id in webru_vary:
                        i = i+1
                        webru_url = await webru(id,"vary",client)
                        streams['streams'].append({'title': f"Server {i} " + channel['title'],'url': webru_url})
            if not streams['streams']:
                raise HTTPException(status_code=404)
            return respond_with(streams)
        else:
            logging.debug(f"Handling movie or series: {id}")
            if "kitsu" in id:
                if ANIMEWORLD == "1":
                    animeworld_urls = await animeworld(id,client)
                    print(animeworld_urls)
                    if animeworld_urls:
                        i = 0
                        for url in animeworld_urls:
                            if url:
                                if i == 0:
                                    title = "Original"
                                elif i == 1:
                                     title = "Italian"
                                streams['streams'].append({'title': f'{HF}Animeworld {title}', 'url': url})
                                i+=1
            else:
                if MYSTERIUS == "1":
                    results = await cool(id,client)
                    if results:
                        for resolution, link in results.items():
                            streams['streams'].append({'title': f'{HF}Mysterious {resolution}', 'url': link})
                if STREAMINGCOMMUNITY == "1":
                    url_streaming_community,url_720_streaming_community,quality_sc = await streaming_community(id,client)
                    if url_streaming_community is not None:
                        if quality_sc == "1080":
                            streams['streams'].append({'title': f'{HF}StreamingCommunity 1080p Max', 'url': url_streaming_community})
                            streams['streams'].append({'title': f'{HF}StreamingCommunity 720p Max', 'url': url_720_streaming_community})
                        else:
                            streams['streams'].append({'title': f'{HF}StreamingCommunity 720p Max', 'url': url_streaming_community})
                if LORDCHANNEL == "1":
                    url_lordchannel,quality_lordchannel = await lordchannel(id,client)
                    if quality_lordchannel == "FULL HD" and url_lordchannel !=  None:
                        streams['streams'].append({'title': f'{HF}LordChannel 1080p', 'url': url_lordchannel})
                    elif url_lordchannel !=  None:
                        streams['streams'].append({'title': f'{HF}LordChannel 720p', 'url': url_lordchannel})            
                if FILMPERTUTTI == "1":
                    url_filmpertutti = await filmpertutti(id,client)
                    if url_filmpertutti is not None:
                        streams['streams'].append({'title': 'Filmpertutti', 'url': url_filmpertutti})
                if TUTTIFILM == "1":
                    url_tuttifilm = await tantifilm(id,client)
                    if url_tuttifilm:
                        if not isinstance(url_tuttifilm, str):
                            for title, url in url_tuttifilm.items():    
                                streams['streams'].append({'title': f'{HF}Tantifilm {title}', 'url': url,  'behaviorHints': {'proxyHeaders': {"request": {"Referer": "https://d000d.com/"}}, 'notWebReady': True}})
                if STREAMINGWATCH == "1":
                    url_streamingwatch = await streamingwatch(id,client)
                    if url_streamingwatch:
                        streams['streams'].append({'title': f'{HF}StreamingWatch 720p', 'url': url_streamingwatch})
        if not streams['streams']:
            raise HTTPException(status_code=404)

    return respond_with(streams)


if __name__ == '__main__':
    import uvicorn
    uvicorn.run("run:app", host=HOST, port=PORT, log_level="info")