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

use std::borrow::Cow;

use crate::{
	ConstnessOverride,
	GeneratorEnv,
	TypeRef,
};

#[derive(Clone, Debug, PartialEq)]
pub enum DefinitionLocation {
	Module,
	Type,
	Custom(String),
}

#[derive(Debug)]
pub struct ReturnTypeWrapper<'tu> {
	type_ref: TypeRef<'tu>,
	const_hint: ConstnessOverride,
	definition_location: DefinitionLocation,
	gen_env: &'tu GeneratorEnv<'tu>,
}

impl<'tu> ReturnTypeWrapper<'tu> {
	pub fn new(type_ref: TypeRef<'tu>, definition_location: DefinitionLocation, gen_env: &'tu GeneratorEnv<'tu>) -> Self {
		Self::new_ext(type_ref, ConstnessOverride::No, definition_location, gen_env)
	}

	pub fn new_ext(type_ref: TypeRef<'tu>, const_hint: ConstnessOverride, definition_location: DefinitionLocation, gen_env: &'tu GeneratorEnv<'tu>) -> Self {
		Self { type_ref, const_hint, definition_location, gen_env }
	}

	pub fn type_ref(&self) -> &TypeRef<'tu> {
		&self.type_ref
	}

	pub fn const_hint(&self) -> ConstnessOverride {
		self.const_hint
	}

	pub fn definition_location(&self) -> Cow<str> {
		match &self.definition_location {
			DefinitionLocation::Module => {
				self.gen_env.module().into()
			}
			DefinitionLocation::Type => {
				self.type_ref.rust_module()
			}
			DefinitionLocation::Custom(module) => {
				module.into()
			}
		}
	}
}