from __future__ import print_function

import datetime
import os.path
import argparse
import json

from google.auth.transport.requests import Request
from google.oauth2.credentials import Credentials
from google_auth_oauthlib.flow import InstalledAppFlow
from googleapiclient.discovery import build

SCOPES = [
    'https://www.googleapis.com/auth/calendar.readonly',
    "https://www.googleapis.com/auth/youtube"
]

def authorize():
    creds = None

    if os.path.exists('token.json'):
        creds = Credentials.from_authorized_user_file('token.json', SCOPES)

    if not creds or not creds.valid:
        if creds and creds.expired and creds.refresh_token:
            creds.refresh(Request())
        else:
            flow = InstalledAppFlow.from_client_secrets_file('credentials.json', SCOPES)
            creds = flow.run_local_server(port=7384)

        with open('token.json', 'w') as token:
            token.write(creds.to_json())

    return creds

def construct_calendar_service(creds):
    return build('calendar', 'v3', credentials=creds)

def list_all_calendars(calendar_service):
    page_token = None
    list_of_entries = []

    while True:
        calendar_list = calendar_service.calendarList().list(pageToken=page_token).execute()
        
        list_of_entries.extend(calendar_list['items'])

        page_token = calendar_list.get('nextPageToken')

        if not page_token:
            break

    return list_of_entries

def get_calendar_events(calendar_service, calendar_id, num_events):
    now = datetime.datetime.utcnow().isoformat() + "Z"

    events_result = calendar_service.events().list(
        calendarId=calendar_id,
        timeMin=now,
        maxResults=num_events,
        singleEvents=True, 
        orderBy='startTime'
    ).execute()

    events = events_result.get('items', [])

    return events

def construct_youtube_service(creds):
    return build('youtube', 'v3', credentials=creds)

def list_all_stream_sources(youtube_service):
    page_token = None
    list_of_entries = []

    while True:
        stream_list = youtube_service.liveStreams().list(
            part="snippet",
            mine=True,
            pageToken=page_token
        ).execute()

        list_of_entries.extend(stream_list["items"])

        page_token = stream_list.get("nextPageToken")

        if not page_token:
            break

    return list_of_entries

def list_all_video_categories(youtube_service):
    category_list = youtube_service.videoCategories().list(
        part="snippet",
        regionCode="US"
    ).execute()

    return [
        {
            "name": element["snippet"]["title"],
            "id": element["id"]
        }
        for element in category_list["items"] if element["snippet"]["assignable"]
    ]

def list_all_live_broadcasts(youtube_service, broadcastStatus):
    page_token = None
    list_of_entries = []

    while True:
        broadcast_list = youtube_service.liveBroadcasts().list(
            part="snippet",
            broadcastStatus=broadcastStatus,
            pageToken=page_token
        ).execute()

        list_of_entries.extend(broadcast_list["items"])

        page_token = broadcast_list.get("nextPageToken")

        if not page_token:
            break

    return list_of_entries

def main():
    with open("config.json", "r") as config_file:
        config_json = json.loads(config_file.read())

    creds = authorize()

    calendar_service = construct_calendar_service(creds)
    youtube_service = construct_youtube_service(creds)