fetcher 0.15.3

Data automation and pipelining framework
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

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

//! This module contains the [`Set`] field transform

use rand::seq::IndexedRandom;
use std::convert::Infallible;

use super::TransformField;
use crate::{
	StaticStr,
	actions::transforms::result::{OptionUnwrapTransformResultExt, TransformResult as TrRes},
};

/// Set a field to a hardcoded string, a random string, or nothing at all
#[derive(Debug)]
pub enum Set {
	/// One single always the same hardcoded string
	Single(StaticStr),

	/// A random string from a list of hardcoded strings
	Random(Vec<StaticStr>),

	/// Empty the field. In other words: make it [`None`]
	Empty,
}

impl Set {
	/// Creates a new [`Set::single`] with the provided string
	pub fn single<S>(string: S) -> Self
	where
		S: Into<StaticStr>,
	{
		Self::Single(string.into())
	}

	/// Creates a new [`Set::random`] with the provided strings
	pub fn random<I, S>(iter: I) -> Self
	where
		I: IntoIterator<Item = S>,
		S: Into<StaticStr>,
	{
		Self::Random(iter.into_iter().map(Into::into).collect())
	}
}

impl TransformField for Set {
	type Err = Infallible;

	async fn transform_field(&mut self, _value: Option<&str>) -> Result<TrRes<String>, Self::Err> {
		Ok(match self {
			Set::Single(x) => TrRes::New(x.to_string()),
			Set::Random(vec) => vec
				.choose(&mut rand::rng())
				.map(ToString::to_string)
				.unwrap_or_empty(),
			Set::Empty => TrRes::Empty,
		})
	}
}