menuTask.go

package main

import (
	"fmt"
	"slices"

	"github.com/VladiTNT/Todo-tcell/tcellUtils"
	"github.com/gdamore/tcell/v2"
)

type taskMenu struct {
	screen        tcell.Screen
	currentInput  *[]rune
	isHoursGood   bool
	isDaysGood    bool
	currentMaxLen int
	inputTitle    []rune
	inputHours    []rune
	inputDays     []rune
	inputInfo     []rune
}

func (t *taskMenu) Draw() {
	t.screen.Clear()
	// Ui Title
	tcellUtils.DrawStringHoriz("-[ Task Edit Menu ]-", 49, 1, t.screen)
	tcellUtils.DrawStringHoriz("In this menu you can add and edit tasks, write each field and press <KeyDown> to jump to the next one. Press <KeyUp>", 4, 2, t.screen)
	tcellUtils.DrawStringHoriz("to jump to the previous field. If you are done, press <Enter> to add/modify the task, or <Esc> to exit.", 4, 3, t.screen)

	// Ui Margins
	tcellUtils.DrawLine('=', 2, 122, 5, t.screen)
	tcellUtils.DrawColumn('|', 4, 6, 16, t.screen)
	tcellUtils.DrawColumn('=', 3, 6, 16, t.screen)
	tcellUtils.DrawColumn('|', 2, 6, 16, t.screen)

	tcellUtils.DrawColumn('|', 120, 6, 16, t.screen)
	tcellUtils.DrawColumn('=', 121, 6, 16, t.screen)
	tcellUtils.DrawColumn('|', 122, 6, 16, t.screen)
	tcellUtils.DrawLine('=', 2, 122, 17, t.screen)

	// Side bars
	tcellUtils.DrawColumn('|', 0, 0, 31, t.screen)
	tcellUtils.DrawColumn('|', 1, 0, 31, t.screen)

	tcellUtils.DrawColumn('|', 123, 0, 31, t.screen)
	tcellUtils.DrawColumn('|', 124, 0, 31, t.screen)

	tcellUtils.DrawLine('=', 2, 122, 0, t.screen)

	tcellUtils.DrawLine('=', 2, 122, 26, t.screen)
	for i := 27; i < 32; i++ {
		tcellUtils.DrawLine('-', 2, 122, i, t.screen)
	}

	// Ui Text
	tcellUtils.DrawStringHoriz("Input Title:", 6, 7, t.screen)
	tcellUtils.DrawStringHoriz("Add hours until deadline:", 6, 9, t.screen)
	tcellUtils.DrawStringHoriz("Add days until deadline:", 6, 11, t.screen)
	tcellUtils.DrawStringHoriz("Extra notes:", 6, 13, t.screen)

	// InputTitle
	tcellUtils.DrawLine('.', 19, 118, 7, t.screen)
	for i, r := range t.inputTitle {
		t.screen.SetContent(19+i, 7, r, nil, tcell.StyleDefault.Foreground(tcell.ColorLightGreen))
	}

	validRunes := []rune{'0', '1', '2', '3', '4', '5', '6', '7', '8', '9'}

	// InputHours
	t.isHoursGood = true
	tcellUtils.DrawLine('.', 32, 35, 9, t.screen)
	for i, r := range t.inputHours {
		t.screen.SetContent(32+i, 9, r, nil, tcell.StyleDefault.Foreground(tcell.ColorLightGreen))
		if !slices.Contains(validRunes, r) {
			t.isHoursGood = false
			tcellUtils.DrawColorHoriz("Error: invalid rune, must be a number!", 6, 10, t.screen, tcell.ColorRed)
		} else {
			t.isHoursGood = true
		}
	}

	// InputDays
	t.isDaysGood = true
	tcellUtils.DrawLine('.', 31, 33, 11, t.screen)
	for i, r := range t.inputDays {
		t.screen.SetContent(31+i, 11, r, nil, tcell.StyleDefault.Foreground(tcell.ColorLightGreen))
		if !slices.Contains(validRunes, r) {
			t.isDaysGood = false
			tcellUtils.DrawColorHoriz("Error: invalid rune, must be a number!", 6, 12, t.screen, tcell.ColorRed)
		} else {
			t.isDaysGood = true
		}
	}

	// InputInfo
	tcellUtils.DrawLine('.', 19, 118, 13, t.screen)
	tcellUtils.DrawLine('.', 19, 118, 14, t.screen)
	tcellUtils.DrawLine('.', 19, 118, 15, t.screen)
	for i, r := range t.inputInfo {
		switch {
		case i < 100:
			t.screen.SetContent(19+i, 13, r, nil, tcell.StyleDefault.Foreground(tcell.ColorLightGreen))
		case i > 199:
			t.screen.SetContent(19+i-200, 15, r, nil, tcell.StyleDefault.Foreground(tcell.ColorLightGreen))
		default:
			t.screen.SetContent(19+i-100, 14, r, nil, tcell.StyleDefault.Foreground(tcell.ColorLightGreen))
		}
	}

	// Cursor
	switch t.currentInput {
	case &t.inputTitle:
		t.screen.SetContent(19+len(*t.currentInput), 7, '_', nil, tcell.StyleDefault.Foreground(tcell.ColorGold))
	case &t.inputHours:
		t.screen.SetContent(32+len(*t.currentInput), 9, '_', nil, tcell.StyleDefault.Foreground(tcell.ColorGold))
	case &t.inputDays:
		t.screen.SetContent(31+len(*t.currentInput), 11, '_', nil, tcell.StyleDefault.Foreground(tcell.ColorGold))
	case &t.inputInfo:
		switch {
		case len(*t.currentInput) < 100:
			t.screen.SetContent(len(*t.currentInput)+19, 13, '_', nil, tcell.StyleDefault.Foreground(tcell.ColorGold))
		case len(*t.currentInput) > 199:
			t.screen.SetContent(len(*t.currentInput)+19-200, 15, '_', nil, tcell.StyleDefault.Foreground(tcell.ColorGold))
		default:
			t.screen.SetContent(len(*t.currentInput)+19-100, 14, '_', nil, tcell.StyleDefault.Foreground(tcell.ColorGold))
		}
	}

	// Fields info at the bottom
	titleColorStatus := tcell.ColorRed
	titleStatus := "✖"
	if len(t.inputTitle) > 0 {
		titleColorStatus = tcell.ColorLightGreen
		titleStatus = "✔"
	}
	titleString := fmt.Sprintf("- Title[%d/100]: %s", len(t.inputTitle), titleStatus)
	tcellUtils.DrawColorHoriz(titleString, 6, 19, t.screen, titleColorStatus)

	daysColorStatus := tcell.ColorGray
	daysStatus := "✖"
	if len(t.inputDays) > 0 && t.isDaysGood {
		daysStatus = "✔"
		daysColorStatus = tcell.ColorLightGreen
	}
	daysString := fmt.Sprintf("- Days (optional): %s", daysStatus)
	tcellUtils.DrawColorHoriz(daysString, 6, 21, t.screen, daysColorStatus)

	hoursColorStatus := tcell.ColorRed
	hoursStatus := "✖"

	if len(t.inputDays) > 0 && t.isDaysGood {
		hoursColorStatus = tcell.ColorGray
	}

	if len(t.inputHours) > 0 && t.isHoursGood {
		hoursStatus = "✔"
		hoursColorStatus = tcell.ColorLightGreen
	}
	hoursString := fmt.Sprintf("- Hours: %s", hoursStatus)
	tcellUtils.DrawColorHoriz(hoursString, 6, 20, t.screen, hoursColorStatus)

	infoColorStatus := tcell.ColorGray
	infoStatus := "✖"
	if len(t.inputInfo) > 0 {
		infoStatus = "✔"
		infoColorStatus = tcell.ColorLightGreen
	}
	infoString := fmt.Sprintf("- Info[%d/300] (optional): %s", len(t.inputInfo), infoStatus)
	tcellUtils.DrawColorHoriz(infoString, 6, 22, t.screen, infoColorStatus)

	canAddColorStatus := tcell.ColorRed
	canAdd := "=> Task can't be added, requires a valid title and a deadline."
	if titleStatus == "✔" && t.isHoursGood && t.isDaysGood && (hoursStatus == "✔" || daysStatus == "✔") {
		canAdd = "=> Task can be added! Press <Enter> to add."
		canAddColorStatus = tcell.ColorLightGreen
	}
	tcellUtils.DrawColorHoriz(canAdd, 6, 24, t.screen, canAddColorStatus)

	t.screen.Show()
}

func (t *taskMenu) Menu(a *App, isEdit bool) {
	t.currentInput = &t.inputTitle
	t.currentMaxLen = 100
	quit := false

	for !quit {
		t.Draw()

		isGood := len(t.inputTitle) > 0 && (len(t.inputHours) > 0 || len(t.inputDays) > 0)

		ev := t.screen.PollEvent()
		switch ev := ev.(type) {
		case *tcell.EventKey:
			switch ev.Key() {
			case tcell.KeyEsc:
				quit = true
			case tcell.KeyEnter:
				if isGood && t.isHoursGood && t.isDaysGood {
					t.AddTask(a, isEdit)
					quit = true
				}
			case tcell.KeyUp:
				switch t.currentInput {
				case &t.inputHours:
					t.currentInput = &t.inputTitle
					t.currentMaxLen = 100
				case &t.inputDays:
					t.currentInput = &t.inputHours
					t.currentMaxLen = 4
				case &t.inputInfo:
					t.currentInput = &t.inputDays
					t.currentMaxLen = 3
				}
			case tcell.KeyDown:
				switch t.currentInput {
				case &t.inputTitle:
					t.currentInput = &t.inputHours
					t.currentMaxLen = 4
				case &t.inputHours:
					if t.isHoursGood {
						t.currentInput = &t.inputDays
						t.currentMaxLen = 3
					}
				case &t.inputDays:
					if t.isDaysGood {
						t.currentInput = &t.inputInfo
						t.currentMaxLen = 300
					}
				}
			case tcell.KeyBackspace, tcell.KeyBackspace2:
				if len(*t.currentInput) > 0 {
					*t.currentInput = (*t.currentInput)[:len(*t.currentInput)-1]
				}
			default:
				if ev.Rune() != 0 && len(*t.currentInput) < t.currentMaxLen {
					(*t.currentInput) = append(*t.currentInput, ev.Rune())
				}
			}
		}
	}

	t.inputTitle = []rune{}
	t.inputHours = []rune{}
	t.inputDays = []rune{}
	t.inputInfo = []rune{}
}