data.go

package kpp

import (
	"errors"
	"io"
	"io/fs"
	"path"
)

type TestData struct {
	fs fs.FS

	group *TestGroup
	name  string

	inputSize, answerSize int64

	illusExt string
}

func (d TestData) Name() string {
	return d.name
}

func (d TestData) Group() string {
	return d.group.Path()
}

func (d TestData) Input() (io.ReadCloser, error) {
	return d.fs.Open(path.Join("data", d.group.Path(), d.name+".in"))
}

func (d TestData) InputSize() int64 {
	return d.inputSize
}

func (d TestData) Answer() (io.ReadCloser, error) {
	return d.fs.Open(path.Join("data", d.group.Path(), d.name+".ans"))
}

func (d TestData) AnswerSize() int64 {
	return d.answerSize
}

func (d TestData) readText(name string) (string, error) {
	f, err := d.fs.Open(path.Join("data", d.group.Path(), name))
	if errors.Is(err, fs.ErrNotExist) {
		return "", nil
	}
	if err != nil {
		return "", err
	}
	defer f.Close()
	b, err := io.ReadAll(f)
	return string(b), nil
}

func (d TestData) Hint() (string, error) {
	return d.readText(d.name + ".hint")
}

func (d TestData) Description() (string, error) {
	return d.readText(d.name + ".desc")
}

func (d TestData) Illustration() (io.ReadCloser, string, error) {
	if d.illusExt == "" {
		return nil, "", nil
	}
	f, err := d.fs.Open(path.Join("data", d.group.Path(), d.name+"."+d.illusExt))
	return f, d.illusExt, err
}

type TestGroup struct {
	parent *TestGroup

	name string
}

func (g TestGroup) Path() string {
	if g.parent == nil {
		return g.name
	}
	return path.Join(g.parent.Path(), g.name)
}