fix: check state is null

slang-i18n · Oct 30, 2020 · 611d59e · 611d59e
1 parent 0bdfb50
commit 611d59e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/src/generator.dart b/lib/src/generator.dart
@@ -117,7 +117,7 @@ void _generateHeader(StringBuffer buffer, List<I18nData> allLocales) {
   buffer.writeln('\tstatic Future<void> useDeviceLocale() async {');
   buffer.writeln(
       '\t\t$localeVar = await FastI18n.findDeviceLocale($mapVar.keys.toList(), $baseLocaleVar);');
-  buffer.writeln('\n\t\tif ($translationProviderKey != null)');
+  buffer.writeln('\n\t\tif ($translationProviderKey.currentState != null)');
   buffer.writeln(
       '\t\t\t$translationProviderKey.currentState.setLocale($localeVar);');
   buffer.writeln('\t}');
@@ -126,7 +126,7 @@ void _generateHeader(StringBuffer buffer, List<I18nData> allLocales) {
   buffer.writeln('\tstatic void setLocale(String locale) {');
   buffer.writeln(
       '\t\t$localeVar = FastI18n.selectLocale(locale, $mapVar.keys.toList(), $baseLocaleVar);');
-  buffer.writeln('\n\t\tif ($translationProviderKey != null)');
+  buffer.writeln('\n\t\tif ($translationProviderKey.currentState != null)');
   buffer.writeln(
       '\t\t\t$translationProviderKey.currentState.setLocale($localeVar);');
   buffer.writeln('\t}');