'''
Set as cron or task for times of allowing and not allowing user access to server.
Unsharing will kill any current stream from user before unsharing.

Share or unshare libraries.

optional arguments:
  -h, --help            show this help message and exit
  -s [], --share []     To share or to unshare.:
                        (choices: share, share_all, unshare, unshare_all)
  -u [], --user []      Space separated list of case sensitive names to process. Allowed names are:
                        (choices: All users names)
  -l  [ ...], --libraries  [ ...]
                        Space separated list of case sensitive names to process. Allowed names are:
                        (choices: All library names)
                        (default: All Libraries)

Usage:

   plex_api_share.py -s share -u USER -l Movies
       - Shared libraries: ['Movies'] with USER

   plex_api_share.py -s share -u USER -l Movies "TV Shows"
       - Shared libraries: ['Movies', 'TV Shows'] with USER
          * Double Quote libraries with spaces

   plex_api_share.py -s share_all -u USER
       - Shared all libraries with USER.

   plex_api_share.py -s unshare -u USER -l Movies
       - Unshared libraries: ['Movies'] from USER

   plex_api_share.py -s unshare_all -u USER
       - Kill users current stream.
       - Unshared all libraries with USER.
       - USER is still exists as a Friend or Home User

'''


import argparse
from plexapi.server import PlexServer


PLEX_URL = 'http://localhost:32400'
PLEX_TOKEN = 'xxxxx'
plex = PlexServer(PLEX_URL, PLEX_TOKEN)

user_lst = [x.title for x in plex.myPlexAccount().users()]
sections_lst = [x.title for x in plex.library.sections()]

MESSAGE = "GET TO BED!"


def share(user, libraries):
    plex.myPlexAccount().updateFriend(user=user, server=plex, sections=libraries)
    if not libraries:
        print('Shared all libraries with {user}.'.format(user=user))
    else:
        print('Shared libraries: {libraries} with {user}.'.format(libraries=libraries, user=user))


def unshare(user, libraries):
    plex.myPlexAccount().updateFriend(user=user, server=plex, remove_sections=True)
    print('Unshared libraries: {libraries} from {user}.'.format(libraries=libraries, user=user))


def kill_session(user, libraries):
    for session in plex.sessions():
        # Check for users stream and if user is watching a to be unshared library.
        # If to be unshared library is not being watched then stream is not killed.
        if session.usernames[0] in user and session.librarySectionID != plex.library.section(libraries).key:
            title = (session.grandparentTitle + ' - ' if session.type == 'episode' else '') + session.title
            print('{user} is watching {title} and it\'s past their bedtime. Killing stream.'.format(
                user=user, title=title))
            session.stop(reason=MESSAGE)


if __name__ == "__main__":

    parser = argparse.ArgumentParser(description="Share or unshare libraries.",
                                     formatter_class=argparse.RawTextHelpFormatter)
    parser.add_argument('-s', '--share', nargs='?', type=str, required=True, choices=['share', 'unshare'], metavar='',
                        help='To share or to unshare.: \n'
                             '(choices: %(choices)s)')
    parser.add_argument('-u', '--user', nargs='?', type=str, required=True, choices=user_lst, metavar='',
                        help='Space separated list of case sensitive names to process. Allowed names are: \n'
                             '(choices: %(choices)s)')
    parser.add_argument('-l', '--libraries', nargs='+', default='', choices=sections_lst, metavar='',
                        help='Space separated list of case sensitive names to process. Allowed names are: \n'
                             '(choices: %(choices)s \n(default: All Libraries)')

    opts = parser.parse_args()

    if opts.share == 'share':
        share(opts.user, opts.libraries)
    elif opts.share == 'share_all':
        unshare(opts.user, sections_lst)
    elif opts.share == 'unshare':
        kill_session(opts.user, opts.libraries)
        unshare(opts.user, opts.libraries)
    elif opts.share == 'unshare_all':
        kill_session(opts.user, '')
        unshare(opts.user, sections_lst)
    else:
        print('I don\'t know what else you want.')