mudblock/lib/core/features/profile.dart

import 'package:encrypt/encrypt.dart' as enc;
import 'package:flutter/foundation.dart';

import '../secrets.dart';
import '../storage.dart';
import '../string_utils.dart';
import 'keyboard_shortcuts.dart';
import 'plugin.dart';
import 'settings.dart';

class MUDProfile extends PluginBase {
  final String id;

  String name;
  String host;
  int port;
  bool mccpEnabled;
  String username;
  String password;
  AuthMethod authMethod;

  Settings settings = Settings.empty();
  KeyboardShortcutMap keyboardShortcuts = KeyboardShortcutMap.empty();

  @override
  IStorage<Map<String, dynamic>> get storage => _storage;

  final IStorage<Map<String, dynamic>> _storage;

  MUDProfile({
    required this.id,
    required this.name,
    required this.host,
    required this.port,
    this.mccpEnabled = true,
    this.username = '',
    this.password = '',
    this.authMethod = AuthMethod.none,
  }) : _storage = ProfileStorage(id);

  factory MUDProfile.empty() => MUDProfile(
        id: uuid(),
        name: '',
        host: '',
        port: 23,
      );

  MUDProfile copyWith({
    String? id,
    String? name,
    String? host,
    int? port,
    bool? mccpEnabled,
    String? username,
    String? password,
    AuthMethod? authMethod,
  }) =>
      MUDProfile(
        id: id ?? this.id,
        name: name ?? this.name,
        host: host ?? this.host,
        port: port ?? this.port,
        mccpEnabled: mccpEnabled ?? this.mccpEnabled,
        username: username ?? this.username,
        password: password ?? this.password,
        authMethod: authMethod ?? this.authMethod,
      );

  factory MUDProfile.fromJson(Map<String, dynamic> json) {
    return MUDProfile(
      id: json['id'],
      name: json['name'],
      host: json['host'],
      port: json['port'],
      mccpEnabled: json['mccpEnabled'],
      username: json['username'],
      password: decrypt(json['password']),
      // TODO generalize getting enum from string
      authMethod: AuthMethod.values.firstWhere(
        (e) => e.name == json['authMethod'],
        orElse: () => AuthMethod.none,
      ),
    );
  }

  Map<String, dynamic> toJson() => {
        'id': id,
        'name': name,
        'host': host,
        'port': port,
        'mccpEnabled': mccpEnabled,
        'username': username,
        'password': encrypt(password),
        'authMethod': authMethod.name,
      };

  @override
  List<Future<void>> additionalLoaders() => [
        getKeyboardShortcuts(),
        getSettings(),
      ];

  Future<KeyboardShortcutMap> loadKeyboardShortcuts() async {
    debugPrint('MUDProfile.loadKeyboardShortcuts: $id');
    final shortcuts = await storage.readFile('keyboard_shortcuts');
    debugPrint('MUDProfile.loadKeyboardShortcuts: $shortcuts');
    if (shortcuts == null) {
      return KeyboardShortcutMap.empty();
    }
    return KeyboardShortcutMap.fromJson(shortcuts);
  }

  Future<void> saveKeyboardShortcuts(KeyboardShortcutMap shortcuts) async {
    debugPrint('MUDProfile.saveKeyboardShortcuts: $id');
    keyboardShortcuts = shortcuts;
    notifyListeners();
    return storage.writeFile('keyboard_shortcuts', shortcuts.toJson());
  }

  Future<void> getKeyboardShortcuts() async {
    final shortcuts = await loadKeyboardShortcuts();
    keyboardShortcuts = shortcuts;
    notifyListeners();
    debugPrint('KeyboardShortcuts loaded');
  }

  Future<Settings> loadSettings() async {
    debugPrint('MUDProfile.loadSettings');
    final settings = await storage.readFile('settings');
    debugPrint('MUDProfile.loadSettings: $settings');
    if (settings == null) {
      return Settings.empty();
    }
    return Settings.fromJson(settings);
  }

  Future<void> getSettings() async {
    final settings = await loadSettings();
    this.settings = settings;
    notifyListeners();
    debugPrint('Settings loaded: ${settings.showTimestamps}');
  }

  Future<void> saveSettings(Settings settings) async {
    debugPrint('MUDProfile.saveSettings');
    this.settings = settings;
    notifyListeners();
    return storage.writeFile('settings', settings.toJson());
  }

  Future<void> save() async {
    debugPrint('MUDProfile.save: $id');
    if (!storage.initialized) {
      await storage.init();
    }
    return storage.writeFile(id, toJson());
  }

  Future<void> delete() async {
    debugPrint('MUDProfile.delete: $id');
    if (!storage.initialized) {
      await storage.init();
    }
    return storage.deleteDirectory('.');
  }

  static final encKey = enc.Key.fromUtf8(pwdKey);
  static final encrypter = enc.Encrypter(enc.AES(encKey, padding: null));

  static String encrypt(String password) {
    final iv = enc.IV.fromLength(16);
    if (password.isEmpty) {
      return '';
    }
    final encrypted = encrypter.encrypt(password, iv: iv);
    return '${encrypted.base64}:${iv.base64}';
  }

  static String decrypt(String password) {
    if (password.isEmpty) {
      return '';
    }
    try {
      final ivStr = password.substring(password.indexOf(':') + 1);
      final iv = enc.IV.fromBase64(ivStr);
      password = password.substring(0, password.indexOf(':'));
      final encrypted = enc.Encrypted.fromBase64(password);
      final decrypted = encrypter.decrypt(encrypted, iv: iv);
      return decrypted;
    } catch (e, stack) {
      debugPrint('MUDProfile.decrypt: $e\n$stack');
      return password;
    }
  }

}

enum AuthMethod {
  none,
  diku,
}