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=False):
    return base64.urlsafe_b64encode(pyproto.ProtoBuf(d).toBuf()) \
            .decode('ascii').replace("=", "%3D" if padding else "")

def continuation(subject, params, query=None):
    return proto({
        80226972: {
            2: subject,
            3: proto(params, padding=True),
            11: query,
        }
    }, padding=True)

# 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 continuation(subject, {
            2: "playlists", #type_s
            4: 1, #type_i
            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({1: proto({1: 0})}) #offset
        })
    elif typ in ("search",):
        return continuation(subject, {
            2: "search",#type_s
            7: 1, #unknown_const1
            23: 0, #unknown_const2
            6: 2, #list_or_grid (2=list, None/1=grid)
        }, query)
    elif typ in ("videos", "streams", "shorts"):
        sortorder = dict(newest=1, popular=2)
        typekey = dict(videos=15, streams=14, shorts=10)
        return continuation(subject, { 110: { 3: {
            typekey[typ]: {
                2: { 1: "00000000-0000-0000-0000-000000000000" },
                3: sortorder.get(sort,1)
            }
        }}})
    else:
        raise NotImplementedError
# }}} CHANNEL

# PLAYLIST {{{
def make_playlist_params(playlist_id, offset):
    # mix takes video id instead of offset. an id not in the playlist requests the beginning.
    mix_playlist = playlist_id.startswith("RD")
    return continuation("VL" + playlist_id, {
        15: "PT:" + proto(
            { 2: '___________' } if mix_playlist else { 1: offset }
        )
    })
# }}} PLAYLIST