]> git.lizzy.rs Git - rust.git/blob - src/librustc_save_analysis/sig.rs
Auto merge of #71292 - marmeladema:queries-local-def-id, r=eddyb
[rust.git] / src / librustc_save_analysis / sig.rs
1 // A signature is a string representation of an item's type signature, excluding
2 // any body. It also includes ids for any defs or refs in the signature. For
3 // example:
4 //
5 // ```
6 // fn foo(x: String) {
7 //     println!("{}", x);
8 // }
9 // ```
10 // The signature string is something like "fn foo(x: String) {}" and the signature
11 // will have defs for `foo` and `x` and a ref for `String`.
12 //
13 // All signature text should parse in the correct context (i.e., in a module or
14 // impl, etc.). Clients may want to trim trailing `{}` or `;`. The text of a
15 // signature is not guaranteed to be stable (it may improve or change as the
16 // syntax changes, or whitespace or punctuation may change). It is also likely
17 // not to be pretty - no attempt is made to prettify the text. It is recommended
18 // that clients run the text through Rustfmt.
19 //
20 // This module generates Signatures for items by walking the AST and looking up
21 // references.
22 //
23 // Signatures do not include visibility info. I'm not sure if this is a feature
24 // or an ommission (FIXME).
25 //
26 // FIXME where clauses need implementing, defs/refs in generics are mostly missing.
27
28 use crate::{id_from_def_id, id_from_node_id, SaveContext};
29
30 use rls_data::{SigElement, Signature};
31
32 use rustc_ast::ast::{self, Extern, NodeId};
33 use rustc_ast_pretty::pprust;
34 use rustc_hir::def::{DefKind, Res};
35
36 pub fn item_signature(item: &ast::Item, scx: &SaveContext<'_, '_>) -> Option<Signature> {
37     if !scx.config.signatures {
38         return None;
39     }
40     item.make(0, None, scx).ok()
41 }
42
43 pub fn foreign_item_signature(
44     item: &ast::ForeignItem,
45     scx: &SaveContext<'_, '_>,
46 ) -> Option<Signature> {
47     if !scx.config.signatures {
48         return None;
49     }
50     item.make(0, None, scx).ok()
51 }
52
53 /// Signature for a struct or tuple field declaration.
54 /// Does not include a trailing comma.
55 pub fn field_signature(field: &ast::StructField, scx: &SaveContext<'_, '_>) -> Option<Signature> {
56     if !scx.config.signatures {
57         return None;
58     }
59     field.make(0, None, scx).ok()
60 }
61
62 /// Does not include a trailing comma.
63 pub fn variant_signature(variant: &ast::Variant, scx: &SaveContext<'_, '_>) -> Option<Signature> {
64     if !scx.config.signatures {
65         return None;
66     }
67     variant.make(0, None, scx).ok()
68 }
69
70 pub fn method_signature(
71     id: NodeId,
72     ident: ast::Ident,
73     generics: &ast::Generics,
74     m: &ast::FnSig,
75     scx: &SaveContext<'_, '_>,
76 ) -> Option<Signature> {
77     if !scx.config.signatures {
78         return None;
79     }
80     make_method_signature(id, ident, generics, m, scx).ok()
81 }
82
83 pub fn assoc_const_signature(
84     id: NodeId,
85     ident: ast::Name,
86     ty: &ast::Ty,
87     default: Option<&ast::Expr>,
88     scx: &SaveContext<'_, '_>,
89 ) -> Option<Signature> {
90     if !scx.config.signatures {
91         return None;
92     }
93     make_assoc_const_signature(id, ident, ty, default, scx).ok()
94 }
95
96 pub fn assoc_type_signature(
97     id: NodeId,
98     ident: ast::Ident,
99     bounds: Option<&ast::GenericBounds>,
100     default: Option<&ast::Ty>,
101     scx: &SaveContext<'_, '_>,
102 ) -> Option<Signature> {
103     if !scx.config.signatures {
104         return None;
105     }
106     make_assoc_type_signature(id, ident, bounds, default, scx).ok()
107 }
108
109 type Result = std::result::Result<Signature, &'static str>;
110
111 trait Sig {
112     fn make(&self, offset: usize, id: Option<NodeId>, scx: &SaveContext<'_, '_>) -> Result;
113 }
114
115 fn extend_sig(
116     mut sig: Signature,
117     text: String,
118     defs: Vec<SigElement>,
119     refs: Vec<SigElement>,
120 ) -> Signature {
121     sig.text = text;
122     sig.defs.extend(defs.into_iter());
123     sig.refs.extend(refs.into_iter());
124     sig
125 }
126
127 fn replace_text(mut sig: Signature, text: String) -> Signature {
128     sig.text = text;
129     sig
130 }
131
132 fn merge_sigs(text: String, sigs: Vec<Signature>) -> Signature {
133     let mut result = Signature { text, defs: vec![], refs: vec![] };
134
135     let (defs, refs): (Vec<_>, Vec<_>) = sigs.into_iter().map(|s| (s.defs, s.refs)).unzip();
136
137     result.defs.extend(defs.into_iter().flat_map(|ds| ds.into_iter()));
138     result.refs.extend(refs.into_iter().flat_map(|rs| rs.into_iter()));
139
140     result
141 }
142
143 fn text_sig(text: String) -> Signature {
144     Signature { text, defs: vec![], refs: vec![] }
145 }
146
147 fn push_extern(text: &mut String, ext: Extern) {
148     match ext {
149         Extern::None => {}
150         Extern::Implicit => text.push_str("extern "),
151         Extern::Explicit(abi) => text.push_str(&format!("extern \"{}\" ", abi.symbol)),
152     }
153 }
154
155 impl Sig for ast::Ty {
156     fn make(&self, offset: usize, _parent_id: Option<NodeId>, scx: &SaveContext<'_, '_>) -> Result {
157         let id = Some(self.id);
158         match self.kind {
159             ast::TyKind::Slice(ref ty) => {
160                 let nested = ty.make(offset + 1, id, scx)?;
161                 let text = format!("[{}]", nested.text);
162                 Ok(replace_text(nested, text))
163             }
164             ast::TyKind::Ptr(ref mt) => {
165                 let prefix = match mt.mutbl {
166                     ast::Mutability::Mut => "*mut ",
167                     ast::Mutability::Not => "*const ",
168                 };
169                 let nested = mt.ty.make(offset + prefix.len(), id, scx)?;
170                 let text = format!("{}{}", prefix, nested.text);
171                 Ok(replace_text(nested, text))
172             }
173             ast::TyKind::Rptr(ref lifetime, ref mt) => {
174                 let mut prefix = "&".to_owned();
175                 if let &Some(ref l) = lifetime {
176                     prefix.push_str(&l.ident.to_string());
177                     prefix.push(' ');
178                 }
179                 if let ast::Mutability::Mut = mt.mutbl {
180                     prefix.push_str("mut ");
181                 };
182
183                 let nested = mt.ty.make(offset + prefix.len(), id, scx)?;
184                 let text = format!("{}{}", prefix, nested.text);
185                 Ok(replace_text(nested, text))
186             }
187             ast::TyKind::Never => Ok(text_sig("!".to_owned())),
188             ast::TyKind::CVarArgs => Ok(text_sig("...".to_owned())),
189             ast::TyKind::Tup(ref ts) => {
190                 let mut text = "(".to_owned();
191                 let mut defs = vec![];
192                 let mut refs = vec![];
193                 for t in ts {
194                     let nested = t.make(offset + text.len(), id, scx)?;
195                     text.push_str(&nested.text);
196                     text.push(',');
197                     defs.extend(nested.defs.into_iter());
198                     refs.extend(nested.refs.into_iter());
199                 }
200                 text.push(')');
201                 Ok(Signature { text, defs, refs })
202             }
203             ast::TyKind::Paren(ref ty) => {
204                 let nested = ty.make(offset + 1, id, scx)?;
205                 let text = format!("({})", nested.text);
206                 Ok(replace_text(nested, text))
207             }
208             ast::TyKind::BareFn(ref f) => {
209                 let mut text = String::new();
210                 if !f.generic_params.is_empty() {
211                     // FIXME defs, bounds on lifetimes
212                     text.push_str("for<");
213                     text.push_str(
214                         &f.generic_params
215                             .iter()
216                             .filter_map(|param| match param.kind {
217                                 ast::GenericParamKind::Lifetime { .. } => {
218                                     Some(param.ident.to_string())
219                                 }
220                                 _ => None,
221                             })
222                             .collect::<Vec<_>>()
223                             .join(", "),
224                     );
225                     text.push('>');
226                 }
227
228                 if let ast::Unsafe::Yes(_) = f.unsafety {
229                     text.push_str("unsafe ");
230                 }
231                 push_extern(&mut text, f.ext);
232                 text.push_str("fn(");
233
234                 let mut defs = vec![];
235                 let mut refs = vec![];
236                 for i in &f.decl.inputs {
237                     let nested = i.ty.make(offset + text.len(), Some(i.id), scx)?;
238                     text.push_str(&nested.text);
239                     text.push(',');
240                     defs.extend(nested.defs.into_iter());
241                     refs.extend(nested.refs.into_iter());
242                 }
243                 text.push(')');
244                 if let ast::FnRetTy::Ty(ref t) = f.decl.output {
245                     text.push_str(" -> ");
246                     let nested = t.make(offset + text.len(), None, scx)?;
247                     text.push_str(&nested.text);
248                     text.push(',');
249                     defs.extend(nested.defs.into_iter());
250                     refs.extend(nested.refs.into_iter());
251                 }
252
253                 Ok(Signature { text, defs, refs })
254             }
255             ast::TyKind::Path(None, ref path) => path.make(offset, id, scx),
256             ast::TyKind::Path(Some(ref qself), ref path) => {
257                 let nested_ty = qself.ty.make(offset + 1, id, scx)?;
258                 let prefix = if qself.position == 0 {
259                     format!("<{}>::", nested_ty.text)
260                 } else if qself.position == 1 {
261                     let first = pprust::path_segment_to_string(&path.segments[0]);
262                     format!("<{} as {}>::", nested_ty.text, first)
263                 } else {
264                     // FIXME handle path instead of elipses.
265                     format!("<{} as ...>::", nested_ty.text)
266                 };
267
268                 let name = pprust::path_segment_to_string(path.segments.last().ok_or("Bad path")?);
269                 let res = scx.get_path_res(id.ok_or("Missing id for Path")?);
270                 let id = id_from_def_id(res.def_id());
271                 if path.segments.len() - qself.position == 1 {
272                     let start = offset + prefix.len();
273                     let end = start + name.len();
274
275                     Ok(Signature {
276                         text: prefix + &name,
277                         defs: vec![],
278                         refs: vec![SigElement { id, start, end }],
279                     })
280                 } else {
281                     let start = offset + prefix.len() + 5;
282                     let end = start + name.len();
283                     // FIXME should put the proper path in there, not elipses.
284                     Ok(Signature {
285                         text: prefix + "...::" + &name,
286                         defs: vec![],
287                         refs: vec![SigElement { id, start, end }],
288                     })
289                 }
290             }
291             ast::TyKind::TraitObject(ref bounds, ..) => {
292                 // FIXME recurse into bounds
293                 let nested = pprust::bounds_to_string(bounds);
294                 Ok(text_sig(nested))
295             }
296             ast::TyKind::ImplTrait(_, ref bounds) => {
297                 // FIXME recurse into bounds
298                 let nested = pprust::bounds_to_string(bounds);
299                 Ok(text_sig(format!("impl {}", nested)))
300             }
301             ast::TyKind::Array(ref ty, ref v) => {
302                 let nested_ty = ty.make(offset + 1, id, scx)?;
303                 let expr = pprust::expr_to_string(&v.value).replace('\n', " ");
304                 let text = format!("[{}; {}]", nested_ty.text, expr);
305                 Ok(replace_text(nested_ty, text))
306             }
307             ast::TyKind::Typeof(_)
308             | ast::TyKind::Infer
309             | ast::TyKind::Err
310             | ast::TyKind::ImplicitSelf
311             | ast::TyKind::MacCall(_) => Err("Ty"),
312         }
313     }
314 }
315
316 impl Sig for ast::Item {
317     fn make(&self, offset: usize, _parent_id: Option<NodeId>, scx: &SaveContext<'_, '_>) -> Result {
318         let id = Some(self.id);
319
320         match self.kind {
321             ast::ItemKind::Static(ref ty, m, ref expr) => {
322                 let mut text = "static ".to_owned();
323                 if m == ast::Mutability::Mut {
324                     text.push_str("mut ");
325                 }
326                 let name = self.ident.to_string();
327                 let defs = vec![SigElement {
328                     id: id_from_node_id(self.id, scx),
329                     start: offset + text.len(),
330                     end: offset + text.len() + name.len(),
331                 }];
332                 text.push_str(&name);
333                 text.push_str(": ");
334
335                 let ty = ty.make(offset + text.len(), id, scx)?;
336                 text.push_str(&ty.text);
337
338                 if let Some(expr) = expr {
339                     text.push_str(" = ");
340                     let expr = pprust::expr_to_string(expr).replace('\n', " ");
341                     text.push_str(&expr);
342                 }
343
344                 text.push(';');
345
346                 Ok(extend_sig(ty, text, defs, vec![]))
347             }
348             ast::ItemKind::Const(_, ref ty, ref expr) => {
349                 let mut text = "const ".to_owned();
350                 let name = self.ident.to_string();
351                 let defs = vec![SigElement {
352                     id: id_from_node_id(self.id, scx),
353                     start: offset + text.len(),
354                     end: offset + text.len() + name.len(),
355                 }];
356                 text.push_str(&name);
357                 text.push_str(": ");
358
359                 let ty = ty.make(offset + text.len(), id, scx)?;
360                 text.push_str(&ty.text);
361
362                 if let Some(expr) = expr {
363                     text.push_str(" = ");
364                     let expr = pprust::expr_to_string(expr).replace('\n', " ");
365                     text.push_str(&expr);
366                 }
367
368                 text.push(';');
369
370                 Ok(extend_sig(ty, text, defs, vec![]))
371             }
372             ast::ItemKind::Fn(_, ast::FnSig { ref decl, header }, ref generics, _) => {
373                 let mut text = String::new();
374                 if let ast::Const::Yes(_) = header.constness {
375                     text.push_str("const ");
376                 }
377                 if header.asyncness.is_async() {
378                     text.push_str("async ");
379                 }
380                 if let ast::Unsafe::Yes(_) = header.unsafety {
381                     text.push_str("unsafe ");
382                 }
383                 push_extern(&mut text, header.ext);
384                 text.push_str("fn ");
385
386                 let mut sig = name_and_generics(text, offset, generics, self.id, self.ident, scx)?;
387
388                 sig.text.push('(');
389                 for i in &decl.inputs {
390                     // FIXME should descend into patterns to add defs.
391                     sig.text.push_str(&pprust::pat_to_string(&i.pat));
392                     sig.text.push_str(": ");
393                     let nested = i.ty.make(offset + sig.text.len(), Some(i.id), scx)?;
394                     sig.text.push_str(&nested.text);
395                     sig.text.push(',');
396                     sig.defs.extend(nested.defs.into_iter());
397                     sig.refs.extend(nested.refs.into_iter());
398                 }
399                 sig.text.push(')');
400
401                 if let ast::FnRetTy::Ty(ref t) = decl.output {
402                     sig.text.push_str(" -> ");
403                     let nested = t.make(offset + sig.text.len(), None, scx)?;
404                     sig.text.push_str(&nested.text);
405                     sig.defs.extend(nested.defs.into_iter());
406                     sig.refs.extend(nested.refs.into_iter());
407                 }
408                 sig.text.push_str(" {}");
409
410                 Ok(sig)
411             }
412             ast::ItemKind::Mod(ref _mod) => {
413                 let mut text = "mod ".to_owned();
414                 let name = self.ident.to_string();
415                 let defs = vec![SigElement {
416                     id: id_from_node_id(self.id, scx),
417                     start: offset + text.len(),
418                     end: offset + text.len() + name.len(),
419                 }];
420                 text.push_str(&name);
421                 // Could be either `mod foo;` or `mod foo { ... }`, but we'll just pick one.
422                 text.push(';');
423
424                 Ok(Signature { text, defs, refs: vec![] })
425             }
426             ast::ItemKind::TyAlias(_, ref generics, _, ref ty) => {
427                 let text = "type ".to_owned();
428                 let mut sig = name_and_generics(text, offset, generics, self.id, self.ident, scx)?;
429
430                 sig.text.push_str(" = ");
431                 let ty = match ty {
432                     Some(ty) => ty.make(offset + sig.text.len(), id, scx)?,
433                     None => return Err("Ty"),
434                 };
435                 sig.text.push_str(&ty.text);
436                 sig.text.push(';');
437
438                 Ok(merge_sigs(sig.text.clone(), vec![sig, ty]))
439             }
440             ast::ItemKind::Enum(_, ref generics) => {
441                 let text = "enum ".to_owned();
442                 let mut sig = name_and_generics(text, offset, generics, self.id, self.ident, scx)?;
443                 sig.text.push_str(" {}");
444                 Ok(sig)
445             }
446             ast::ItemKind::Struct(_, ref generics) => {
447                 let text = "struct ".to_owned();
448                 let mut sig = name_and_generics(text, offset, generics, self.id, self.ident, scx)?;
449                 sig.text.push_str(" {}");
450                 Ok(sig)
451             }
452             ast::ItemKind::Union(_, ref generics) => {
453                 let text = "union ".to_owned();
454                 let mut sig = name_and_generics(text, offset, generics, self.id, self.ident, scx)?;
455                 sig.text.push_str(" {}");
456                 Ok(sig)
457             }
458             ast::ItemKind::Trait(is_auto, unsafety, ref generics, ref bounds, _) => {
459                 let mut text = String::new();
460
461                 if is_auto == ast::IsAuto::Yes {
462                     text.push_str("auto ");
463                 }
464
465                 if let ast::Unsafe::Yes(_) = unsafety {
466                     text.push_str("unsafe ");
467                 }
468                 text.push_str("trait ");
469                 let mut sig = name_and_generics(text, offset, generics, self.id, self.ident, scx)?;
470
471                 if !bounds.is_empty() {
472                     sig.text.push_str(": ");
473                     sig.text.push_str(&pprust::bounds_to_string(bounds));
474                 }
475                 // FIXME where clause
476                 sig.text.push_str(" {}");
477
478                 Ok(sig)
479             }
480             ast::ItemKind::TraitAlias(ref generics, ref bounds) => {
481                 let mut text = String::new();
482                 text.push_str("trait ");
483                 let mut sig = name_and_generics(text, offset, generics, self.id, self.ident, scx)?;
484
485                 if !bounds.is_empty() {
486                     sig.text.push_str(" = ");
487                     sig.text.push_str(&pprust::bounds_to_string(bounds));
488                 }
489                 // FIXME where clause
490                 sig.text.push_str(";");
491
492                 Ok(sig)
493             }
494             ast::ItemKind::Impl {
495                 unsafety,
496                 polarity,
497                 defaultness,
498                 constness,
499                 ref generics,
500                 ref of_trait,
501                 ref self_ty,
502                 items: _,
503             } => {
504                 let mut text = String::new();
505                 if let ast::Defaultness::Default(_) = defaultness {
506                     text.push_str("default ");
507                 }
508                 if let ast::Unsafe::Yes(_) = unsafety {
509                     text.push_str("unsafe ");
510                 }
511                 text.push_str("impl");
512                 if let ast::Const::Yes(_) = constness {
513                     text.push_str(" const");
514                 }
515
516                 let generics_sig = generics.make(offset + text.len(), id, scx)?;
517                 text.push_str(&generics_sig.text);
518
519                 text.push(' ');
520
521                 let trait_sig = if let Some(ref t) = *of_trait {
522                     if let ast::ImplPolarity::Negative(_) = polarity {
523                         text.push('!');
524                     }
525                     let trait_sig = t.path.make(offset + text.len(), id, scx)?;
526                     text.push_str(&trait_sig.text);
527                     text.push_str(" for ");
528                     trait_sig
529                 } else {
530                     text_sig(String::new())
531                 };
532
533                 let ty_sig = self_ty.make(offset + text.len(), id, scx)?;
534                 text.push_str(&ty_sig.text);
535
536                 text.push_str(" {}");
537
538                 Ok(merge_sigs(text, vec![generics_sig, trait_sig, ty_sig]))
539
540                 // FIXME where clause
541             }
542             ast::ItemKind::ForeignMod(_) => Err("extern mod"),
543             ast::ItemKind::GlobalAsm(_) => Err("glboal asm"),
544             ast::ItemKind::ExternCrate(_) => Err("extern crate"),
545             // FIXME should implement this (e.g., pub use).
546             ast::ItemKind::Use(_) => Err("import"),
547             ast::ItemKind::MacCall(..) | ast::ItemKind::MacroDef(_) => Err("Macro"),
548         }
549     }
550 }
551
552 impl Sig for ast::Path {
553     fn make(&self, offset: usize, id: Option<NodeId>, scx: &SaveContext<'_, '_>) -> Result {
554         let res = scx.get_path_res(id.ok_or("Missing id for Path")?);
555
556         let (name, start, end) = match res {
557             Res::PrimTy(..) | Res::SelfTy(..) | Res::Err => {
558                 return Ok(Signature {
559                     text: pprust::path_to_string(self),
560                     defs: vec![],
561                     refs: vec![],
562                 });
563             }
564             Res::Def(DefKind::AssocConst | DefKind::Variant | DefKind::Ctor(..), _) => {
565                 let len = self.segments.len();
566                 if len < 2 {
567                     return Err("Bad path");
568                 }
569                 // FIXME: really we should descend into the generics here and add SigElements for
570                 // them.
571                 // FIXME: would be nice to have a def for the first path segment.
572                 let seg1 = pprust::path_segment_to_string(&self.segments[len - 2]);
573                 let seg2 = pprust::path_segment_to_string(&self.segments[len - 1]);
574                 let start = offset + seg1.len() + 2;
575                 (format!("{}::{}", seg1, seg2), start, start + seg2.len())
576             }
577             _ => {
578                 let name = pprust::path_segment_to_string(self.segments.last().ok_or("Bad path")?);
579                 let end = offset + name.len();
580                 (name, offset, end)
581             }
582         };
583
584         let id = id_from_def_id(res.def_id());
585         Ok(Signature { text: name, defs: vec![], refs: vec![SigElement { id, start, end }] })
586     }
587 }
588
589 // This does not cover the where clause, which must be processed separately.
590 impl Sig for ast::Generics {
591     fn make(&self, offset: usize, _parent_id: Option<NodeId>, scx: &SaveContext<'_, '_>) -> Result {
592         if self.params.is_empty() {
593             return Ok(text_sig(String::new()));
594         }
595
596         let mut text = "<".to_owned();
597
598         let mut defs = Vec::with_capacity(self.params.len());
599         for param in &self.params {
600             let mut param_text = String::new();
601             if let ast::GenericParamKind::Const { .. } = param.kind {
602                 param_text.push_str("const ");
603             }
604             param_text.push_str(&param.ident.as_str());
605             defs.push(SigElement {
606                 id: id_from_node_id(param.id, scx),
607                 start: offset + text.len(),
608                 end: offset + text.len() + param_text.as_str().len(),
609             });
610             if let ast::GenericParamKind::Const { ref ty } = param.kind {
611                 param_text.push_str(": ");
612                 param_text.push_str(&pprust::ty_to_string(&ty));
613             }
614             if !param.bounds.is_empty() {
615                 param_text.push_str(": ");
616                 match param.kind {
617                     ast::GenericParamKind::Lifetime { .. } => {
618                         let bounds = param
619                             .bounds
620                             .iter()
621                             .map(|bound| match bound {
622                                 ast::GenericBound::Outlives(lt) => lt.ident.to_string(),
623                                 _ => panic!(),
624                             })
625                             .collect::<Vec<_>>()
626                             .join(" + ");
627                         param_text.push_str(&bounds);
628                         // FIXME add lifetime bounds refs.
629                     }
630                     ast::GenericParamKind::Type { .. } => {
631                         param_text.push_str(&pprust::bounds_to_string(&param.bounds));
632                         // FIXME descend properly into bounds.
633                     }
634                     ast::GenericParamKind::Const { .. } => {
635                         // Const generics cannot contain bounds.
636                     }
637                 }
638             }
639             text.push_str(&param_text);
640             text.push(',');
641         }
642
643         text.push('>');
644         Ok(Signature { text, defs, refs: vec![] })
645     }
646 }
647
648 impl Sig for ast::StructField {
649     fn make(&self, offset: usize, _parent_id: Option<NodeId>, scx: &SaveContext<'_, '_>) -> Result {
650         let mut text = String::new();
651         let mut defs = None;
652         if let Some(ident) = self.ident {
653             text.push_str(&ident.to_string());
654             defs = Some(SigElement {
655                 id: id_from_node_id(self.id, scx),
656                 start: offset,
657                 end: offset + text.len(),
658             });
659             text.push_str(": ");
660         }
661
662         let mut ty_sig = self.ty.make(offset + text.len(), Some(self.id), scx)?;
663         text.push_str(&ty_sig.text);
664         ty_sig.text = text;
665         ty_sig.defs.extend(defs.into_iter());
666         Ok(ty_sig)
667     }
668 }
669
670 impl Sig for ast::Variant {
671     fn make(&self, offset: usize, parent_id: Option<NodeId>, scx: &SaveContext<'_, '_>) -> Result {
672         let mut text = self.ident.to_string();
673         match self.data {
674             ast::VariantData::Struct(ref fields, r) => {
675                 let id = parent_id.unwrap();
676                 let name_def = SigElement {
677                     id: id_from_node_id(id, scx),
678                     start: offset,
679                     end: offset + text.len(),
680                 };
681                 text.push_str(" { ");
682                 let mut defs = vec![name_def];
683                 let mut refs = vec![];
684                 if r {
685                     text.push_str("/* parse error */ ");
686                 } else {
687                     for f in fields {
688                         let field_sig = f.make(offset + text.len(), Some(id), scx)?;
689                         text.push_str(&field_sig.text);
690                         text.push_str(", ");
691                         defs.extend(field_sig.defs.into_iter());
692                         refs.extend(field_sig.refs.into_iter());
693                     }
694                 }
695                 text.push('}');
696                 Ok(Signature { text, defs, refs })
697             }
698             ast::VariantData::Tuple(ref fields, id) => {
699                 let name_def = SigElement {
700                     id: id_from_node_id(id, scx),
701                     start: offset,
702                     end: offset + text.len(),
703                 };
704                 text.push('(');
705                 let mut defs = vec![name_def];
706                 let mut refs = vec![];
707                 for f in fields {
708                     let field_sig = f.make(offset + text.len(), Some(id), scx)?;
709                     text.push_str(&field_sig.text);
710                     text.push_str(", ");
711                     defs.extend(field_sig.defs.into_iter());
712                     refs.extend(field_sig.refs.into_iter());
713                 }
714                 text.push(')');
715                 Ok(Signature { text, defs, refs })
716             }
717             ast::VariantData::Unit(id) => {
718                 let name_def = SigElement {
719                     id: id_from_node_id(id, scx),
720                     start: offset,
721                     end: offset + text.len(),
722                 };
723                 Ok(Signature { text, defs: vec![name_def], refs: vec![] })
724             }
725         }
726     }
727 }
728
729 impl Sig for ast::ForeignItem {
730     fn make(&self, offset: usize, _parent_id: Option<NodeId>, scx: &SaveContext<'_, '_>) -> Result {
731         let id = Some(self.id);
732         match self.kind {
733             ast::ForeignItemKind::Fn(_, ref sig, ref generics, _) => {
734                 let decl = &sig.decl;
735                 let mut text = String::new();
736                 text.push_str("fn ");
737
738                 let mut sig = name_and_generics(text, offset, generics, self.id, self.ident, scx)?;
739
740                 sig.text.push('(');
741                 for i in &decl.inputs {
742                     // FIXME should descend into patterns to add defs.
743                     sig.text.push_str(&pprust::pat_to_string(&i.pat));
744                     sig.text.push_str(": ");
745                     let nested = i.ty.make(offset + sig.text.len(), Some(i.id), scx)?;
746                     sig.text.push_str(&nested.text);
747                     sig.text.push(',');
748                     sig.defs.extend(nested.defs.into_iter());
749                     sig.refs.extend(nested.refs.into_iter());
750                 }
751                 sig.text.push(')');
752
753                 if let ast::FnRetTy::Ty(ref t) = decl.output {
754                     sig.text.push_str(" -> ");
755                     let nested = t.make(offset + sig.text.len(), None, scx)?;
756                     sig.text.push_str(&nested.text);
757                     sig.defs.extend(nested.defs.into_iter());
758                     sig.refs.extend(nested.refs.into_iter());
759                 }
760                 sig.text.push(';');
761
762                 Ok(sig)
763             }
764             ast::ForeignItemKind::Static(ref ty, m, _) => {
765                 let mut text = "static ".to_owned();
766                 if m == ast::Mutability::Mut {
767                     text.push_str("mut ");
768                 }
769                 let name = self.ident.to_string();
770                 let defs = vec![SigElement {
771                     id: id_from_node_id(self.id, scx),
772                     start: offset + text.len(),
773                     end: offset + text.len() + name.len(),
774                 }];
775                 text.push_str(&name);
776                 text.push_str(": ");
777
778                 let ty_sig = ty.make(offset + text.len(), id, scx)?;
779                 text.push(';');
780
781                 Ok(extend_sig(ty_sig, text, defs, vec![]))
782             }
783             ast::ForeignItemKind::TyAlias(..) => {
784                 let mut text = "type ".to_owned();
785                 let name = self.ident.to_string();
786                 let defs = vec![SigElement {
787                     id: id_from_node_id(self.id, scx),
788                     start: offset + text.len(),
789                     end: offset + text.len() + name.len(),
790                 }];
791                 text.push_str(&name);
792                 text.push(';');
793
794                 Ok(Signature { text, defs, refs: vec![] })
795             }
796             ast::ForeignItemKind::MacCall(..) => Err("macro"),
797         }
798     }
799 }
800
801 fn name_and_generics(
802     mut text: String,
803     offset: usize,
804     generics: &ast::Generics,
805     id: NodeId,
806     name: ast::Ident,
807     scx: &SaveContext<'_, '_>,
808 ) -> Result {
809     let name = name.to_string();
810     let def = SigElement {
811         id: id_from_node_id(id, scx),
812         start: offset + text.len(),
813         end: offset + text.len() + name.len(),
814     };
815     text.push_str(&name);
816     let generics: Signature = generics.make(offset + text.len(), Some(id), scx)?;
817     // FIXME where clause
818     let text = format!("{}{}", text, generics.text);
819     Ok(extend_sig(generics, text, vec![def], vec![]))
820 }
821
822 fn make_assoc_type_signature(
823     id: NodeId,
824     ident: ast::Ident,
825     bounds: Option<&ast::GenericBounds>,
826     default: Option<&ast::Ty>,
827     scx: &SaveContext<'_, '_>,
828 ) -> Result {
829     let mut text = "type ".to_owned();
830     let name = ident.to_string();
831     let mut defs = vec![SigElement {
832         id: id_from_node_id(id, scx),
833         start: text.len(),
834         end: text.len() + name.len(),
835     }];
836     let mut refs = vec![];
837     text.push_str(&name);
838     if let Some(bounds) = bounds {
839         text.push_str(": ");
840         // FIXME should descend into bounds
841         text.push_str(&pprust::bounds_to_string(bounds));
842     }
843     if let Some(default) = default {
844         text.push_str(" = ");
845         let ty_sig = default.make(text.len(), Some(id), scx)?;
846         text.push_str(&ty_sig.text);
847         defs.extend(ty_sig.defs.into_iter());
848         refs.extend(ty_sig.refs.into_iter());
849     }
850     text.push(';');
851     Ok(Signature { text, defs, refs })
852 }
853
854 fn make_assoc_const_signature(
855     id: NodeId,
856     ident: ast::Name,
857     ty: &ast::Ty,
858     default: Option<&ast::Expr>,
859     scx: &SaveContext<'_, '_>,
860 ) -> Result {
861     let mut text = "const ".to_owned();
862     let name = ident.to_string();
863     let mut defs = vec![SigElement {
864         id: id_from_node_id(id, scx),
865         start: text.len(),
866         end: text.len() + name.len(),
867     }];
868     let mut refs = vec![];
869     text.push_str(&name);
870     text.push_str(": ");
871
872     let ty_sig = ty.make(text.len(), Some(id), scx)?;
873     text.push_str(&ty_sig.text);
874     defs.extend(ty_sig.defs.into_iter());
875     refs.extend(ty_sig.refs.into_iter());
876
877     if let Some(default) = default {
878         text.push_str(" = ");
879         text.push_str(&pprust::expr_to_string(default));
880     }
881     text.push(';');
882     Ok(Signature { text, defs, refs })
883 }
884
885 fn make_method_signature(
886     id: NodeId,
887     ident: ast::Ident,
888     generics: &ast::Generics,
889     m: &ast::FnSig,
890     scx: &SaveContext<'_, '_>,
891 ) -> Result {
892     // FIXME code dup with function signature
893     let mut text = String::new();
894     if let ast::Const::Yes(_) = m.header.constness {
895         text.push_str("const ");
896     }
897     if m.header.asyncness.is_async() {
898         text.push_str("async ");
899     }
900     if let ast::Unsafe::Yes(_) = m.header.unsafety {
901         text.push_str("unsafe ");
902     }
903     push_extern(&mut text, m.header.ext);
904     text.push_str("fn ");
905
906     let mut sig = name_and_generics(text, 0, generics, id, ident, scx)?;
907
908     sig.text.push('(');
909     for i in &m.decl.inputs {
910         // FIXME should descend into patterns to add defs.
911         sig.text.push_str(&pprust::pat_to_string(&i.pat));
912         sig.text.push_str(": ");
913         let nested = i.ty.make(sig.text.len(), Some(i.id), scx)?;
914         sig.text.push_str(&nested.text);
915         sig.text.push(',');
916         sig.defs.extend(nested.defs.into_iter());
917         sig.refs.extend(nested.refs.into_iter());
918     }
919     sig.text.push(')');
920
921     if let ast::FnRetTy::Ty(ref t) = m.decl.output {
922         sig.text.push_str(" -> ");
923         let nested = t.make(sig.text.len(), None, scx)?;
924         sig.text.push_str(&nested.text);
925         sig.defs.extend(nested.defs.into_iter());
926         sig.refs.extend(nested.refs.into_iter());
927     }
928     sig.text.push_str(" {}");
929
930     Ok(sig)
931 }