gremlin-cli 0.1.0

A minimal Rust CLI for Apache TinkerPop™
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
116
117
118
119
120
121
122
123

use crate::{actions::Action, command::Command, context::GremlinContext};
use futures::FutureExt;
use gremlin_client::{aio::GremlinClient, ConnectionOptions, GraphSON, TlsOptions};
use std::str::FromStr;
use structopt::StructOpt;

use anyhow::{anyhow, Error};
use structopt::clap::AppSettings;

pub struct ConnectAction;

#[derive(Debug)]
pub enum Serializer {
    GraphSONV1,
    GraphSONV2,
    GraphSONV3,
}

impl FromStr for Serializer {
    type Err = Error;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "graphson_v1" => Ok(Serializer::GraphSONV1),
            "graphson_v2" => Ok(Serializer::GraphSONV2),
            "graphson_v3" => Ok(Serializer::GraphSONV3),
            _ => Err(anyhow!(
                "Serializer {} not valid, possible values [graphson_v1, graphson_v2, graphson_v3]",
                s
            )),
        }
    }
}

impl From<Serializer> for GraphSON {
    fn from(serializer: Serializer) -> Self {
        match serializer {
            Serializer::GraphSONV1 => GraphSON::V1,
            Serializer::GraphSONV2 => GraphSON::V2,
            Serializer::GraphSONV3 => GraphSON::V3,
        }
    }
}
#[derive(Debug, StructOpt)]
#[structopt(name = "connect", no_version, global_settings = &[AppSettings::DisableVersion, AppSettings::NoBinaryName, AppSettings::ColoredHelp])]
struct Connect {
    #[structopt(short, long, default_value = "localhost")]
    host: String,
    #[structopt(short, long, default_value = "8182")]
    port: u16,
    #[structopt(long)]
    ssl: bool,

    #[structopt(long)]
    insecure: bool,

    #[structopt(long)]
    user: Option<String>,

    #[structopt(long)]
    password: Option<String>,

    #[structopt(long, default_value = "graphson_v3")]
    serializer: Serializer,
}

impl ConnectAction {
    pub fn new() -> ConnectAction {
        ConnectAction
    }
}

impl Action for ConnectAction {
    fn name(&self) -> &str {
        "connect"
    }

    fn description(&self) -> &str {
        "Connect to the Gremlin Server."
    }

    fn handle(
        &self,
        _: &crate::context::GremlinContext,
        _: String,
        args: Vec<String>,
    ) -> Vec<Command> {
        match Connect::from_iter_safe(args[0..args.len()].to_vec()) {
            Ok(connect) => {
                let task = |_ctx: &GremlinContext| {
                    let future = async move {
                        let mut options_builder = ConnectionOptions::builder()
                            .host(connect.host.as_str())
                            .port(connect.port)
                            .ssl(connect.ssl)
                            .serializer(connect.serializer.into())
                            .tls_options(TlsOptions {
                                accept_invalid_certs: connect.insecure,
                            });

                        if let (Some(username), Some(password)) = (connect.user, connect.password) {
                            options_builder = options_builder.credentials(&username, &password);
                        }
                        match GremlinClient::connect(options_builder.build()).await {
                            Ok(client) => vec![
                                Command::Update(Box::new(move |ctx| GremlinContext {
                                    client: Some(client),
                                    ..ctx
                                })),
                                Command::Print(Some("Connected!".into())),
                            ],
                            Err(err) => vec![Command::Print(Some(err.to_string()))],
                        }
                    };
                    future.boxed()
                };

                vec![Command::Exec(Box::new(task))]
            }
            Err(e) => vec![Command::Print(Some(format!("{}", e)))],
        }
    }
}