blob: 5a1c0f71c25a55fb405447b401c9bbe5fae7035b [file] [log] [blame] [edit]
// Copyright (C) 2017 GerritForge Ltd
//
// 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.gerritforge.analytics.gitcommits.model
import java.net.URLEncoder
import scala.xml._
import java.time.format.DateTimeFormatter
import java.time.{LocalDate, ZoneOffset}
import com.gerritforge.analytics.common.api.GerritConnectivity
import com.gerritforge.analytics.support.ops.AnalyticsDateTimeFormatter
import java.nio.charset.StandardCharsets
case class GerritEndpointConfig(
baseUrl: Option[String] = None,
prefix: Option[String] = None,
outputDir: String =
s"file://${System.getProperty("java.io.tmpdir")}/analytics-${System.nanoTime()}",
elasticIndex: Option[String] = None,
since: Option[LocalDate] = None,
until: Option[LocalDate] = None,
aggregate: Option[String] = None,
emailAlias: Option[String] = None,
username: Option[String] = None,
password: Option[String] = None,
ignoreSSLCert: Option[Boolean] = None,
extractBranches: Option[Boolean] = None,
manifest: Option[String] = None,
manifestBranch: Option[String] = None,
manifestLabel: Option[String] = None
) {
lazy val projectsFromManifest: Option[Set[GerritProjectWithRef]] = manifest.map { mf =>
val mfDoc = XML.loadFile(mf)
val defaultRef = (mfDoc \ "default" \@ "revision").trim
val mfProjects = mfDoc \ "project"
mfProjects.flatMap { projectNode =>
for {
name <- projectNode.attribute("name").map(_.text.stripSuffix(".git"))
revision = projectNode.attribute("revision").map(_.text).orElse(Some(defaultRef))
} yield GerritProjectWithRef(URLEncoder.encode(name, "UTF-8"), name, revision)
}.toSet
}
val gerritApiConnection: GerritConnectivity =
new GerritConnectivity(username, password, ignoreSSLCert.getOrElse(false))
val gerritProjectsUrl: Option[String] = baseUrl.map { url =>
s"${url}/projects/" + prefix.fold("")("?p=" + _)
}
def queryOpt(opt: (String, Option[String])): Option[String] = {
opt match {
case (name: String, value: Option[String]) => value.map(name + "=" + _)
}
}
@transient
private lazy val format: DateTimeFormatter =
AnalyticsDateTimeFormatter.yyyy_MM_dd.withZone(ZoneOffset.UTC)
val queryString = Seq(
"since" -> since.map(format.format),
"until" -> until.map(format.format),
"aggregate" -> aggregate,
"extract-branches" -> extractBranches.map(_.toString)
).flatMap(queryOpt).mkString("?", "&", "")
def contributorsUrl(project: GerritProjectWithRef): Option[String] =
baseUrl.map { url =>
val refFilter = project.refName.fold("")(ref => s"&starting-revision=$ref")
s"$url/projects/${project.id}/analytics~contributors$queryString$refFilter"
}
}