import base64
from . import pyproto

from dataclasses import dataclass
from typing import Optional

from pure_protobuf.dataclasses_ import field, message
from pure_protobuf.types import int64

def b64e(b, padding=True):
    return base64.urlsafe_b64encode(b).decode('ascii') \
            .replace("=", "%3D" if padding else "")

def proto(d, padding=True):
    return base64.urlsafe_b64encode(pyproto.ProtoBuf(d).toBuf()) \
            .decode('ascii').replace("=", "%3D" if padding else "")

# SEARCH {{{
@message
@dataclass
class Extras:
    verbatim: Optional[bool] = field(1, default=None) # don't fix spelling
@message
@dataclass
class Filters: # adapted from invidious
    date:      Optional[int64] = field(1, default=None)
    type:      Optional[int64] = field(2, default=None)
    length:    Optional[int64] = field(3, default=None)
    is_hd:     Optional[bool]  = field(4, default=None)
    subtitles: Optional[bool]  = field(5, default=None)
    ccommons:  Optional[bool]  = field(6, default=None)
    is_3d:     Optional[bool]  = field(7, default=None)
    live:      Optional[bool]  = field(8, default=None)
    purchased: Optional[bool]  = field(9, default=None)
    is_4k:     Optional[bool]  = field(14, default=None)
    is_360:    Optional[bool]  = field(15, default=None)
    location:  Optional[bool]  = field(23, default=None)
    is_hdr:    Optional[bool]  = field(25, default=None)
@message
@dataclass
class SearchRequest:
    sorted: Optional[int64] = field(1, default=None)
    filter: Optional[Filters] = field(2, default=None)
    extras: Optional[Extras] = field(8, default=None)

def make_sp(sort=None, date=None, type=None, len=None, features=[], extras=[]):
    sortorder  = dict(relevance=0, rating=1, date=2, views=3)
    datefilter = dict(hour=1, day=2, week=3, month=4, year=5)
    typefilter = dict(video=1, channel=2, playlist=3, movie=4, show=5)
    lenfilter  = dict(short=1, long=2)

    return b64e(SearchRequest(
        sorted=sortorder.get(sort),
        filter=Filters(
            date=datefilter.get(date),
            type=typefilter.get(type),
            length=lenfilter.get(len),
            **{f:True for f in features},
        ) if date or type or len or features else None,
        extras=Extras(**{f:True for f in extras}),
    ).dumps())
# }}} SEARCH

# CHANNEL {{{
def make_channel_params(subject, typ="videos", sort=None, query=None):
    if typ in ("playlists",):
        sortorder = dict(newest=3, modified=4)
        return proto({
            # Continuation
            80226972: {#params
                # Params
                2: subject,#subject
                3: proto({#params
                    # Subparams
                    2: "playlists",#type_s
                    4: 1,#type_i
                    #15: None,#page
                    3: sortorder.get(sort),#sort optional
                    7: 1, #unknown_const1
                    23: 0, #unknown_const2
                    6: 2, #list_or_grid (2=list, None/1=grid)
                    61: proto({#field61 optional
                        # ChannelDataContainer
                        1: proto({#data
                            #ChannelDataInner
                            1: 0#offset
                        }, padding=False)
                    }, padding=False)
                }),
                #11: query or None,#query
            }
        })
    elif typ in ("search",):
        return proto({
            # Continuation
            80226972: {#params
                # Params
                2: subject,#subject
                3: proto({#params
                    # Subparams
                    2: "search",#type_s
                    #4: None,#type_i
                    15: proto({#page optional
                        # SearchOffset
                        3: 0,#offset
                    }, padding=False) if 0 else None,
                    #3: None,#sort optional
                    7: 1, #unknown_const1
                    23: 0, #unknown_const2
                    6: 2, #list_or_grid (2=list, None/1=grid)
                    #61: None#field61 optional
                }),
                11: query,#query
            }
        })
    elif typ in ("videos", "streams", "shorts"):
        sortorder = dict(newest=1, popular=2)
        typekey = dict(videos=15, streams=14, shorts=10)
        return proto({
            # Continuation
            80226972: {#params
                # Params
                2: subject,#subject
                3: proto({#params
                    # NewChannelDataContainerOuter
                    110: {#data
                        # NewChannelDataContainer
                        3: {#data
                            # NewChannelDataInner
                            typekey[typ]: {
                                # NewChannelDataSort
                                2: {#uuid
                                    # NewChannelDataUuid
                                    1: "00000000-0000-0000-0000-000000000000",#uuid
                                },
                                3: sortorder.get(sort,1)#sort
                            },
                        }
                    }
                })
            }
        })
    else:
        raise NotImplementedError
# }}} CHANNEL

# PLAYLIST {{{
@message
@dataclass
class PlaylistData:
    offset: Optional[int64] = field(1)  # normal playlists
    after: Optional[str] = field(2)  # mix playlist (RDCL*, RDCM*): video id of last loaded item (how to get first page!?)
@message
@dataclass
class PlaylistSubparams:
    data: str = field(15)
def make_playlist_params(playlist_id, offset):
    # TODO: special type of playlists: (see https://github.com/nlitsme/youtube_tool)
    # - PL*: normal playlists
    # - special channel playlists (suffix=channel_id without UC):
    #   - UU*: newest uploads (works)
    #   - PU*: popular uploads (works)
    #     Note: youtubekids (e.g. PLMYEtVRpaqY00V9W81Cwmzp6N6vZqfUKD4) fallback-only (youtube webui 404's)
    #   - FL*: favourited uploads (works; 404's if private)
    #   - LL*: liked uploads (untested)
    #   - RDCMUC*: mix for channel (fallback only; e.g. RDCMUCYO_jab_esuFRV4b17AJtAw, undisplayable on youtube (only works on watch))
    #   - CL, EL, MQ, TT: unknown
    # - OLAK5uy_ music album playlist (works, e.g. OLAK5uy_m4xAFdmMC5rX3Ji3g93pQe3hqLZw_9LhM)
    # - RD*: mix playlist (works with after=; e.g. RDCLAK5uy_kLWIr9gv1XLlPbaDS965-Db4TrBoUTxQ8, RDCLAK5uy_mx8mvGaCkyI12AedAu8yIbBpEYmg6Ayow, RDCLAK5uy_n340faD9pPGRVEVwozYTzIpfXuqmQwcgA, RDCLAK5uy_kQmVjdr90p2onIIAnijgU8u9iUUAOXM2I)
    # https://www.youtube.com/watch?v=FTQbiNvZqaY&list=RDCLAK5uy_khNGopKCT_t38MZ1W7z4kERrqprkXovxo&start_radio=1
    # - EC, UL, TL: unknown
    # - own user playlists (obviously not working):
    #   - RDMM: my mix
    #   - WL: watch later
    #   - LL: own liked videos
    #   - LM: own liked youtube-music videos
    after = None
    if playlist_id.startswith("RD"):
        # NOTE: this only fixes RDCLAK5uy_, not RDCMUC
        after = '___________'  # any video id that doesn't exist within the playlist requests the beginning
        offset = None

    return b64e(Continuation(
        params=Params(
            subject="VL" + playlist_id,
            params=b64e(PlaylistSubparams(
                data="PT:" + b64e(PlaylistData(
                    offset=offset,
                    after=after,
                ).dumps(), padding=False)
            ).dumps()),
        ),
    ).dumps())
# }}} PLAYLIST