cake_wallet/lib/utils/proxy_wrapper.dart

import 'dart:io';

import 'package:cake_wallet/store/settings_store.dart';
import 'package:cake_wallet/view_model/settings/tor_connection.dart';
import 'package:socks5_proxy/socks.dart';
import 'package:tor/tor.dart';

class ProxyWrapper {
  ProxyWrapper({
    required this.settingsStore,
  });

  late SettingsStore settingsStore;

  HttpClient? _torClient;

  // Factory method to get the singleton instance of TorSingleton
  // static ProxyWrapper get instance => _instance;

  static int get port => Tor.instance.port;

  static bool get enabled => Tor.instance.enabled;

  bool started = false;
  // bool torEnabled = false;
  // bool torOnly = false;

  // Method to get or create the Tor proxy instance
  Future<HttpClient> getProxyHttpClient({int? portOverride}) async {
    if (!started) {
      started = true;
      _torClient = HttpClient();

      // Assign connection factory.
      SocksTCPClient.assignToHttpClient(_torClient!, [
        ProxySettings(
          InternetAddress.loopbackIPv4,
          portOverride ?? Tor.instance.port,
          password: null,
        ),
      ]);
    }

    return _torClient!;
  }

  Future<HttpClientResponse> get(
    Uri uri, {
    Map<String, String>? headers,
    int? portOverride,
    bool torOnly = false,
    Uri? clearnetUri,
  }) async {
    HttpClient? client;
    late bool torEnabled;
    if (settingsStore.torConnectionMode == TorConnectionMode.onionOnly ||
        settingsStore.torConnectionMode == TorConnectionMode.enabled) {
      client = await getProxyHttpClient(portOverride: portOverride);
      torEnabled = true;
    } else {
      client = HttpClient();
      torEnabled = false;
    }

    if (settingsStore.torConnectionMode == TorConnectionMode.onionOnly) {
      if (!uri.path.contains(".onion")) {
        throw Exception("Cannot connect to clearnet");
      }
    }

    HttpClientResponse? response;

    try {
      final request = await client.getUrl(uri);
      if (headers != null) {
        headers.forEach((key, value) {
          request.headers.add(key, value);
        });
      }
      response = await request.close();
    } catch (e) {
      if (!torOnly &&
          torEnabled &&
          settingsStore.torConnectionMode != TorConnectionMode.onionOnly) {
        // try again without tor:
        client = HttpClient();
        final request = await client.getUrl(clearnetUri ?? uri);
        if (headers != null) {
          headers.forEach((key, value) {
            request.headers.add(key, value);
          });
        }
        response = await request.close();
      } else {
        throw e;
      }
    }

    return response;
  }

  Future<HttpClientResponse> post(
    Uri uri, {
    Map<String, String>? headers,
    int? portOverride,
    bool torOnly = false,
    Uri? clearnetUri,
  }) async {
    HttpClient? client;
    late bool torEnabled;
    if (settingsStore.torConnectionMode == TorConnectionMode.onionOnly ||
        settingsStore.torConnectionMode == TorConnectionMode.enabled) {
      client = await getProxyHttpClient(portOverride: portOverride);
      torEnabled = true;
    } else {
      client = HttpClient();
      torEnabled = false;
    }

    if (settingsStore.torConnectionMode == TorConnectionMode.onionOnly) {
      if (!uri.path.contains(".onion")) {
        throw Exception("Cannot connect to clearnet");
      }
    }

    HttpClientResponse? response;

    try {
      final request = await client.postUrl(uri);
      if (headers != null) {
        headers.forEach((key, value) {
          request.headers.add(key, value);
        });
      }
      response = await request.close();
    } catch (e) {
      if (!torOnly &&
          torEnabled &&
          settingsStore.torConnectionMode != TorConnectionMode.onionOnly) {
        // try again without tor:
        client = HttpClient();
        final request = await client.postUrl(clearnetUri ?? uri);
        if (headers != null) {
          headers.forEach((key, value) {
            request.headers.add(key, value);
          });
        }
        response = await request.close();
      } else {
        throw e;
      }
    }

    return response;
  }
}