mirror of
https://github.com/go-task/task.git
synced 2024-12-12 10:45:49 +02:00
117 lines
2.8 KiB
Go
117 lines
2.8 KiB
Go
package taskfile
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/go-task/task/v3/internal/execext"
|
|
"github.com/go-task/task/v3/internal/filepathext"
|
|
"github.com/go-task/task/v3/internal/logger"
|
|
"github.com/go-task/task/v3/taskfile/ast"
|
|
)
|
|
|
|
// A FileNode is a node that reads a taskfile from the local filesystem.
|
|
type FileNode struct {
|
|
*BaseNode
|
|
Entrypoint string
|
|
}
|
|
|
|
func NewFileNode(l *logger.Logger, entrypoint, dir string, opts ...NodeOption) (*FileNode, error) {
|
|
var err error
|
|
base := NewBaseNode(opts...)
|
|
entrypoint, dir, err = resolveFileNodeEntrypointAndDir(l, entrypoint, dir)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
base.dir = dir
|
|
return &FileNode{
|
|
BaseNode: base,
|
|
Entrypoint: entrypoint,
|
|
}, nil
|
|
}
|
|
|
|
func (node *FileNode) Location() string {
|
|
return node.Entrypoint
|
|
}
|
|
|
|
func (node *FileNode) Remote() bool {
|
|
return false
|
|
}
|
|
|
|
func (node *FileNode) Read(ctx context.Context) ([]byte, error) {
|
|
f, err := os.Open(node.Location())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
return io.ReadAll(f)
|
|
}
|
|
|
|
// resolveFileNodeEntrypointAndDir resolves checks the values of entrypoint and dir and
|
|
// populates them with default values if necessary.
|
|
func resolveFileNodeEntrypointAndDir(l *logger.Logger, entrypoint, dir string) (string, string, error) {
|
|
var err error
|
|
if entrypoint != "" {
|
|
entrypoint, err = Exists(l, entrypoint)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
if dir == "" {
|
|
dir = filepath.Dir(entrypoint)
|
|
}
|
|
return entrypoint, dir, nil
|
|
}
|
|
if dir == "" {
|
|
dir, err = os.Getwd()
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
}
|
|
entrypoint, err = ExistsWalk(l, dir)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
dir = filepath.Dir(entrypoint)
|
|
return entrypoint, dir, nil
|
|
}
|
|
|
|
func (node *FileNode) ResolveIncludeEntrypoint(include ast.Include) (string, error) {
|
|
// If the file is remote, we don't need to resolve the path
|
|
if strings.Contains(include.Taskfile, "://") {
|
|
return include.Taskfile, nil
|
|
}
|
|
|
|
path, err := execext.Expand(include.Taskfile)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if filepathext.IsAbs(path) {
|
|
return path, nil
|
|
}
|
|
|
|
// NOTE: Uses the directory of the entrypoint (Taskfile), not the current working directory
|
|
// This means that files are included relative to one another
|
|
entrypointDir := filepath.Dir(node.Entrypoint)
|
|
return filepathext.SmartJoin(entrypointDir, path), nil
|
|
}
|
|
|
|
func (node *FileNode) ResolveIncludeDir(include ast.Include) (string, error) {
|
|
path, err := execext.Expand(include.Dir)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if filepathext.IsAbs(path) {
|
|
return path, nil
|
|
}
|
|
|
|
// NOTE: Uses the directory of the entrypoint (Taskfile), not the current working directory
|
|
// This means that files are included relative to one another
|
|
entrypointDir := filepath.Dir(node.Entrypoint)
|
|
return filepathext.SmartJoin(entrypointDir, path), nil
|
|
}
|