Binato/server/MultiplayerMatch.js

const osu = require("osu-packet"),
      getUserById = require("./util/getUserById.js"),
      StatusUpdate = require("./Packets/StatusUpdate.js");

// TODO: Cache the player's slot position in their user class for a small optimisation

module.exports = class {
    constructor(MatchHost, MatchData = {matchId: -1,inProgress: false,matchType: 0,activeMods: 0,gameName: "",gamePassword: '',beatmapName: '',beatmapId: 1250198,beatmapChecksum: '',slots: [],host: 0,playMode: 0,matchScoringType: 0,matchTeamType: 0,specialModes: 0,seed: 0}) {
        this.matchId = global.getAndAddToHistoricalMultiplayerMatches();

        this.inProgress = MatchData.inProgress;
        this.matchStartCountdownActive = false;

        this.matchType = MatchData.matchType;

        this.activeMods = MatchData.activeMods;

        this.gameName = MatchData.gameName;
        if (MatchData.gamePassword == '') MatchData.gamePassword == null;
        this.gamePassword = MatchData.gamePassword;

        this.beatmapName = MatchData.beatmapName;
        this.beatmapId = MatchData.beatmapId;
        this.beatmapChecksum = MatchData.beatmapChecksum;

        this.slots = MatchData.slots;
        for (let i = 0; i < this.slots.length; i++) {
            this.slots[i].mods = 0;
        }

        this.host = MatchData.host;

        this.playMode = MatchData.playMode;

        this.matchScoringType = MatchData.matchScoringType;
        this.matchTeamType = MatchData.matchTeamType;
        this.specialModes = MatchData.specialModes;

        this.seed = MatchData.seed;

        this.matchStreamName = `mp_${this.matchId}`;

        this.matchLoadSlots = null;
        this.matchSkippedSlots = null;

        this.playerScores = null;

        this.multiplayerExtras = null;

        const osuPacketWriter = new osu.Bancho.Writer;

        // Update the status of the current user
        StatusUpdate(MatchHost, MatchHost.id);
        osuPacketWriter.MatchNew(this.createOsuMatchJSON());

        // Queue match creation for user
        MatchHost.addActionToQueue(osuPacketWriter.toBuffer);

        global.StreamsHandler.addStream(this.matchStreamName, true, this.matchId);

        // Update the match listing for users in the multiplayer lobby
        global.MultiplayerManager.updateMatchListing();
    }

    createOsuMatchJSON() {
        return {
            matchId: this.matchId,
            inProgress: this.inProgress,
            matchType: this.matchType,
            activeMods: this.activeMods,
            gameName: this.gameName,
            gamePassword: this.gamePassword,
            beatmapName: this.beatmapName,
            beatmapId: this.beatmapId,
            beatmapChecksum: this.beatmapChecksum,
            slots: this.slots,
            host: this.host,
            playMode: this.playMode,
            matchScoringType: this.matchScoringType,
            matchTeamType: this.matchTeamType,
            specialModes: this.specialModes,
            seed: this.seed
        };
    }

    leaveMatch(MatchUser) {
        try {
            let userInMatch = false;
            // Loop through all slots in the match
            for (let i = 0; i < this.slots.length; i++) {
                const slot = this.slots[i];
                // Check if the user is in this slot
                if (slot.playerId == MatchUser.id) {
                    userInMatch = true;
                    break;
                }
            }

            // Make sure we don't run more than once
            // Again, client double firing packets.
            if (!userInMatch) return;

            let osuPacketWriter = new osu.Bancho.Writer;
    
            // Loop through all slots in the match
            for (let i = 0; i < this.slots.length; i++) {
                const slot = this.slots[i];
                // Make sure the user is in this slot
                if (slot.playerId != MatchUser.id) continue;
    
                // Set the slot's status to avaliable
                slot.playerId = -1;
                slot.status = 1;
                
                break;
            }
    
            osuPacketWriter.MatchUpdate(this.createOsuMatchJSON());
    
            // Remove the leaving user from the match's stream
            global.StreamsHandler.removeUserFromStream(this.matchStreamName, MatchUser.id);

            // Inform all users in the match that the leaving user has left
            global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);

            osuPacketWriter = new osu.Bancho.Writer;

            // Remove user from the multiplayer channel for the match
            osuPacketWriter.ChannelRevoked("#multiplayer");

            MatchUser.addActionToQueue(osuPacketWriter.toBuffer);
            
            return this;
        } catch (e) { }
    }

    updateMatch(MatchData) {
        // Update match with new data
        this.inProgress = MatchData.inProgress;

        this.matchType = MatchData.matchType;

        this.activeMods = MatchData.activeMods;

        this.gameName = MatchData.gameName;
        if (MatchData.gamePassword == '') MatchData.gamePassword == null;
        this.gamePassword = MatchData.gamePassword;

        this.beatmapName = MatchData.beatmapName;
        this.beatmapId = MatchData.beatmapId;
        this.beatmapChecksum = MatchData.beatmapChecksum;

        this.host = MatchData.host;

        this.playMode = MatchData.playMode;

        this.matchScoringType = MatchData.matchScoringType;
        this.matchTeamType = MatchData.matchTeamType;
        this.specialModes = MatchData.specialModes;

        this.seed = MatchData.seed;
        const osuPacketWriter = new osu.Bancho.Writer;

        osuPacketWriter.MatchUpdate(this.createOsuMatchJSON());

        // Send this new match data to all users in the match
        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);

        // Update the match listing in the lobby to reflect these changes
        global.MultiplayerManager.updateMatchListing();
    }

    sendMatchUpdate() {
        const osuPacketWriter = new osu.Bancho.Writer;

        osuPacketWriter.MatchUpdate(this.createOsuMatchJSON());

        // Update all users in the match with new match information
        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);
    }

    moveToSlot(MatchUser, SlotToMoveTo) {
        const osuPacketWriter = new osu.Bancho.Writer;

        let currentUserData, slotIndex;
        // Loop through all slots in the match
        for (let i = 0; i < this.slots.length; i++) {
            const slot = this.slots[i];
            // Make sure the user in this slot is the user we want
            if (slot.playerId != MatchUser.id) continue;

            currentUserData = slot;
            slotIndex = i;
            break;
        }

        // Set the new slot's data to the user's old slot data
        this.slots[SlotToMoveTo].playerId = currentUserData.playerId;
        MatchUser.matchSlotId = SlotToMoveTo;
        this.slots[SlotToMoveTo].status = currentUserData.status;

        // Set the old slot's data to open
        this.slots[slotIndex].playerId = -1;
        this.slots[slotIndex].status = 1;

        osuPacketWriter.MatchUpdate(this.createOsuMatchJSON());

        // Send this change to all users in the match
        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);

        // Update the match listing in the lobby to reflect this change
        global.MultiplayerManager.updateMatchListing();
    }

    setReadyState(MatchUser, ReadyState) {
        // Get the match the user is in
        const osuPacketWriter = new osu.Bancho.Writer;

        // Loop though all slots in the match
        for (let i = 0; i < this.slots.length; i++) {
            const slot = this.slots[i];
            // Check if the player in this slot is this user
            if (slot.playerId == MatchUser.id) {
                // Turn on or off the user's ready state
                if (ReadyState) slot.status = 8; // Ready
                else slot.status = 4; // Not Ready
                break;
            }
        }

        osuPacketWriter.MatchUpdate(this.createOsuMatchJSON());

        // Send this update to all users in the stream
        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);
    }

    lockMatchSlot(MatchUser, MatchUserToKick) {
        const osuPacketWriter = new osu.Bancho.Writer;

        // Make sure the user attempting to kick / lock is the host of the match
        if (this.host != MatchUser.id) return;

        // Make sure the user that is attempting to be kicked is not the host
        if (this.slots[MatchUserToKick].playerId === this.host) return;

        // Get the data of the slot at the index sent by the client
        const slot = this.slots[MatchUserToKick];
        let cachedPlayerId = slot.playerId;

        // If the slot is empty lock instead of kicking
        if (slot.playerId === -1) { // Slot is empty, lock it
            if (slot.status === 1) slot.status = 2;
            else slot.status = 1;
        }
        // The slot isn't empty, prepare to kick the player
        else {
            const kickedPlayer = getUserById(slot.playerId);
            kickedPlayer.matchSlotId = -1;
            slot.playerId = -1;
            slot.status = 1;
        }

        osuPacketWriter.MatchUpdate(this.createOsuMatchJSON());

        // Inform all users in the match of the change
        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);

        // Update the match listing in the lobby listing to reflect this change
        global.MultiplayerManager.updateMatchListing();

        if (cachedPlayerId !== null && cachedPlayerId !== -1) {
            // Remove the kicked user from the match stream
            global.StreamsHandler.removeUserFromStream(this.matchStreamName, cachedPlayerId);
        }
    }

    missingBeatmap(MatchUser) {
        const osuPacketWriter = new osu.Bancho.Writer;

        // Loop through all slots in the match
        for (let i = 0; i < this.slots.length; i++) {
            const slot = this.slots[i];
            // Make sure the user in the slot is the user we want to update
            if (slot.playerId != MatchUser.id) continue;

            // User is missing the beatmap set the status to reflect it
            slot.status = 16;
            break;
        }

        osuPacketWriter.MatchUpdate(this.createOsuMatchJSON());

        // Inform all users in the match of this change
        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);
    }

    notMissingBeatmap(MatchUser) {
        const osuPacketWriter = new osu.Bancho.Writer;

        // Loop through all slots in the match
        for (let i = 0; i < this.slots.length; i++) {
            const slot = this.slots[i];
            // Make sure the user in the slot is the user we want to update
            if (slot.playerId != MatchUser.id) continue;

            // The user is not missing the beatmap, set the status to normal
            else slot.status = 4;
            break;
        }

        osuPacketWriter.MatchUpdate(this.createOsuMatchJSON());

        // Inform all users in the match of this change
        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);
    }

    matchSkip(MatchUser) {
        if (this.matchSkippedSlots == null) {
            this.matchSkippedSlots = [];

            const skippedSlots = this.matchSkippedSlots;

            for (let i = 0; i < this.slots.length; i++) {
                const slot = this.slots[i];
                // Make sure the slot has a user in it
                if (slot.playerId === -1 || slot.status === 1 || slot.status === 2) continue;
    
                // Add the slot's user to the loaded checking array
                skippedSlots.push({playerId: slot.playerId, skipped: false}); 
            }
        }

        const skippedSlots = this.matchSkippedSlots;

        for (let i = 0; i < skippedSlots.length; i++) {
            // If loadslot belongs to this user then set loaded to true
            if (skippedSlots[i].playerId == MatchUser.id) {
                skippedSlots[i].skipped = true;
            }
        }

        let allSkipped = true;
        for (let i = 0; i < skippedSlots.length; i++) {
            if (skippedSlots[i].skipped) continue;

            // A user hasn't finished playing
            allSkipped = false;
        }

        // All players have finished playing, finish the match
        if (allSkipped) {
            const osuPacketWriter = new osu.Bancho.Writer;
            osuPacketWriter.MatchPlayerSkipped(MatchUser.id);
            osuPacketWriter.MatchSkip();
            global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);

            this.matchSkippedSlots = null;
        } else {
            const osuPacketWriter = new osu.Bancho.Writer;
            osuPacketWriter.MatchPlayerSkipped(MatchUser.id);

            global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);
        }
    }

    transferHost(MatchUserToTransferHostTo) {
        const osuPacketWriter = new osu.Bancho.Writer;

        // Set the lobby's host to the new user
        this.host = this.slots[MatchUserToTransferHostTo].playerId;

        osuPacketWriter.MatchUpdate(this.createOsuMatchJSON());

        // Inform all clients in the match of the change
        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);
    }

    // TODO: Fix not being able to add DT when freemod is active
    updateMods(MatchUser, MatchMods) {
        // Check if freemod is enabled
        if (this.specialModes === 1) {
            const osuPacketWriter = new osu.Bancho.Writer;
            for (let i = 0; i < this.slots.length; i++) {
                const slot = this.slots[i];
                if (slot.playerId === MatchUser.id) {
                    slot.mods = MatchMods;
                    break;
                }
            }

            osuPacketWriter.MatchUpdate(this.createOsuMatchJSON());

            global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);
        } else {
            // Make sure the person updating mods is the host of the match
            if (this.host !== MatchUser.id) return;
            const osuPacketWriter = new osu.Bancho.Writer;

            // Change the matches mods to these new mods
            // TODO: Do this per user if freemod is enabled
            this.activeMods = MatchMods;

            osuPacketWriter.MatchUpdate(this.createOsuMatchJSON());

            // Inform all users in the match of the change
            global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);
        }

        // Update match listing in the lobby to reflect this change
        global.MultiplayerManager.updateMatchListing();
    }

    startMatch() {
        // Make sure the match is not already in progress
        // The client sometimes double fires the start packet
        if (this.inProgress) return;
        this.inProgress = true;
        // Create array for monitoring users until they are ready to play
        this.matchLoadSlots = [];
        const loadedSlots = this.matchLoadSlots;
        // Loop through all slots in the match
        for (let i = 0; i < this.slots.length; i++) {
            const slot = this.slots[i];
            // Make sure the slot has a user in it
            if (slot.playerId === -1 || slot.status === 1 || slot.status === 2) continue;

            // Add the slot's user to the loaded checking array
            loadedSlots.push({
                playerId: slot.playerId,
                loaded: false
            }); 
        }
        const osuPacketWriter = new osu.Bancho.Writer;

        // Loop through all slots in the match
        for (let i = 0; i < this.slots.length; i++) {
            const slot = this.slots[i];
            // Make sure the slot has a user in it
            if (slot.playerId === -1 || slot.status === 1 || slot.status === 2) continue;

            // Set the user's status to playing
            slot.status = 32;
        }

        osuPacketWriter.MatchStart(this.createOsuMatchJSON());

        // Inform all users in the match that it has started
        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);

        // Update all users in the match with new info
        this.sendMatchUpdate();

        // Update match listing in lobby to show the game is in progress
        global.MultiplayerManager.updateMatchListing();
    }

    matchPlayerLoaded(MatchUser) {
        const loadedSlots = this.matchLoadSlots;

        // Loop through all user load check items
        for (let i = 0; i < loadedSlots.length; i++) {
            // If loadslot belongs to this user then set loaded to true
            if (loadedSlots[i].playerId == MatchUser.id) {
                loadedSlots[i].loaded = true;
            }
        }

        // Loop through all loaded slots and check if all users are loaded
        let allLoaded = true;
        for (let i = 0; i < loadedSlots.length; i++) {
            if (loadedSlots[i].loaded) continue;

            // A user wasn't loaded, keep waiting.
            allLoaded = false;
            break;
        }

        // All players have loaded the beatmap, start playing.
        if (allLoaded) {
            let osuPacketWriter = new osu.Bancho.Writer;
            osuPacketWriter.MatchAllPlayersLoaded();
            global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);

            // Blank out user loading array
            this.matchLoadSlots = null;

            this.playerScores = [];
            for (let i = 0; i < this.slots.length; i++) {
                const slot = this.slots[i];
                if (slot.playerId === -1 || slot.status === 1 || slot.status === 2) continue;

                this.playerScores.push({playerId: slot.playerId, slotId: i, score: 0, isCurrentlyFailed: false});
            }
        }
    }

    onPlayerFinishMatch(MatchUser) {
        // If user loading slots do not exist
        if (this.matchLoadSlots == null) {
            this.matchLoadSlots = [];
            // Repopulate user loading slots again
            const loadedSlots = this.matchLoadSlots;
            for (let i = 0; i < this.slots.length; i++) {
                const slot = this.slots[i];
                // Make sure the slot has a user
                if (slot.playerId === -1 || slot.status === 1 || slot.status === 2) continue;
    
                // Populate user loading slots with this user's id and load status
                loadedSlots.push({
                    playerId: slot.playerId,
                    loaded: false
                }); 
            }
        } 
        
        const loadedSlots = this.matchLoadSlots;

        // Loop through all loaded slots to make sure all users have finished playing
        for (let i = 0; i < loadedSlots.length; i++) {
            if (loadedSlots[i].playerId == MatchUser.id) {
                loadedSlots[i].loaded = true;
            }
        }

        let allLoaded = true;
        for (let i = 0; i < loadedSlots.length; i++) {
            if (loadedSlots[i].loaded) continue;

            // A user hasn't finished playing
            allLoaded = false;
        }

        // All players have finished playing, finish the match
        if (allLoaded) this.finishMatch();
    }

    finishMatch() {
        if (!this.inProgress) return;
        this.matchLoadSlots = null;
        this.inProgress = false;
        let osuPacketWriter = new osu.Bancho.Writer;

        // Loop through all slots in the match
        for (let i = 0; i < this.slots.length; i++) {
            const slot = this.slots[i];
            // Make sure the slot has a user
            if (slot.playerId === -1 || slot.status === 1 || slot.status === 2) continue;

            // Set the user's status back to normal from playing
            slot.status = 4;
        }

        osuPacketWriter.MatchComplete();

        // Inform all users in the match that it is complete
        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);

        // Update all users in the match with new info
        this.sendMatchUpdate();

        // Update match info in the lobby to reflect that the match has finished
        global.MultiplayerManager.updateMatchListing();

        if (this.multiplayerExtras != null) this.multiplayerExtras.onMatchFinished(JSON.parse(JSON.stringify(this.playerScores)));

        this.playerScores = null;
    }

    updatePlayerScore(MatchPlayer, MatchScoreData) {
        const osuPacketWriter = new osu.Bancho.Writer;

        // Make sure the user's slot ID is not invalid
        if (this.matchSlotId == -1) return;

        // Get the user's current slotID and append it to the givien data, just incase.
        MatchScoreData.id = MatchPlayer.matchSlotId;

        // Update the playerScores array accordingly
        for (let i = 0; i < this.playerScores.length; i++) {
            if (this.playerScores[i].playerId == MatchPlayer.id) {
                this.playerScores[i].score = MatchScoreData.totalScore;
                this.playerScores[i].isCurrentlyFailed = MatchScoreData.currentHp == 254;
                break;
            }
        }
        
        osuPacketWriter.MatchScoreUpdate(MatchScoreData);

        // Send the newly updated score to all users in the match
        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);
    }

    matchFailed(MatchUser) {
        const osuPacketWriter = new osu.Bancho.Writer;

        // Make sure the user's slot ID is not invalid
        if (MatchUser.matchSlotId == -1) return;

        osuPacketWriter.MatchPlayerFailed(MatchUser.id);

        global.StreamsHandler.sendToStream(this.matchStreamName, osuPacketWriter.toBuffer, null);
    }
}