import win32com.client
import pywintypes
import os
import shutil
import http_server_39 as server
#import http.server as server
import socketserver
import threading
import asyncio
import websockets
import logging, json

logging.basicConfig()

powerpoint = None
cache = r'''C:\Windows\Temp'''

class Handler(server.CGIHTTPRequestHandler):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, directory=os.path.dirname(os.path.realpath(__file__)))

def run_http():
    http_server = server.HTTPServer(("", 80), Handler)
    http_server.serve_forever()

async def first(websocket, path):
    slideshow_view_first()
    await websocket.send(True)

STATE = {"value": "?", "visible": "1"}

USERS = set()


def state_event():
    return json.dumps({"type": "state", **STATE})


def users_event():
    return json.dumps({"type": "users", "count": len(USERS)})


async def notify_state():
    STATE["value"] = str(current_slide()) + "/" + str(total_slides())
    if USERS:  # asyncio.wait doesn't accept an empty list
        message = state_event()
        await asyncio.wait([user.send(message) for user in USERS])


async def notify_users():
    if USERS:  # asyncio.wait doesn't accept an empty list
        message = users_event()
        await asyncio.wait([user.send(message) for user in USERS])


async def register(websocket):
    USERS.add(websocket)
    await notify_users()


async def unregister(websocket):
    USERS.remove(websocket)
    await notify_users()


async def ws_handle(websocket, path):
    # register(websocket) sends user_event() to websocket
    await register(websocket)
    try:
        await websocket.send(state_event())
        async for message in websocket:
            data = json.loads(message)
            if data["action"] == "prev":
                slideshow_view_previous()
                await notify_state()
            elif data["action"] == "next":
                slideshow_view_next()
                await notify_state()
            elif data["action"] == "first":
                slideshow_view_first()
                await notify_state()
            elif data["action"] == "last":
                slideshow_view_last()
                await notify_state()
            else:
                logging.error("unsupported event: {}", data)
    finally:
        await unregister(websocket)

def run_ws():
    start_server = websockets.serve(ws_handle, "0.0.0.0", 5678)
    asyncio.get_event_loop().run_until_complete(start_server)
    asyncio.get_event_loop().run_forever()

def start_server():
    STATE["value"] = current_slide()
    http_daemon = threading.Thread(name="http_daemon", target=run_http)
    http_daemon.setDaemon(True)
    http_daemon.start()

    run_ws()
    #ws_daemon = threading.Thread(name="ws_daemon", target=run_ws)
    #ws_daemon.setDaemon(True)
    #ws_daemon.start()

    #try:
    #    ws_daemon.start()
    #    http_daemon.start()
    #except (KeyboardInterrupt, SystemExit):
    #    cleanup_stop_thread()
    #    sys.exit()


def get_slideshow_view():
    global powerpoint

    if powerpoint is None:
        powerpoint = win32com.client.Dispatch('Powerpoint.Application')

    if powerpoint is None:
        return

    ssw = powerpoint.SlideShowWindows
    if ssw.Count == 0:
        return

    # https://docs.microsoft.com/en-us/office/vba/api/powerpoint.slideshowwindow.view
    ssv = ssw[0].View

    return ssv

def get_activepresentation():
    global powerpoint

    if powerpoint is None:
        powerpoint = win32com.client.Dispatch('Powerpoint.Application')

    if powerpoint is None:
        return

    activepres = powerpoint.ActivePresentation
    return activepres

def total_slides():
    ssp = get_activepresentation()
    if ssp:
        return len(ssp.Slides)

def current_slide():
    ssv = get_slideshow_view()
    if ssv:
        return ssv.CurrentShowPosition

def export(slide):
    global cache
    ssp = get_activepresentation()
    if ssp:
        for (slide, name) in [(slide, "current"), (slide+1, "next")]:
            if slide < len(ssp.Slides):
                ssp.Slides(slide).Export(os.path.dirname(os.path.realpath(__file__)) + r'''\\''' + name + r'''0.jpg''', "JPG")
                attempts = 0
                while attempts < 3:
                    try:
                        os.replace(os.path.dirname(os.path.realpath(__file__)) + r'''\\''' + name + r'''0.jpg''', os.path.dirname(os.path.realpath(__file__)) + r'''\\''' + name + '''.jpg''')
                    except:
                        pass
                    attempts += 1
            else:
                    shutil.copyfileobj(open(os.path.dirname(os.path.realpath(__file__)) + r'''\blank.jpg''', 'rb'), open(os.path.dirname(os.path.realpath(__file__)) + r'''\\''' + name + r'''next.jpg''', 'wb'))

def slideshow_view_first():
    ssv = get_slideshow_view()
    if ssv:
        ssv.First()
        export(ssv.CurrentShowPosition)

def slideshow_view_previous():
    ssv = get_slideshow_view()
    if ssv:
        ssv.Previous()
        export(ssv.CurrentShowPosition)

def slideshow_view_next():
    ssv = get_slideshow_view()
    if ssv:
        ssv.Next()
        export(ssv.CurrentShowPosition)
            

def slideshow_view_last():
    ssv = get_slideshow_view()
    if ssv:
        ssv.Last()
        export(ssv.CurrentShowPosition)

def slideshow_view_black():
    ssv = get_slideshow_view()
    if ssv:
        ssv.State = 3

def slideshow_view_white():
    ssv = get_slideshow_view()
    if ssv:
        ssv.State = 4

def slideshow_view_normal():
    ssv = get_slideshow_view()
    if ssv:
        ssv.State = 1

if __name__ == "__main__":
    start_server()