package tasks

import (
	"context"
	"fmt"
	"os"
	"strings"

	"github.com/LazyBachelor/LazyPM/internal/service"
	"github.com/LazyBachelor/LazyPM/pkg/task"
	taskui "github.com/LazyBachelor/LazyPM/pkg/task/ui"
	"github.com/charmbracelet/huh"
)

const codingDescription = `You are tasked with writing a simple function.

Write a function that takes two integers and returns their sum.
The function should be named "Add" and be part of the "coding" package.`

var textFileContent = codingDescription + `
Please write your code below this line!
############################################################
`

type CodingTask struct {
	svc *service.Services
}

func NewCodingTask(svc *service.Services) *CodingTask {
	return &CodingTask{svc: svc}
}

func (t *CodingTask) Config() task.TaskConfig {
	return BaseConfig().WithStatisticsStoragePath("./.pm/coding-task-stats.json")
}

func (t *CodingTask) Details() taskui.TaskDetails {
	return BaseDetails().WithTitle("Coding Task").WithDescription(codingDescription)
}

func (t *CodingTask) Questions(interfaceType task.InterfaceType) (questions taskui.Questions) {
	return BaseQuestions(interfaceType).
		With(ReplQuestion(interfaceType,
			huh.NewInput().Title("Please write your code in the 'code.txt' file and save it.").
				Placeholder("Write your code here...")),
		)
}

func (t *CodingTask) Setup(ctx context.Context) error {
	if err := ClearIssues(t.svc); err != nil {
		return err
	}

	content := textFileContent

	if err := os.WriteFile("./code.txt", []byte(content), 0644); err != nil {
		return err
	}

	return nil
}

func (t *CodingTask) Validate(ctx context.Context) (bool, error) {

	file, err := os.ReadFile("./code.txt")
	if err != nil {
		return false, err
	}

	if string(file) == "" {
		return false, fmt.Errorf("the file is empty")
	}

	code, ok := strings.CutPrefix(string(file), textFileContent)
	if !ok {
		return false, fmt.Errorf("the file content is not in the expected format")
	}

	code = strings.TrimSpace(code)

	if !strings.Contains(code, "package coding") {
		return false, fmt.Errorf("the code does not belong to the 'coding' package")
	}

	if !strings.Contains(code, "func Add") {
		return false, fmt.Errorf("the function 'Add' is not defined")
	}

	if !strings.Contains(code, "return") {
		return false, fmt.Errorf("the function does not contain a return statement")
	}

	return true, nil
}