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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
//! A general sorting pass
import doc::item_utils;
import std::sort;
export item_lteq, mk_pass;
type item_lteq = pure fn~(v1: &doc::itemtag, v2: &doc::itemtag) -> bool;
fn mk_pass(name: ~str, +lteq: item_lteq) -> pass {
{
name: name,
f: fn~(srv: astsrv::srv, doc: doc::doc) -> doc::doc {
run(srv, doc, lteq)
}
}
}
#[allow(non_implicitly_copyable_typarams)]
fn run(
_srv: astsrv::srv,
doc: doc::doc,
lteq: item_lteq
) -> doc::doc {
let fold = fold::fold({
fold_mod: fold_mod,
.. *fold::default_any_fold(lteq)
});
fold.fold_doc(fold, doc)
}
#[allow(non_implicitly_copyable_typarams)]
fn fold_mod(
fold: fold::fold<item_lteq>,
doc: doc::moddoc
) -> doc::moddoc {
let doc = fold::default_any_fold_mod(fold, doc);
doc::moddoc_({
items: sort::merge_sort(fold.ctxt, doc.items),
.. *doc
})
}
#[test]
fn test() {
pure fn name_lteq(item1: &doc::itemtag, item2: &doc::itemtag) -> bool {
(*item1).name() <= (*item2).name()
}
let source = ~"mod z { mod y { } fn x() { } } mod w { }";
do astsrv::from_str(source) |srv| {
let doc = extract::from_srv(srv, ~"");
let doc = mk_pass(~"", name_lteq).f(srv, doc);
assert doc.cratemod().mods()[0].name() == ~"w";
assert doc.cratemod().mods()[1].items[0].name() == ~"x";
assert doc.cratemod().mods()[1].items[1].name() == ~"y";
assert doc.cratemod().mods()[1].name() == ~"z";
}
}
#[test]
fn should_be_stable() {
pure fn always_eq(_item1: &doc::itemtag, _item2: &doc::itemtag) -> bool {
true
}
let source = ~"mod a { mod b { } } mod c { mod d { } }";
do astsrv::from_str(source) |srv| {
let doc = extract::from_srv(srv, ~"");
let doc = mk_pass(~"", always_eq).f(srv, doc);
assert doc.cratemod().mods()[0].items[0].name() == ~"b";
assert doc.cratemod().mods()[1].items[0].name() == ~"d";
let doc = mk_pass(~"", always_eq).f(srv, doc);
assert doc.cratemod().mods()[0].items[0].name() == ~"b";
assert doc.cratemod().mods()[1].items[0].name() == ~"d";
}
}
|
