@@ -48,9 +48,7 @@ class _FunctionsFrameworkBuilder implements Builder {
4848 Future <void > build (BuildStep buildStep) async {
4949 final entries = < String , FactoryData > {};
5050
51- final input = buildStep.inputId;
52-
53- final libraryElement = await buildStep.resolver.libraryFor (input);
51+ final libraryElement = await buildStep.inputLibrary;
5452 final validator = await FunctionTypeValidator .create (buildStep.resolver);
5553
5654 for (var annotatedElement in _fromLibrary (libraryElement)) {
@@ -89,7 +87,7 @@ class _FunctionsFrameworkBuilder implements Builder {
8987
9088 final importDirectives = [
9189 "'package:functions_framework/serve.dart'" ,
92- "'${input .uri }' as $functionsLibraryPrefix " ,
90+ "'${buildStep . inputId .uri }' as $functionsLibraryPrefix " ,
9391 ]..sort ();
9492
9593 var output = '''
@@ -122,7 +120,9 @@ ${cases.join('\n')}
122120''' ;
123121
124122 try {
125- output = DartFormatter ().format (output);
123+ output = DartFormatter (
124+ languageVersion: libraryElement.languageVersion.effective,
125+ ).format (output);
126126 } on FormatterException catch (e, stack) {
127127 log.warning ('Could not format output.' , e, stack);
128128 }
0 commit comments