src/main/scala/com/googlesource/gerrit/plugins/analytics/common/BranchesExtractor.scala

// Copyright (C) 2017 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package com.googlesource.gerrit.plugins.analytics.common

import com.googlesource.gerrit.plugins.analytics.common.ManagedResource.use
import org.eclipse.jgit.api.Git
import org.eclipse.jgit.lib.{Constants, ObjectId, Repository}
import org.eclipse.jgit.revwalk.RevWalk

import scala.collection.JavaConversions._

case class BranchesExtractor(repo: Repository) {
  lazy val branchesOfCommit: Map[ObjectId, Set[String]] = {

    use(new Git(repo)) { git =>
      git.branchList.call.foldLeft(Map.empty[ObjectId, Set[String]]) { (branchesAcc, ref) =>
        val branchName = ref.getName.drop(Constants.R_HEADS.length)

        use(new RevWalk(repo)) { rw: RevWalk =>
          rw.markStart(rw.parseCommit(ref.getObjectId))
          rw.foldLeft(branchesAcc) { (thisBranchAcc, rev) =>
            val sha1 = rev.getId
            thisBranchAcc.get(sha1) match {
              case Some(set) => thisBranchAcc + (sha1 -> (set + branchName))
              case None      => thisBranchAcc + (sha1 -> Set(branchName))
            }
          }
        }
      }
    }
  }
}