AppFlowy/frontend/appflowy_flutter/lib/startup/tasks/rust_sdk.dart

import 'dart:convert';
import 'dart:io';

import 'package:appflowy/env/backend_env.dart';
import 'package:appflowy/env/env.dart';
import 'package:appflowy_backend/appflowy_backend.dart';
import 'package:path_provider/path_provider.dart';
import 'package:path/path.dart' as path;

import '../startup.dart';

class InitRustSDKTask extends LaunchTask {
  const InitRustSDKTask({
    this.directory,
  });

  // Customize the RustSDK initialization path
  final Directory? directory;

  @override
  LaunchTaskType get type => LaunchTaskType.dataProcessing;

  @override
  Future<void> initialize(LaunchContext context) async {
    final dir = directory ?? await appFlowyApplicationDataDirectory();

    // Pass the environment variables to the Rust SDK
    final env = getAppFlowyEnv();
    await context.getIt<FlowySDK>().init(dir, jsonEncode(env.toJson()));
  }

  @override
  Future<void> dispose() async {}
}

AppFlowyEnv getAppFlowyEnv() {
  if (isCloudEnabled) {
    final supabaseConfig = SupabaseConfiguration(
      url: Env.supabaseUrl,
      anon_key: Env.supabaseAnonKey,
    );

    final appflowyCloudConfig = AppFlowyCloudConfiguration(
      base_url: Env.afCloudBaseUrl,
      ws_base_url: Env.afCloudWSBaseUrl,
      gotrue_url: Env.afCloudGoTrueUrl,
    );

    return AppFlowyEnv(
      cloud_type: Env.cloudType,
      supabase_config: supabaseConfig,
      appflowy_cloud_config: appflowyCloudConfig,
    );
  } else {
    // Use the default configuration if the cloud feature is disabled
    final supabaseConfig = SupabaseConfiguration(url: '', anon_key: '');
    final appflowyCloudConfig = AppFlowyCloudConfiguration(
      base_url: '',
      ws_base_url: '',
      gotrue_url: '',
    );

    return AppFlowyEnv(
      cloud_type: 0,
      supabase_config: supabaseConfig,
      appflowy_cloud_config: appflowyCloudConfig,
    );
  }
}

/// The default directory to store the user data. The directory can be
/// customized by the user via the [ApplicationDataStorage]
Future<Directory> appFlowyApplicationDataDirectory() async {
  switch (integrationMode()) {
    case IntegrationMode.develop:
      final Directory documentsDir = await getApplicationSupportDirectory()
        ..create();
      return Directory(path.join(documentsDir.path, 'data_dev')).create();
    case IntegrationMode.release:
      final Directory documentsDir = await getApplicationSupportDirectory();
      return Directory(path.join(documentsDir.path, 'data')).create();
    case IntegrationMode.unitTest:
    case IntegrationMode.integrationTest:
      return Directory(path.join(Directory.current.path, '.sandbox'));
  }
}