// Copyright 2017 Baliance. All rights reserved.
package main

import (
	"fmt"
	"log"
	"math/rand"

	"baliance.com/gooxml/chart"
	"baliance.com/gooxml/color"
	"baliance.com/gooxml/measurement"
	"baliance.com/gooxml/spreadsheet"

	"baliance.com/gooxml/schema/soo/sml"
)

func main() {
	ss := spreadsheet.New()
	sheet := ss.AddSheet()

	hdrStyle := ss.StyleSheet.AddCellStyle()

	f := ss.StyleSheet.Fills().AddFill()
	pf := f.SetPatternFill()
	pf.SetFgColor(color.LightGray)
	hdrStyle.SetFill(f)

	fnt := ss.StyleSheet.AddFont()
	fnt.SetBold(true)
	hdrStyle.SetFont(fnt)

	row := sheet.AddRow()
	row.Cell("A").SetString("Item")
	row.Cell("A").SetStyle(hdrStyle)
	row.Cell("B").SetString("Price")
	row.Cell("B").SetStyle(hdrStyle)
	row.Cell("C").SetString("# Sold")
	row.Cell("C").SetStyle(hdrStyle)
	row.Cell("D").SetString("Total")
	row.Cell("D").SetStyle(hdrStyle)

	// Set some column widths
	sheet.Column(1).SetWidth(1.5 * measurement.Inch)
	sheet.Column(4).SetWidth(2 * measurement.Inch)

	for r := 0; r < 5; r++ {
		row := sheet.AddRow()
		row.AddCell().SetString(fmt.Sprintf("Product %d", r+1))
		row.AddCell().SetNumber(float64(rand.Intn(50)) / 10.0)
		row.AddCell().SetNumber(float64(rand.Intn(50) + 1))
		row.AddCell().SetFormulaRaw(fmt.Sprintf("C%d*B%d", r+2, r+2))
	}

	// add an auto-filter
	sheet.SetAutoFilter("A1:D6")

	// conditional formatting
	// total column
	cf := sheet.AddConditionalFormatting([]string{"D2:D6"})
	rule := cf.AddRule()
	db := rule.SetDataBar()
	db.AddFormatValue(sml.ST_CfvoTypeMin, "0")
	db.AddFormatValue(sml.ST_CfvoTypeMax, "0")
	db.SetColor(color.Blue)

	// sold column
	cf = sheet.AddConditionalFormatting([]string{"B2:B6"})
	rule = cf.AddRule()
	cs := rule.SetColorScale()
	cs.AddFormatValue(sml.ST_CfvoTypeMin, "0")
	cs.AddFormatValue(sml.ST_CfvoTypePercentile, "50")
	cs.AddFormatValue(sml.ST_CfvoTypeMax, "0")
	cs.AddGradientStop(color.SuccessGreen)
	cs.AddGradientStop(color.Orange)
	cs.AddGradientStop(color.Red)

	// Charts need to reside in a drawing
	dwng := ss.AddDrawing()
	chrt1, anc1 := dwng.AddChart(spreadsheet.AnchorTypeTwoCell)
	chrt2, anc2 := dwng.AddChart(spreadsheet.AnchorTypeTwoCell)
	addBar3DChart(chrt1)
	addLineChart(chrt2)
	anc1.SetWidth(9)
	anc1.MoveTo(6, 1)
	anc2.MoveTo(0, 9)

	// and finally add the chart to the sheet
	sheet.SetDrawing(dwng)

	if err := ss.Validate(); err != nil {
		log.Fatalf("error validating sheet: %s", err)
	}
	ss.SaveToFile("complex.xlsx")
}

func addBar3DChart(chrt chart.Chart) {
	chrt.AddTitle().SetText("Bar Chart")
	lc := chrt.AddBar3DChart()
	priceSeries := lc.AddSeries()
	priceSeries.SetText("Price")
	// Set a category axis reference on the first series to pull the product names
	priceSeries.CategoryAxis().SetLabelReference(`'Sheet 1'!A2:A6`)
	priceSeries.Values().SetReference(`'Sheet 1'!B2:B6`)

	soldSeries := lc.AddSeries()
	soldSeries.SetText("Sold")
	soldSeries.Values().SetReference(`'Sheet 1'!C2:C6`)

	totalSeries := lc.AddSeries()
	totalSeries.SetText("Total")
	totalSeries.Values().SetReference(`'Sheet 1'!D2:D6`)

	// the line chart accepts up to two axes
	ca := chrt.AddCategoryAxis()
	va := chrt.AddValueAxis()
	lc.AddAxis(ca)
	lc.AddAxis(va)

	ca.SetCrosses(va)
	va.SetCrosses(ca)
}

func addLineChart(chrt chart.Chart) {
	chrt.AddTitle().SetText("Line Chart")
	lc := chrt.AddLine3DChart()
	priceSeries := lc.AddSeries()
	priceSeries.SetText("Price")
	// Set a category axis reference on the first series to pull the product names
	priceSeries.CategoryAxis().SetLabelReference(`'Sheet 1'!A2:A6`)
	priceSeries.Values().SetReference(`'Sheet 1'!B2:B6`)

	soldSeries := lc.AddSeries()
	soldSeries.SetText("Sold")
	soldSeries.Values().SetReference(`'Sheet 1'!C2:C6`)

	totalSeries := lc.AddSeries()
	totalSeries.SetText("Total")
	totalSeries.Values().SetReference(`'Sheet 1'!D2:D6`)

	// the line chart accepts up to two axes
	ca := chrt.AddCategoryAxis()
	va := chrt.AddValueAxis()
	lc.AddAxis(ca)
	lc.AddAxis(va)

	ca.SetCrosses(va)
	va.SetCrosses(ca)
}