// MIT License

// Copyright (c) 2022 Tree Xie

// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:

// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

package charts

type LegendPainter struct {
	p   *Painter
	opt *LegendPainterOption
}

const IconRect = "rect"
const IconLineDot = "lineDot"

type LegendPainterOption struct {
	Theme ColorPalette
	// Text array of legend
	Data []string
	// Distance between legend component and the left side of the container.
	// It can be pixel value: 20, percentage value: 20%,
	// or position value: right, center.
	Left string
	// Distance between legend component and the top side of the container.
	// It can be pixel value: 20.
	Top string
	// Legend marker and text aligning, it can be left or right, default is left.
	Align string
	// The layout orientation of legend, it can be horizontal or vertical, default is horizontal.
	Orient string
	// Icon of the legend.
	Icon string
	// Font size of legend text
	FontSize float64
	// FontColor color of legend text
	FontColor Color
}

func NewLegendPainter(p *Painter, opt LegendPainterOption) *LegendPainter {
	return &LegendPainter{
		p:   p,
		opt: &opt,
	}
}

func (l *LegendPainter) Render() (Box, error) {
	opt := l.opt
	theme := opt.Theme
	if theme == nil {
		theme = l.p.theme
	}
	p := l.p
	p.SetTextStyle(Style{
		FontSize:  opt.FontSize,
		FontColor: opt.FontColor,
	})
	measureList := make([]Box, len(opt.Data))
	maxTextWidth := 0
	for index, text := range opt.Data {
		b := p.MeasureText(text)
		if b.Width() > maxTextWidth {
			maxTextWidth = b.Width()
		}
		measureList[index] = b
	}
	x := 0
	y := 0
	offset := 20
	textOffset := 2
	legendWidth := 30
	legendHeight := 20
	drawIcon := func(top, left int) int {
		if opt.Icon == IconRect {
			p.Rect(Box{
				Top:    top - legendHeight + 4,
				Left:   left,
				Right:  left + legendWidth,
				Bottom: top - 2,
			})
		} else {
			p.LegendLineDot(Box{
				Top:    top,
				Left:   left,
				Right:  left + legendWidth,
				Bottom: top + legendHeight,
			})
		}
		return left + legendWidth
	}
	for index, text := range opt.Data {
		color := theme.GetSeriesColor(index)
		p.SetDrawingStyle(Style{
			FillColor:   color,
			StrokeColor: color,
		})
		if opt.Align != AlignRight {
			x = drawIcon(y, x)
			x += textOffset
		}
		p.Text(text, x, y)
		x += measureList[index].Width()
		if opt.Align == AlignRight {
			x += textOffset
			x = drawIcon(0, x)
		}
		if opt.Orient == OrientVertical {
			y += offset
			x = 0
		} else {
			x += offset
			y = 0
		}
	}
	width := 0
	height := 0
	for _, item := range measureList {
		if opt.Orient == OrientVertical {
			height += item.Height()
		} else {
			width += item.Width()
		}
	}
	if opt.Orient == OrientVertical {
		width = maxTextWidth + textOffset + legendWidth
		height = offset * len(opt.Data)
	} else {
		height = legendHeight
		offsetValue := (len(opt.Data) - 1) * (offset + textOffset)
		allLegendWidth := len(opt.Data) * legendWidth
		width += (offsetValue + allLegendWidth)
	}

	return Box{
		Right:  width,
		Bottom: height,
	}, nil
}