regorus 0.9.1

A fast, lightweight Rego (OPA policy language) interpreter
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
#![allow(
    clippy::redundant_pub_crate,
    clippy::unused_trait_names,
    clippy::pattern_type_mismatch
)]

//! Functions responsible for building destructuring plans.

use alloc::collections::{BTreeMap, BTreeSet};
use alloc::string::{String, ToString};
use alloc::vec::Vec;

use crate::ast::{Expr, ExprRef};
use crate::compiler::destructuring_planner::context::OverlayBindingContext;
use crate::compiler::destructuring_planner::utils::extract_literal_key;
use crate::compiler::destructuring_planner::{
    DestructuringPlan, ScopingMode, VariableBindingContext,
};

/// Create a destructuring plan for an expression using a variable binding context.
pub fn create_destructuring_plan<T: VariableBindingContext>(
    expr: &ExprRef,
    context: &T,
    scoping: ScopingMode,
) -> Option<DestructuringPlan> {
    let mut newly_bound = BTreeSet::new();
    create_destructuring_plan_with_tracking(expr, context, scoping, &mut newly_bound)
}

/// Create a destructuring plan while tracking newly bound variables.
pub(crate) fn create_destructuring_plan_with_tracking<T: VariableBindingContext>(
    expr: &ExprRef,
    context: &T,
    scoping: ScopingMode,
    newly_bound: &mut BTreeSet<String>,
) -> Option<DestructuringPlan> {
    let overlay = OverlayBindingContext {
        base: context,
        newly_bound,
    };

    match expr.as_ref() {
        // Variable binding
        Expr::Var { span: name, .. } => {
            if name.text() == "_" {
                return Some(DestructuringPlan::Ignore);
            }
            if overlay.is_var_unbound(name.text(), scoping) {
                newly_bound.insert(name.text().to_string());
                Some(DestructuringPlan::Var(name.clone()))
            } else {
                // Already bound - treat as equality check
                Some(DestructuringPlan::EqualityExpr(expr.clone()))
            }
        }

        Expr::String { value, .. }
        | Expr::RawString { value, .. }
        | Expr::Number { value, .. }
        | Expr::Bool { value, .. }
        | Expr::Null { value, .. } => Some(DestructuringPlan::EqualityValue(value.clone())),

        // Array destructuring
        Expr::Array { items, .. } => {
            let mut element_plans = Vec::new();
            for item in items {
                if let Some(plan) =
                    create_destructuring_plan_with_tracking(item, context, scoping, newly_bound)
                {
                    element_plans.push(plan);
                } else {
                    // If any element can't be destructured, fail the whole array
                    return None;
                }
            }
            Some(DestructuringPlan::Array { element_plans })
        }

        // Object destructuring
        Expr::Object { fields, .. } => {
            let mut field_plans = BTreeMap::new();
            let mut dynamic_fields = Vec::new();
            for (_, key_expr, value_expr) in fields {
                if let Some(value_plan) = create_destructuring_plan_with_tracking(
                    value_expr,
                    context,
                    scoping,
                    newly_bound,
                ) {
                    if let Some(key_value) = extract_literal_key(key_expr) {
                        field_plans.insert(key_value, value_plan);
                    } else {
                        dynamic_fields.push((key_expr.clone(), value_plan));
                    }
                } else {
                    return None;
                }
            }

            Some(DestructuringPlan::Object {
                field_plans,
                dynamic_fields,
            })
        }

        // For all others, treat as equality checks
        _ => Some(DestructuringPlan::EqualityExpr(expr.clone())),
    }
}