fluffychat/lib/pangea/models/bot_options_model.dart

import 'dart:developer';

import 'package:fluffychat/pangea/constants/model_keys.dart';
import 'package:fluffychat/pangea/utils/error_handler.dart';
import 'package:flutter/foundation.dart';
import 'package:matrix/matrix.dart';

import '../constants/pangea_event_types.dart';

class BotOptionsModel {
  int? languageLevel;
  String topic;
  List<String> keywords;
  bool safetyModeration;
  String mode;
  String? custom;
  String? discussionTopic;
  String? discussionKeywords;
  bool discussionTriggerReactionEnabled;
  String discussionTriggerReactionKey;
  String? customSystemPrompt;
  bool customTriggerReactionEnabled;
  String customTriggerReactionKey;

  BotOptionsModel({
    ////////////////////////////////////////////////////////////////////////////
    // General Bot Options
    ////////////////////////////////////////////////////////////////////////////
    this.languageLevel,
    this.topic = "General Conversation",
    this.keywords = const [],
    this.safetyModeration = true,
    this.mode = "discussion",

    ////////////////////////////////////////////////////////////////////////////
    // Discussion Mode Options
    ////////////////////////////////////////////////////////////////////////////
    this.discussionTopic,
    this.discussionKeywords,
    this.discussionTriggerReactionEnabled = true,
    this.discussionTriggerReactionKey = "⏩",

    ////////////////////////////////////////////////////////////////////////////
    // Custom Mode Options
    ////////////////////////////////////////////////////////////////////////////
    this.customSystemPrompt,
    this.customTriggerReactionEnabled = true,
    this.customTriggerReactionKey = "⏩",
  });

  factory BotOptionsModel.fromJson(json) {
    return BotOptionsModel(
      //////////////////////////////////////////////////////////////////////////
      // General Bot Options
      //////////////////////////////////////////////////////////////////////////
      languageLevel: json[ModelKey.languageLevel],
      safetyModeration: json[ModelKey.safetyModeration] ?? true,
      mode: json[ModelKey.mode] ?? "discussion",

      //////////////////////////////////////////////////////////////////////////
      // Discussion Mode Options
      //////////////////////////////////////////////////////////////////////////
      discussionTopic: json[ModelKey.discussionTopic],
      discussionKeywords: json[ModelKey.discussionKeywords],
      discussionTriggerReactionEnabled:
          json[ModelKey.discussionTriggerReactionEnabled],
      discussionTriggerReactionKey: json[ModelKey.discussionTriggerReactionKey],

      //////////////////////////////////////////////////////////////////////////
      // Custom Mode Options
      //////////////////////////////////////////////////////////////////////////
      customSystemPrompt: json[ModelKey.customSystemPrompt],
      customTriggerReactionEnabled: json[ModelKey.customTriggerReactionEnabled],
      customTriggerReactionKey: json[ModelKey.customTriggerReactionKey],
    );
  }

  Map<String, dynamic> toJson() {
    final data = <String, dynamic>{};
    try {
      // data[ModelKey.isConversationBotChat] = isConversationBotChat;
      data[ModelKey.languageLevel] = languageLevel;
      data[ModelKey.safetyModeration] = safetyModeration;
      data[ModelKey.mode] = mode;
      data[ModelKey.custom] = custom;
      data[ModelKey.discussionTopic] = discussionTopic;
      data[ModelKey.discussionKeywords] = discussionKeywords;
      data[ModelKey.discussionTriggerReactionEnabled] =
          discussionTriggerReactionEnabled;
      data[ModelKey.discussionTriggerReactionKey] =
          discussionTriggerReactionKey;
      data[ModelKey.customSystemPrompt] = customSystemPrompt;
      data[ModelKey.customTriggerReactionEnabled] =
          customTriggerReactionEnabled;
      data[ModelKey.customTriggerReactionKey] = customTriggerReactionKey;
      return data;
    } catch (e, s) {
      debugger(when: kDebugMode);
      ErrorHandler.logError(e: e, s: s);
      return data;
    }
  }

  //TODO: define enum with all possible values
  updateBotOption(String key, dynamic value) {
    switch (key) {
      case ModelKey.languageLevel:
        languageLevel = value;
        break;
      case ModelKey.safetyModeration:
        safetyModeration = value;
        break;
      case ModelKey.mode:
        mode = value;
        break;
      case ModelKey.custom:
        custom = value;
        break;
      case ModelKey.discussionTopic:
        discussionTopic = value;
        break;
      case ModelKey.discussionKeywords:
        discussionKeywords = value;
        break;
      case ModelKey.discussionTriggerReactionEnabled:
        discussionTriggerReactionEnabled = value;
        break;
      case ModelKey.discussionTriggerReactionKey:
        discussionTriggerReactionKey = value;
        break;
      case ModelKey.customSystemPrompt:
        customSystemPrompt = value;
        break;
      case ModelKey.customTriggerReactionEnabled:
        customTriggerReactionEnabled = value;
        break;
      case ModelKey.customTriggerReactionKey:
        customTriggerReactionKey = value;
        break;
      default:
        throw Exception('Invalid key for bot options - $key');
    }
  }

  StateEvent get toStateEvent => StateEvent(
        content: toJson(),
        type: PangeaEventTypes.botOptions,
      );
}