Preserve pedantic include in analysis_options.yaml

Dockerfile

@@ -1,4 +1,4 @@
-FROM google/dart:2.0.0 as build
+FROM google/dart:2.1.1 as build
 
 # Build Environment Vars
 ARG BUILD_ID

lib/src/update.dart

@@ -19,48 +19,65 @@ import 'package:yaml/yaml.dart';
 import 'package:abide/src/constants.dart';
 import 'package:abide/src/util.dart';
 
-Future<Null> updateAnalysisOption(YamlMap abideYaml,
-    {bool uncommentClean}) async {
-  final YamlMap currentAnalysisOptions =
-      loadAnalysisOptions(renameDeprecatedFilename: true);
-  final String currentAnalysisOptionsString = loadAnalysisOptionsAsString();
+Future<String> updateAnalysisOption(YamlMap abideYaml,
+    {String pathToAnalysisOptionsFile,
+    bool uncommentClean,
+    bool writeToFile = true}) async {
+  YamlMap currentAnalysisOptions = loadAnalysisOptions(
+      pathToAnalysisOptionsFile: pathToAnalysisOptionsFile,
+      renameDeprecatedFilename: true);
+
+  final String currentAnalysisOptionsString = loadAnalysisOptionsAsString(
+      pathToAnalysisOptionsFile: pathToAnalysisOptionsFile);
+
   Map<String, Map<String, int>> lintErrorCounts = <String, Map<String, int>>{};
   // Write a version with ALL lint rules enabled
   // and then run dart analyzer to get counts of lint per rule
   // those counts are passed into the 2nd write run to decide
   // whether or not to comment out a lint rule with a lot of
-  // new lint warnings
-  writeAnalyisOptionsFile(
-      all: true,
-      abideYaml: abideYaml,
-      currentAnalysisOptions: currentAnalysisOptions,
-      currentAnalysisOptionsString: currentAnalysisOptionsString);
-  lintErrorCounts = await getLintErrorCounts(abideYaml: abideYaml);
-  writeAnalyisOptionsFile(
+  // new lint warnings. If we're not writing to a file, then lint
+  // counts won't work, so skip that step.
+  if (writeToFile) {
+    writeAnalyisOptionsFile(
+        all: true,
+        abideYaml: abideYaml,
+        currentAnalysisOptions: currentAnalysisOptions,
+        currentAnalysisOptionsString: currentAnalysisOptionsString);
+    lintErrorCounts = await getLintErrorCounts(abideYaml: abideYaml);
+  }
+  return writeAnalyisOptionsFile(
       abideYaml: abideYaml,
       currentAnalysisOptions: currentAnalysisOptions,
       currentAnalysisOptionsString: currentAnalysisOptionsString,
       lintErrorCounts: lintErrorCounts,
-      uncommentClean: uncommentClean);
+      uncommentClean: uncommentClean,
+      writeToFile: writeToFile);
 }
 
-void writeAnalyisOptionsFile(
+String writeAnalyisOptionsFile(
     {YamlMap abideYaml,
     YamlMap currentAnalysisOptions,
     String currentAnalysisOptionsString,
     bool all = false,
     bool uncommentClean = false,
-    Map<String, Map<String, int>> lintErrorCounts:
+    bool writeToFile = true,
+    Map<String, Map<String, int>> lintErrorCounts =
         const <String, Map<String, int>>{}}) {
   currentAnalysisOptions ??= new YamlMap();
 
+  final String currentInclude = getYamlValue(currentAnalysisOptions, 'include');
   final bool currentImplicitCasts = getYamlValue(
       currentAnalysisOptions, 'analyzer:strong-mode:implicit-casts', true);
   final bool currentImplicitDynamic = getYamlValue(
       currentAnalysisOptions, 'analyzer:strong-mode:implicit-dynamic', true);
 
   final StringBuffer sb = new StringBuffer('''
 # analysis_options.yaml docs: https://www.dartlang.org/guides/language/analysis-options 
+''');
+  if (currentInclude != null && currentInclude.isNotEmpty) {
+    sb.writeln('include: $currentInclude');
+  }
+  sb.write('''
 analyzer:
   # Strong mode is required. Applies to the current project.
   strong-mode:
@@ -198,15 +215,18 @@ linter:
 $output
 ''');
   }
-
-  new File(analysisOptionsFilename).writeAsStringSync(sb.toString());
+  String finalOutput = sb.toString();
+  if (writeToFile) {
+    new File(analysisOptionsFilename).writeAsStringSync(finalOutput);
+  }
   if (!all) {
     print('Wrote $analysisOptionsFilename');
   }
   if (nMissingRecommendations > 0) {
     print(
         'There were missing recommendations. Please inform the maintainers of the abide tool to perform an abide upgrade.');
   }
+  return finalOutput;
 }
 
 String _lintResultFor(String lint, Map<String, Map<String, int>> lintErrors) {

lib/src/util.dart

@@ -62,12 +62,19 @@ YamlMap loadPubspec() => loadYamlFile(pubspecFilename);
 YamlMap loadSmithy() =>
     loadYamlFile(smithyFilename) ?? loadYamlFile(smithyFilename2);
 
-YamlMap loadAnalysisOptions({bool renameDeprecatedFilename: false}) =>
-    loadYamlFile(findAnalysisOptionsFile(
+YamlMap loadAnalysisOptions(
+    {String pathToAnalysisOptionsFile, bool renameDeprecatedFilename: false}) {
+  if (pathToAnalysisOptionsFile != null) {
+    return loadYamlFile(pathToAnalysisOptionsFile);
+  } else {
+    return loadYamlFile(findAnalysisOptionsFile(
         renameDeprecatedFilename: renameDeprecatedFilename));
+  }
+}
 
-String loadAnalysisOptionsAsString() {
-  final String filename = findAnalysisOptionsFile();
+String loadAnalysisOptionsAsString({String pathToAnalysisOptionsFile}) {
+  final String filename =
+      pathToAnalysisOptionsFile ?? findAnalysisOptionsFile();
   if (filename == null) {
     return '';
   }

test/fixtures/analysis/includes_pedantic/analysis_options.yaml

@@ -0,0 +1 @@
+include: package:pedantic/analysis_options.yaml

test/vm/check_test.dart

@@ -43,7 +43,7 @@ Future<Null> main() async {
     }
   });
 
-  group('Verifiy pubspec check', () {
+  group('Verify pubspec check', () {
     final YamlMap pubspec = loadPubspec();
     final YamlMap pubspecWithMissingDeps =
         loadYamlFile('test/fixtures/pubspec_missing_deps.yaml');

test/vm/update_test.dart

@@ -0,0 +1,38 @@
+// Copyright 2017-2019 Workiva Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+@TestOn('vm')
+import 'dart:async';
+import 'package:test/test.dart';
+import 'package:yaml/yaml.dart';
+
+import 'package:abide/src/util.dart';
+import 'package:abide/src/update.dart';
+
+Future<Null> main() async {
+  YamlMap abideYaml = await loadAbideYaml();
+
+  group('Update', () {
+    test('preserves an existing include key', () async {
+      String result = await updateAnalysisOption(abideYaml,
+          pathToAnalysisOptionsFile:
+              'test/fixtures/analysis/includes_pedantic/analysis_options.yaml',
+          uncommentClean: false,
+          writeToFile: false);
+      YamlMap yamlOutput = loadYaml(result);
+      expect(yamlOutput['include'] == 'package:pedantic/analysis_options.yaml',
+          isTrue);
+    });
+  });
+}