akiyosi.goneovim/editor/palette.go

package editor

import (
	"fmt"
	"math"
	"path/filepath"
	"runtime"
	"sort"
	"strings"

	"github.com/akiyosi/goneovim/util"
	"github.com/akiyosi/qt/core"
	"github.com/akiyosi/qt/gui"
	"github.com/akiyosi/qt/svg"
	"github.com/akiyosi/qt/widgets"
)

// Palette is the popup for cmdline
type Palette struct {
	ws               *Workspace
	background       *RGBA
	widget           *widgets.QWidget
	foreground       *RGBA
	scrollCol        *widgets.QWidget
	scrollBar        *widgets.QWidget
	patternWidget    *widgets.QWidget
	resultWidget     *widgets.QWidget
	resultMainWidget *widgets.QWidget
	pattern          *widgets.QLabel
	inactiveFg       *RGBA
	resultType       string
	patternText      string
	itemTypes        []string
	resultItems      []*PaletteResultItem
	showTotal        int
	itemHeight       int
	patternPadding   int
	max              int
	scrollBarPos     int
	width            int
	padding          int
	cursorX          int
	isHTMLText       bool
	hidden           bool
}

// PaletteResultItem is the result item
type PaletteResultItem struct {
	p          *Palette
	widget     *widgets.QWidget
	icon       *svg.QSvgWidget
	base       *widgets.QLabel
	iconType   string
	baseText   string
	iconHidden bool
	hidden     bool
	selected   bool
}

func initPalette() *Palette {
	width := 600
	padding := 8

	mainLayout := widgets.NewQVBoxLayout()
	mainLayout.SetContentsMargins(0, 0, 0, 0)
	mainLayout.SetSpacing(0)
	mainLayout.SetSizeConstraint(widgets.QLayout__SetMinAndMaxSize)
	widget := widgets.NewQWidget(nil, 0)
	widget.SetLayout(mainLayout)
	widget.SetContentsMargins(1, 1, 1, 1)
	// widget.SetFixedWidth(width)
	widget.SetObjectName("palette")

	widget.SetGraphicsEffect(util.DropShadow(0, 15, 130, 120))

	resultMainLayout := widgets.NewQHBoxLayout()
	resultMainLayout.SetContentsMargins(0, 0, 0, 0)
	resultMainLayout.SetSpacing(0)
	resultMainLayout.SetSizeConstraint(widgets.QLayout__SetMinAndMaxSize)

	resultLayout := widgets.NewQVBoxLayout()
	resultLayout.SetContentsMargins(0, 0, 0, 0)
	resultLayout.SetSpacing(0)
	resultLayout.SetSizeConstraint(widgets.QLayout__SetMinAndMaxSize)
	resultWidget := widgets.NewQWidget(nil, 0)
	resultWidget.SetLayout(resultLayout)
	resultWidget.SetStyleSheet("background-color: rgba(0, 0, 0, 0); white-space: pre-wrap;")
	resultWidget.SetContentsMargins(0, 0, 0, 0)

	scrollCol := widgets.NewQWidget(nil, 0)
	scrollCol.SetContentsMargins(0, 0, 0, 0)
	scrollCol.SetFixedWidth(5)
	scrollBar := widgets.NewQWidget(scrollCol, 0)
	scrollBar.SetFixedWidth(5)

	resultMainWidget := widgets.NewQWidget(nil, 0)
	resultMainWidget.SetStyleSheet(" * { background-color: rgba(0, 0, 0, 0); }")
	resultMainWidget.SetContentsMargins(0, 0, 0, 0)
	resultMainLayout.AddWidget(resultWidget, 0, 0)
	resultMainLayout.AddWidget(scrollCol, 0, 0)
	resultMainWidget.SetLayout(resultMainLayout)

	pattern := widgets.NewQLabel(nil, 0)
	pattern.SetContentsMargins(padding, padding, padding, padding)
	pattern.SetFixedWidth(width - padding*2)
	pattern.SetSizePolicy2(widgets.QSizePolicy__Preferred, widgets.QSizePolicy__Maximum)
	patternLayout := widgets.NewQVBoxLayout()
	patternLayout.AddWidget(pattern, 0, 0)
	patternLayout.SetContentsMargins(0, 0, 0, 0)
	patternLayout.SetSpacing(0)
	patternLayout.SetSizeConstraint(widgets.QLayout__SetMinAndMaxSize)
	patternWidget := widgets.NewQWidget(nil, 0)
	patternWidget.SetLayout(patternLayout)
	patternWidget.SetContentsMargins(padding, padding, padding, padding)

	mainLayout.AddWidget(patternWidget, 0, 0)
	mainLayout.AddWidget(resultMainWidget, 0, 0)

	palette := &Palette{
		width:            width,
		widget:           widget,
		padding:          padding,
		resultWidget:     resultWidget,
		resultMainWidget: resultMainWidget,
		pattern:          pattern,
		patternPadding:   padding,
		patternWidget:    patternWidget,
		scrollCol:        scrollCol,
		scrollBar:        scrollBar,
	}

	resultItems := []*PaletteResultItem{}
	max := editor.config.Palette.MaxNumberOfResultItems
	for i := 0; i < max; i++ {
		itemWidget := widgets.NewQWidget(nil, 0)
		itemWidget.SetContentsMargins(0, 0, 0, 0)
		itemLayout := util.NewVFlowLayout(padding, padding*2, 0, 0, 9999)
		itemLayout.SetSizeConstraint(widgets.QLayout__SetMinAndMaxSize)
		itemWidget.SetLayout(itemLayout)
		// itemWidget.SetStyleSheet("background-color: rgba(0, 0, 0, 0);")
		resultLayout.AddWidget(itemWidget, 0, 0)
		icon := svg.NewQSvgWidget(nil)
		icon.SetFixedWidth(editor.iconSize - 1)
		icon.SetFixedHeight(editor.iconSize - 1)
		icon.SetContentsMargins(0, 0, 0, 0)
		// icon.SetStyleSheet("background-color: rgba(0, 0, 0, 0);")
		base := widgets.NewQLabel(nil, 0)
		base.SetText("base")
		base.SetContentsMargins(0, padding, 0, padding)
		// base.SetStyleSheet("background-color: rgba(0, 0, 0, 0); white-space: pre-wrap;")
		base.SetSizePolicy2(widgets.QSizePolicy__Preferred, widgets.QSizePolicy__Maximum)
		itemLayout.AddWidget(icon)
		itemLayout.AddWidget(base)
		resultItem := &PaletteResultItem{
			p:      palette,
			widget: itemWidget,
			icon:   icon,
			base:   base,
		}
		resultItems = append(resultItems, resultItem)
	}
	palette.max = max
	palette.resultItems = resultItems
	return palette
}

func (p *Palette) setParent(win *Window) {
	if win.isExternal {
		p.widget.SetParent(win.extwin)
	} else {
		p.widget.SetParent(p.ws.widget)
	}

}

func (p *Palette) setColor() {
	if p.foreground.equals(editor.colors.widgetFg) && p.background.equals(editor.colors.widgetBg) && p.inactiveFg.equals(editor.colors.inactiveFg) {
		return
	}

	p.foreground = editor.colors.widgetFg
	p.background = editor.colors.widgetBg
	p.inactiveFg = editor.colors.inactiveFg

	fg := p.foreground.String()
	bg := editor.colors.widgetBg
	if bg.HSV().V < 0.1 {
		bg = warpColor(bg, -15)
	} else if bg.HSV().V < 0.1 {
		bg = warpColor(bg, -5)
	}
	inactiveFg := editor.colors.inactiveFg

	transparent := transparent() * transparent()
	if editor.config.Palette.Transparent < 1.0 {
		transparent = editor.config.Palette.Transparent * editor.config.Palette.Transparent
	}

	p.widget.SetStyleSheet(fmt.Sprintf(" .QWidget { background-color: rgba(%d, %d, %d, %f); } * { color: %s; } ", bg.R, bg.G, bg.B, transparent, fg))
	p.scrollBar.SetStyleSheet(fmt.Sprintf("background-color: rgba(%d, %d, %d, %f);", inactiveFg.R, inactiveFg.G, inactiveFg.B, transparent))
	for _, item := range p.resultItems {
		item.widget.SetStyleSheet(fmt.Sprintf(" .QWidget { background-color: rgba(0, 0, 0, 0.0); } * { color: %s; } ", fg))
	}
	if transparent < 1.0 {
		p.patternWidget.SetStyleSheet("background-color: rgba(0, 0, 0, 0);")
		p.pattern.SetStyleSheet("background-color: rgba(0, 0, 0, 0);")
	} else {
		p.pattern.SetStyleSheet(fmt.Sprintf("background-color: %s;", bg.String()))
	}
}

func (p *Palette) resize() {
	parentWidget := p.widget.ParentWidget()
	eWidth := 0
	if parentWidget == nil {
		eWidth = editor.window.Width()
	} else {
		eWidth = parentWidget.Width()
	}
	width := int(math.Trunc(float64(eWidth) * 0.7))
	cursorBoundary := p.padding*4 + p.textLength() + p.patternPadding
	if cursorBoundary > width {
		width = cursorBoundary
	}
	if width > eWidth {
		width = eWidth
		p.pattern.SetAlignment(core.Qt__AlignRight | core.Qt__AlignCenter)
	} else if width <= eWidth {
		if p.pattern.Alignment() != core.Qt__AlignLeft {
			p.pattern.SetAlignment(core.Qt__AlignLeft)
		}
	}

	if p.width == width {
		return
	}
	p.width = width
	p.pattern.SetFixedWidth(p.width - p.padding*2)
	p.widget.SetMaximumWidth(p.width)
	p.widget.SetMinimumWidth(p.width)

	x := eWidth - p.width
	if x < 0 {
		x = 0
	}
	p.widget.Move2(x/2, 10)

	p.showTotal = 0
	for i := p.showTotal; i < len(p.resultItems); i++ {
		p.resultItems[i].hide()
	}
}

func (p *Palette) resizeResultItems() {
	if p.showTotal != 0 {
		return
	}
	itemHeight := p.resultItems[0].widget.SizeHint().Height()
	p.itemHeight = itemHeight
	p.showTotal = int(float64(p.ws.height)/float64(itemHeight)*editor.config.Palette.AreaRatio) - 1
}

func (p *Palette) show() {
	if !p.hidden {
		return
	}
	p.setColor()
	p.resizeResultItems()
	p.hidden = false
	p.widget.Raise()
	p.widget.SetWindowOpacity(1.0)
	p.resize()
	p.widget.Show()
}

func (p *Palette) hide() {
	if p.hidden {
		return
	}
	p.hidden = true
	p.widget.Hide()
}

func (p *Palette) setPattern(text string) {
	p.patternText = text
	p.pattern.SetText(text)
}

func (p *Palette) cursorMove(x int) {
	X := p.textLength()
	var stickOutLen int
	boundary := p.pattern.Width() - (p.padding * 2)
	if X >= boundary {
		stickOutLen = X - boundary
	}
	pos := p.cursorPos(x) - stickOutLen
	if pos < 0 {
		pos = 0
	}

	p.cursorX = pos

	px := float64(p.cursorX + p.patternPadding)
	py := float64(p.patternPadding + p.ws.cursor.horizontalShift)
	c := p.ws.cursor
	if !c.isInPalette {
		c.SetParent(p.pattern)
		p.ws.cursor.isInPalette = true
	}
	if !(c.x == px && c.y == py) {
		c.xprime = c.x
		c.yprime = c.y
		c.x = px
		c.y = py
		c.doAnimate = true
		c.animateMove()
	}

	p.ws.cursor.update()

	p.redrawAllContentInWindows()
}

func (p *Palette) redrawAllContentInWindows() {
	if runtime.GOOS != "windows" {
		return
	}

	p.hide()
	p.resultWidget.Hide()
	p.resultMainWidget.Hide()
	p.show()
	p.resultWidget.Show()
	p.resultMainWidget.Show()
}

func (p *Palette) updateFont() {
	p.widget.SetFont(p.ws.screen.font.qfont)
	p.pattern.SetFont(p.ws.screen.font.qfont)
}

func (p *Palette) textLength() int {
	font := gui.NewQFontMetricsF(editor.font.qfont)
	l := 0
	if p.isHTMLText {
		t := gui.NewQTextDocument(nil)
		t.SetHtml(p.patternText)
		l = int(
			font.HorizontalAdvance(
				t.ToPlainText(),
				-1,
			),
		)
	} else {
		l = int(
			font.HorizontalAdvance(
				p.patternText,
				-1,
			),
		)
	}

	return l
}

func (p *Palette) cursorPos(x int) int {
	font := gui.NewQFontMetricsF(p.ws.screen.font.qfont)
	l := 0
	if p.isHTMLText {
		t := gui.NewQTextDocument(nil)
		t.SetHtml(p.patternText)
		l = int(
			font.HorizontalAdvance(
				t.ToPlainText()[:x],
				-1,
			),
		)
	} else {
		l = int(
			font.HorizontalAdvance(
				p.patternText[:x],
				-1,
			),
		)
	}

	return l
}

func (p *Palette) showSelected(selected int) {
	if p.resultType == "file_line" {
		n := 0
		for i := 0; i <= selected; i++ {
			for n++; n < len(p.itemTypes) && p.itemTypes[n] == "file"; n++ {
			}
		}
		selected = n
	}
	for i, resultItem := range p.resultItems {
		resultItem.setSelected(selected == i)
	}
}

func (f *PaletteResultItem) update() {
	c := editor.colors.selectedBg
	// transparent := editor.config.Editor.Transparent
	transparent := transparent()
	if f.selected {
		f.widget.SetStyleSheet(fmt.Sprintf(".QWidget {background-color: rgba(%d, %d, %d, %f);}", c.R, c.G, c.B, transparent))
	} else {
		f.widget.SetStyleSheet("")
	}
	// f.p.widget.Hide()
	// f.p.widget.Show()

}

func (f *PaletteResultItem) setSelected(selected bool) {
	if f.selected == selected {
		return
	}
	f.selected = selected
	f.update()
}

func (f *PaletteResultItem) show() {
	// if f.hidden {
	f.hidden = false
	f.widget.Show()
	// }
}

func (f *PaletteResultItem) hide() {
	if !f.hidden {
		f.hidden = true
		f.widget.Hide()
	}
}

func (f *PaletteResultItem) setItem(text string, itemType string, match []int) {
	iconType := ""
	path := false
	if itemType == "dir" {
		iconType = "folder"
		path = true
	} else if itemType == "file" {
		iconType = getFileType(text)
		path = true
	} else if itemType == "file_line" {
		iconType = "empty"
	}
	if iconType != "" {
		if iconType != f.iconType {
			f.iconType = iconType
			f.updateIcon()
		}
		f.showIcon()
	} else {
		f.hideIcon()
	}

	formattedText := formatText(text, match, path)
	if formattedText != f.baseText {
		f.baseText = formattedText
		f.base.SetText(f.baseText)
	}
}

func formatText(text string, matchIndex []int, path bool) string {
	sort.Ints(matchIndex)

	color := ""
	if editor.colors.matchFg != nil {
		color = editor.colors.matchFg.Hex()
	}

	match := len(matchIndex) > 0
	if !path || strings.HasPrefix(text, "term://") {
		formattedText := ""
		i := 0
		for _, char := range text {
			if color != "" && len(matchIndex) > 0 && i == matchIndex[0] {
				formattedText += fmt.Sprintf("<font color='%s'>%s</font>", color, string(char))
				matchIndex = matchIndex[1:]
			} else if color != "" && match {
				switch string(char) {
				case " ":
					formattedText += "&nbsp;"
				case "\t":
					formattedText += "&nbsp;&nbsp;&nbsp;&nbsp;"
				case "<":
					formattedText += "&lt;"
				case ">":
					formattedText += "&gt;"
				default:
					formattedText += string(char)
				}
			} else {
				formattedText += string(char)
			}
			i++
		}
		return formattedText
	}

	dirText := ""
	dir := filepath.Dir(text)
	if dir == "." {
		dir = ""
	}
	if dir != "" {
		i := strings.Index(text, dir)
		if i != -1 {
			for j, char := range dir {
				if color != "" && len(matchIndex) > 0 && i+j == matchIndex[0] {
					dirText += fmt.Sprintf("<font color='%s'>%s</font>", color, string(char))
					matchIndex = matchIndex[1:]
				} else {
					dirText += string(char)
				}
			}
		}
	}

	baseText := ""
	base := filepath.Base(text)
	if base != "" {
		i := strings.LastIndex(text, base)
		if i != -1 {
			for j, char := range base {
				if color != "" && len(matchIndex) > 0 && i+j == matchIndex[0] {
					baseText += fmt.Sprintf("<font color='%s'>%s</font>", color, string(char))
					matchIndex = matchIndex[1:]
				} else {
					baseText += string(char)
				}
			}
		}
	}

	return fmt.Sprintf("%s <font color='#838383'>%s</font>", baseText, dirText)
}

func (f *PaletteResultItem) updateIcon() {
	svgContent := editor.getSvg(f.iconType, nil)
	f.icon.Load2(core.NewQByteArray2(svgContent, len(svgContent)))
}

func (f *PaletteResultItem) showIcon() {
	if f.iconHidden {
		f.iconHidden = false
		f.icon.Show()
	}
}

func (f *PaletteResultItem) hideIcon() {
	if !f.iconHidden {
		f.iconHidden = true
		f.icon.Hide()
	}
}