import 'package:matrix/matrix.dart';

import 'package:fluffychat/pangea/common/constants/model_keys.dart';
import 'package:fluffychat/pangea/common/utils/error_handler.dart';
import 'package:fluffychat/pangea/join_codes/space_code_extension.dart';

extension JoinRuleExtension on Client {
  Future<StateEvent> pangeaJoinRules(
    String joinRule, {
    List<Map<String, dynamic>>? allow,
  }) async {
    String? joinCode;
    try {
      joinCode = await requestSpaceCode();
    } catch (e, s) {
      ErrorHandler.logError(
        e: e,
        s: s,
        data: {
          'joinRule': joinRule,
        },
      );
    }

    return StateEvent(
      type: EventTypes.RoomJoinRules,
      content: {
        ModelKey.joinRule: joinRule,
        if (joinCode != null) ModelKey.accessCode: joinCode,
        if (allow != null) 'allow': allow,
      },
    );
  }
}

extension JoinRuleExtensionOnRoom on Room {
  Future<void> addJoinCode() async {
    if (!canChangeStateEvent(EventTypes.RoomJoinRules)) {
      throw Exception('Cannot change join rules for this room');
    }

    final currentJoinRules = getState(EventTypes.RoomJoinRules)?.content ?? {};
    if (currentJoinRules[ModelKey.accessCode] != null) return;

    final joinCode = await client.requestSpaceCode();
    currentJoinRules[ModelKey.accessCode] = joinCode;

    await client.setRoomStateWithKey(
      id,
      EventTypes.RoomJoinRules,
      '',
      currentJoinRules,
    );
  }

  /// Keep the room's current join rule state event content (except for what's intentionally replaced)
  /// since space's access codes were stored there. Don't want to accidentally remove them.
  Future<void> pangeaSetJoinRules(
    String joinRule, {
    List<Map<String, dynamic>>? allow,
  }) async {
    final currentJoinRule = getState(
          EventTypes.RoomJoinRules,
        )?.content ??
        {};

    if (currentJoinRule[ModelKey.joinRule] == joinRule &&
        (currentJoinRule['allow'] == allow)) {
      return; // No change needed
    }

    currentJoinRule[ModelKey.joinRule] = joinRule;
    currentJoinRule['allow'] = allow;

    await client.setRoomStateWithKey(
      id,
      EventTypes.RoomJoinRules,
      '',
      currentJoinRule,
    );
  }
}