rust/src/libsyntax/print/pprust.rs

1908 lines
55 KiB
Rust
Raw Normal View History

import parse::comments;
import parse::lexer;
import codemap::codemap;
import pp::{break_offset, word, printer,
2011-09-12 23:13:28 +00:00
space, zerobreak, hardbreak, breaks, consistent,
inconsistent, eof};
import diagnostic;
import ast::{required, provided};
import ast_util::{operator_prec};
2012-05-10 14:24:56 +00:00
import dvec::{dvec, extensions};
import parse::classify::*;
2012-07-17 18:22:11 +00:00
import util::interner;
2012-08-07 00:50:45 +00:00
type ident_interner = @interner::interner<@~str>;
// The ps is stored here to prevent recursive type.
2012-01-19 22:24:03 +00:00
enum ann_node {
2012-01-20 01:56:05 +00:00
node_block(ps, ast::blk),
node_item(ps, @ast::item),
node_expr(ps, @ast::expr),
node_pat(ps, @ast::pat),
}
type pp_ann = {pre: fn@(ann_node), post: fn@(ann_node)};
fn no_ann() -> pp_ann {
fn ignore(_node: ann_node) { }
2012-08-02 00:30:05 +00:00
return {pre: ignore, post: ignore};
}
type ps =
2011-07-27 12:19:39 +00:00
@{s: pp::printer,
cm: option<codemap>,
intr: @interner::interner<@~str>,
comments: option<~[comments::cmnt]>,
literals: option<~[comments::lit]>,
2012-03-27 01:35:18 +00:00
mut cur_cmnt: uint,
mut cur_lit: uint,
2012-05-10 14:24:56 +00:00
boxes: dvec<pp::breaks>,
2011-07-27 12:19:39 +00:00
ann: pp_ann};
2012-05-10 14:24:56 +00:00
fn ibox(s: ps, u: uint) {
s.boxes.push(pp::inconsistent);
pp::ibox(s.s, u);
}
2012-05-10 14:24:56 +00:00
fn end(s: ps) {
s.boxes.pop();
pp::end(s.s);
}
2011-08-12 02:14:38 +00:00
fn rust_printer(writer: io::writer) -> ps {
2012-08-02 00:30:05 +00:00
return @{s: pp::mk_printer(writer, default_columns),
2012-08-07 00:50:45 +00:00
cm: none::<codemap>,
intr: @interner::mk::<@~str>(|x| str::hash(*x),
|x,y| str::eq(*x, *y)),
comments: none::<~[comments::cmnt]>,
literals: none::<~[comments::lit]>,
mut cur_cmnt: 0u,
mut cur_lit: 0u,
boxes: dvec(),
ann: no_ann()};
}
fn unexpanded_rust_printer(writer: io::writer, intr: ident_interner) -> ps {
return @{s: pp::mk_printer(writer, default_columns),
cm: none::<codemap>,
intr: intr,
comments: none::<~[comments::cmnt]>,
literals: none::<~[comments::lit]>,
mut cur_cmnt: 0u,
mut cur_lit: 0u,
boxes: dvec(),
ann: no_ann()};
}
2011-07-27 12:19:39 +00:00
const indent_unit: uint = 4u;
const alt_indent_unit: uint = 2u;
2011-07-27 12:19:39 +00:00
const default_columns: uint = 78u;
// Requires you to pass an input filename and reader so that
// it can scan the input text for comments and literals to
// copy forward.
fn print_crate(cm: codemap, intr: @interner::interner<@~str>,
2012-07-17 18:22:11 +00:00
span_diagnostic: diagnostic::span_handler,
crate: @ast::crate, filename: ~str, in: io::reader,
out: io::writer, ann: pp_ann, is_expanded: bool) {
2012-04-15 10:57:24 +00:00
let r = comments::gather_comments_and_literals(span_diagnostic,
filename, in);
2011-07-27 12:19:39 +00:00
let s =
@{s: pp::mk_printer(out, default_columns),
cm: some(cm),
intr: intr,
2011-07-27 12:19:39 +00:00
comments: some(r.cmnts),
// If the code is post expansion, don't use the table of
// literals, since it doesn't correspond with the literals
// in the AST anymore.
literals: if is_expanded { none } else { some(r.lits) },
2012-03-27 01:35:18 +00:00
mut cur_cmnt: 0u,
mut cur_lit: 0u,
2012-05-10 14:24:56 +00:00
boxes: dvec(),
2011-07-27 12:19:39 +00:00
ann: ann};
2012-02-21 23:34:26 +00:00
print_crate_(s, crate);
}
fn print_crate_(s: ps, &&crate: @ast::crate) {
print_mod(s, crate.node.module, crate.node.attrs);
print_remaining_comments(s);
2011-05-29 02:16:18 +00:00
eof(s.s);
}
2012-08-02 00:30:05 +00:00
fn ty_to_str(ty: @ast::ty) -> ~str { return to_str(ty, print_type); }
2012-08-02 00:30:05 +00:00
fn pat_to_str(pat: @ast::pat) -> ~str { return to_str(pat, print_pat); }
2012-08-02 00:30:05 +00:00
fn expr_to_str(e: @ast::expr) -> ~str { return to_str(e, print_expr); }
2012-08-07 00:50:45 +00:00
fn unexpanded_tt_to_str(tt: ast::token_tree, intr: ident_interner)
-> ~str {
let buffer = io::mem_buffer();
let s = unexpanded_rust_printer(io::mem_buffer_writer(buffer), intr);
print_tt(s, tt);
eof(s.s);
io::mem_buffer_str(buffer)
}
2012-08-02 00:30:05 +00:00
fn stmt_to_str(s: ast::stmt) -> ~str { return to_str(s, print_stmt); }
2012-08-02 00:30:05 +00:00
fn item_to_str(i: @ast::item) -> ~str { return to_str(i, print_item); }
2012-08-02 00:30:05 +00:00
fn attr_to_str(i: ast::attribute) -> ~str {
return to_str(i, print_attribute);
}
fn typarams_to_str(tps: ~[ast::ty_param]) -> ~str {
2012-08-02 00:30:05 +00:00
return to_str(tps, print_type_params)
2012-02-02 06:41:41 +00:00
}
fn path_to_str(&&p: @ast::path) -> ~str {
2012-08-02 00:30:05 +00:00
return to_str(p, |a,b| print_path(a, b, false));
}
fn fun_to_str(decl: ast::fn_decl, name: ast::ident,
params: ~[ast::ty_param]) -> ~str {
let buffer = io::mem_buffer();
let s = rust_printer(io::mem_buffer_writer(buffer));
print_fn(s, decl, name, params);
2012-01-20 02:09:06 +00:00
end(s); // Close the head box
end(s); // Close the outer box
2011-05-29 02:16:18 +00:00
eof(s.s);
io::mem_buffer_str(buffer)
}
2012-01-20 02:09:06 +00:00
#[test]
fn test_fun_to_str() {
let decl: ast::fn_decl = {
inputs: ~[],
output: @{id: 0,
node: ast::ty_nil,
span: ast_util::dummy_sp()},
2012-01-20 02:09:06 +00:00
purity: ast::impure_fn,
cf: ast::return_val
2012-01-20 02:09:06 +00:00
};
assert fun_to_str(decl, "a", ~[]) == "fn a()";
2012-01-20 02:09:06 +00:00
}
fn block_to_str(blk: ast::blk) -> ~str {
let buffer = io::mem_buffer();
let s = rust_printer(io::mem_buffer_writer(buffer));
2011-05-31 16:54:18 +00:00
// containing cbox, will be closed by print-block at }
cbox(s, indent_unit);
2011-05-31 16:54:18 +00:00
// head-ibox, will be closed by print-block after {
ibox(s, 0u);
print_block(s, blk);
2011-05-29 02:16:18 +00:00
eof(s.s);
io::mem_buffer_str(buffer)
}
fn meta_item_to_str(mi: ast::meta_item) -> ~str {
2012-08-02 00:30:05 +00:00
return to_str(@mi, print_meta_item);
}
fn attribute_to_str(attr: ast::attribute) -> ~str {
2012-08-02 00:30:05 +00:00
return to_str(attr, print_attribute);
}
fn variant_to_str(var: ast::variant) -> ~str {
2012-08-02 00:30:05 +00:00
return to_str(var, print_variant);
2012-01-26 01:22:08 +00:00
}
#[test]
fn test_variant_to_str() {
let var = ast_util::respan(ast_util::dummy_sp(), {
name: "principle_skinner",
attrs: ~[],
args: ~[],
2012-01-26 01:22:08 +00:00
id: 0,
disr_expr: none
});
let varstr = variant_to_str(var);
assert varstr == "principle_skinner";
}
2012-05-10 14:24:56 +00:00
fn cbox(s: ps, u: uint) {
s.boxes.push(pp::consistent);
pp::cbox(s.s, u);
}
2012-05-10 14:24:56 +00:00
fn box(s: ps, u: uint, b: pp::breaks) {
s.boxes.push(b);
pp::box(s.s, u, b);
}
fn nbsp(s: ps) { word(s.s, ~" "); }
fn word_nbsp(s: ps, w: ~str) { word(s.s, w); nbsp(s); }
2011-05-29 02:16:18 +00:00
fn word_space(s: ps, w: ~str) { word(s.s, w); space(s.s); }
2011-05-29 02:16:18 +00:00
fn popen(s: ps) { word(s.s, ~"("); }
2011-05-29 02:16:18 +00:00
fn pclose(s: ps) { word(s.s, ~")"); }
2011-05-29 02:16:18 +00:00
fn head(s: ps, w: ~str) {
2011-05-29 02:16:18 +00:00
// outer-box is consistent
cbox(s, indent_unit);
2011-05-29 02:16:18 +00:00
// head-box is inconsistent
ibox(s, str::len(w) + 1);
2011-05-29 02:16:18 +00:00
// keyword that starts the head
2011-05-30 21:10:54 +00:00
word_nbsp(s, w);
2011-05-29 02:16:18 +00:00
}
fn bopen(s: ps) {
word(s.s, ~"{");
end(s); // close the head-box
}
2011-05-29 02:16:18 +00:00
fn bclose_(s: ps, span: codemap::span, indented: uint) {
maybe_print_comment(s, span.hi);
break_offset_if_not_bol(s, 1u, -(indented as int));
word(s.s, ~"}");
end(s); // close the outer-box
}
fn bclose(s: ps, span: codemap::span) { bclose_(s, span, indent_unit); }
2011-05-29 02:16:18 +00:00
fn is_begin(s: ps) -> bool {
2012-08-06 19:34:08 +00:00
match s.s.last_token() { pp::BEGIN(_) => true, _ => false }
}
fn is_end(s: ps) -> bool {
2012-08-06 19:34:08 +00:00
match s.s.last_token() { pp::END => true, _ => false }
}
fn is_bol(s: ps) -> bool {
2012-08-02 00:30:05 +00:00
return s.s.last_token() == pp::EOF ||
s.s.last_token() == pp::hardbreak_tok();
}
fn in_cbox(s: ps) -> bool {
2012-05-10 14:24:56 +00:00
let len = s.boxes.len();
2012-08-02 00:30:05 +00:00
if len == 0u { return false; }
return s.boxes[len - 1u] == pp::consistent;
}
fn hardbreak_if_not_bol(s: ps) { if !is_bol(s) { hardbreak(s.s); } }
fn space_if_not_bol(s: ps) { if !is_bol(s) { space(s.s); } }
fn break_offset_if_not_bol(s: ps, n: uint, off: int) {
2011-07-27 12:19:39 +00:00
if !is_bol(s) {
break_offset(s.s, n, off);
} else {
2011-07-27 12:19:39 +00:00
if off != 0 && s.s.last_token() == pp::hardbreak_tok() {
// We do something pretty sketchy here: tuck the nonzero
// offset-adjustment we were going to deposit along with the
// break into the previous hardbreak.
s.s.replace_last_token(pp::hardbreak_tok_offset(off));
}
}
}
// Synthesizes a comment that was not textually present in the original source
// file.
fn synth_comment(s: ps, text: ~str) {
word(s.s, ~"/*");
space(s.s);
word(s.s, text);
space(s.s);
word(s.s, ~"*/");
}
fn commasep<IN>(s: ps, b: breaks, elts: ~[IN], op: fn(ps, IN)) {
box(s, 0u, b);
let mut first = true;
2012-06-30 23:19:07 +00:00
for elts.each |elt| {
if first { first = false; } else { word_space(s, ~","); }
op(s, elt);
}
end(s);
}
fn commasep_cmnt<IN>(s: ps, b: breaks, elts: ~[IN], op: fn(ps, IN),
2012-01-23 22:59:00 +00:00
get_span: fn(IN) -> codemap::span) {
box(s, 0u, b);
let len = vec::len::<IN>(elts);
let mut i = 0u;
2012-06-30 23:19:07 +00:00
for elts.each |elt| {
maybe_print_comment(s, get_span(elt).hi);
op(s, elt);
i += 1u;
2011-07-27 12:19:39 +00:00
if i < len {
word(s.s, ~",");
maybe_print_trailing_comment(s, get_span(elt),
some(get_span(elts[i]).hi));
space_if_not_bol(s);
}
}
end(s);
}
fn commasep_exprs(s: ps, b: breaks, exprs: ~[@ast::expr]) {
2012-08-02 00:30:05 +00:00
fn expr_span(&&expr: @ast::expr) -> codemap::span { return expr.span; }
commasep_cmnt(s, b, exprs, print_expr, expr_span);
}
fn print_mod(s: ps, _mod: ast::_mod, attrs: ~[ast::attribute]) {
print_inner_attributes(s, attrs);
2012-06-30 23:19:07 +00:00
for _mod.view_items.each |vitem| {
print_view_item(s, vitem);
}
2012-06-30 23:19:07 +00:00
for _mod.items.each |item| { print_item(s, item); }
}
fn print_foreign_mod(s: ps, nmod: ast::foreign_mod,
attrs: ~[ast::attribute]) {
print_inner_attributes(s, attrs);
2012-06-30 23:19:07 +00:00
for nmod.view_items.each |vitem| {
print_view_item(s, vitem);
}
2012-06-30 23:19:07 +00:00
for nmod.items.each |item| { print_foreign_item(s, item); }
}
fn print_region(s: ps, region: @ast::region) {
2012-08-06 19:34:08 +00:00
match region.node {
2012-08-04 02:59:04 +00:00
ast::re_anon => word_space(s, ~"&"),
ast::re_named(name) => {
word(s.s, ~"&");
word(s.s, *name);
}
}
}
fn print_type(s: ps, &&ty: @ast::ty) {
print_type_ex(s, ty, false);
}
fn print_type_ex(s: ps, &&ty: @ast::ty, print_colons: bool) {
maybe_print_comment(s, ty.span.lo);
ibox(s, 0u);
2012-08-06 19:34:08 +00:00
match ty.node {
2012-08-04 02:59:04 +00:00
ast::ty_nil => word(s.s, ~"()"),
ast::ty_bot => word(s.s, ~"!"),
ast::ty_box(mt) => { word(s.s, ~"@"); print_mt(s, mt); }
ast::ty_uniq(mt) => { word(s.s, ~"~"); print_mt(s, mt); }
ast::ty_vec(mt) => {
word(s.s, ~"[");
2012-08-06 19:34:08 +00:00
match mt.mutbl {
2012-08-04 02:59:04 +00:00
ast::m_mutbl => word_space(s, ~"mut"),
ast::m_const => word_space(s, ~"const"),
ast::m_imm => ()
}
print_type(s, mt.ty);
word(s.s, ~"]");
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::ty_ptr(mt) => { word(s.s, ~"*"); print_mt(s, mt); }
ast::ty_rptr(region, mt) => {
2012-08-06 19:34:08 +00:00
match region.node {
2012-08-04 02:59:04 +00:00
ast::re_anon => word(s.s, ~"&"),
_ => { print_region(s, region); word(s.s, ~"/"); }
}
print_mt(s, mt);
}
2012-08-04 02:59:04 +00:00
ast::ty_rec(fields) => {
word(s.s, ~"{");
fn print_field(s: ps, f: ast::ty_field) {
2011-07-27 12:19:39 +00:00
cbox(s, indent_unit);
print_mutability(s, f.node.mt.mutbl);
2012-06-10 07:49:59 +00:00
word(s.s, *f.node.ident);
word_space(s, ~":");
print_type(s, f.node.mt.ty);
2011-07-27 12:19:39 +00:00
end(s);
}
2012-08-02 00:30:05 +00:00
fn get_span(f: ast::ty_field) -> codemap::span { return f.span; }
2011-07-27 12:19:39 +00:00
commasep_cmnt(s, consistent, fields, print_field, get_span);
word(s.s, ~",}");
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::ty_tup(elts) => {
popen(s);
commasep(s, inconsistent, elts, print_type);
pclose(s);
}
ast::ty_fn(proto, bounds, d) => {
print_ty_fn(s, some(proto), bounds, d, none, none);
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::ty_path(path, _) => print_path(s, path, print_colons),
ast::ty_fixed_length(t, v) => {
print_type(s, t);
word(s.s, ~"/");
print_vstore(s, ast::vstore_fixed(v));
2012-04-10 00:32:49 +00:00
}
2012-08-04 02:59:04 +00:00
ast::ty_mac(_) => {
fail ~"print_type doesn't know how to print a ty_mac";
}
2012-08-04 02:59:04 +00:00
ast::ty_infer => {
fail ~"print_type shouldn't see a ty_infer";
}
}
end(s);
}
fn print_foreign_item(s: ps, item: @ast::foreign_item) {
hardbreak_if_not_bol(s);
maybe_print_comment(s, item.span.lo);
print_outer_attributes(s, item.attrs);
2012-08-06 19:34:08 +00:00
match item.node {
2012-08-04 02:59:04 +00:00
ast::foreign_item_fn(decl, typarams) => {
print_fn(s, decl, item.ident, typarams);
end(s); // end head-ibox
word(s.s, ~";");
end(s); // end the outer fn box
}
}
}
fn print_item(s: ps, &&item: @ast::item) {
hardbreak_if_not_bol(s);
maybe_print_comment(s, item.span.lo);
print_outer_attributes(s, item.attrs);
2011-07-27 12:19:39 +00:00
let ann_node = node_item(s, item);
s.ann.pre(ann_node);
2012-08-06 19:34:08 +00:00
match item.node {
2012-08-04 02:59:04 +00:00
ast::item_const(ty, expr) => {
head(s, ~"const");
word_space(s, *item.ident + ~":");
print_type(s, ty);
2011-07-27 12:19:39 +00:00
space(s.s);
end(s); // end the head-ibox
word_space(s, ~"=");
2011-07-27 12:19:39 +00:00
print_expr(s, expr);
word(s.s, ~";");
2011-07-27 12:19:39 +00:00
end(s); // end the outer cbox
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::item_fn(decl, typarams, body) => {
print_fn(s, decl, item.ident, typarams);
word(s.s, ~" ");
print_block_with_attrs(s, body, item.attrs);
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::item_mod(_mod) => {
head(s, ~"mod");
2012-06-10 07:49:59 +00:00
word_nbsp(s, *item.ident);
2011-07-27 12:19:39 +00:00
bopen(s);
print_mod(s, _mod, item.attrs);
bclose(s, item.span);
}
2012-08-04 02:59:04 +00:00
ast::item_foreign_mod(nmod) => {
head(s, ~"extern");
word_nbsp(s, ~"mod");
2012-06-10 07:49:59 +00:00
word_nbsp(s, *item.ident);
2011-07-27 12:19:39 +00:00
bopen(s);
print_foreign_mod(s, nmod, item.attrs);
2011-07-27 12:19:39 +00:00
bclose(s, item.span);
}
2012-08-04 02:59:04 +00:00
ast::item_ty(ty, params) => {
2011-07-27 12:19:39 +00:00
ibox(s, indent_unit);
ibox(s, 0u);
word_nbsp(s, ~"type");
2012-06-10 07:49:59 +00:00
word(s.s, *item.ident);
2011-07-27 12:19:39 +00:00
print_type_params(s, params);
end(s); // end the inner ibox
space(s.s);
word_space(s, ~"=");
print_type(s, ty);
word(s.s, ~";");
2011-07-27 12:19:39 +00:00
end(s); // end the outer ibox
}
2012-08-04 02:59:04 +00:00
ast::item_enum(variants, params) => {
let mut newtype =
2011-08-15 23:38:23 +00:00
vec::len(variants) == 1u &&
str::eq(item.ident, variants[0].node.name);
if newtype {
match variants[0].node.kind {
ast::tuple_variant_kind(args) if args.len() == 1 => {}
_ => newtype = false
}
}
2011-07-27 12:19:39 +00:00
if newtype {
ibox(s, indent_unit);
word_space(s, ~"enum");
} else {
head(s, ~"enum");
}
2012-06-10 07:49:59 +00:00
word(s.s, *item.ident);
2011-07-27 12:19:39 +00:00
print_type_params(s, params);
space(s.s);
if newtype {
word_space(s, ~"=");
match variants[0].node.kind {
ast::tuple_variant_kind(args) => print_type(s, args[0].ty),
_ => fail ~"newtype syntax with struct?"
}
word(s.s, ~";");
2011-07-27 12:19:39 +00:00
end(s);
} else {
bopen(s);
2012-06-30 23:19:07 +00:00
for variants.each |v| {
space_if_not_bol(s);
2011-07-27 12:19:39 +00:00
maybe_print_comment(s, v.span.lo);
print_outer_attributes(s, v.node.attrs);
ibox(s, indent_unit);
2012-01-26 01:22:08 +00:00
print_variant(s, v);
word(s.s, ~",");
end(s);
maybe_print_trailing_comment(s, v.span, none::<uint>);
}
2011-05-29 02:16:18 +00:00
bclose(s, item.span);
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::item_class(tps, traits, items, m_ctor, m_dtor) => {
head(s, ~"class");
2012-06-10 07:49:59 +00:00
word_nbsp(s, *item.ident);
print_type_params(s, tps);
if vec::len(traits) != 0u {
word_space(s, ~":");
commasep(s, inconsistent, traits, |s, p|
2012-06-30 23:19:07 +00:00
print_path(s, p.path, false));
}
bopen(s);
hardbreak_if_not_bol(s);
do option::iter(m_ctor) |ctor| {
maybe_print_comment(s, ctor.span.lo);
print_outer_attributes(s, ctor.node.attrs);
// Doesn't call head because there shouldn't be a space after new.
cbox(s, indent_unit);
ibox(s, 4);
word(s.s, ~"new(");
print_fn_args(s, ctor.node.dec, ~[]);
word(s.s, ~")");
space(s.s);
print_block(s, ctor.node.body);
}
2012-06-30 23:19:07 +00:00
do option::iter(m_dtor) |dtor| {
hardbreak_if_not_bol(s);
maybe_print_comment(s, dtor.span.lo);
print_outer_attributes(s, dtor.node.attrs);
head(s, ~"drop");
print_block(s, dtor.node.body);
}
2012-06-30 23:19:07 +00:00
for items.each |ci| {
/*
FIXME (#1893): collect all private items and print
them in a single "priv" section
tjc: I'm not going to fix this yet b/c we might
change how exports work, including for class items
*/
hardbreak_if_not_bol(s);
maybe_print_comment(s, ci.span.lo);
let pr = ast_util::class_member_visibility(ci);
2012-08-06 19:34:08 +00:00
match pr {
2012-08-04 02:59:04 +00:00
ast::private => {
head(s, ~"priv");
bopen(s);
hardbreak_if_not_bol(s);
}
2012-08-04 02:59:04 +00:00
_ => ()
}
2012-08-06 19:34:08 +00:00
match ci.node {
2012-08-04 02:59:04 +00:00
ast::instance_var(nm, t, mt, _,_) => {
word_nbsp(s, ~"let");
2012-08-06 19:34:08 +00:00
match mt {
2012-08-04 02:59:04 +00:00
ast::class_mutable => word_nbsp(s, ~"mut"),
_ => ()
}
2012-06-10 07:49:59 +00:00
word(s.s, *nm);
word_nbsp(s, ~":");
print_type(s, t);
word(s.s, ~";");
}
2012-08-04 02:59:04 +00:00
ast::class_method(m) => {
print_method(s, m);
}
}
2012-08-06 19:34:08 +00:00
match pr {
2012-08-04 02:59:04 +00:00
ast::private => bclose(s, ci.span),
_ => ()
}
}
bclose(s, item.span);
}
2012-08-04 02:59:04 +00:00
ast::item_impl(tps, traits, ty, methods) => {
head(s, ~"impl");
2012-06-10 07:49:59 +00:00
word(s.s, *item.ident);
print_type_params(s, tps);
space(s.s);
if vec::len(traits) != 0u {
word_nbsp(s, ~"of");
do commasep(s, inconsistent, traits) |s, p| {
print_path(s, p.path, false);
}
space(s.s);
}
word_nbsp(s, ~"for");
print_type(s, ty);
space(s.s);
bopen(s);
2012-06-30 23:19:07 +00:00
for methods.each |meth| {
print_method(s, meth);
}
bclose(s, item.span);
}
2012-08-04 02:59:04 +00:00
ast::item_trait(tps, traits, methods) => {
head(s, ~"trait");
2012-06-10 07:49:59 +00:00
word(s.s, *item.ident);
2012-01-30 10:36:18 +00:00
print_type_params(s, tps);
if vec::len(traits) != 0u {
word_space(s, ~":");
commasep(s, inconsistent, traits, |s, p|
print_path(s, p.path, false));
}
word(s.s, ~" ");
bopen(s);
for methods.each |meth| { print_trait_method(s, meth); }
bclose(s, item.span);
}
2012-08-04 02:59:04 +00:00
ast::item_mac({node: ast::mac_invoc_tt(pth, tts), _}) => {
head(s, path_to_str(pth) + ~"! " + *item.ident);
bopen(s);
2012-07-28 00:38:01 +00:00
for tts.each |tt| { print_tt(s, tt); }
bclose(s, item.span);
}
2012-08-04 02:59:04 +00:00
ast::item_mac(_) => {
fail ~"invalid item-position syntax bit"
}
}
s.ann.post(ann_node);
}
2012-07-28 00:38:01 +00:00
/// This doesn't deserve to be called "pretty" printing, but it should be
/// meaning-preserving. A quick hack that might help would be to look at the
/// spans embedded in the TTs to decide where to put spaces and newlines.
/// But it'd be better to parse these according to the grammar of the
/// appropriate macro, transcribe back into the grammar we just parsed from,
/// and then pretty-print the resulting AST nodes (so, e.g., we print
/// expression arguments as expressions). It can be done! I think.
fn print_tt(s: ps, tt: ast::token_tree) {
2012-08-06 19:34:08 +00:00
match tt {
ast::tt_delim(tts) => for tts.each() |tt_elt| { print_tt(s, tt_elt); },
2012-08-04 02:59:04 +00:00
ast::tt_tok(_, tk) => {
2012-08-06 19:34:08 +00:00
match tk {
2012-08-04 02:59:04 +00:00
parse::token::IDENT(*) => { // don't let idents run together
if s.s.token_tree_last_was_ident { word(s.s, ~" ") }
s.s.token_tree_last_was_ident = true;
}
2012-08-04 02:59:04 +00:00
_ => { s.s.token_tree_last_was_ident = false; }
2012-07-28 00:38:01 +00:00
}
word(s.s, parse::token::to_str(*s.intr, tk));
2012-07-28 00:38:01 +00:00
}
2012-08-04 02:59:04 +00:00
ast::tt_seq(_, tts, sep, zerok) => {
2012-07-28 00:38:01 +00:00
word(s.s, ~"$(");
for tts.each() |tt_elt| { print_tt(s, tt_elt); }
word(s.s, ~")");
2012-08-06 19:34:08 +00:00
match sep {
2012-08-04 02:59:04 +00:00
some(tk) => word(s.s, parse::token::to_str(*s.intr, tk)),
none => ()
2012-07-28 00:38:01 +00:00
}
word(s.s, if zerok { ~"*" } else { ~"+" });
s.s.token_tree_last_was_ident = false;
2012-07-28 00:38:01 +00:00
}
2012-08-04 02:59:04 +00:00
ast::tt_nonterminal(_, name) => {
2012-07-28 00:38:01 +00:00
word(s.s, ~"$" + *name);
s.s.token_tree_last_was_ident = true;
2012-07-28 00:38:01 +00:00
}
}
}
2012-01-26 01:22:08 +00:00
fn print_variant(s: ps, v: ast::variant) {
2012-06-10 07:49:59 +00:00
word(s.s, *v.node.name);
match v.node.kind {
ast::tuple_variant_kind(args) => {
if vec::len(args) > 0u {
popen(s);
fn print_variant_arg(s: ps, arg: ast::variant_arg) {
print_type(s, arg.ty);
}
commasep(s, consistent, args, print_variant_arg);
pclose(s);
}
2012-01-26 01:22:08 +00:00
}
ast::struct_variant_kind => {}
2012-01-26 01:22:08 +00:00
}
2012-08-06 19:34:08 +00:00
match v.node.disr_expr {
2012-08-04 02:59:04 +00:00
some(d) => {
2012-01-26 01:22:08 +00:00
space(s.s);
word_space(s, ~"=");
2012-01-26 01:22:08 +00:00
print_expr(s, d);
}
2012-08-04 02:59:04 +00:00
_ => ()
2012-01-26 01:22:08 +00:00
}
}
fn print_ty_method(s: ps, m: ast::ty_method) {
hardbreak_if_not_bol(s);
maybe_print_comment(s, m.span.lo);
print_outer_attributes(s, m.attrs);
print_ty_fn(s, none, @~[], m.decl, some(m.ident), some(m.tps));
word(s.s, ~";");
}
fn print_trait_method(s: ps, m: ast::trait_method) {
2012-08-06 19:34:08 +00:00
match m {
2012-08-04 02:59:04 +00:00
required(ty_m) => print_ty_method(s, ty_m),
provided(m) => print_method(s, m)
}
}
fn print_method(s: ps, meth: @ast::method) {
hardbreak_if_not_bol(s);
maybe_print_comment(s, meth.span.lo);
print_outer_attributes(s, meth.attrs);
print_fn(s, meth.decl, meth.ident, meth.tps);
word(s.s, ~" ");
print_block_with_attrs(s, meth.body, meth.attrs);
}
fn print_outer_attributes(s: ps, attrs: ~[ast::attribute]) {
let mut count = 0;
2012-06-30 23:19:07 +00:00
for attrs.each |attr| {
2012-08-06 19:34:08 +00:00
match attr.node.style {
2012-08-04 02:59:04 +00:00
ast::attr_outer => { print_attribute(s, attr); count += 1; }
_ => {/* fallthrough */ }
}
}
2011-07-27 12:19:39 +00:00
if count > 0 { hardbreak_if_not_bol(s); }
}
fn print_inner_attributes(s: ps, attrs: ~[ast::attribute]) {
let mut count = 0;
2012-06-30 23:19:07 +00:00
for attrs.each |attr| {
2012-08-06 19:34:08 +00:00
match attr.node.style {
2012-08-04 02:59:04 +00:00
ast::attr_inner => {
2011-07-27 12:19:39 +00:00
print_attribute(s, attr);
if !attr.node.is_sugared_doc {
word(s.s, ~";");
}
2011-07-27 12:19:39 +00:00
count += 1;
}
2012-08-04 02:59:04 +00:00
_ => {/* fallthrough */ }
}
}
2011-07-27 12:19:39 +00:00
if count > 0 { hardbreak_if_not_bol(s); }
}
fn print_attribute(s: ps, attr: ast::attribute) {
hardbreak_if_not_bol(s);
maybe_print_comment(s, attr.span.lo);
if attr.node.is_sugared_doc {
let meta = attr::attr_meta(attr);
let comment = attr::get_meta_item_value_str(meta).get();
word(s.s, *comment);
} else {
word(s.s, ~"#[");
print_meta_item(s, @attr.node.value);
word(s.s, ~"]");
}
}
fn print_stmt(s: ps, st: ast::stmt) {
maybe_print_comment(s, st.span.lo);
2012-08-06 19:34:08 +00:00
match st.node {
2012-08-04 02:59:04 +00:00
ast::stmt_decl(decl, _) => {
print_decl(s, decl);
}
2012-08-04 02:59:04 +00:00
ast::stmt_expr(expr, _) => {
space_if_not_bol(s);
print_expr(s, expr);
}
2012-08-04 02:59:04 +00:00
ast::stmt_semi(expr, _) => {
space_if_not_bol(s);
print_expr(s, expr);
word(s.s, ~";");
}
}
if parse::classify::stmt_ends_with_semi(st) { word(s.s, ~";"); }
maybe_print_trailing_comment(s, st.span, none::<uint>);
}
fn print_block(s: ps, blk: ast::blk) {
2011-08-15 21:42:33 +00:00
print_possibly_embedded_block(s, blk, block_normal, indent_unit);
}
fn print_block_with_attrs(s: ps, blk: ast::blk, attrs: ~[ast::attribute]) {
print_possibly_embedded_block_(s, blk, block_normal, indent_unit, attrs);
}
enum embed_type { block_block_fn, block_normal, }
2011-08-15 21:42:33 +00:00
fn print_possibly_embedded_block(s: ps, blk: ast::blk, embedded: embed_type,
2011-07-27 12:19:39 +00:00
indented: uint) {
print_possibly_embedded_block_(
s, blk, embedded, indented, ~[]);
}
fn print_possibly_embedded_block_(s: ps, blk: ast::blk, embedded: embed_type,
indented: uint, attrs: ~[ast::attribute]) {
2012-08-06 19:34:08 +00:00
match blk.node.rules {
2012-08-04 02:59:04 +00:00
ast::unchecked_blk => word(s.s, ~"unchecked"),
ast::unsafe_blk => word(s.s, ~"unsafe"),
ast::default_blk => ()
}
maybe_print_comment(s, blk.span.lo);
2011-07-27 12:19:39 +00:00
let ann_node = node_block(s, blk);
s.ann.pre(ann_node);
2012-08-06 19:34:08 +00:00
match embedded {
2012-08-04 02:59:04 +00:00
block_block_fn => end(s),
block_normal => bopen(s)
2011-08-15 21:42:33 +00:00
}
print_inner_attributes(s, attrs);
2012-06-30 23:19:07 +00:00
for blk.node.view_items.each |vi| { print_view_item(s, vi); }
for blk.node.stmts.each |st| {
print_stmt(s, *st);
}
2012-08-06 19:34:08 +00:00
match blk.node.expr {
2012-08-04 02:59:04 +00:00
some(expr) => {
2011-07-27 12:19:39 +00:00
space_if_not_bol(s);
print_expr(s, expr);
2011-07-27 12:19:39 +00:00
maybe_print_trailing_comment(s, expr.span, some(blk.span.hi));
}
2012-08-04 02:59:04 +00:00
_ => ()
}
bclose_(s, blk.span, indented);
s.ann.post(ann_node);
}
2012-08-02 00:30:05 +00:00
// return and fail, without arguments cannot appear is the discriminant of if,
// alt, do, & while unambiguously without being parenthesized
fn print_maybe_parens_discrim(s: ps, e: @ast::expr) {
2012-08-06 19:34:08 +00:00
let disambig = match e.node {
2012-08-04 02:59:04 +00:00
ast::expr_ret(none) | ast::expr_fail(none) => true,
_ => false
};
if disambig { popen(s); }
print_expr(s, e);
if disambig { pclose(s); }
}
fn print_if(s: ps, test: @ast::expr, blk: ast::blk,
elseopt: option<@ast::expr>, chk: bool) {
head(s, ~"if");
if chk { word_nbsp(s, ~"check"); }
print_maybe_parens_discrim(s, test);
2011-06-16 21:08:17 +00:00
space(s.s);
print_block(s, blk);
fn do_else(s: ps, els: option<@ast::expr>) {
2012-08-06 19:34:08 +00:00
match els {
2012-08-04 02:59:04 +00:00
some(_else) => {
2012-08-06 19:34:08 +00:00
match _else.node {
2011-07-27 12:19:39 +00:00
// "another else-if"
2012-08-04 02:59:04 +00:00
ast::expr_if(i, t, e) => {
2011-07-27 12:19:39 +00:00
cbox(s, indent_unit - 1u);
ibox(s, 0u);
word(s.s, ~" else if ");
print_maybe_parens_discrim(s, i);
2011-07-27 12:19:39 +00:00
space(s.s);
print_block(s, t);
do_else(s, e);
}
// "final else"
2012-08-04 02:59:04 +00:00
ast::expr_block(b) => {
2011-07-27 12:19:39 +00:00
cbox(s, indent_unit - 1u);
ibox(s, 0u);
word(s.s, ~" else ");
2011-07-27 12:19:39 +00:00
print_block(s, b);
}
// BLEAH, constraints would be great here
2012-08-04 02:59:04 +00:00
_ => {
fail ~"print_if saw if with weird alternative";
}
2011-06-16 21:08:17 +00:00
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
_ => {/* fall through */ }
2011-06-16 21:08:17 +00:00
}
}
do_else(s, elseopt);
}
fn print_mac(s: ps, m: ast::mac) {
2012-08-06 19:34:08 +00:00
match m.node {
2012-08-04 02:59:04 +00:00
ast::mac_invoc(path, arg, body) => {
word(s.s, ~"#");
print_path(s, path, false);
2012-08-06 19:34:08 +00:00
match arg {
2012-08-04 02:59:04 +00:00
some(@{node: ast::expr_vec(_, _), _}) => (),
_ => word(s.s, ~" ")
}
2012-06-30 23:19:07 +00:00
option::iter(arg, |a| print_expr(s, a));
// FIXME: extension 'body' (#2339)
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::mac_invoc_tt(pth, tts) => {
head(s, path_to_str(pth) + ~"!");
bopen(s);
for tts.each() |tt| { print_tt(s, tt); }
bclose(s, m.span);
}
2012-08-04 02:59:04 +00:00
ast::mac_ellipsis => word(s.s, ~"..."),
ast::mac_var(v) => word(s.s, fmt!{"$%u", v}),
_ => { /* fixme */ }
}
}
2012-04-10 00:32:49 +00:00
fn print_vstore(s: ps, t: ast::vstore) {
2012-08-06 19:34:08 +00:00
match t {
2012-08-04 02:59:04 +00:00
ast::vstore_fixed(some(i)) => word(s.s, fmt!{"%u", i}),
ast::vstore_fixed(none) => word(s.s, ~"_"),
ast::vstore_uniq => word(s.s, ~"~"),
ast::vstore_box => word(s.s, ~"@"),
2012-08-06 19:34:08 +00:00
ast::vstore_slice(r) => match r.node {
2012-08-04 02:59:04 +00:00
ast::re_anon => word(s.s, ~"&"),
ast::re_named(name) => {
word(s.s, ~"&");
word(s.s, *name);
word(s.s, ~".");
}
}
2012-08-04 02:59:04 +00:00
}
2012-04-10 00:32:49 +00:00
}
fn print_expr(s: ps, &&expr: @ast::expr) {
fn print_field(s: ps, field: ast::field) {
ibox(s, indent_unit);
if field.node.mutbl == ast::m_mutbl { word_nbsp(s, ~"mut"); }
word(s.s, *field.node.ident);
word_space(s, ~":");
print_expr(s, field.node.expr);
end(s);
}
2012-08-02 00:30:05 +00:00
fn get_span(field: ast::field) -> codemap::span { return field.span; }
maybe_print_comment(s, expr.span.lo);
ibox(s, indent_unit);
2011-07-27 12:19:39 +00:00
let ann_node = node_expr(s, expr);
s.ann.pre(ann_node);
2012-08-06 19:34:08 +00:00
match expr.node {
ast::expr_vstore(e, v) => match v {
2012-08-04 02:59:04 +00:00
ast::vstore_fixed(_) => {
print_expr(s, e);
2012-08-04 02:59:04 +00:00
word(s.s, ~"/");
print_vstore(s, v);
}
2012-08-04 02:59:04 +00:00
_ => {
print_vstore(s, v);
2012-08-04 02:59:04 +00:00
print_expr(s, e);
}
},
2012-08-04 02:59:04 +00:00
ast::expr_vec(exprs, mutbl) => {
2011-07-27 12:19:39 +00:00
ibox(s, indent_unit);
word(s.s, ~"[");
if mutbl == ast::m_mutbl {
word(s.s, ~"mut");
2011-08-15 23:38:23 +00:00
if vec::len(exprs) > 0u { nbsp(s); }
}
2011-07-27 12:19:39 +00:00
commasep_exprs(s, inconsistent, exprs);
word(s.s, ~"]");
2011-07-27 12:19:39 +00:00
end(s);
}
ast::expr_repeat(element, count, mutbl) => {
ibox(s, indent_unit);
word(s.s, ~"[");
if mutbl == ast::m_mutbl {
word(s.s, ~"mut");
nbsp(s);
}
print_expr(s, element);
word(s.s, ~",");
word(s.s, ~"..");
print_expr(s, count);
word(s.s, ~"]");
end(s);
}
2012-08-04 02:59:04 +00:00
ast::expr_rec(fields, wth) => {
word(s.s, ~"{");
2011-07-27 12:19:39 +00:00
commasep_cmnt(s, consistent, fields, print_field, get_span);
2012-08-06 19:34:08 +00:00
match wth {
2012-08-04 02:59:04 +00:00
some(expr) => {
2011-08-15 23:38:23 +00:00
if vec::len(fields) > 0u { space(s.s); }
2011-07-27 12:19:39 +00:00
ibox(s, indent_unit);
word_space(s, ~"with");
print_expr(s, expr);
2011-07-27 12:19:39 +00:00
end(s);
}
2012-08-04 02:59:04 +00:00
_ => word(s.s, ~",")
}
word(s.s, ~"}");
2011-07-27 12:19:39 +00:00
}
ast::expr_struct(path, fields, wth) => {
print_path(s, path, true);
word(s.s, ~"{");
commasep_cmnt(s, consistent, fields, print_field, get_span);
2012-08-06 19:34:08 +00:00
match wth {
some(expr) => {
if vec::len(fields) > 0u { space(s.s); }
ibox(s, indent_unit);
word_space(s, ~"with");
print_expr(s, expr);
end(s);
}
_ => word(s.s, ~",")
}
word(s.s, ~"}");
}
2012-08-04 02:59:04 +00:00
ast::expr_tup(exprs) => {
popen(s);
2011-08-15 10:08:05 +00:00
commasep_exprs(s, inconsistent, exprs);
pclose(s);
}
2012-08-04 02:59:04 +00:00
ast::expr_call(func, args, has_block) => {
let mut base_args = args;
let blk = if has_block {
let blk_arg = vec::pop(base_args);
2012-08-06 19:34:08 +00:00
match blk_arg.node {
2012-08-04 02:59:04 +00:00
ast::expr_loop_body(_) => word_nbsp(s, ~"for"),
ast::expr_do_body(_) => word_nbsp(s, ~"do"),
_ => ()
}
some(blk_arg)
} else { none };
print_expr_parens_if_not_bot(s, func);
if !has_block || vec::len(base_args) > 0u {
popen(s);
commasep_exprs(s, inconsistent, base_args);
pclose(s);
}
if has_block {
nbsp(s);
print_expr(s, option::get(blk));
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::expr_binary(op, lhs, rhs) => {
2011-07-27 12:19:39 +00:00
let prec = operator_prec(op);
print_op_maybe_parens(s, lhs, prec);
2011-07-27 12:19:39 +00:00
space(s.s);
word_space(s, ast_util::binop_to_str(op));
print_op_maybe_parens(s, rhs, prec + 1u);
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::expr_unary(op, expr) => {
word(s.s, ast_util::unop_to_str(op));
print_op_maybe_parens(s, expr, parse::prec::unop_prec);
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::expr_addr_of(m, expr) => {
word(s.s, ~"&");
print_mutability(s, m);
print_expr(s, expr);
}
2012-08-04 02:59:04 +00:00
ast::expr_lit(lit) => print_literal(s, lit),
ast::expr_cast(expr, ty) => {
print_op_maybe_parens(s, expr, parse::prec::as_prec);
2011-07-27 12:19:39 +00:00
space(s.s);
word_space(s, ~"as");
print_type_ex(s, ty, true);
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::expr_if(test, blk, elseopt) => {
2011-07-27 12:19:39 +00:00
print_if(s, test, blk, elseopt, false);
}
2012-08-04 02:59:04 +00:00
ast::expr_while(test, blk) => {
head(s, ~"while");
print_maybe_parens_discrim(s, test);
2011-07-27 12:19:39 +00:00
space(s.s);
print_block(s, blk);
}
2012-08-04 02:59:04 +00:00
ast::expr_loop(blk) => {
head(s, ~"loop");
space(s.s);
print_block(s, blk);
}
2012-08-07 20:35:51 +00:00
ast::expr_match(expr, arms, mode) => {
2011-07-27 12:19:39 +00:00
cbox(s, alt_indent_unit);
ibox(s, 4u);
2012-08-06 19:34:08 +00:00
word_nbsp(s, ~"match");
if mode == ast::alt_check { word_nbsp(s, ~"check"); }
print_maybe_parens_discrim(s, expr);
2011-07-27 12:19:39 +00:00
space(s.s);
bopen(s);
let len = arms.len();
for arms.eachi |i, arm| {
space(s.s);
cbox(s, alt_indent_unit);
2011-07-27 12:19:39 +00:00
ibox(s, 0u);
let mut first = true;
2012-06-30 23:19:07 +00:00
for arm.pats.each |p| {
2011-07-27 12:19:39 +00:00
if first {
first = false;
} else { space(s.s); word_space(s, ~"|"); }
2011-07-27 12:19:39 +00:00
print_pat(s, p);
}
space(s.s);
2012-08-06 19:34:08 +00:00
match arm.guard {
2012-08-04 02:59:04 +00:00
some(e) => {
word_space(s, ~"if");
print_expr(s, e);
space(s.s);
}
none => ()
}
word_space(s, ~"=>");
// Extract the expression from the extra block the parser adds
assert arm.body.node.view_items.is_empty();
assert arm.body.node.stmts.is_empty();
assert arm.body.node.rules == ast::default_blk;
2012-08-06 19:34:08 +00:00
match arm.body.node.expr {
some(expr) => {
end(s); // close the ibox for the pattern
print_expr(s, expr);
if !expr_is_simple_block(expr)
&& i < len - 1 {
word(s.s, ~",");
}
end(s); // close enclosing cbox
}
none => fail
}
}
2011-07-27 12:19:39 +00:00
bclose_(s, expr.span, alt_indent_unit);
}
2012-08-04 02:59:04 +00:00
ast::expr_fn(proto, decl, body, cap_clause) => {
// containing cbox, will be closed by print-block at }
cbox(s, indent_unit);
// head-box, will be closed by print-block at start
ibox(s, 0u);
2012-05-25 06:44:58 +00:00
print_purity(s, decl.purity);
word(s.s, proto_to_str(proto));
print_fn_args_and_ret(s, decl, *cap_clause);
space(s.s);
print_block(s, body);
}
2012-08-04 02:59:04 +00:00
ast::expr_fn_block(decl, body, cap_clause) => {
print_fn_block_args(s, decl, *cap_clause);
// The parser always adds an extra implicit block around lambdas
assert body.node.stmts.is_empty();
assert body.node.expr.is_some();
space(s.s);
print_expr(s, body.node.expr.get());
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::expr_loop_body(body) => {
print_expr(s, body);
}
2012-08-04 02:59:04 +00:00
ast::expr_do_body(body) => {
2012-06-19 00:42:09 +00:00
print_expr(s, body);
}
2012-08-04 02:59:04 +00:00
ast::expr_block(blk) => {
2011-07-27 12:19:39 +00:00
// containing cbox, will be closed by print-block at }
cbox(s, indent_unit);
// head-box, will be closed by print-block after {
ibox(s, 0u);
print_block(s, blk);
}
2012-08-04 02:59:04 +00:00
ast::expr_copy(e) => { word_space(s, ~"copy"); print_expr(s, e); }
ast::expr_unary_move(e) => { word_space(s, ~"move"); print_expr(s, e); }
ast::expr_move(lhs, rhs) => {
2011-07-27 12:19:39 +00:00
print_expr(s, lhs);
space(s.s);
word_space(s, ~"<-");
2011-07-27 12:19:39 +00:00
print_expr(s, rhs);
}
2012-08-04 02:59:04 +00:00
ast::expr_assign(lhs, rhs) => {
2011-07-27 12:19:39 +00:00
print_expr(s, lhs);
space(s.s);
word_space(s, ~"=");
2011-07-27 12:19:39 +00:00
print_expr(s, rhs);
}
2012-08-04 02:59:04 +00:00
ast::expr_swap(lhs, rhs) => {
2011-07-27 12:19:39 +00:00
print_expr(s, lhs);
space(s.s);
word_space(s, ~"<->");
2011-07-27 12:19:39 +00:00
print_expr(s, rhs);
}
2012-08-04 02:59:04 +00:00
ast::expr_assign_op(op, lhs, rhs) => {
2011-07-27 12:19:39 +00:00
print_expr(s, lhs);
space(s.s);
word(s.s, ast_util::binop_to_str(op));
word_space(s, ~"=");
2011-07-27 12:19:39 +00:00
print_expr(s, rhs);
}
2012-08-04 02:59:04 +00:00
ast::expr_field(expr, id, tys) => {
2011-09-13 15:08:59 +00:00
// Deal with '10.x'
if ends_in_lit_int(expr) {
2011-09-13 15:08:59 +00:00
popen(s); print_expr(s, expr); pclose(s);
} else {
print_expr_parens_if_not_bot(s, expr);
2011-09-13 15:08:59 +00:00
}
word(s.s, ~".");
2012-06-10 07:49:59 +00:00
word(s.s, *id);
if vec::len(tys) > 0u {
word(s.s, ~"::<");
commasep(s, inconsistent, tys, print_type);
word(s.s, ~">");
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::expr_index(expr, index) => {
print_expr_parens_if_not_bot(s, expr);
word(s.s, ~"[");
2011-07-27 12:19:39 +00:00
print_expr(s, index);
word(s.s, ~"]");
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::expr_path(path) => print_path(s, path, true),
ast::expr_fail(maybe_fail_val) => {
word(s.s, ~"fail");
2012-08-06 19:34:08 +00:00
match maybe_fail_val {
2012-08-04 02:59:04 +00:00
some(expr) => { word(s.s, ~" "); print_expr(s, expr); }
_ => ()
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::expr_break => word(s.s, ~"break"),
ast::expr_again => word(s.s, ~"again"),
ast::expr_ret(result) => {
2012-08-02 00:30:05 +00:00
word(s.s, ~"return");
2012-08-06 19:34:08 +00:00
match result {
2012-08-04 02:59:04 +00:00
some(expr) => { word(s.s, ~" "); print_expr(s, expr); }
_ => ()
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::expr_log(lvl, lexp, expr) => {
2012-08-06 19:34:08 +00:00
match check lvl {
2012-08-04 02:59:04 +00:00
1 => { word_nbsp(s, ~"log"); print_expr(s, expr); }
0 => { word_nbsp(s, ~"log_err"); print_expr(s, expr); }
2 => {
word_nbsp(s, ~"log");
popen(s);
print_expr(s, lexp);
word(s.s, ~",");
space_if_not_bol(s);
print_expr(s, expr);
pclose(s);
}
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::expr_assert(expr) => {
word_nbsp(s, ~"assert");
2011-07-27 12:19:39 +00:00
print_expr(s, expr);
}
2012-08-04 02:59:04 +00:00
ast::expr_mac(m) => print_mac(s, m),
}
s.ann.post(ann_node);
end(s);
}
fn print_expr_parens_if_not_bot(s: ps, ex: @ast::expr) {
2012-08-06 19:34:08 +00:00
let parens = match ex.node {
ast::expr_fail(_) | ast::expr_ret(_) |
ast::expr_binary(_, _, _) | ast::expr_unary(_, _) |
ast::expr_move(_, _) | ast::expr_copy(_) |
2012-05-14 23:55:01 +00:00
ast::expr_assign(_, _) |
ast::expr_assign_op(_, _, _) | ast::expr_swap(_, _) |
ast::expr_log(_, _, _) | ast::expr_assert(_) |
ast::expr_call(_, _, true) |
2012-08-04 02:59:04 +00:00
ast::expr_vstore(_, _) => true,
_ => false
};
if parens { popen(s); }
print_expr(s, ex);
if parens { pclose(s); }
}
fn print_local_decl(s: ps, loc: @ast::local) {
print_pat(s, loc.node.pat);
2012-08-06 19:34:08 +00:00
match loc.node.ty.node {
2012-08-04 02:59:04 +00:00
ast::ty_infer => (),
_ => { word_space(s, ~":"); print_type(s, loc.node.ty); }
}
}
fn print_decl(s: ps, decl: @ast::decl) {
maybe_print_comment(s, decl.span.lo);
2012-08-06 19:34:08 +00:00
match decl.node {
2012-08-04 02:59:04 +00:00
ast::decl_local(locs) => {
2011-07-27 12:19:39 +00:00
space_if_not_bol(s);
ibox(s, indent_unit);
word_nbsp(s, ~"let");
2012-03-27 01:35:18 +00:00
// if any are mut, all are mut
2012-06-30 23:19:07 +00:00
if vec::any(locs, |l| l.node.is_mutbl) {
assert vec::all(locs, |l| l.node.is_mutbl);
word_nbsp(s, ~"mut");
}
fn print_local(s: ps, &&loc: @ast::local) {
ibox(s, indent_unit);
print_local_decl(s, loc);
end(s);
2012-08-06 19:34:08 +00:00
match loc.node.init {
2012-08-04 02:59:04 +00:00
some(init) => {
2011-07-27 12:19:39 +00:00
nbsp(s);
2012-08-06 19:34:08 +00:00
match init.op {
2012-08-04 02:59:04 +00:00
ast::init_assign => word_space(s, ~"="),
ast::init_move => word_space(s, ~"<-")
}
2011-07-27 12:19:39 +00:00
print_expr(s, init.expr);
}
2012-08-04 02:59:04 +00:00
_ => ()
}
}
commasep(s, consistent, locs, print_local);
2011-07-27 12:19:39 +00:00
end(s);
}
2012-08-04 02:59:04 +00:00
ast::decl_item(item) => print_item(s, item)
}
}
2012-06-10 07:49:59 +00:00
fn print_ident(s: ps, ident: ast::ident) { word(s.s, *ident); }
fn print_for_decl(s: ps, loc: @ast::local, coll: @ast::expr) {
print_local_decl(s, loc);
space(s.s);
word_space(s, ~"in");
print_expr(s, coll);
}
fn print_path(s: ps, &&path: @ast::path, colons_before_params: bool) {
maybe_print_comment(s, path.span.lo);
if path.global { word(s.s, ~"::"); }
let mut first = true;
2012-06-30 23:19:07 +00:00
for path.idents.each |id| {
if first { first = false; } else { word(s.s, ~"::"); }
2012-06-10 07:49:59 +00:00
word(s.s, *id);
}
if path.rp.is_some() || !path.types.is_empty() {
if colons_before_params { word(s.s, ~"::"); }
2012-08-06 19:34:08 +00:00
match path.rp {
2012-08-04 02:59:04 +00:00
none => { /* ok */ }
some(r) => {
word(s.s, ~"/");
print_region(s, r);
}
}
if !path.types.is_empty() {
word(s.s, ~"<");
commasep(s, inconsistent, path.types, print_type);
word(s.s, ~">");
}
}
}
fn print_pat(s: ps, &&pat: @ast::pat) {
maybe_print_comment(s, pat.span.lo);
2011-07-27 12:19:39 +00:00
let ann_node = node_pat(s, pat);
s.ann.pre(ann_node);
/* Pat isn't normalized, but the beauty of it
is that it doesn't matter */
2012-08-06 19:34:08 +00:00
match pat.node {
2012-08-04 02:59:04 +00:00
ast::pat_wild => word(s.s, ~"_"),
ast::pat_ident(binding_mode, path, sub) => {
2012-08-06 19:34:08 +00:00
match binding_mode {
ast::bind_by_ref(mutbl) => {
word_nbsp(s, ~"ref");
print_mutability(s, mutbl);
}
ast::bind_by_implicit_ref |
ast::bind_by_value => {}
}
print_path(s, path, true);
2012-08-06 19:34:08 +00:00
match sub {
some(p) => { word(s.s, ~"@"); print_pat(s, p); }
2012-08-04 02:59:04 +00:00
none => ()
}
}
2012-08-04 02:59:04 +00:00
ast::pat_enum(path, args_) => {
print_path(s, path, true);
2012-08-06 19:34:08 +00:00
match args_ {
2012-08-04 02:59:04 +00:00
none => word(s.s, ~"(*)"),
some(args) => {
if vec::len(args) > 0u {
popen(s);
commasep(s, inconsistent, args, print_pat);
pclose(s);
} else { }
}
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::pat_rec(fields, etc) => {
word(s.s, ~"{");
fn print_field(s: ps, f: ast::field_pat) {
2011-07-27 12:19:39 +00:00
cbox(s, indent_unit);
2012-06-10 07:49:59 +00:00
word(s.s, *f.ident);
word_space(s, ~":");
print_pat(s, f.pat);
end(s);
}
fn get_span(f: ast::field_pat) -> codemap::span { return f.pat.span; }
commasep_cmnt(s, consistent, fields, print_field, get_span);
if etc {
if vec::len(fields) != 0u { word_space(s, ~","); }
word(s.s, ~"_");
}
word(s.s, ~"}");
}
ast::pat_struct(path, fields, etc) => {
print_path(s, path, true);
word(s.s, ~"{");
fn print_field(s: ps, f: ast::field_pat) {
cbox(s, indent_unit);
word(s.s, *f.ident);
word_space(s, ~":");
2011-07-27 12:19:39 +00:00
print_pat(s, f.pat);
end(s);
2011-07-11 12:13:20 +00:00
}
2012-08-02 00:30:05 +00:00
fn get_span(f: ast::field_pat) -> codemap::span { return f.pat.span; }
2011-07-27 12:19:39 +00:00
commasep_cmnt(s, consistent, fields, print_field, get_span);
if etc {
if vec::len(fields) != 0u { word_space(s, ~","); }
word(s.s, ~"_");
}
word(s.s, ~"}");
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::pat_tup(elts) => {
2011-08-15 11:15:19 +00:00
popen(s);
commasep(s, inconsistent, elts, print_pat);
pclose(s);
}
2012-08-04 02:59:04 +00:00
ast::pat_box(inner) => { word(s.s, ~"@"); print_pat(s, inner); }
ast::pat_uniq(inner) => { word(s.s, ~"~"); print_pat(s, inner); }
ast::pat_lit(e) => print_expr(s, e),
ast::pat_range(begin, end) => {
print_expr(s, begin);
space(s.s);
word_space(s, ~"to");
print_expr(s, end);
}
}
s.ann.post(ann_node);
}
fn print_fn(s: ps, decl: ast::fn_decl, name: ast::ident,
typarams: ~[ast::ty_param]) {
2012-08-06 19:34:08 +00:00
match decl.purity {
2012-08-04 02:59:04 +00:00
ast::impure_fn => head(s, ~"fn"),
_ => head(s, purity_to_str(decl.purity) + ~" fn")
}
2012-06-10 07:49:59 +00:00
word(s.s, *name);
print_type_params(s, typarams);
print_fn_args_and_ret(s, decl, ~[]);
}
2012-05-04 19:33:04 +00:00
fn print_fn_args(s: ps, decl: ast::fn_decl,
cap_items: ~[ast::capture_item]) {
2012-05-04 19:33:04 +00:00
commasep(s, inconsistent, decl.inputs, print_arg);
if cap_items.is_not_empty() {
let mut first = decl.inputs.is_empty();
2012-06-30 23:19:07 +00:00
for cap_items.each |cap_item| {
if first { first = false; } else { word_space(s, ~","); }
if cap_item.is_move { word_nbsp(s, ~"move") }
else { word_nbsp(s, ~"copy") }
2012-06-10 07:49:59 +00:00
word(s.s, *cap_item.name);
}
}
}
2012-05-04 19:33:04 +00:00
fn print_fn_args_and_ret(s: ps, decl: ast::fn_decl,
cap_items: ~[ast::capture_item]) {
popen(s);
2012-05-04 19:33:04 +00:00
print_fn_args(s, decl, cap_items);
pclose(s);
maybe_print_comment(s, decl.output.span.lo);
2011-07-27 12:19:39 +00:00
if decl.output.node != ast::ty_nil {
space_if_not_bol(s);
word_space(s, ~"->");
print_type(s, decl.output);
}
}
2012-05-04 19:33:04 +00:00
fn print_fn_block_args(s: ps, decl: ast::fn_decl,
cap_items: ~[ast::capture_item]) {
word(s.s, ~"|");
2012-05-04 19:33:04 +00:00
print_fn_args(s, decl, cap_items);
word(s.s, ~"|");
if decl.output.node != ast::ty_infer {
space_if_not_bol(s);
word_space(s, ~"->");
print_type(s, decl.output);
}
2011-08-15 21:42:33 +00:00
maybe_print_comment(s, decl.output.span.lo);
}
fn mode_to_str(m: ast::mode) -> ~str {
2012-08-06 19:34:08 +00:00
match m {
2012-08-04 02:59:04 +00:00
ast::expl(ast::by_mutbl_ref) => ~"&",
ast::expl(ast::by_move) => ~"-",
ast::expl(ast::by_ref) => ~"&&",
ast::expl(ast::by_val) => ~"++",
ast::expl(ast::by_copy) => ~"+",
ast::infer(_) => ~""
}
}
fn print_arg_mode(s: ps, m: ast::mode) {
let ms = mode_to_str(m);
if ms != ~"" { word(s.s, ms); }
}
fn print_bounds(s: ps, bounds: @~[ast::ty_param_bound]) {
if vec::len(*bounds) > 0u {
word(s.s, ~":");
2012-06-30 23:19:07 +00:00
for vec::each(*bounds) |bound| {
nbsp(s);
2012-08-06 19:34:08 +00:00
match bound {
2012-08-04 02:59:04 +00:00
ast::bound_copy => word(s.s, ~"copy"),
ast::bound_send => word(s.s, ~"send"),
ast::bound_const => word(s.s, ~"const"),
ast::bound_owned => word(s.s, ~"owned"),
ast::bound_trait(t) => print_type(s, t)
}
}
2011-08-03 04:26:54 +00:00
}
}
fn print_type_params(s: ps, &&params: ~[ast::ty_param]) {
2011-08-15 23:38:23 +00:00
if vec::len(params) > 0u {
word(s.s, ~"<");
fn printParam(s: ps, param: ast::ty_param) {
2012-06-10 07:49:59 +00:00
word(s.s, *param.ident);
print_bounds(s, param.bounds);
}
commasep(s, inconsistent, params, printParam);
word(s.s, ~">");
}
}
fn print_meta_item(s: ps, &&item: @ast::meta_item) {
ibox(s, indent_unit);
2012-08-06 19:34:08 +00:00
match item.node {
2012-08-04 02:59:04 +00:00
ast::meta_word(name) => word(s.s, *name),
ast::meta_name_value(name, value) => {
2012-06-10 07:49:59 +00:00
word_space(s, *name);
word_space(s, ~"=");
2011-07-27 12:19:39 +00:00
print_literal(s, @value);
}
2012-08-04 02:59:04 +00:00
ast::meta_list(name, items) => {
2012-06-10 07:49:59 +00:00
word(s.s, *name);
2011-07-27 12:19:39 +00:00
popen(s);
commasep(s, consistent, items, print_meta_item);
pclose(s);
}
}
end(s);
}
fn print_view_path(s: ps, &&vp: @ast::view_path) {
2012-08-06 19:34:08 +00:00
match vp.node {
2012-08-04 02:59:04 +00:00
ast::view_path_simple(ident, path, _) => {
2012-04-23 11:04:46 +00:00
if path.idents[vec::len(path.idents)-1u] != ident {
2012-06-10 07:49:59 +00:00
word_space(s, *ident);
word_space(s, ~"=");
}
print_path(s, path, false);
}
2012-08-04 02:59:04 +00:00
ast::view_path_glob(path, _) => {
print_path(s, path, false);
word(s.s, ~"::*");
}
2012-08-04 02:59:04 +00:00
ast::view_path_list(path, idents, _) => {
print_path(s, path, false);
word(s.s, ~"::{");
2012-06-30 23:19:07 +00:00
do commasep(s, inconsistent, idents) |s, w| {
2012-06-10 07:49:59 +00:00
word(s.s, *w.node.name)
}
word(s.s, ~"}");
}
}
}
fn print_view_paths(s: ps, vps: ~[@ast::view_path]) {
commasep(s, inconsistent, vps, print_view_path);
}
fn print_view_item(s: ps, item: @ast::view_item) {
hardbreak_if_not_bol(s);
maybe_print_comment(s, item.span.lo);
print_outer_attributes(s, item.attrs);
2012-08-06 19:34:08 +00:00
match item.node {
2012-08-04 02:59:04 +00:00
ast::view_item_use(id, mta, _) => {
head(s, ~"use");
2012-06-10 07:49:59 +00:00
word(s.s, *id);
2011-08-15 23:38:23 +00:00
if vec::len(mta) > 0u {
2011-07-27 12:19:39 +00:00
popen(s);
commasep(s, consistent, mta, print_meta_item);
pclose(s);
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::view_item_import(vps) => {
head(s, ~"import");
print_view_paths(s, vps);
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::view_item_export(vps) => {
head(s, ~"export");
print_view_paths(s, vps);
}
}
word(s.s, ~";");
end(s); // end inner head-block
end(s); // end outer head-block
}
fn print_op_maybe_parens(s: ps, expr: @ast::expr, outer_prec: uint) {
let add_them = need_parens(expr, outer_prec);
2011-07-27 12:19:39 +00:00
if add_them { popen(s); }
print_expr(s, expr);
2011-07-27 12:19:39 +00:00
if add_them { pclose(s); }
}
fn print_mutability(s: ps, mutbl: ast::mutability) {
2012-08-06 19:34:08 +00:00
match mutbl {
2012-08-04 02:59:04 +00:00
ast::m_mutbl => word_nbsp(s, ~"mut"),
ast::m_const => word_nbsp(s, ~"const"),
ast::m_imm => {/* nothing */ }
}
2011-06-15 17:25:23 +00:00
}
fn print_mt(s: ps, mt: ast::mt) {
print_mutability(s, mt.mutbl);
print_type(s, mt.ty);
}
2012-05-04 19:33:04 +00:00
fn print_arg(s: ps, input: ast::arg) {
ibox(s, indent_unit);
print_arg_mode(s, input.mode);
2012-08-06 19:34:08 +00:00
match input.ty.node {
2012-08-04 02:59:04 +00:00
ast::ty_infer => word(s.s, *input.ident),
_ => {
2012-06-10 07:49:59 +00:00
if str::len(*input.ident) > 0u {
word_space(s, *input.ident + ~":");
2012-05-04 19:33:04 +00:00
}
print_type(s, input.ty);
}
}
end(s);
}
fn print_ty_fn(s: ps, opt_proto: option<ast::proto>,
bounds: @~[ast::ty_param_bound],
decl: ast::fn_decl, id: option<ast::ident>,
tps: option<~[ast::ty_param]>) {
ibox(s, indent_unit);
word(s.s, opt_proto_to_str(opt_proto));
print_bounds(s, bounds);
2012-08-06 19:34:08 +00:00
match id { some(id) => { word(s.s, ~" "); word(s.s, *id); } _ => () }
match tps { some(tps) => print_type_params(s, tps), _ => () }
2011-06-03 00:36:28 +00:00
zerobreak(s.s);
2011-05-29 02:16:18 +00:00
popen(s);
commasep(s, inconsistent, decl.inputs, print_arg);
pclose(s);
maybe_print_comment(s, decl.output.span.lo);
if decl.output.node != ast::ty_nil {
space_if_not_bol(s);
ibox(s, indent_unit);
word_space(s, ~"->");
if decl.cf == ast::noreturn { word_nbsp(s, ~"!"); }
else { print_type(s, decl.output); }
end(s);
}
end(s);
}
fn maybe_print_trailing_comment(s: ps, span: codemap::span,
next_pos: option<uint>) {
let mut cm;
2012-08-06 19:34:08 +00:00
match s.cm { some(ccm) => cm = ccm, _ => return }
match next_comment(s) {
2012-08-04 02:59:04 +00:00
some(cmnt) => {
2012-08-02 00:30:05 +00:00
if cmnt.style != comments::trailing { return; }
2011-07-27 12:19:39 +00:00
let span_line = codemap::lookup_char_pos(cm, span.hi);
let comment_line = codemap::lookup_char_pos(cm, cmnt.pos);
let mut next = cmnt.pos + 1u;
2012-08-06 19:34:08 +00:00
match next_pos { none => (), some(p) => next = p }
2011-07-27 12:19:39 +00:00
if span.hi < cmnt.pos && cmnt.pos < next &&
span_line.line == comment_line.line {
print_comment(s, cmnt);
s.cur_cmnt += 1u;
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
_ => ()
}
}
fn print_remaining_comments(s: ps) {
// If there aren't any remaining comments, then we need to manually
// make sure there is a line break at the end.
if option::is_none(next_comment(s)) { hardbreak(s.s); }
loop {
2012-08-06 19:34:08 +00:00
match next_comment(s) {
2012-08-04 02:59:04 +00:00
some(cmnt) => { print_comment(s, cmnt); s.cur_cmnt += 1u; }
_ => break
}
}
}
fn print_literal(s: ps, &&lit: @ast::lit) {
maybe_print_comment(s, lit.span.lo);
2012-08-06 19:34:08 +00:00
match next_lit(s, lit.span.lo) {
2012-08-04 02:59:04 +00:00
some(ltrl) => {
word(s.s, ltrl.lit);
2012-08-02 00:30:05 +00:00
return;
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
_ => ()
}
2012-08-06 19:34:08 +00:00
match lit.node {
2012-08-04 02:59:04 +00:00
ast::lit_str(st) => print_string(s, *st),
ast::lit_int(ch, ast::ty_char) => {
word(s.s, ~"'" + char::escape_default(ch as char) + ~"'");
}
2012-08-04 02:59:04 +00:00
ast::lit_int(i, t) => {
if i < 0_i64 {
word(s.s,
~"-" + u64::to_str(-i as u64, 10u)
+ ast_util::int_ty_to_str(t));
} else {
word(s.s,
u64::to_str(i as u64, 10u)
+ ast_util::int_ty_to_str(t));
}
}
2012-08-04 02:59:04 +00:00
ast::lit_uint(u, t) => {
word(s.s,
u64::to_str(u, 10u)
+ ast_util::uint_ty_to_str(t));
}
2012-08-04 02:59:04 +00:00
ast::lit_int_unsuffixed(i) => {
if i < 0_i64 {
word(s.s, ~"-" + u64::to_str(-i as u64, 10u));
} else {
word(s.s, u64::to_str(i as u64, 10u));
}
}
2012-08-04 02:59:04 +00:00
ast::lit_float(f, t) => {
2012-06-10 07:49:59 +00:00
word(s.s, *f + ast_util::float_ty_to_str(t));
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
ast::lit_nil => word(s.s, ~"()"),
ast::lit_bool(val) => {
if val { word(s.s, ~"true"); } else { word(s.s, ~"false"); }
2011-07-27 12:19:39 +00:00
}
}
}
2012-08-02 00:30:05 +00:00
fn lit_to_str(l: @ast::lit) -> ~str { return to_str(l, print_literal); }
fn next_lit(s: ps, pos: uint) -> option<comments::lit> {
2012-08-06 19:34:08 +00:00
match s.literals {
2012-08-04 02:59:04 +00:00
some(lits) => {
while s.cur_lit < vec::len(lits) {
let ltrl = lits[s.cur_lit];
2012-08-02 00:30:05 +00:00
if ltrl.pos > pos { return none; }
s.cur_lit += 1u;
2012-08-02 00:30:05 +00:00
if ltrl.pos == pos { return some(ltrl); }
}
2012-08-02 00:30:05 +00:00
return none;
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
_ => return none
}
}
fn maybe_print_comment(s: ps, pos: uint) {
loop {
2012-08-06 19:34:08 +00:00
match next_comment(s) {
2012-08-04 02:59:04 +00:00
some(cmnt) => {
2011-07-27 12:19:39 +00:00
if cmnt.pos < pos {
print_comment(s, cmnt);
s.cur_cmnt += 1u;
} else { break; }
}
2012-08-04 02:59:04 +00:00
_ => break
}
}
}
fn print_comment(s: ps, cmnt: comments::cmnt) {
2012-08-06 19:34:08 +00:00
match cmnt.style {
2012-08-04 02:59:04 +00:00
comments::mixed => {
2011-08-15 23:38:23 +00:00
assert (vec::len(cmnt.lines) == 1u);
2011-07-27 12:19:39 +00:00
zerobreak(s.s);
word(s.s, cmnt.lines[0]);
2011-07-27 12:19:39 +00:00
zerobreak(s.s);
}
2012-08-04 02:59:04 +00:00
comments::isolated => {
2011-07-27 12:19:39 +00:00
pprust::hardbreak_if_not_bol(s);
2012-06-30 23:19:07 +00:00
for cmnt.lines.each |line| {
// Don't print empty lines because they will end up as trailing
// whitespace
2011-09-02 22:34:58 +00:00
if str::is_not_empty(line) { word(s.s, line); }
hardbreak(s.s);
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
comments::trailing => {
word(s.s, ~" ");
2011-08-15 23:38:23 +00:00
if vec::len(cmnt.lines) == 1u {
word(s.s, cmnt.lines[0]);
hardbreak(s.s);
2011-07-27 12:19:39 +00:00
} else {
ibox(s, 0u);
2012-06-30 23:19:07 +00:00
for cmnt.lines.each |line| {
2011-09-02 22:34:58 +00:00
if str::is_not_empty(line) { word(s.s, line); }
hardbreak(s.s);
}
2011-07-27 12:19:39 +00:00
end(s);
}
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
comments::blank_line => {
2011-07-27 12:19:39 +00:00
// We need to do at least one, possibly two hardbreaks.
let is_semi =
2012-08-06 19:34:08 +00:00
match s.s.last_token() {
2012-08-04 02:59:04 +00:00
pp::STRING(s, _) => *s == ~";",
_ => false
};
if is_semi || is_begin(s) || is_end(s) { hardbreak(s.s); }
2011-07-27 12:19:39 +00:00
hardbreak(s.s);
}
}
}
fn print_string(s: ps, st: ~str) {
word(s.s, ~"\"");
word(s.s, str::escape_default(st));
word(s.s, ~"\"");
}
fn to_str<T>(t: T, f: fn@(ps, T)) -> ~str {
let buffer = io::mem_buffer();
let s = rust_printer(io::mem_buffer_writer(buffer));
f(s, t);
eof(s.s);
io::mem_buffer_str(buffer)
}
fn next_comment(s: ps) -> option<comments::cmnt> {
2012-08-06 19:34:08 +00:00
match s.comments {
2012-08-04 02:59:04 +00:00
some(cmnts) => {
2011-08-15 23:38:23 +00:00
if s.cur_cmnt < vec::len(cmnts) {
2012-08-02 00:30:05 +00:00
return some(cmnts[s.cur_cmnt]);
} else { return none::<comments::cmnt>; }
2011-07-27 12:19:39 +00:00
}
2012-08-04 02:59:04 +00:00
_ => return none::<comments::cmnt>
}
}
fn opt_proto_to_str(opt_p: option<ast::proto>) -> ~str {
2012-08-06 19:34:08 +00:00
match opt_p {
2012-08-04 02:59:04 +00:00
none => ~"fn",
some(p) => proto_to_str(p)
}
}
pure fn purity_to_str(p: ast::purity) -> ~str {
2012-08-06 19:34:08 +00:00
match p {
ast::impure_fn => ~"impure",
ast::unsafe_fn => ~"unsafe",
ast::pure_fn => ~"pure",
ast::extern_fn => ~"extern"
2012-05-25 06:44:58 +00:00
}
}
fn print_purity(s: ps, p: ast::purity) {
2012-08-06 19:34:08 +00:00
match p {
2012-08-04 02:59:04 +00:00
ast::impure_fn => (),
_ => word_nbsp(s, purity_to_str(p))
2012-05-25 06:44:58 +00:00
}
}
fn proto_to_str(p: ast::proto) -> ~str {
2012-08-06 19:34:08 +00:00
return match p {
2012-08-04 02:59:04 +00:00
ast::proto_bare => ~"extern fn",
ast::proto_block => ~"fn&",
ast::proto_uniq => ~"fn~",
ast::proto_box => ~"fn@"
};
}
//
// Local Variables:
// mode: rust
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// End:
//