feat: use custom RoleIdentification if Riot roles are badly identified

server/app/Services/RoleIdentificationService.js

@@ -193,7 +193,13 @@ class RoleIdentificationService {
       identified[best[0]] = best[1]
     }
 
-    return positions
+    // Rename UTILITY to SUPPORT
+    const {
+      UTILITY: SUPPORT,
+      ...rest
+    } = positions
+
+    return { ...rest, SUPPORT }
   }
 }
 

server/app/Transformers/BasicMatchTransformer.js

@@ -1,6 +1,7 @@
 'use strict'
 
 const MatchTransformer = use('App/Transformers/MatchTransformer')
+const { queuesWithRole } = use('App/helpers')
 
 /**
  * BasicMatchTransformer class
@@ -65,8 +66,9 @@ class BasicMatchTransformer extends MatchTransformer {
         enemyTeam.push(playerInfos)
       }
     }
-    allyTeam.sort(this.sortTeamByRole)
+
-    enemyTeam.sort(this.sortTeamByRole)
+    // Roles
+    super.getMatchRoles(match, allyTeam, enemyTeam, player.teamId, playerData)
 
     return {
       account_id: identity.player.currentAccountId,

server/app/Transformers/DetailedMatchTransformer.js

@@ -23,6 +23,9 @@ class DetailedMatchTransformer extends MatchTransformer {
     const firstTeam = this.getTeamData(match, match.teams[0])
     const secondTeam = this.getTeamData(match, match.teams[1])
 
+    // Roles
+    super.getMatchRoles(match, firstTeam.players, secondTeam.players)
+
     return {
       gameId: match.gameId,
       season: match.seasonId,

server/app/Transformers/LiveMatchTransformer.js

@@ -1,7 +1,6 @@
 'use strict'
 
 const MatchTransformer = use('App/Transformers/MatchTransformer')
-const RoleIdentificationService = use('App/Services/RoleIdentificationService')
 const SummonerService = use('App/Services/SummonerService')
 const { queuesWithRole } = use('App/helpers')
 
@@ -24,13 +23,6 @@ class LiveMatchTransformer extends MatchTransformer {
     return participant
   }
 
-  _getTeamRoles(team) {
-    const teamJunglers = team.filter(p => p.jungle)
-    const jungle = teamJunglers.length === 1 ? teamJunglers[0].champion : null
-
-    return RoleIdentificationService.getRoles(this.championRoles, team.map(p => p.champion), jungle)
-  }
-
   /**
    * Transform raw data from Riot API
    * @param match data from Riot API, one live match
@@ -50,8 +42,8 @@ class LiveMatchTransformer extends MatchTransformer {
         p.teamId === 100 ? blueTeam.push(playerRole) : redTeam.push(playerRole)
       })
 
-      blueRoles = this._getTeamRoles(blueTeam)
+      blueRoles = super.getTeamRoles(blueTeam)
-      redRoles = this._getTeamRoles(redTeam)
+      redRoles = super.getTeamRoles(redTeam)
     }
 
     for (const participant of match.participants) {
@@ -62,9 +54,6 @@ class LiveMatchTransformer extends MatchTransformer {
       if (needsRole) {
         const roles = participant.teamId === 100 ? blueRoles : redRoles
         participant.role = Object.entries(roles).find(([, champion]) => participant.championId === champion)[0]
-        if (participant.role === 'UTILITY') {
-          participant.role = 'SUPPORT'
-        }
       }
     }
 

server/app/Transformers/MatchTransformer.js

@@ -207,6 +207,66 @@ class MatchTransformer {
     return timeline.lane
   }
 
+  /**
+   * Return the 5 roles of a team based on champions
+   * @param team 5 champions + smite from a team
+   */
+  getTeamRoles(team) {
+    const teamJunglers = team.filter(p => p.jungle)
+    const jungle = teamJunglers.length === 1 ? teamJunglers[0].champion : null
+
+    return RoleIdentificationService.getRoles(this.championRoles, team.map(p => p.champion), jungle)
+  }
+
+  /**
+   * Update roles for a team if Riot's ones are badly identified
+   * @param {Object} team 5 players data of the team
+   * @param {Array} champs 5 champions + smite from the team
+   * @param {Object} playerData data of the searched player, only for basic matches 
+   */
+  updateTeamRoles(team, champs, playerData = null) {
+    const actualRoles = [...new Set(team.map(p => p.role))]
+    if (actualRoles.length === 5) {
+      return
+    }
+
+    champs = this.getTeamRoles(champs)
+    for (const summoner of team) {
+      summoner.role = Object.entries(champs).find(([, champion]) => summoner.champion.id === champion)[0]
+
+      if (playerData && summoner.champion.id === playerData.champion.id) {
+        playerData.role = summoner.role
+      }
+    }
+  }
+
+  /**
+   * 
+   * @param {Object} match from Riot Api
+   * @param {Array} allyTeam 5 players of the first team
+   * @param {Array} enemyTeam 5 players of the second team
+   * @param {Number} allyTeamId team id of the searched player, only for basic matches
+   * @param {Object} playerData data of the searched player, only for basic matches
+   */
+  getMatchRoles(match, allyTeam, enemyTeam, allyTeamId = 100, playerData = null) {
+    if (!this.championRoles || !queuesWithRole.includes(match.queueId)) {
+      return
+    }
+
+    let allyChamps = []
+    let enemyChamps = []
+    match.participants.map(p => {
+      const playerRole = { champion: p.championId, jungle: p.spell1Id === 11 || p.spell2Id === 11 }
+      p.teamId === allyTeamId ? allyChamps.push(playerRole) : enemyChamps.push(playerRole)
+    })
+
+    this.updateTeamRoles(allyTeam, allyChamps, playerData)
+    this.updateTeamRoles(enemyTeam, enemyChamps)
+
+    allyTeam.sort(this.sortTeamByRole)
+    enemyTeam.sort(this.sortTeamByRole)
+  }
+
   /**
    * Get Summoner Spell Data from CDragon
    * @param id of the summonerSpell