Run commands zoned

LinwoodDev · Sep 30, 2024 · 4ac7cb3 · 4ac7cb3
1 parent f8f793c
commit 4ac7cb3
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 4 deletions.
diff --git a/packages/consoler/example/bin/example.dart b/packages/consoler/example/bin/example.dart
@@ -7,6 +7,7 @@ Future<void> main(List<String> arguments) async {
         DefaultProgramConfiguration(description: 'Consoler example'),
   );
   consoler.registerProgram('nothing', NothingProgram());
+  consoler.registerProgram('warn', WarningProgram(consoler));
   consoler.run();
   for (var i = 10; i >= 0; i--) {
     consoler.print('Countdown: $i');

diff --git a/packages/consoler/example/lib/example.dart b/packages/consoler/example/lib/example.dart
@@ -7,3 +7,16 @@ class NothingProgram extends ConsoleProgram {
   @override
   String? getDescription() => null;
 }
+
+class WarningProgram extends ConsoleProgram {
+  final Consoler consoler;
+
+  WarningProgram(this.consoler);
+  @override
+  void run(String label, List<String> args) {
+    consoler.print('Warning: This is a warning!', level: LogLevel.warning);
+  }
+
+  @override
+  String? getDescription() => null;
+}
diff --git a/packages/consoler/lib/src/consoler.dart b/packages/consoler/lib/src/consoler.dart
@@ -82,6 +82,11 @@ final class Consoler<T extends ConsoleProgram> {
     stdout.write('\r\n$prefix');
   }
 
+  T runPrintZone<T>(T Function() action) => runZoned(action,
+          zoneSpecification: ZoneSpecification(print: (_, __, ___, message) {
+        print(message);
+      }));
+
   void print(Object? message, {LogLevel? level}) {
     if (level != null && level.index < minLogLevel.index) return;
     stdout.write('\r');
@@ -99,11 +104,11 @@ final class Consoler<T extends ConsoleProgram> {
     return _programs.remove(name) != null;
   }
 
-  void _onInput(String input) {
+  Future<void> _onInput(String input) async {
     final splitted = _splitBySpaces(input);
     final command = splitted.firstOrNull;
-    (_programs[command] ?? _programs[null])
-        ?.run(command ?? '', splitted.isEmpty ? const [] : splitted.sublist(1));
-    sendPrefix();
+
+    runPrintZone(() => (_programs[command] ?? _programs[null])?.run(
+        command ?? '', splitted.isEmpty ? const [] : splitted.sublist(1)));
   }
 }