print.go

package print

import (
	"errors"

	"go.starlark.net/starlark"

	"github.com/tilt-dev/tilt/internal/tiltfile/starkit"
	"github.com/tilt-dev/tilt/pkg/logger"
)

// Additional functions for print output.
type Extension struct{}

func NewExtension() Extension {
	return Extension{}
}

func (Extension) OnStart(env *starkit.Environment) error {
	err := env.AddBuiltin("warn", warn)
	if err != nil {
		return err
	}
	err = env.AddBuiltin("fail", fail)
	if err != nil {
		return err
	}
	return nil
}

func fail(thread *starlark.Thread, fn *starlark.Builtin, args starlark.Tuple, kwargs []starlark.Tuple) (starlark.Value, error) {
	var msg string
	err := starkit.UnpackArgs(thread, fn.Name(), args, kwargs, "msg", &msg)
	if err != nil {
		return nil, err
	}

	return nil, errors.New(msg)
}

func warn(thread *starlark.Thread, fn *starlark.Builtin, args starlark.Tuple, kwargs []starlark.Tuple) (starlark.Value, error) {
	var msg string
	err := starkit.UnpackArgs(thread, fn.Name(), args, kwargs, "msg", &msg)
	if err != nil {
		return nil, err
	}

	ctx, err := starkit.ContextFromThread(thread)
	if err != nil {
		return nil, err
	}

	logger.Get(ctx).Warnf("%s", msg)

	return starlark.None, nil
}