fix(gebura): wrap TraceMode

tuihub · Dec 29, 2023 · 8f8fa03 · 8f8fa03
1 parent 1570485
commit 8f8fa03
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/lib/bloc/gebura/gebura_bloc.dart b/lib/bloc/gebura/gebura_bloc.dart
@@ -8,7 +8,7 @@ import 'package:tuihub_protos/librarian/v1/common.pb.dart';
 
 import '../../common/bloc_event_status_mixin.dart';
 import '../../ffi/ffi.dart';
-import '../../ffi/rust_ffi/rust_ffi.dart';
+import '../../ffi/ffi_model.dart';
 import '../../model/gebura_model.dart';
 import '../../repo/grpc/api_helper.dart';
 import '../../repo/local/gebura.dart';

diff --git a/lib/ffi/ffi_model.dart b/lib/ffi/ffi_model.dart
@@ -1,5 +1,3 @@
-import 'rust_ffi/rust_ffi.dart';
-
 abstract class FFIBase {
   Future<(int, int, bool)> processRunner(
     TraceMode mode,
@@ -13,3 +11,8 @@ abstract class FFIBase {
     throw UnimplementedError();
   }
 }
+
+enum TraceMode {
+  Simple,
+  ByName,
+}
diff --git a/lib/ffi/native_ffi.dart b/lib/ffi/native_ffi.dart
@@ -2,7 +2,7 @@ import 'dart:ffi';
 import 'dart:io';
 
 import 'ffi_model.dart';
-import 'rust_ffi/rust_ffi.dart';
+import 'rust_ffi/rust_ffi.dart' as rust_ffi;
 
 class NativeFFI {
   NativeFFI._();
@@ -30,7 +30,7 @@ class NativeFFI {
 }
 
 class FFI extends FFIBase {
-  static final _ffi = RustFfiImpl(NativeFFI.dyLib);
+  static final _ffi = rust_ffi.RustFfiImpl(NativeFFI.dyLib);
 
   @override
   Future<(int, int, bool)> processRunner(
@@ -49,7 +49,9 @@ class FFI extends FFIBase {
       workingDir: workingDir,
       sleepCount: sleepCount,
       sleepMillis: sleepMillis,
-      mode: mode,
+      mode: mode == TraceMode.Simple
+          ? rust_ffi.TraceMode.Simple
+          : rust_ffi.TraceMode.ByName,
     );
   }
 }