fuel-streams 0.0.28

A library for working with streams of Fuel blockchain data
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

use std::str::FromStr;

/// FuelStreamsNetworks; shortened to FuelNetworks for brievity and public familiarity
use serde::{Deserialize, Serialize};
use url::Url;

#[derive(Debug, Clone, Default)]
pub enum FuelNetworkUserRole {
    Admin,
    #[default]
    Default,
}

#[derive(Debug, Copy, Clone, Default, Deserialize, Serialize)]
#[serde(rename_all = "lowercase")]
pub enum FuelNetwork {
    #[default]
    Local,
    Staging,
    Mainnet,
}

impl FromStr for FuelNetwork {
    type Err = String;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "local" => Ok(FuelNetwork::Local),
            "staging" => Ok(FuelNetwork::Staging),
            "mainnet" => Ok(FuelNetwork::Mainnet),
            _ => Err(format!("unknown network: {}", s)),
        }
    }
}

impl std::fmt::Display for FuelNetwork {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            FuelNetwork::Local => write!(f, "local"),
            FuelNetwork::Staging => write!(f, "staging"),
            FuelNetwork::Mainnet => write!(f, "mainnet"),
        }
    }
}

impl FuelNetwork {
    pub fn load_from_env() -> Self {
        match std::env::var("NETWORK").as_deref() {
            Ok("mainnet") => FuelNetwork::Mainnet,
            Ok("staging") => FuelNetwork::Staging,
            _ => FuelNetwork::Local,
        }
    }

    pub fn to_nats_url(&self) -> String {
        match self {
            FuelNetwork::Local => "nats://localhost:4222",
            FuelNetwork::Staging => "nats://stream-staging.fuel.network:4222",
            FuelNetwork::Mainnet => "nats://stream-mainnet.fuel.network:4222",
        }
        .to_string()
    }

    pub fn to_web_url(&self) -> Url {
        match self {
            FuelNetwork::Local => {
                Url::parse("http://localhost:9003").expect("working url")
            }
            FuelNetwork::Staging => {
                Url::parse("https://stream-staging.fuel.network")
                    .expect("working url")
            }
            FuelNetwork::Mainnet => {
                Url::parse("https://stream-mainnet.fuel.network")
                    .expect("working url")
            }
        }
    }

    pub fn to_ws_url(&self) -> Url {
        match self {
            FuelNetwork::Local => {
                Url::parse("ws://0.0.0.0:9003").expect("working url")
            }
            FuelNetwork::Staging => {
                Url::parse("wss://stream-staging.fuel.network")
                    .expect("working url")
            }
            FuelNetwork::Mainnet => {
                Url::parse("wss://stream-mainnet.fuel.network")
                    .expect("working url")
            }
        }
    }
}