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

use crate::{Id, ValidId, ValidVocabularyId, VocabularyId};
use rdf_types::{Generator, Vocabulary};
use std::collections::HashMap;
use std::hash::Hash;

pub struct Environment<'n, N: Vocabulary, G> {
	vocabulary: &'n mut N,
	generator: G,
	map: HashMap<N::BlankId, ValidVocabularyId<N>>,
}

impl<'n, N: Vocabulary, G> Environment<'n, N, G> {
	pub fn new(vocabulary: &'n mut N, generator: G) -> Self {
		Self {
			vocabulary,
			generator,
			map: HashMap::new(),
		}
	}
}

impl<'n, V: Vocabulary, G: Generator<V>> Environment<'n, V, G>
where
	V::Iri: Clone,
	V::BlankId: Clone + Hash + Eq,
{
	pub fn assign(&mut self, blank_id: V::BlankId) -> ValidId<V::Iri, V::BlankId> {
		use std::collections::hash_map::Entry;
		match self.map.entry(blank_id) {
			Entry::Occupied(entry) => entry.get().clone(),
			Entry::Vacant(entry) => {
				let id = self.generator.next(self.vocabulary);
				entry.insert(id.clone());
				id
			}
		}
	}

	pub fn assign_node_id(&mut self, r: Option<&VocabularyId<V>>) -> Id<V::Iri, V::BlankId> {
		match r {
			Some(Id::Valid(ValidId::Blank(id))) => self.assign(id.clone()).into(),
			Some(r) => r.clone(),
			None => self.next().into(),
		}
	}

	#[allow(clippy::should_implement_trait)]
	pub fn next(&mut self) -> ValidId<V::Iri, V::BlankId> {
		self.generator.next(self.vocabulary)
	}
}