import re
import sys

def main():
    with open('dump.txt', 'r', encoding='utf-8', errors='ignore') as f:
        content = f.read()
    
    # Find the [Ordinal/Name Pointer] Table
    match = re.search(r'\[Ordinal/Name Pointer\] Table.*?\n(.*?)(\n\n|\Z)', content, re.DOTALL)
    if not match:
        print("Could not find Name Pointer Table in dump.txt")
        return
        
    lines = match.group(1).split('\n')
    
    hooks = [
        "EOS_P2P_SendPacket",
        "EOS_P2P_GetNextReceivedPacketSize",
        "EOS_P2P_ReceivePacket",
        "EOS_P2P_AcceptConnection",
        "EOS_P2P_CloseConnection",
        "EOS_P2P_CloseConnections",
        "EOS_P2P_QueryNATType",
        "EOS_P2P_GetNATType",
        "EOS_P2P_SetRelayControl",
        "EOS_P2P_GetRelayControl",
        "EOS_P2P_SetPortRange",
        "EOS_P2P_GetPortRange",
        "EOS_P2P_AddNotifyPeerConnectionRequest",
        "EOS_P2P_RemoveNotifyPeerConnectionRequest",
        "EOS_P2P_AddNotifyPeerConnectionClosed",
        "EOS_P2P_RemoveNotifyPeerConnectionClosed",
        "EOS_P2P_AddNotifyPeerConnectionEstablished",
        "EOS_P2P_RemoveNotifyPeerConnectionEstablished",
        "EOS_P2P_AddNotifyPeerConnectionInterrupted",
        "EOS_P2P_RemoveNotifyPeerConnectionInterrupted",
        "EOS_P2P_AddNotifyIncomingPacketQueueFull",
        "EOS_P2P_RemoveNotifyIncomingPacketQueueFull",
        "EOS_P2P_ClearPacketQueue",
        "EOS_P2P_GetPacketQueueInfo",
        "EOS_P2P_SetPacketQueueSize",
        "EOS_Auth_Login",
        "EOS_Connect_Login",
        "EOS_Connect_CreateUser",
        "EOS_EpicAccountId_FromString",
        "EOS_ProductUserId_FromString",
        "EOS_EpicAccountId_IsValid",
        "EOS_ProductUserId_IsValid",
        "EOS_ProductUserId_ToString",
        "EOS_Connect_AddNotifyAuthExpiration",
        "EOS_Lobby_CreateLobby",
        "EOS_Lobby_JoinLobby",
        "EOS_Lobby_CreateLobbySearch",
        "EOS_LobbySearch_Find",
        "EOS_LobbySearch_GetSearchResultCount",
        "EOS_LobbySearch_CopySearchResultByIndex",
        "EOS_LobbyDetails_CopyInfo",
        "EOS_LobbyDetails_GetLobbyOwner",
        "EOS_LobbyDetails_Release",
        "EOS_LobbySearch_Release",
        "EOS_Lobby_AddNotifyLobbyUpdateReceived",
        "EOS_Lobby_RemoveNotifyLobbyUpdateReceived",
        "EOS_LobbySearch_RemoveParameter",
        "EOS_LobbySearch_SetLobbyId",
        "EOS_LobbySearch_SetMaxResults",
        "EOS_LobbySearch_SetParameter",
        "EOS_LobbySearch_SetTargetUserId",
        "EOS_LobbyModification_AddAttribute",
        "EOS_LobbyModification_AddMemberAttribute",
        "EOS_LobbyModification_Release",
        "EOS_LobbyModification_RemoveAttribute",
        "EOS_LobbyModification_RemoveMemberAttribute",
        "EOS_LobbyModification_SetAllowedPlatformIds",
        "EOS_LobbyModification_SetBucketId",
        "EOS_LobbyModification_SetInvitesAllowed",
        "EOS_LobbyModification_SetMaxMembers",
        "EOS_LobbyModification_SetPermissionLevel",
        "EOS_Lobby_UpdateLobbyModification",
        "EOS_LobbyDetails_CopyAttributeByIndex",
        "EOS_LobbyDetails_CopyAttributeByKey",
        "EOS_LobbyDetails_CopyMemberAttributeByIndex",
        "EOS_LobbyDetails_CopyMemberAttributeByKey",
        "EOS_LobbyDetails_CopyMemberInfo",
        "EOS_LobbyDetails_GetAttributeCount",
        "EOS_LobbyDetails_GetMemberAttributeCount",
        "EOS_LobbyDetails_GetMemberByIndex",
        "EOS_LobbyDetails_GetMemberCount",
        "EOS_LobbyDetails_Info_Release",
        "EOS_LobbyDetails_MemberInfo_Release",
        "EOS_Lobby_CopyLobbyDetailsHandle",
        "EOS_Lobby_CopyLobbyDetailsHandleByInviteId",
        "EOS_Lobby_CopyLobbyDetailsHandleByUiEventId",
        "EOS_Lobby_UpdateLobby",
        "EOS_Lobby_DestroyLobby",
        "EOS_Lobby_LeaveLobby",
        "EOS_Lobby_AddNotifyLobbyMemberStatusReceived",
        "EOS_Lobby_RemoveNotifyLobbyMemberStatusReceived",
        "EOS_Lobby_AddNotifyLobbyMemberUpdateReceived",
        "EOS_Lobby_RemoveNotifyLobbyMemberUpdateReceived",
        "EOS_Lobby_IsRTCRoomConnected",
        "EOS_Lobby_GetRTCRoomName",
        "EOS_Lobby_Attribute_Release",
        "EOS_Connect_CopyIdToken",
        "EOS_Connect_IdToken_Release",
        "EOS_Connect_VerifyIdToken",
        "EOS_Platform_Tick",
        "EOS_Lobby_IsLobbyOwner",
        "EOS_Lobby_GetInviteCount",
        "EOS_Lobby_CopyInviteIdByIndex",
        "EOS_Lobby_GetConnectString",
        "EOS_Lobby_ParseConnectString",
        "EOS_Platform_GetActiveCountryCode",
        "EOS_Platform_GetActiveLocaleCode",
        "EOS_Connect_RemoveNotifyLoginStatusChanged",
        "EOS_Connect_GetLoginStatus",
        "EOS_Connect_AddNotifyLoginStatusChanged",
        "EOS_Auth_GetLoginStatus",
        "EOS_Lobby_JoinLobbyById",
        "EOS_Logging_SetCallback",
        "EOS_Logging_SetLogLevel",
        "EOS_PlayerDataStorage_QueryFile",
        "EOS_Sanctions_QueryActivePlayerSanctions",
        "EOS_Sanctions_GetPlayerSanctionCount",
        "EOS_Sanctions_CopyPlayerSanctionByIndex",
    ]
    
    with open('proxy.def', 'w') as out:
        out.write('LIBRARY "EOSSDK-Win64-Shipping.dll"\n')
        out.write('EXPORTS\n')
        for line in lines:
            line = line.strip()
            if not line: continue
            # Parse line: [   0] +base[   1]  0000 EOS_Achievements_AddNotifyAchievementsUnlocked
            parts = line.split()
            if len(parts) >= 4:
                func_name = parts[-1]
                if func_name in hooks:
                    # Export our hook directly
                    out.write(f'    {func_name}\n')
                else:
                    # Forward to the original dll
                    out.write(f'    {func_name}=EOSSDK-Win64-Shipping_orig.{func_name}\n')
                    
    print("proxy.def generated successfully.")

if __name__ == "__main__":
    main()