import com.audienceproject.crossbow.{*, given}

val data = Seq(("a", 1), ("b", 2), ("c", 3))
val df = DataFrame.fromSeq(data)

df.printSchema()

/**
 * _0: string
 * _1: int
 */

df.as[(String, Int)].foreach(println)

/**
 * (a, 1)
 * (b, 2)
 * (c, 3)
 */

val df = Seq((1, 2), (3, 4)).toDataFrame("x", "y")
df.select($"x" + $"y" / 2d as "avg", ($"x", $"y") as "tuple")

// Lambda functions
val pythagoras = lambda[(Int, Int), Double]:
  (a, b) => math.sqrt(a * a + b * b)
df.select(pythagoras($"x", $"y"))

df.filter($"x" >= 2 && $"y" % 2 =!= 0)

val df = Seq(("foo", 1), ("foo", 2), ("bar", 3)).toDataFrame("someKey", "x")
df.groupBy($"someKey").agg(sum($"x") / count() as "avg", collect($"x") as "xs")

// Custom aggregators
val product = reducer[Int, Int](1)(_ * _)
df.groupBy($"someKey").agg(product($"x") as "product")

df.sortBy($"x")

// Sorting on 'x' first, then 'y'
df.sortBy(($"someKey", $"x"))

// Sorting with explicit ordering (e.g. integer descending)
df.sortBy($"x")(using Order.by(Ordering.Int.reverse))

val otherDf = Seq(("foo", 1, 10d), ("foo", 2, 20d), ("bar", 3, 30d)).toDataFrame("someKey", "x", "y")

// Inner join
df.join(otherDf, $"someKey")

// Inner join on multiple columns
df.join(otherDf, ($"someKey", $"x"))

// Other join types
df.join(otherDf, $"someKey", JoinType.LeftOuter)