import asyncio
import willump

async def get_summoner_data():
    summoner = await (await wllp.request("GET", "/lol-summoner/v1/current-summoner")).json()
    print(f"summonerName:    {summoner['displayName']}")
    print(f"summonerLevel:   {summoner['summonerLevel']}")
    print(f"profileIconId:   {summoner['profileIconId']}")
    print(f"summonerId:      {summoner['summonerId']}")
    print(f"puuid:           {summoner['puuid']}")
    print(f"---")


async def main():
    global wllp
    wllp = await willump.start()
    await get_summoner_data()
    await wllp.close()

if __name__ == '__main__':
  loop = asyncio.get_event_loop()
  loop.run_until_complete(main())
  loop.close()

from lcu_driver import Connector
connector = Connector()

@connector.ready
async def connect(connection):
    print(connection.address)
    print('LCU API is ready to be used.')

@connector.close
async def disconnect(connection):
    print('The client was closed')
    await connector.stop()

@connector.ws.register('/lol-lobby/v2/lobby', event_types=('CREATE',))
async def icon_changed(connection, event):
    print(f"The summoner {event.data['localMember']['summonerName']} created a lobby.")

connector.start()

from lcu_driver import Connector
connector = Connector()

async def get_summoner_data(connection):
  summoner = await connection.request('GET', '/lol-summoner/v1/current-summoner')
  data = await summoner.json()
  print(f"summonerName:    {data['displayName']}")
  print(f"summonerLevel:   {data['summonerLevel']}")
  print(f"profileIconId:   {data['profileIconId']}")
  print(f"summonerId:      {data['summonerId']}")
  print(f"puuid:           {data['puuid']}")
  print(f"---")

@connector.ready
async def connect(connection):
    await get_summoner_data(connection)

connector.start()

async def createLobby(connection):
  queue = {'queueId': 430}
  await connection.request('POST', '/lol-lobby/v2/lobby', data=queue)

async def creatCustomLabby(connection):
  # 房间数据
  LobbyConfig = {
    'customGameLobby': {
      'configuration': {
        'gameMode': 'PRACTICETOOL', 
        'gameMutator': '', 
        'gameServerRegion': '', 
        'mapId': 11, 
        'mutators': {'id': 1}, 
        'spectatorPolicy': 'AllAllowed', 
        'teamSize': 5
      },
      'lobbyName': 'PRACTICETOOL',
      'lobbyPassword': ''
    },
    'isCustom': True
  }

  # 发送创建房间请求
  await connection.request('POST', '/lol-lobby/v2/lobby', data=LobbyConfig)

bots = {
  'championId': 25,
  'botDifficulty': 'MEDIUM',
  'teamId': '200'
}
await connection.request('POST', '/lol-lobby/v1/lobby/custom/bots', data=bots)

async def addBots(connection):
  team2 = [122, 86, 1, 51, 25]
  for id in team2:
    bots = { 'championId': id, 'botDifficulty': 'MEDIUM', 'teamId': '200' }
    await connection.request('POST', '/lol-lobby/v1/lobby/custom/bots', data=bots)

async def addBots(connection):
  # 获取可用的机器人列表
  activedata = await connection.request('GET', '/lol-lobby/v2/lobby/custom/available-bots')
  champions = { bot['name']: bot['id'] for bot in await activedata.json() }
  
  # 队伍2的机器人
  team2 = ['诺克萨斯之手', '德玛西亚之力', '曙光女神', '皮城女警', '众星之子']
  
  for name in team2:
    bots = { 'championId': champions[name], 'botDifficulty': 'MEDIUM', 'teamId': '200' }
    await connection.request('POST', '/lol-lobby/v1/lobby/custom/bots', data=bots)

data = await connection.request('GET', '/lol-lobby/v2/lobby/custom/available-bots')
champions = {bots['name']: bots['id'] for bots in await data.json()}
print(champions)

data = await connection.request('GET', '/riotclient/get_region_locale')
print(await data.json())

connection.address

path = await connection.request('GET', '/data-store/v1/install-dir')
print(path)

connection.installation_path

path.encode('gbk').decode('utf-8')

client_path = connection.installation_path.encode('gbk').decode('utf-8')
lockfile_path = os.path.join(client_path, 'lockfile')
print(lockfile_path)

def get_lockfile(path):
  # 读取数据
  if os.path.isfile(path):
    file = open(path, 'r')
    text = file.readline().split(':')
    file.close()
    print(f"riot    {text[3]}")
    return text
  else:
    print(f"{path} \n lockfile 文件不存在")
  return None

data = await connection.request('GET', '/lol-game-queues/v1/queues')
print(await data.json())

id = 900
data = await connection.request('GET', f"/lol-game-queues/v1/queues/{id}")
print(await data.json())

queueType = 'URF'
data = await connection.request('GET', f"/lol-game-queues/v1/queues/type/{queueType}")
print(await data.json())

data = await connection.request('GET', '/lol-maps/v1/maps')
print(await data.json())

data = await connection.request('GET', '/lol-lobby/v2/lobby/members')
print(await data.json())

profileIconId = 4804
image = await connection.request('GET', f"/lol-game-data/assets/v1/profile-icons/{profileIconId}.jpg")
raw = base64.b64encode(await image.read())
print(raw)

data = await connection.request('GET', f"/lol-champions/v1/inventories/{summonerData['summonerId']}/champions")
# print(data)

def CheckProcess():
    process = psutil.pids()
    for pid in process:
        if psutil.Process(pid).name() == 'LeagueClient.exe':
            return pid
    else:
        return None

if CheckProcess() == None:
  QMessageBox.warning(Window(),'LeagueCustomLobby', '无法读取游戏数据\n需要先启动游戏，再运行此程序。')
  return