// Copyright 2021, Chef.  All rights reserved.
// https://github.com/q191201771/naza
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)

// Package chartbar 控制台绘制ascii柱状图
package chartbar

const (
	OrderOrigin    Order = iota + 1 // 原始序
	OrderAscCount                   // 按计数值升序排序
	OrderDescCount                  // 按计数值降序排序
	OrderAscName                    // 按字段名称升序排序
	OrderDescName                   // 按字段名称降序排序

	NoNumLimit = -1
)

type Item struct {
	Name string  // key
	Num  float64 // value

	count int // bar
}

type Order int

type Option struct {
	MaxBarLength    int    // 柱状图形的最大长度
	DrawIconBlock   string // 柱状图实体绘制内容
	DrawIconPadding string // 柱状图空余部分绘制内容
	HideName        bool   // 是否隐藏图形旁边的Name字段
	HideNum         bool   // 是否隐藏图形旁边的Num字段

	Order          Order // 排序方式
	PrefixNumLimit int   // 只显示前`PrefixNumLimit`个元素，注意，可以和`SuffixNumLimit`同时使用
	SuffixNumLimit int   // 只显示后`SuffixNumLimit`个元素
}

var defaultOption = Option{
	// 50 "▇" " "
	// 18 "口" "　"
	MaxBarLength:    50,
	DrawIconBlock:   "▇",
	DrawIconPadding: " ",
	HideName:        false, //
	HideNum:         false,

	Order:          OrderDescCount,
	PrefixNumLimit: NoNumLimit,
	SuffixNumLimit: NoNumLimit,
}

// ---------------------------------------------------------------------------------------------------------------------

var DefaultCtx = NewCtx()

type ModOption func(option *Option)

func NewCtx(modOptions ...ModOption) *Ctx {
	option := defaultOption
	for _, fn := range modOptions {
		fn(&option)
	}
	return &Ctx{
		option: option,
	}
}

// NewCtxWith
//
// 在`ctx`参数基础上使用`modOptions`生成新的 Ctx
func NewCtxWith(ctx *Ctx, modOptions ...ModOption) *Ctx {
	option := ctx.option
	for _, fn := range modOptions {
		fn(&option)
	}
	return &Ctx{
		option: option,
	}
}