import ctypes
import os
from uuid import uuid4

GS_OK = 0
GS_FAILED = -1
GS_OUT_OF_MEMORY = -2
GS_INVALID = -3
GS_WRONG_STATE = -4
GS_IO_ERROR = -5
GS_NOT_SUPPORTED_4K = -6

class SERVER_DATA(ctypes.Structure):
    _fields_ = [("address", ctypes.c_char_p),
                ("gpuType", ctypes.c_char_p),
                ("gfeVersion", ctypes.c_char_p),
                ("paired", ctypes.c_bool),
                ("supports4K", ctypes.c_bool),
                ("currentGame", ctypes.c_int),
                ("serverMajorVersion", ctypes.c_int)]


class APP_LIST(ctypes.Structure):
    pass

APP_LIST._fields_ = [("name", ctypes.c_char_p),
                     ("id", ctypes.c_int),
                     ("next", ctypes.POINTER(APP_LIST))]

class _HTTP_DATA(ctypes.Structure):
    _fields_ = [("memory", ctypes.POINTER(ctypes.c_ubyte)),
                ("size", ctypes.c_size_t)]

class LibGameStream:
    def __init__(self, libpath = ""):
        self.commonlib = ctypes.cdll.LoadLibrary(os.path.join(libpath, "libmoonlight-common.so.0"))
        self.gslib = ctypes.cdll.LoadLibrary(os.path.join(libpath, "libgamestream.so.0"))
        self.connected = False
        self.address = ""
        self.key_dir = ""
    
    def discover_server(self):
        addr = ctypes.create_string_buffer('\000' * 40)
        self.gslib.gs_discover_server(addr)
        return addr.value
    
    def connect_server(self, address, key_dir = ""):
        self.server = ctypes.pointer(SERVER_DATA(address, False, False, 0, 0))
        self.address = address
        
        if key_dir == "":
            if "XDG_CONFIG_DIR" in os.environ:
                key_dir = os.path.join(os.environ["XDG_CONFIG_DIR"], "moonlight")
            else:
                key_dir = os.path.join(os.environ["HOME"], ".cache", "moonlight")
        
        self.key_dir = key_dir        
        ret = self.gslib.gs_init(self.server, ctypes.c_char_p(key_dir))
        if ret == GS_OK:
            self.connected = True
            return True
        return False
    
    def isPaired(self):
        if not self.connected:
            return False
        
        return self.server[0].paired
    
    def applist(self):
        if not self.connected:
            return None
        
        lst = []
        applst_ptr = ctypes.POINTER(APP_LIST)
        applst = applst_ptr()
        
        ret = self.gslib.gs_applist(self.server, ctypes.byref(applst))
        if ret != GS_OK:
            return None
        
        while applst:
            lst.append((applst[0].id, applst[0].name))
            applst = applst[0].next
        
        return lst
        
    def poster(self, appId, toFolder):
        unique_id = ""        
        with open(os.path.join(self.key_dir, "uniqueid.dat"), "r") as f:
            unique_id = f.read()
        
        uid = uuid4()
        url = "https://%s:47984/appasset?uniqueid=%s&uuid=%s&appid=%d&AssetType=2&AssetIdx=0" % (self.address, unique_id, str(uid), appId)
        
        self.gslib.http_create_data.restype = ctypes.POINTER(_HTTP_DATA)
        data = self.gslib.http_create_data();
        self.gslib.http_request(ctypes.c_char_p(url), data)
        
        barray = bytearray(data[0].memory[0:data[0].size])
        
        with open(os.path.join(toFolder, str(appId) + ".png"), "wb") as f:
            f.write(barray)
        
        self.gslib.http_free_data(data);
            
    def pair(self, pin):
        if not self.connected:
            return None
        
        ret = self.gslib.gs_pair(self.server, ctypes.c_char_p(pin))
        return ret == GS_OK