std_basename.go

// Code generated by __generator__/interpreter.go at once

package builtin

import (
	"path/filepath"
	"strings"

	"github.com/ysugimoto/falco/interpreter/context"
	"github.com/ysugimoto/falco/interpreter/function/errors"
	"github.com/ysugimoto/falco/interpreter/value"
)

const Std_basename_Name = "std.basename"

var Std_basename_ArgumentTypes = []value.Type{value.StringType}

func Std_basename_Validate(args []value.Value) error {
	if len(args) != 1 {
		return errors.ArgumentNotEnough(Std_basename_Name, 1, args)
	}
	for i := range args {
		if args[i].Type() != Std_basename_ArgumentTypes[i] {
			return errors.TypeMismatch(Std_basename_Name, i+1, Std_basename_ArgumentTypes[i], args[i].Type())
		}
	}
	return nil
}

// Fastly built-in function implementation of std.basename
// Arguments may be:
// - STRING
// Reference: https://developer.fastly.com/reference/vcl/functions/strings/std-basename/
func Std_basename(ctx *context.Context, args ...value.Value) (value.Value, error) {
	// Argument validations
	if err := Std_basename_Validate(args); err != nil {
		return value.Null, err
	}

	s := value.Unwrap[*value.String](args[0])
	switch s.Value {
	case ".", "":
		return &value.String{Value: "."}, nil
	case "..":
		return &value.String{Value: ".."}, nil
	case "/":
		return &value.String{Value: "/"}, nil
	default:
		base := filepath.Base(strings.TrimSuffix(s.Value, "/"))
		return &value.String{Value: base}, nil
	}
}