com.societegenerale.githubcrawler.ownership.OwnershipParserImpl.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of github-crawler-core Show documentation

There is a newer version: 2.2.0

package com.societegenerale.githubcrawler.ownership

import com.societegenerale.githubcrawler.model.team.Membership
import com.societegenerale.githubcrawler.remote.RemoteGitHub
import org.slf4j.LoggerFactory

/**
 * Example of implementation for repository ownership computation.
 * Just instantiate a class implementing OwnershipParser interface in your Spring Boot config. If none is found, a dummy one will be instantiated by default.
 */
class OwnershipParserImpl(private val githubClient: RemoteGitHub, private val membershipParser: MembershipParser, private val organizationName: String) : OwnershipParser {

    companion object {
        private val log = LoggerFactory.getLogger(OwnershipParserImpl::class.java)
    }

    private val UNDEFINED = "Undefined"

    private var memberIdToTeamName: Membership = Membership()

    private var isParserInitiated: Boolean = false


    override fun computeOwnershipFor(repositoryFullName: String, lastCommitNumber: Int): String {

        //TODO see if we can make this an init block - currently, problem for tests, because it requires a remote server to be available
        // to fetch teams...  and the OwnershipParserImpl is instanciated (as part of the whole config) BEFORE the fake remote webserver is available
        synchronized(isParserInitiated) {
            if (!isParserInitiated) {
                memberIdToTeamName = membershipParser.computeMembership()
                isParserInitiated = true
            }
        }


        if (memberIdToTeamName.isEmpty()) {
            log.info("Membership is empty, unable to compute repository owner...")
            return UNDEFINED
        }

        val commits = githubClient.fetchCommits(organizationName, repositoryFullName, lastCommitNumber)
        val commitsWithStats = commits.map { (sha) -> githubClient.fetchCommit(organizationName, repositoryFullName, sha) }
        log.debug("${commitsWithStats.size} fetched for $organizationName and $repositoryFullName (with max fetch to $lastCommitNumber)")
        return commitsWithStats
                .filter { it.author != null }
                .flatMap { (_, author, stats) -> memberIdToTeamName.getTeams(author!!.login).map { team -> Pair(team, stats.total) } }
                .groupBy { it.first }
                .mapValues { it.value.sumBy { stat -> stat.second } }
                .entries
                .sortedByDescending { it.value }
                .firstOrNull()?.key ?: UNDEFINED
    }

}