test

var (
	// Decls represent the standard variables and functions available for testing templates.
	Decls = cel.Declarations(
		decls.NewVar("destination.ip", decls.String),
		decls.NewVar("origin.ip", decls.String),
		decls.NewVar("request.auth.claims", decls.NewMapType(decls.String, decls.Dyn)),
		decls.NewVar("request.time", decls.Timestamp),
		decls.NewVar("resource.name", decls.String),
		decls.NewVar("resource.type", decls.String),
		decls.NewVar("resource.labels", decls.NewMapType(decls.String, decls.String)),
		decls.NewFunction("locationCode",
			decls.NewOverload("location_code_string",
				[]*exprpb.Type{decls.String},
				decls.String,
			),
		),
		decls.NewFunction("rangeLow",
			decls.NewOverload("range_low",
				[]*exprpb.Type{decls.String},
				decls.Int,
			),
		),
		decls.NewFunction("rangeHigh",
			decls.NewOverload("range_high",
				[]*exprpb.Type{decls.String},
				decls.Int,
			),
		),
	)

	// Funcs are the custom function implementations used within templates.
	Funcs = []*functions.Overload{
		{
			Operator: "location_code_string",
			Unary: func(ip ref.Val) ref.Val {
				switch ip.(types.String) {
				case types.String("10.0.0.1"):
					return types.String("us")

				case types.String("10.0.0.2"):
					return types.String("de")
				default:
					return types.String("ir")
				}
			},
		},
		{
			Operator: "range_low",
			Unary: func(arg ref.Val) ref.Val {
				r := arg.(types.String).Value().(string)
				r = strings.TrimSpace(r)
				if r == "" {
					return types.Int(0)
				}
				if strings.Contains(r, "-") {
					rs := strings.Split(r, "-")
					if len(rs) != 2 {
						return types.ValOrErr(arg, "invalid port")
					}
					r = rs[0]
				}
				v, err := strconv.Atoi(r)
				if err != nil {
					return types.ValOrErr(arg, "invalid port")
				}
				return types.Int(v)
			},
		},
		{
			Operator: "range_high",
			Unary: func(arg ref.Val) ref.Val {
				r := arg.(types.String).Value().(string)
				r = strings.TrimSpace(r)
				if r == "" {
					return types.Int(65535)
				}
				if strings.Contains(r, "-") {
					rs := strings.Split(r, "-")
					if len(rs) != 2 {
						return types.ValOrErr(arg, "invalid port")
					}
					r = rs[1]
				}
				v, err := strconv.Atoi(r)
				if err != nil {
					return types.ValOrErr(arg, "invalid port")
				}
				return types.Int(v)
			},
		},
	}
)