task.go

package main

import (
	"errors"
	"os"
	"strconv"
	"time"

	"github.com/aquasecurity/table"
	"github.com/fatih/color"
)

type Task struct {
	Title       string
	Completed   bool
	CreatedAt   time.Time
	CompletedAt *time.Time
}

type Tasks []Task

func (t *Tasks) Add(title string) {
	task := Task{
		Title:     title,
		Completed: false,
		CreatedAt: time.Now(),
	}
	*t = append(*t, task)
}

func (t *Tasks) ValidateIndex(index int) error {
	if index < 0 || index >= len(*t) {
		return errors.New("invalid index")
	}
	return nil
}

func (t *Tasks) Delete(index int) error {
	if err := t.ValidateIndex(index); err != nil {
		return err
	}
	*t = append((*t)[:index], (*t)[index+1:]...)
	return nil
}

// Delete All Task
func (t *Tasks) DeleteAll() error {
	*t = nil
	return nil
}

func (t *Tasks) Toggle(index int) error {
	task := *t

	if err := task.ValidateIndex(index); err != nil {
		return err
	}
	isCompleted := task[index].Completed

	if !isCompleted {
		completionTime := time.Now()
		task[index].CompletedAt = &completionTime
	}

	task[index].Completed = !isCompleted

	return nil
}

func (t *Tasks) Edit(index int, title string) error {
	task := *t

	if err := task.ValidateIndex(index); err != nil {
		return err
	}

	task[index].Title = title

	return nil
}

func (t *Tasks) Print() {
	table := table.New(os.Stdout)
	table.SetRowLines(false)
	table.SetHeaders("#", "Task", "Completed", "Created At", "Completed At")
	for index, todo := range *t {
		completed := "No"
		completedAt := ""

		if todo.Completed {
			completed = "Yes"
			if todo.CompletedAt != nil {
				completedAt = todo.CompletedAt.Format(time.RFC1123)
			}
		}

		if todo.Completed {
			table.AddRow(
				color.GreenString(strconv.Itoa(index)),
				color.GreenString(todo.Title),
				color.GreenString(completed),
				color.GreenString(todo.CreatedAt.Format(time.RFC1123)),
				color.GreenString(completedAt),
			)
		} else {
			table.AddRow(
				color.RedString(strconv.Itoa(index)),
				color.RedString(todo.Title),
				color.RedString(completed),
				color.RedString(todo.CreatedAt.Format(time.RFC1123)),
				color.RedString(completedAt),
			)
		}
	}
	table.Render()
}