Function optMapExpander

parser/parser_test.go:2597–2623

(meh ExprHelper, target ast.Expr, args []ast.Expr)

var optMapMacro = NewReceiverMacro("optMap", 2, optMapExpander)

func optMapExpander(meh ExprHelper, target ast.Expr, args []ast.Expr) (ast.Expr, *common.Error) {
	varIdent := args[0]
	varName := ""
	switch varIdent.Kind() {
	case ast.IdentKind:
		varName = varIdent.AsIdent()
	default:
		return nil, meh.NewError(varIdent.ID(), "optMap() variable name must be a simple identifier")
	}
	mapExpr := args[1]
	return meh.NewCall(
		operators.Conditional,
		meh.NewMemberCall("hasValue", target),
		meh.NewCall("optional.of",
			meh.NewComprehension(
				meh.NewList(),
				"#unused",
				varName,
				meh.NewMemberCall("value", meh.Copy(target)),
				meh.NewLiteral(types.False),
				meh.NewIdent(varName),
				mapExpr,
			),
		),
		meh.NewCall("optional.none"),
	), nil
}

func newBenchmarkCategoryParser(tb testing.TB) *Parser {
	tb.Helper()