Vaani/lib/shared/extensions/obfuscation.dart

import 'package:flutter/foundation.dart';
import 'package:http/http.dart' as http;
import 'package:shelfsdk/audiobookshelf_api.dart' as shelfsdk;
import 'package:vaani/settings/models/api_settings.dart';
import 'package:vaani/settings/models/audiobookshelf_server.dart';
import 'package:vaani/settings/models/authenticated_user.dart';

// bool kReleaseMode = true;

extension ObfuscateString on String {
  String obfuscate() {
    if (!kReleaseMode) {
      return this;
    }
    return 'obfuscated';
  }
}

extension ObfuscateURI on Uri {
  /// keeps everything except the base url for security reasons
  Uri obfuscate() {
    if (!kReleaseMode) {
      return this;
    }

    // do not obfuscate the local host
    if ([null, 'localhost'].contains(host)) {
      return this;
    }

    // do not obfuscate file urls
    if (scheme == 'file') {
      return this;
    }

    return replace(
      userInfo: userInfo == '' ? '' : 'userInfoObfuscated',
      host: 'hostObfuscated',
    );
  }
}

extension ObfuscateList<T> on List<T> {
  List<T> obfuscate() {
    return map((e) {
      if (e is AuthenticatedUser) {
        return e.obfuscate() as T;
      } else if (e is AudiobookShelfServer) {
        return e.obfuscate() as T;
      } else if (e is Uri) {
        return e.obfuscate() as T;
      } else {
        return e;
      }
    }).toList();
  }
}

extension ObfuscateSet<T> on Set<T> {
  Set<T> obfuscate() {
    return toList().obfuscate().toSet();
  }
}

extension ObfuscateAuthenticatedUser on AuthenticatedUser {
  AuthenticatedUser obfuscate() {
    if (!kReleaseMode) {
      return this;
    }
    return copyWith(
      username: username == null ? null : 'usernameObfuscated',
      authToken: 'authTokenObfuscated',
      server: server.obfuscate(),
    );
  }
}

extension ObfuscateServer on AudiobookShelfServer {
  AudiobookShelfServer obfuscate() {
    if (!kReleaseMode) {
      return this;
    }
    return copyWith(
      serverUrl: serverUrl.obfuscate(),
    );
  }
}

extension ObfuscateApiSettings on ApiSettings {
  ApiSettings obfuscate() {
    if (!kReleaseMode) {
      return this;
    }
    return copyWith(
      activeServer: activeServer?.obfuscate(),
      activeUser: activeUser?.obfuscate(),
    );
  }
}

extension ObfuscateRequest on http.BaseRequest {
  http.BaseRequest obfuscate() {
    if (!kReleaseMode) {
      return this;
    }
    return http.Request(
      method,
      url.obfuscate(),
    );
  }
}

extension ObfuscateResponse on http.Response {
  http.Response obfuscate() {
    if (!kReleaseMode) {
      return this;
    }
    return http.Response(
      obfuscateBody(),
      statusCode,
      headers: headers,
      request: request?.obfuscate(),
    );
  }

  String obfuscateBody() {
    if (!kReleaseMode) {
      return body;
    }
    // replace any email addresses with emailObfuscated
    // replace any phone numbers with phoneObfuscated
    // replace any urls with urlObfuscated
    // replace any tokens with tokenObfuscated
    // token regex is `"token": "..."`
    return body
        .replaceAll(
          RegExp(r'(\b\w+@\w+\.\w+\b)|'
              r'(\b\d{3}-\d{3}-\d{4}\b)|'
              r'(\bhttps?://\S+\b)'),
          'obfuscated',
        )
        .replaceAll(
          RegExp(r'"?token"?:?\s*"[^"]+"'),
          '"token": "tokenObfuscated"',
        );
  }
}

extension ObfuscateLoginResponse on shelfsdk.LoginResponse {
  shelfsdk.LoginResponse obfuscate() {
    if (!kReleaseMode) {
      return this;
    }
    return copyWith(
      user: user.obfuscate(),
    );
  }
}

extension ObfuscateUser on shelfsdk.User {
  shelfsdk.User obfuscate() {
    if (!kReleaseMode) {
      return this;
    }
    return shelfsdk.User.fromJson(
      toJson()..['token'] = 'tokenObfuscated',
    );
  }
}