mirror of
https://github.com/golang/go
synced 2024-11-21 19:54:41 -07:00
parent
65063bc61d
commit
10242e806f
@ -18,9 +18,9 @@ import (
|
||||
|
||||
// Disabled formatting - enable eventually and remove the flag.
|
||||
const (
|
||||
oneLineFuncDecls = false;
|
||||
compositeLitBlank = false;
|
||||
stringListMode = exprListMode(0); // previously: noIndent
|
||||
oneLineFuncDecls = false;
|
||||
compositeLitBlank = false;
|
||||
stringListMode = exprListMode(0); // previously: noIndent
|
||||
)
|
||||
|
||||
|
||||
@ -37,8 +37,10 @@ const (
|
||||
func (p *printer) linebreak(line, min, max int, ws whiteSpace, newSection bool) (printedBreak bool) {
|
||||
n := line - p.pos.Line;
|
||||
switch {
|
||||
case n < min: n = min;
|
||||
case n > max: n = max;
|
||||
case n < min:
|
||||
n = min;
|
||||
case n > max:
|
||||
n = max;
|
||||
}
|
||||
if n > 0 {
|
||||
p.print(ws);
|
||||
@ -118,12 +120,13 @@ func (p *printer) stringList(list []*ast.BasicLit, multiLine *bool) {
|
||||
}
|
||||
|
||||
|
||||
type exprListMode uint;
|
||||
type exprListMode uint
|
||||
|
||||
const (
|
||||
blankStart exprListMode = 1 << iota; // print a blank before the list
|
||||
commaSep; // elements are separated by commas
|
||||
commaTerm; // elements are terminated by comma
|
||||
noIndent; // no extra indentation in multi-line lists
|
||||
blankStart exprListMode = 1<<iota; // print a blank before the list
|
||||
commaSep; // elements are separated by commas
|
||||
commaTerm; // elements are terminated by comma
|
||||
noIndent; // no extra indentation in multi-line lists
|
||||
)
|
||||
|
||||
|
||||
@ -150,7 +153,7 @@ func (p *printer) exprList(prev token.Position, list []ast.Expr, mode exprListMo
|
||||
// all list entries on a single line
|
||||
for i, x := range list {
|
||||
if i > 0 {
|
||||
if mode & commaSep != 0 {
|
||||
if mode&commaSep != 0 {
|
||||
p.print(token.COMMA);
|
||||
}
|
||||
p.print(blank);
|
||||
@ -179,7 +182,7 @@ func (p *printer) exprList(prev token.Position, list []ast.Expr, mode exprListMo
|
||||
prev := line;
|
||||
line = x.Pos().Line;
|
||||
if i > 0 {
|
||||
if mode & commaSep != 0 {
|
||||
if mode&commaSep != 0 {
|
||||
p.print(token.COMMA);
|
||||
}
|
||||
if prev < line {
|
||||
@ -200,7 +203,7 @@ func (p *printer) exprList(prev token.Position, list []ast.Expr, mode exprListMo
|
||||
// expression list - be conservative and check anyway
|
||||
p.print(unindent);
|
||||
}
|
||||
p.print(formfeed); // terminating comma needs a line break to look good
|
||||
p.print(formfeed); // terminating comma needs a line break to look good
|
||||
} else if ws == ignore && mode&noIndent == 0 {
|
||||
p.print(unindent);
|
||||
}
|
||||
@ -304,7 +307,7 @@ func (p *printer) fieldList(lbrace token.Position, list []*ast.Field, rbrace tok
|
||||
p.print("// contains unexported fields");
|
||||
}
|
||||
|
||||
} else { // interface
|
||||
} else { // interface
|
||||
|
||||
var ml bool;
|
||||
for i, f := range list {
|
||||
@ -353,7 +356,7 @@ func needsBlanks(expr ast.Expr) bool {
|
||||
return false;
|
||||
case *ast.IndexExpr:
|
||||
// index expressions don't need blanks if the indexed expressions are simple
|
||||
return needsBlanks(x.X)
|
||||
return needsBlanks(x.X);
|
||||
case *ast.CallExpr:
|
||||
// call expressions need blanks if they have more than one
|
||||
// argument or if the function expression needs blanks
|
||||
@ -539,7 +542,7 @@ func (p *printer) expr1(expr ast.Expr, prec1 int, multiLine *bool) (optSemi bool
|
||||
p.print(blank);
|
||||
}
|
||||
p.print(x.Lbrace, token.LBRACE);
|
||||
p.exprList(x.Lbrace, x.Elts, commaSep|commaTerm, multiLine);
|
||||
p.exprList(x.Lbrace, x.Elts, commaSep | commaTerm, multiLine);
|
||||
p.print(x.Rbrace, token.RBRACE);
|
||||
|
||||
case *ast.Ellipsis:
|
||||
@ -603,7 +606,7 @@ func (p *printer) expr(x ast.Expr, multiLine *bool) (optSemi bool) {
|
||||
// ----------------------------------------------------------------------------
|
||||
// Statements
|
||||
|
||||
const maxStmtNewlines = 2 // maximum number of newlines between statements
|
||||
const maxStmtNewlines = 2 // maximum number of newlines between statements
|
||||
|
||||
// Print the statement list indented, but without a newline after the last statement.
|
||||
// Extra line breaks between statements in the source are respected but at most one
|
||||
@ -695,7 +698,7 @@ func (p *printer) stmt(stmt ast.Stmt, multiLine *bool) (optSemi bool) {
|
||||
|
||||
case *ast.DeclStmt:
|
||||
p.decl(s.Decl, inStmtList, multiLine);
|
||||
optSemi = true; // decl prints terminating semicolon if necessary
|
||||
optSemi = true; // decl prints terminating semicolon if necessary
|
||||
|
||||
case *ast.EmptyStmt:
|
||||
// nothing to do
|
||||
@ -775,7 +778,7 @@ func (p *printer) stmt(stmt ast.Stmt, multiLine *bool) (optSemi bool) {
|
||||
}
|
||||
p.print(s.Colon, token.COLON);
|
||||
p.stmtList(s.Body, 1);
|
||||
optSemi = true; // "block" without {}'s
|
||||
optSemi = true; // "block" without {}'s
|
||||
|
||||
case *ast.SwitchStmt:
|
||||
p.print(token.SWITCH);
|
||||
@ -793,7 +796,7 @@ func (p *printer) stmt(stmt ast.Stmt, multiLine *bool) (optSemi bool) {
|
||||
}
|
||||
p.print(s.Colon, token.COLON);
|
||||
p.stmtList(s.Body, 1);
|
||||
optSemi = true; // "block" without {}'s
|
||||
optSemi = true; // "block" without {}'s
|
||||
|
||||
case *ast.TypeSwitchStmt:
|
||||
p.print(token.SWITCH);
|
||||
@ -822,7 +825,7 @@ func (p *printer) stmt(stmt ast.Stmt, multiLine *bool) (optSemi bool) {
|
||||
}
|
||||
p.print(s.Colon, token.COLON);
|
||||
p.stmtList(s.Body, 1);
|
||||
optSemi = true; // "block" without {}'s
|
||||
optSemi = true; // "block" without {}'s
|
||||
|
||||
case *ast.SelectStmt:
|
||||
p.print(token.SELECT, blank);
|
||||
@ -862,9 +865,10 @@ func (p *printer) stmt(stmt ast.Stmt, multiLine *bool) (optSemi bool) {
|
||||
// ----------------------------------------------------------------------------
|
||||
// Declarations
|
||||
|
||||
type declContext uint;
|
||||
type declContext uint
|
||||
|
||||
const (
|
||||
atTop declContext = iota;
|
||||
atTop declContext = iota;
|
||||
inGroup;
|
||||
inStmtList;
|
||||
)
|
||||
@ -876,9 +880,9 @@ const (
|
||||
//
|
||||
func (p *printer) spec(spec ast.Spec, n int, context declContext, multiLine *bool) {
|
||||
var (
|
||||
optSemi bool; // true if a semicolon is optional
|
||||
comment *ast.CommentGroup; // a line comment, if any
|
||||
extraTabs int; // number of extra tabs before comment, if any
|
||||
optSemi bool; // true if a semicolon is optional
|
||||
comment *ast.CommentGroup; // a line comment, if any
|
||||
extraTabs int; // number of extra tabs before comment, if any
|
||||
)
|
||||
|
||||
switch s := spec.(type) {
|
||||
@ -893,7 +897,7 @@ func (p *printer) spec(spec ast.Spec, n int, context declContext, multiLine *boo
|
||||
|
||||
case *ast.ValueSpec:
|
||||
p.leadComment(s.Doc);
|
||||
p.identList(s.Names, multiLine); // always present
|
||||
p.identList(s.Names, multiLine); // always present
|
||||
if n == 1 {
|
||||
if s.Type != nil {
|
||||
p.print(blank);
|
||||
@ -984,25 +988,25 @@ func (p *printer) genDecl(d *ast.GenDecl, context declContext, multiLine *bool)
|
||||
func (p *printer) isOneLiner(b *ast.BlockStmt) bool {
|
||||
switch {
|
||||
case len(b.List) > 1 || p.commentBefore(b.Rbrace):
|
||||
return false; // too many statements or there is a comment - all bets are off
|
||||
return false; // too many statements or there is a comment - all bets are off
|
||||
case len(b.List) == 0:
|
||||
return true; // empty block and no comments
|
||||
return true; // empty block and no comments
|
||||
}
|
||||
|
||||
// test-print the statement and see if it would fit
|
||||
var buf bytes.Buffer;
|
||||
_, err := p.Config.Fprint(&buf, b.List[0]);
|
||||
if err != nil {
|
||||
return false; // don't try
|
||||
return false; // don't try
|
||||
}
|
||||
|
||||
if buf.Len() > 40 {
|
||||
return false; // too long
|
||||
return false; // too long
|
||||
}
|
||||
|
||||
for _, ch := range buf.Bytes() {
|
||||
if ch < ' ' {
|
||||
return false; // contains control chars (tabs, newlines)
|
||||
return false; // contains control chars (tabs, newlines)
|
||||
}
|
||||
}
|
||||
|
||||
@ -1075,7 +1079,7 @@ func (p *printer) decl(decl ast.Decl, context declContext, multiLine *bool) {
|
||||
// ----------------------------------------------------------------------------
|
||||
// Files
|
||||
|
||||
const maxDeclNewlines = 3 // maximum number of newlines between declarations
|
||||
const maxDeclNewlines = 3 // maximum number of newlines between declarations
|
||||
|
||||
func declToken(decl ast.Decl) (tok token.Token) {
|
||||
tok = token.ILLEGAL;
|
||||
|
@ -20,35 +20,35 @@ import (
|
||||
|
||||
|
||||
const (
|
||||
debug = false; // enable for debugging
|
||||
maxNewlines = 3; // maximum vertical white space
|
||||
debug = false; // enable for debugging
|
||||
maxNewlines = 3; // maximum vertical white space
|
||||
)
|
||||
|
||||
|
||||
type whiteSpace int
|
||||
|
||||
const (
|
||||
ignore = whiteSpace(0);
|
||||
blank = whiteSpace(' ');
|
||||
vtab = whiteSpace('\v');
|
||||
newline = whiteSpace('\n');
|
||||
formfeed = whiteSpace('\f');
|
||||
indent = whiteSpace('>');
|
||||
unindent = whiteSpace('<');
|
||||
ignore = whiteSpace(0);
|
||||
blank = whiteSpace(' ');
|
||||
vtab = whiteSpace('\v');
|
||||
newline = whiteSpace('\n');
|
||||
formfeed = whiteSpace('\f');
|
||||
indent = whiteSpace('>');
|
||||
unindent = whiteSpace('<');
|
||||
)
|
||||
|
||||
|
||||
var (
|
||||
esc = []byte{tabwriter.Escape};
|
||||
htab = []byte{'\t'};
|
||||
htabs = [...]byte{'\t', '\t', '\t', '\t', '\t', '\t', '\t', '\t'};
|
||||
newlines = [...]byte{'\n', '\n', '\n', '\n', '\n', '\n', '\n', '\n'}; // more than maxNewlines
|
||||
esc = []byte{tabwriter.Escape};
|
||||
htab = []byte{'\t'};
|
||||
htabs = [...]byte{'\t', '\t', '\t', '\t', '\t', '\t', '\t', '\t'};
|
||||
newlines = [...]byte{'\n', '\n', '\n', '\n', '\n', '\n', '\n', '\n'}; // more than maxNewlines
|
||||
|
||||
esc_quot = strings.Bytes("""); // shorter than """
|
||||
esc_apos = strings.Bytes("'"); // shorter than "'"
|
||||
esc_amp = strings.Bytes("&");
|
||||
esc_lt = strings.Bytes("<");
|
||||
esc_gt = strings.Bytes(">");
|
||||
esc_quot = strings.Bytes("""); // shorter than """
|
||||
esc_apos = strings.Bytes("'"); // shorter than "'"
|
||||
esc_amp = strings.Bytes("&");
|
||||
esc_lt = strings.Bytes("<");
|
||||
esc_gt = strings.Bytes(">");
|
||||
)
|
||||
|
||||
|
||||
@ -57,38 +57,38 @@ var noPos token.Position
|
||||
|
||||
|
||||
// Use ignoreMultiLine if the multiLine information is not important.
|
||||
var ignoreMultiLine = new(bool);
|
||||
var ignoreMultiLine = new(bool)
|
||||
|
||||
|
||||
type printer struct {
|
||||
// Configuration (does not change after initialization)
|
||||
output io.Writer;
|
||||
output io.Writer;
|
||||
Config;
|
||||
errors chan os.Error;
|
||||
errors chan os.Error;
|
||||
|
||||
// Current state
|
||||
written int; // number of bytes written
|
||||
indent int; // current indentation
|
||||
escape bool; // true if in escape sequence
|
||||
written int; // number of bytes written
|
||||
indent int; // current indentation
|
||||
escape bool; // true if in escape sequence
|
||||
|
||||
// Buffered whitespace
|
||||
buffer []whiteSpace;
|
||||
buffer []whiteSpace;
|
||||
|
||||
// The (possibly estimated) position in the generated output;
|
||||
// in AST space (i.e., pos is set whenever a token position is
|
||||
// known accurately, and updated dependending on what has been
|
||||
// written)
|
||||
pos token.Position;
|
||||
pos token.Position;
|
||||
|
||||
// The value of pos immediately after the last item has been
|
||||
// written using writeItem.
|
||||
last token.Position;
|
||||
last token.Position;
|
||||
|
||||
// HTML support
|
||||
lastTaggedLine int; // last line for which a line tag was written
|
||||
lastTaggedLine int; // last line for which a line tag was written
|
||||
|
||||
// The list of comments; or nil.
|
||||
comment *ast.CommentGroup;
|
||||
comment *ast.CommentGroup;
|
||||
}
|
||||
|
||||
|
||||
@ -96,7 +96,7 @@ func (p *printer) init(output io.Writer, cfg *Config) {
|
||||
p.output = output;
|
||||
p.Config = *cfg;
|
||||
p.errors = make(chan os.Error);
|
||||
p.buffer = make([]whiteSpace, 0, 16); // whitespace sequences are short
|
||||
p.buffer = make([]whiteSpace, 0, 16); // whitespace sequences are short
|
||||
}
|
||||
|
||||
|
||||
@ -126,7 +126,7 @@ func (p *printer) write(data []byte) {
|
||||
p.write0(data[i0 : i+1]);
|
||||
|
||||
// update p.pos
|
||||
p.pos.Offset += i+1 - i0;
|
||||
p.pos.Offset += i+1-i0;
|
||||
p.pos.Line++;
|
||||
p.pos.Column = 1;
|
||||
|
||||
@ -151,21 +151,26 @@ func (p *printer) write(data []byte) {
|
||||
case '"', '\'', '&', '<', '>':
|
||||
if p.Mode & GenHTML != 0 {
|
||||
// write segment ending in b
|
||||
p.write0(data[i0 : i]);
|
||||
p.write0(data[i0:i]);
|
||||
|
||||
// write HTML-escaped b
|
||||
var esc []byte;
|
||||
switch b {
|
||||
case '"': esc = esc_quot;
|
||||
case '\'': esc = esc_apos;
|
||||
case '&': esc = esc_amp;
|
||||
case '<': esc = esc_lt;
|
||||
case '>': esc = esc_gt;
|
||||
case '"':
|
||||
esc = esc_quot;
|
||||
case '\'':
|
||||
esc = esc_apos;
|
||||
case '&':
|
||||
esc = esc_amp;
|
||||
case '<':
|
||||
esc = esc_lt;
|
||||
case '>':
|
||||
esc = esc_gt;
|
||||
}
|
||||
p.write0(esc);
|
||||
|
||||
// update p.pos
|
||||
d := i+1 - i0;
|
||||
d := i+1-i0;
|
||||
p.pos.Offset += d;
|
||||
p.pos.Column += d;
|
||||
|
||||
@ -179,10 +184,10 @@ func (p *printer) write(data []byte) {
|
||||
}
|
||||
|
||||
// write remaining segment
|
||||
p.write0(data[i0 : len(data)]);
|
||||
p.write0(data[i0:len(data)]);
|
||||
|
||||
// update p.pos
|
||||
d := len(data) - i0;
|
||||
d := len(data)-i0;
|
||||
p.pos.Offset += d;
|
||||
p.pos.Column += d;
|
||||
}
|
||||
@ -193,7 +198,7 @@ func (p *printer) writeNewlines(n int) {
|
||||
if n > maxNewlines {
|
||||
n = maxNewlines;
|
||||
}
|
||||
p.write(newlines[0 : n]);
|
||||
p.write(newlines[0:n]);
|
||||
}
|
||||
}
|
||||
|
||||
@ -367,8 +372,8 @@ func split(text []byte) [][]byte {
|
||||
i := 0;
|
||||
for j, c := range text {
|
||||
if c == '\n' {
|
||||
lines[n] = text[i:j]; // exclude newline
|
||||
i = j+1; // discard newline
|
||||
lines[n] = text[i:j]; // exclude newline
|
||||
i = j+1; // discard newline
|
||||
n++;
|
||||
}
|
||||
}
|
||||
@ -393,15 +398,15 @@ func commonPrefix(a, b []byte) []byte {
|
||||
for i < len(a) && i < len(b) && a[i] == b[i] && (a[i] <= ' ' || a[i] == '*') {
|
||||
i++;
|
||||
}
|
||||
return a[0 : i];
|
||||
return a[0:i];
|
||||
}
|
||||
|
||||
|
||||
func stripCommonPrefix(lines [][]byte) {
|
||||
if len(lines) < 2 {
|
||||
return; // at most one line - nothing to do
|
||||
return; // at most one line - nothing to do
|
||||
}
|
||||
|
||||
|
||||
// The heuristic in this function tries to handle a few
|
||||
// common patterns of /*-style comments: Comments where
|
||||
// the opening /* and closing */ are aligned and the
|
||||
@ -434,7 +439,7 @@ func stripCommonPrefix(lines [][]byte) {
|
||||
if i := bytes.Index(prefix, []byte{'*'}); i >= 0 {
|
||||
// Line of stars present.
|
||||
if i > 0 && prefix[i-1] == ' ' {
|
||||
i--; // remove trailing blank from prefix so stars remain aligned
|
||||
i--; // remove trailing blank from prefix so stars remain aligned
|
||||
}
|
||||
prefix = prefix[0:i];
|
||||
lineOfStars = true;
|
||||
@ -447,7 +452,7 @@ func stripCommonPrefix(lines [][]byte) {
|
||||
// for the opening /*, assume up to 3 blanks or a tab. This
|
||||
// whitespace may be found as suffix in the common prefix.
|
||||
first := lines[0];
|
||||
if isBlank(first[2 : len(first)]) {
|
||||
if isBlank(first[2:len(first)]) {
|
||||
// no comment text on the first line:
|
||||
// reduce prefix by up to 3 blanks or a tab
|
||||
// if present - this keeps comment text indented
|
||||
@ -480,7 +485,7 @@ func stripCommonPrefix(lines [][]byte) {
|
||||
// Shorten the computed common prefix by the length of
|
||||
// suffix, if it is found as suffix of the prefix.
|
||||
if bytes.HasSuffix(prefix, suffix) {
|
||||
prefix = prefix[0 : len(prefix) - len(suffix)];
|
||||
prefix = prefix[0 : len(prefix)-len(suffix)];
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -508,7 +513,7 @@ func stripCommonPrefix(lines [][]byte) {
|
||||
// Remove the common prefix from all but the first and empty lines.
|
||||
for i, line := range lines {
|
||||
if i > 0 && len(line) != 0 {
|
||||
lines[i] = line[len(prefix) : len(line)];
|
||||
lines[i] = line[len(prefix):len(line)];
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -636,7 +641,7 @@ func (p *printer) writeWhitespace(n int) {
|
||||
// of lines before the label; effectively leading to wrong
|
||||
// indentation.
|
||||
p.buffer[i], p.buffer[i+1] = unindent, formfeed;
|
||||
i--; // do it again
|
||||
i--; // do it again
|
||||
continue;
|
||||
}
|
||||
fallthrough;
|
||||
@ -675,7 +680,7 @@ func (p *printer) print(args ...) {
|
||||
for i := 0; i < v.NumField(); i++ {
|
||||
f := v.Field(i);
|
||||
|
||||
next := p.pos; // estimated position of next item
|
||||
next := p.pos; // estimated position of next item
|
||||
var data []byte;
|
||||
var tag HtmlTag;
|
||||
isKeyword := false;
|
||||
@ -721,7 +726,7 @@ func (p *printer) print(args ...) {
|
||||
// (note that valid Go programs cannot contain esc ('\xff')
|
||||
// bytes since they do not appear in legal UTF-8 sequences)
|
||||
// TODO(gri): this this more efficiently.
|
||||
data = strings.Bytes("\xff" + string(data) + "\xff");
|
||||
data = strings.Bytes("\xff"+string(data)+"\xff");
|
||||
case token.Token:
|
||||
if p.Styler != nil {
|
||||
data, tag = p.Styler.Token(x);
|
||||
@ -731,7 +736,7 @@ func (p *printer) print(args ...) {
|
||||
isKeyword = x.IsKeyword();
|
||||
case token.Position:
|
||||
if x.IsValid() {
|
||||
next = x; // accurate position of next item
|
||||
next = x; // accurate position of next item
|
||||
}
|
||||
default:
|
||||
panicln("print: unsupported argument type", f.Type().String());
|
||||
@ -782,8 +787,8 @@ func (p *printer) flush(next token.Position, isKeyword bool) {
|
||||
// is used).
|
||||
//
|
||||
type trimmer struct {
|
||||
output io.Writer;
|
||||
buf bytes.Buffer;
|
||||
output io.Writer;
|
||||
buf bytes.Buffer;
|
||||
}
|
||||
|
||||
|
||||
@ -815,7 +820,7 @@ func (p *trimmer) Write(data []byte) (n int, err os.Error) {
|
||||
}
|
||||
|
||||
case '\v':
|
||||
b = '\t'; // convert to htab
|
||||
b = '\t'; // convert to htab
|
||||
fallthrough;
|
||||
|
||||
case '\t', ' ', tabwriter.Escape:
|
||||
@ -828,7 +833,7 @@ func (p *trimmer) Write(data []byte) (n int, err os.Error) {
|
||||
}
|
||||
// collect whitespace but discard tabrwiter.Escapes.
|
||||
if b != tabwriter.Escape {
|
||||
p.buf.WriteByte(b); // WriteByte returns no errors
|
||||
p.buf.WriteByte(b); // WriteByte returns no errors
|
||||
}
|
||||
|
||||
case '\f', '\n':
|
||||
@ -865,15 +870,15 @@ func (p *trimmer) Write(data []byte) (n int, err os.Error) {
|
||||
|
||||
// General printing is controlled with these Config.Mode flags.
|
||||
const (
|
||||
GenHTML uint = 1 << iota; // generate HTML
|
||||
RawFormat; // do not use a tabwriter; if set, UseSpaces is ignored
|
||||
UseSpaces; // use spaces instead of tabs for indentation and alignment
|
||||
GenHTML uint = 1<<iota; // generate HTML
|
||||
RawFormat; // do not use a tabwriter; if set, UseSpaces is ignored
|
||||
UseSpaces; // use spaces instead of tabs for indentation and alignment
|
||||
)
|
||||
|
||||
|
||||
// An HtmlTag specifies a start and end tag.
|
||||
type HtmlTag struct {
|
||||
Start, End string; // empty if tags are absent
|
||||
Start, End string; // empty if tags are absent
|
||||
}
|
||||
|
||||
|
||||
@ -882,18 +887,18 @@ type HtmlTag struct {
|
||||
//
|
||||
type Styler interface {
|
||||
LineTag(line int) ([]byte, HtmlTag);
|
||||
Comment(c *ast.Comment, line []byte) ([]byte, HtmlTag);
|
||||
BasicLit(x *ast.BasicLit) ([]byte, HtmlTag);
|
||||
Ident(id *ast.Ident) ([]byte, HtmlTag);
|
||||
Token(tok token.Token) ([]byte, HtmlTag);
|
||||
Comment(c *ast.Comment, line []byte) ([]byte, HtmlTag);
|
||||
BasicLit(x *ast.BasicLit) ([]byte, HtmlTag);
|
||||
Ident(id *ast.Ident) ([]byte, HtmlTag);
|
||||
Token(tok token.Token) ([]byte, HtmlTag);
|
||||
}
|
||||
|
||||
|
||||
// A Config node controls the output of Fprint.
|
||||
type Config struct {
|
||||
Mode uint; // default: 0
|
||||
Tabwidth int; // default: 8
|
||||
Styler Styler; // default: nil
|
||||
Mode uint; // default: 0
|
||||
Tabwidth int; // default: 8
|
||||
Styler Styler; // default: nil
|
||||
}
|
||||
|
||||
|
||||
@ -942,14 +947,14 @@ func (cfg *Config) Fprint(output io.Writer, node interface{}) (int, os.Error) {
|
||||
p.errors <- os.NewError(fmt.Sprintf("printer.Fprint: unsupported node type %T", n));
|
||||
runtime.Goexit();
|
||||
}
|
||||
p.flush(token.Position{Offset: 1<<30, Line: 1<<30}, false); // flush to "infinity"
|
||||
p.errors <- nil; // no errors
|
||||
p.flush(token.Position{Offset: 1<<30, Line: 1<<30}, false); // flush to "infinity"
|
||||
p.errors <- nil; // no errors
|
||||
}();
|
||||
err := <-p.errors; // wait for completion of goroutine
|
||||
err := <-p.errors; // wait for completion of goroutine
|
||||
|
||||
// flush tabwriter, if any
|
||||
if tw != nil {
|
||||
tw.Flush(); // ignore errors
|
||||
tw.Flush(); // ignore errors
|
||||
}
|
||||
|
||||
return p.written, err;
|
||||
@ -960,6 +965,6 @@ func (cfg *Config) Fprint(output io.Writer, node interface{}) (int, os.Error) {
|
||||
// It calls Config.Fprint with default settings.
|
||||
//
|
||||
func Fprint(output io.Writer, node interface{}) os.Error {
|
||||
_, err := (&Config{Tabwidth: 8}).Fprint(output, node); // don't care about number of bytes written
|
||||
_, err := (&Config{Tabwidth: 8}).Fprint(output, node); // don't care about number of bytes written
|
||||
return err;
|
||||
}
|
||||
|
@ -16,12 +16,12 @@ import (
|
||||
|
||||
|
||||
const (
|
||||
dataDir = "testdata";
|
||||
tabwidth = 8;
|
||||
dataDir = "testdata";
|
||||
tabwidth = 8;
|
||||
)
|
||||
|
||||
|
||||
var update = flag.Bool("update", false, "update golden files");
|
||||
var update = flag.Bool("update", false, "update golden files")
|
||||
|
||||
|
||||
func lineString(text []byte, i int) string {
|
||||
@ -29,13 +29,14 @@ func lineString(text []byte, i int) string {
|
||||
for i < len(text) && text[i] != '\n' {
|
||||
i++;
|
||||
}
|
||||
return string(text[i0 : i]);
|
||||
return string(text[i0:i]);
|
||||
}
|
||||
|
||||
|
||||
type checkMode uint;
|
||||
type checkMode uint
|
||||
|
||||
const (
|
||||
export checkMode = 1<<iota;
|
||||
export checkMode = 1<<iota;
|
||||
rawFormat;
|
||||
)
|
||||
|
||||
@ -50,13 +51,13 @@ func check(t *testing.T, source, golden string, mode checkMode) {
|
||||
|
||||
// filter exports if necessary
|
||||
if mode&export != 0 {
|
||||
ast.FileExports(prog); // ignore result
|
||||
prog.Comments = nil; // don't print comments that are not in AST
|
||||
ast.FileExports(prog); // ignore result
|
||||
prog.Comments = nil; // don't print comments that are not in AST
|
||||
}
|
||||
|
||||
// determine printer configuration
|
||||
cfg := Config{Tabwidth: tabwidth};
|
||||
if mode&rawFormat != 0 {
|
||||
if mode & rawFormat != 0 {
|
||||
cfg.Mode |= RawFormat;
|
||||
}
|
||||
|
||||
@ -105,20 +106,20 @@ func check(t *testing.T, source, golden string, mode checkMode) {
|
||||
|
||||
|
||||
type entry struct {
|
||||
source, golden string;
|
||||
mode checkMode;
|
||||
source, golden string;
|
||||
mode checkMode;
|
||||
}
|
||||
|
||||
// Use gotest -update to create/update the respective golden files.
|
||||
var data = []entry{
|
||||
entry{ "empty.input", "empty.golden", 0 },
|
||||
entry{ "comments.input", "comments.golden", 0 },
|
||||
entry{ "comments.input", "comments.x", export },
|
||||
entry{ "linebreaks.input", "linebreaks.golden", 0 },
|
||||
entry{ "expressions.input", "expressions.golden", 0 },
|
||||
entry{ "expressions.input", "expressions.raw", rawFormat },
|
||||
entry{ "declarations.input", "declarations.golden", 0 },
|
||||
entry{ "statements.input", "statements.golden", 0 },
|
||||
entry{"empty.input", "empty.golden", 0},
|
||||
entry{"comments.input", "comments.golden", 0},
|
||||
entry{"comments.input", "comments.x", export},
|
||||
entry{"linebreaks.input", "linebreaks.golden", 0},
|
||||
entry{"expressions.input", "expressions.golden", 0},
|
||||
entry{"expressions.input", "expressions.raw", rawFormat},
|
||||
entry{"declarations.input", "declarations.golden", 0},
|
||||
entry{"statements.input", "statements.golden", 0},
|
||||
}
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user