pdk-core 1.7.0-alpha.0

PDK Core
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

// Copyright (c) 2025, Salesforce, Inc.,
// All rights reserved.
// For full license text, see the LICENSE.txt file

use crate::policy_context::authentication::{Authentication, AuthenticationHandler};
use classy::extract::context::FilterContext;
use classy::extract::{Extract, FromContext};
use classy::hl::context::{RequestContext, ResponseContext};
use classy::hl::{HeadersHandler, HttpClientResponse, StreamProperties};
use pdk_script::{
    EvaluationError, Evaluator, HandlerAttributesBinding, IntoValue, PayloadBinding, Script, Value,
};
use std::convert::Infallible;

pub struct DefaultBindings {
    stream: StreamProperties,
    auth: Authentication,
}

impl FromContext<FilterContext> for DefaultBindings {
    type Error = Infallible;

    fn from_context(context: &FilterContext) -> Result<Self, Self::Error> {
        let stream = context.extract()?;
        let auth = context.extract()?;
        Ok(Self { stream, auth })
    }
}

impl FromContext<RequestContext> for DefaultBindings {
    type Error = Infallible;

    fn from_context(context: &RequestContext) -> Result<Self, Self::Error> {
        let stream = context.extract()?;
        let auth = context.extract()?;
        Ok(Self { stream, auth })
    }
}

impl<C> FromContext<ResponseContext<C>> for DefaultBindings {
    type Error = Infallible;

    fn from_context(context: &ResponseContext<C>) -> Result<Self, Self::Error> {
        let stream = context.extract()?;
        let auth = context.extract()?;
        Ok(Self { stream, auth })
    }
}

impl DefaultBindings {
    pub fn evaluator<'a>(&'a self, script: &'a Script) -> DefaultEvaluator<'a> {
        DefaultEvaluator {
            provider: self,
            evaluator: script.evaluator(),
        }
    }
}

pub struct DefaultEvaluator<'a> {
    provider: &'a DefaultBindings,
    evaluator: Evaluator<'a>,
}

impl DefaultEvaluator<'_> {
    pub fn bind_vars<K: IntoValue>(&mut self, name: &str, value: K) {
        self.evaluator.bind_vars(name, value)
    }

    pub fn bind_headers(&mut self, binding: &dyn HeadersHandler) {
        let binding = HandlerAttributesBinding::new(binding, &self.provider.stream);
        self.evaluator.bind_attributes(&binding)
    }

    pub fn bind_payload(&mut self, binding: &dyn PayloadBinding) {
        self.evaluator.bind_payload(binding);
    }

    pub fn bind_http_client_response(&mut self, binding: &HttpClientResponse) {
        self.evaluator.bind_attributes(binding);
        self.evaluator.bind_payload(binding);
    }

    pub fn bind_authentication(&mut self) {
        let auth = self.provider.auth.authentication().unwrap_or_default();
        self.evaluator.bind_authentication(&auth)
    }

    pub fn is_ready(&self) -> bool {
        self.evaluator.is_ready()
    }

    pub fn eval(self) -> Result<Value, EvaluationError> {
        self.evaluator.eval()
    }
}