com.colisweb.gdrive.client.sheets.GooglePivotTable.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of google-drive-scala-client_2.13 Show documentation

Google Drive Scala Client is a Scala wrapper around the Google Drive client for Java

The newest version!

package com.colisweb.gdrive.client.sheets

import com.colisweb.gdrive.client.sheets.GooglePivotTable.{GooglePivotGroup, GooglePivotValue}
import com.google.api.services.sheets.v4.model.{DataSourceColumnReference, PivotGroup, PivotTable, PivotValue}

import scala.jdk.CollectionConverters._

final case class GooglePivotTable(
    rows: List[GooglePivotGroup],
    values: List[GooglePivotValue],
    dataSourceId: String
) {

  def toGoogle: PivotTable =
    new PivotTable()
      .setRows(rows.map(_.toGoogle).asJava)
      .setValues(values.map(_.toGoogle).asJava)
      .setDataSourceId(dataSourceId)
}

object GooglePivotTable {

  sealed trait GoogleSummarizedFunction { val value: String }

  case object Average           extends GoogleSummarizedFunction { override val value = "AVERAGE"     }
  case object CountA            extends GoogleSummarizedFunction { override val value = "COUNTA"      }
  case object CountUnique       extends GoogleSummarizedFunction { override val value = "COUNTUNIQUE" }
  case object StandardDeviation extends GoogleSummarizedFunction { override val value = "STDEV"       }

  final case class GooglePivotGroup(columnReference: String, sortOrder: String) {

    def toGoogle: PivotGroup =
      new PivotGroup()
        .setDataSourceColumnReference(new DataSourceColumnReference().setName(columnReference))
        .setSortOrder(sortOrder)
        .setShowTotals(true)
  }

  final case class GooglePivotValue(
      name: String,
      columnReference: String,
      summarizedFunction: GoogleSummarizedFunction
  ) {

    def toGoogle: PivotValue =
      new PivotValue()
        .setDataSourceColumnReference(new DataSourceColumnReference().setName(columnReference))
        .setSummarizeFunction(summarizedFunction.value)
        .setName(name)
  }
}