1use crate::{generate::Import, stub_type::ModuleRef, type_info::*, TypeInfo};
2use std::{collections::HashSet, fmt};
3
4#[derive(Debug, Clone, PartialEq)]
5pub struct Arg {
6 pub name: &'static str,
7 pub r#type: TypeInfo,
8 pub signature: Option<SignatureArg>,
9}
10
11impl Import for Arg {
12 fn import(&self) -> HashSet<ModuleRef> {
13 self.r#type.import.clone()
14 }
15}
16
17impl From<&ArgInfo> for Arg {
18 fn from(info: &ArgInfo) -> Self {
19 Self {
20 name: info.name,
21 r#type: (info.r#type)(),
22 signature: info.signature.clone(),
23 }
24 }
25}
26
27impl fmt::Display for Arg {
28 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
29 if let Some(signature: &SignatureArg) = &self.signature {
30 match signature {
31 SignatureArg::Ident => write!(f, "{}:{}", self.name, self.r#type),
32 SignatureArg::Assign { default: &&'static LazyLock } => {
33 let default: &String = default;
34 write!(f, "{}:{}={}", self.name, self.r#type, default)
35 }
36 SignatureArg::Star => write!(f, "*"),
37 SignatureArg::Args => write!(f, "*{}", self.name),
38 SignatureArg::Keywords => write!(f, "**{}", self.name),
39 }
40 } else {
41 write!(f, "{}:{}", self.name, self.r#type)
42 }
43 }
44}
45