blob: 428aeecfb645dd0f72836272e9102ef661b4e610 [file] [log] [blame]
// Copyright (c) 2014, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.
library dart_style.src.dart_formatter;
import 'dart:math' as math;
import 'package:analyzer/analyzer.dart';
import 'package:analyzer/src/generated/parser.dart';
import 'package:analyzer/src/generated/scanner.dart';
import 'package:analyzer/src/generated/source.dart';
import 'package:analyzer/src/string_source.dart';
import 'error_listener.dart';
import 'formatter_exception.dart';
import 'source_code.dart';
import 'source_visitor.dart';
/// Dart source code formatter.
class DartFormatter {
/// The string that newlines should use.
/// If not explicitly provided, this is inferred from the source text. If the
/// first newline is `\r\n` (Windows), it will use that. Otherwise, it uses
/// Unix-style line endings (`\n`).
String lineEnding;
/// The number of characters allowed in a single line.
final int pageWidth;
/// The number of characters of indentation to prefix the output lines with.
final int indent;
/// Creates a new formatter for Dart code.
/// If [lineEnding] is given, that will be used for any newlines in the
/// output. Otherwise, the line separator will be inferred from the line
/// endings in the source file.
/// If [indent] is given, that many levels of indentation will be prefixed
/// before each resulting line in the output.
DartFormatter({this.lineEnding, int pageWidth, this.indent: 0})
: this.pageWidth = (pageWidth == null) ? 80 : pageWidth;
/// Formats the given [source] string containing an entire Dart compilation
/// unit.
/// If [uri] is given, it is a [String] or [Uri] used to identify the file
/// being formatted in error messages.
String format(String source, {uri}) {
if (uri == null) {
uri = "<unknown>";
} else if (uri is Uri) {
uri = uri.toString();
} else if (uri is String) {
// Do nothing.
} else {
throw new ArgumentError("uri must be `null`, a Uri, or a String.");
return formatSource(
new SourceCode(source, uri: uri, isCompilationUnit: true))
/// Formats the given [source] string containing a single Dart statement.
String formatStatement(String source) {
return formatSource(new SourceCode(source, isCompilationUnit: false)).text;
/// Formats the given [source].
/// Returns a new [SourceCode] containing the formatted code and the resulting
/// selection, if any.
SourceCode formatSource(SourceCode source) {
var errorListener = new ErrorListener();
// Tokenize the source.
var reader = new CharSequenceReader(source.text);
var stringSource = new StringSource(source.text, source.uri);
var scanner = new Scanner(stringSource, reader, errorListener);
var startToken = scanner.tokenize();
var lineInfo = new LineInfo(scanner.lineStarts);
// Infer the line ending if not given one. Do it here since now we know
// where the lines start.
if (lineEnding == null) {
// If the first newline is "\r\n", use that. Otherwise, use "\n".
if (scanner.lineStarts.length > 1 &&
scanner.lineStarts[1] >= 2 &&
source.text[scanner.lineStarts[1] - 2] == '\r') {
lineEnding = "\r\n";
} else {
lineEnding = "\n";
// Parse it.
var parser = new Parser(stringSource, errorListener);
var node;
if (source.isCompilationUnit) {
node = parser.parseCompilationUnit(startToken);
} else {
node = parser.parseStatement(startToken);
// Make sure we consumed all of the source.
var token =;
if (token.type != TokenType.EOF) {
var error = new AnalysisError(
math.max(token.length, 1),
throw new FormatterException([error]);
// Format it.
var visitor = new SourceVisitor(this, lineInfo, source);