nu-zenoh 0.4.0

Nu-Zenoh integration
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
112
113
114
115

//
// Copyright (c) 2025 ZettaScale Technology
//
// This program and the accompanying materials are made available under the
// terms of the Eclipse Public License 2.0 which is available at
// http://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
// which is available at https://www.apache.org/licenses/LICENSE-2.0.
//
// SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
//
// Contributors:
//   ZettaScale Zenoh Team, <zenoh@zettascale.tech>
//
use nu_engine::CallExt;
use nu_protocol::{
    engine::{Call, Command, EngineState, Stack},
    PipelineData, ShellError, Signature, SyntaxShape,
};
use zenoh::Wait;

use crate::{call_ext2::CallExt2, signature_ext::SignatureExt, State};

#[derive(Clone)]
pub(crate) struct Put {
    state: State,
}

impl Put {
    pub(crate) fn new(state: State) -> Self {
        Self { state }
    }
}

impl Command for Put {
    fn name(&self) -> &str {
        "zenoh put"
    }

    fn signature(&self) -> Signature {
        Signature::build(self.name())
            .session()
            .zenoh_category()
            .publication(self.state.options.experimental_options)
            .encoding()
            .required(
                "payload",
                SyntaxShape::OneOf(vec![SyntaxShape::String, SyntaxShape::Binary]),
                "Publication payload",
            )
    }

    fn description(&self) -> &str {
        "Zenoh PUT"
    }

    fn run(
        &self,
        engine_state: &EngineState,
        stack: &mut Stack,
        call: &Call,
        _input: PipelineData,
    ) -> Result<PipelineData, ShellError> {
        let key = call.req::<String>(engine_state, stack, 0)?;
        let value = call.req::<Vec<u8>>(engine_state, stack, 1)?;

        self.state
            .with_session(&call.session(engine_state, stack)?, |sess| {
                let mut put = sess.put(key, value);

                if let Some(encoding) = call.encoding(engine_state, stack)? {
                    put = put.encoding(encoding);
                }

                if let Some(priority) = call.priority(engine_state, stack)? {
                    put = put.priority(priority);
                }

                if let Some(congestion_control) = call.congestion_control(
                    engine_state,
                    stack,
                    self.state.options.experimental_options,
                )? {
                    put = put.congestion_control(congestion_control);
                }

                if let Some(reliability) = call.reliable(engine_state, stack)? {
                    put = put.reliability(reliability);
                }

                if let Some(express) = call.express(engine_state, stack)? {
                    put = put.express(express);
                }

                if let Some(attachment) = call.attachment(engine_state, stack)? {
                    put = put.attachment(attachment.as_bytes());
                }

                if let Some(timestamp) = call.timestamp(engine_state, stack)? {
                    put = put.timestamp(timestamp);
                }

                if let Some(destination) = call.allowed_destination(engine_state, stack)? {
                    put = put.allowed_destination(destination);
                }

                put.wait()
            })?
            .map_err(|e| {
                nu_protocol::LabeledError::new("Put operation failed")
                    .with_label(format!("Zenoh put failed: {e}"), call.head)
            })?;

        Ok(nu_protocol::PipelineData::empty())
    }
}