termdash/widgets/gauge/options.go

// Copyright 2018 Google Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package gauge

// options.go contains configurable options for Gauge.

import (
	"fmt"

	"github.com/mum4k/termdash/align"
	"github.com/mum4k/termdash/cell"
	"github.com/mum4k/termdash/internal/draw"
	"github.com/mum4k/termdash/linestyle"
)

// Option is used to provide options.
type Option interface {
	// set sets the provided option.
	set(*options)
}

// options holds the provided options.
type options struct {
	gaugeChar        rune
	hideTextProgress bool
	height           int
	textLabel        string
	hTextAlign       align.Horizontal
	vTextAlign       align.Vertical
	color            cell.Color
	filledTextColor  cell.Color
	emptyTextColor   cell.Color
	// If set, draws a border around the gauge.
	border            linestyle.LineStyle
	borderCellOpts    []cell.Option
	borderTitle       string
	borderTitleHAlign align.Horizontal
}

// newOptions returns options with the default values set.
func newOptions() *options {
	return &options{
		gaugeChar:       DefaultChar,
		hTextAlign:      DefaultHorizontalTextAlign,
		vTextAlign:      DefaultVerticalTextAlign,
		color:           DefaultColor,
		filledTextColor: DefaultFilledTextColor,
		emptyTextColor:  DefaultEmptyTextColor,
	}
}

// validate validates the provided options.
func (o *options) validate() error {
	if got, min := o.height, 0; got < min {
		return fmt.Errorf("invalid Height %d, must be %d <= Height", got, min)
	}
	return nil
}

// option implements Option.
type option func(*options)

// set implements Option.set.
func (o option) set(opts *options) {
	o(opts)
}

// DefaultChar is the default value for the Char option.
const DefaultChar = draw.DefaultRectChar

// Char sets the rune that is used when drawing the rectangle representing the
// current progress.
func Char(ch rune) Option {
	return option(func(opts *options) {
		opts.gaugeChar = ch
	})
}

// ShowTextProgress configures the Gauge so that it also displays a text
// enumerating the progress. This is the default behavior.
// If the progress is set by a call to Percent(), the displayed text will show
// the percentage, e.g. "50%". If the progress is set by a call to Absolute(),
// the displayed text will whos the absolute numbers, e.g. "5/10".
func ShowTextProgress() Option {
	return option(func(opts *options) {
		opts.hideTextProgress = false
	})
}

// HideTextProgress disables the display of a text enumerating the progress.
func HideTextProgress() Option {
	return option(func(opts *options) {
		opts.hideTextProgress = true
	})
}

// Height sets the height of the drawn Gauge. Must be a positive number.
// Defaults to zero which means the height of the container.
func Height(height int) Option {
	return option(func(opts *options) {
		opts.height = height
	})
}

// TextLabel configures the Gauge to display the provided text.
// If the ShowTextProgress() option is also provided, this label is drawn right
// after the progress text.
func TextLabel(text string) Option {
	return option(func(opts *options) {
		opts.textLabel = text
	})
}

// DefaultColor is the default value for the Color option.
const DefaultColor = cell.ColorGreen

// Color sets the color of the gauge.
func Color(c cell.Color) Option {
	return option(func(opts *options) {
		opts.color = c
	})
}

// DefaultFilledTextColor is the default value for the FilledTextColor option.
const DefaultFilledTextColor = cell.ColorBlack

// FilledTextColor sets color of the text progress and text label for the
// portion of the text that falls within the filled up part of the Gauge. I.e.
// text on the Gauge.
func FilledTextColor(c cell.Color) Option {
	return option(func(opts *options) {
		opts.filledTextColor = c
	})
}

// DefaultEmptyTextColor is the default value for the EmptyTextColor option.
const DefaultEmptyTextColor = cell.ColorDefault

// EmptyTextColor sets color of the text progress and text label for the
// portion of the text that falls outside the filled up part of the Gauge. I.e.
// text in the empty area the Gauge didn't fill yet.
func EmptyTextColor(c cell.Color) Option {
	return option(func(opts *options) {
		opts.emptyTextColor = c
	})
}

// DefaultHorizontalTextAlign is the default value for the HorizontalTextAlign option.
const DefaultHorizontalTextAlign = align.HorizontalCenter

// HorizontalTextAlign sets the horizontal alignment of the text progress and
// text label.
func HorizontalTextAlign(h align.Horizontal) Option {
	return option(func(opts *options) {
		opts.hTextAlign = h
	})
}

// DefaultVerticalTextAlign is the default value for the VerticalTextAlign option.
const DefaultVerticalTextAlign = align.VerticalMiddle

// VerticalTextAlign sets the vertical alignment of the text progress and
// text label.
func VerticalTextAlign(v align.Vertical) Option {
	return option(func(opts *options) {
		opts.vTextAlign = v
	})
}

// Border configures the gauge to have a border of the specified style.
func Border(ls linestyle.LineStyle, cOpts ...cell.Option) Option {
	return option(func(opts *options) {
		opts.border = ls
		opts.borderCellOpts = cOpts
	})
}

// BorderTitle sets a text title within the border.
func BorderTitle(title string) Option {
	return option(func(opts *options) {
		opts.borderTitle = title
	})
}

// BorderTitleAlign sets the horizontal alignment for the border title.
// Defaults to alignment on the left.
func BorderTitleAlign(h align.Horizontal) Option {
	return option(func(opts *options) {
		opts.borderTitleHAlign = h
	})
}