@magmamath/students-features 0.2.13 → 0.2.15

package/src/features/chatbot/model/model.ts

@@ -2,276 +2,267 @@ import { attach, createEvent, restore, sample, type Store, type Effect } from 'e
 import { createControllerEffect, createEntityFieldStore } from '../../../lib/effector'
 
 import {
-    AvailableChatsModel,
-    ChatbotModelProps,
-    ChatMessagesModel,
-    ConversationMessage,
-    EntityIds,
-    SendHelpParams,
-    SendHelpProps,
-    SetHintMessageProps,
-    ThreadIdsModel,
+  AvailableChatsModel,
+  ChatbotModelProps,
+  ChatMessagesModel,
+  ConversationMessage,
+  EntityIds,
+  SendHelpParams,
+  SendHelpProps,
+  SetHintMessageProps,
+  ThreadIdsModel,
 } from '../types/model.types'
 import { GetAlternativesResponse, PostMessageResponse } from '../types/api.types'
 
 import { generateRandomId } from '../../../lib/helpers/helpers'
 import { ChatbotAPI } from './api'
 import { ChatbotTranslation } from './translation'
-import { ChatTextToSpeech } from './t2s'
+import { ChatTextToSpeechModel } from './t2s'
 
 export class ChatbotModel {
-    readonly api: ChatbotAPI
-    readonly translation: ChatbotTranslation
-    readonly t2s: ChatTextToSpeech
-
-    readonly credits: ChatbotModelProps['credits']
-
-    readonly resetAlternatives = createEvent()
-    readonly setAlternatives = createEvent<GetAlternativesResponse | null>()
-    readonly $alternatives: Store<GetAlternativesResponse | null>
-
-    readonly setIsOpen = createEvent<boolean>()
-    readonly toggleOpen = createEvent()
-    readonly $isOpen = restore(this.setIsOpen, false).on(this.toggleOpen, isOpen => !isOpen)
-
-    // separate messages and alternatives class?
-    readonly removeLastMessage = createEvent()
-    readonly setLastMessageIds = createEvent<EntityIds | null>()
-    readonly $lastMessageIds = restore(this.setLastMessageIds, null)
-
-    readonly setScrollActive = createEvent<boolean>()
-    readonly $isScrollActive = restore(this.setScrollActive, false)
-
-    readonly sendHelpRequestFx: Effect<SendHelpParams, PostMessageResponse | ConversationMessage>
-
-    readonly defaultMessages: ChatbotModelProps['messages']
-
-    constructor({
-        credits,
-        api,
+  readonly api: ChatbotAPI
+  readonly translation: ChatbotTranslation
+  readonly t2s: ChatTextToSpeechModel
+
+  readonly credits: ChatbotModelProps['credits']
+
+  readonly resetAlternatives = createEvent()
+  readonly setAlternatives = createEvent<GetAlternativesResponse | null>()
+  readonly $alternatives: Store<GetAlternativesResponse | null>
+
+  readonly setIsOpen = createEvent<boolean>()
+  readonly toggleOpen = createEvent()
+  readonly $isOpen = restore(this.setIsOpen, false).on(this.toggleOpen, isOpen => !isOpen)
+
+  // separate messages and alternatives class?
+  readonly removeLastMessage = createEvent()
+  readonly setLastMessageIds = createEvent<EntityIds | null>()
+  readonly $lastMessageIds = restore(this.setLastMessageIds, null)
+
+  readonly setScrollActive = createEvent<boolean>()
+  readonly $isScrollActive = restore(this.setScrollActive, false)
+
+  readonly sendHelpRequestFx: Effect<SendHelpParams, PostMessageResponse | ConversationMessage>
+
+  readonly defaultMessages: ChatbotModelProps['messages']
+
+  constructor({ credits, api, messages }: ChatbotModelProps) {
+    this.api = new ChatbotAPI(api)
+
+    this.defaultMessages = messages
+    this.credits = credits
+
+    this.$alternatives = restore(this.api.getAlternativesFx.doneData, null)
+      .on(this.setAlternatives, (_, payload) => payload)
+      .reset(this.resetAlternatives)
+
+    // TODO: do not pass this, just what you need
+    this.translation = new ChatbotTranslation(this)
+    this.t2s = new ChatTextToSpeechModel({ $isOpen: this.$isOpen, api: this.api })
+
+    this.sendHelpRequestFx = this.createSendHelpRequestFx()
+    this.initChatReplyWatcher()
+  }
+
+  readonly messages = createEntityFieldStore({
+    mapper: ({ entityId, fieldId, messages }: ChatMessagesModel) => ({
+      entityId,
+      fieldId,
+      data: { messages },
+    }),
+  })
+
+  readonly threadIds = createEntityFieldStore({
+    mapper: ({ entityId, fieldId, threadId }: ThreadIdsModel) => ({
+      entityId,
+      fieldId,
+      data: { threadId },
+    }),
+  })
+
+  readonly availableChats = createEntityFieldStore({
+    mapper: ({ entityId, fieldId, isAvailable = true }: AvailableChatsModel) => ({
+      entityId,
+      fieldId,
+      data: { isAvailable },
+    }),
+  })
+
+  private createSendHelpRequestFx() {
+    return attach({
+      source: [
+        this.credits.$assignmentId,
+        this.credits.$exerciseId,
+        this.messages.$state,
+        this.translation.$isTranslated,
+      ],
+      mapParams: (params: SendHelpParams, [assignmentId, exerciseId, messages, isTranslated]) => ({
+        ...params,
+        assignmentId,
+        exerciseId,
         messages,
-    }: ChatbotModelProps) {
-        this.api = new ChatbotAPI(api)
-
-        this.defaultMessages = messages
-        this.credits = credits
-
-        this.$alternatives = restore(this.api.getAlternativesFx.doneData, null)
-            .on(this.setAlternatives, (_, payload) => payload)
-            .reset(this.resetAlternatives)
-
-        // TODO: do not pass this, just what you need
-        this.translation = new ChatbotTranslation(this)
-        this.t2s = new ChatTextToSpeech(this)
-
-        this.sendHelpRequestFx = this.createSendHelpRequestFx()
-        this.initChatReplyWatcher()
-    }
-
-    readonly messages = createEntityFieldStore({
-        mapper: ({ entityId, fieldId, messages }: ChatMessagesModel) => ({
-            entityId,
-            fieldId,
-            data: { messages },
-        }),
-    })
-
-    readonly threadIds = createEntityFieldStore({
-        mapper: ({ entityId, fieldId, threadId }: ThreadIdsModel) => ({
-            entityId,
-            fieldId,
-            data: { threadId },
-        }),
-    })
+        isTranslated,
+      }),
+      effect: createControllerEffect(
+        async ({
+          payload,
+          message = '',
+          translationTargetLanguage,
+          translatedMessage,
+          assignmentId,
+          exerciseId,
+          messages,
+          isTranslated,
+        }: SendHelpProps) => {
+          this.setLastMessageIds({
+            entityId: assignmentId,
+            fieldId: exerciseId,
+          })
 
-    readonly availableChats = createEntityFieldStore({
-        mapper: ({ entityId, fieldId, isAvailable = true }: AvailableChatsModel) => ({
-            entityId,
-            fieldId,
-            data: { isAvailable },
-        }),
-    })
+          const currentMessages = messages[assignmentId]?.[exerciseId]?.messages || []
 
-    private createSendHelpRequestFx() {
-        return attach({
-            source: [
-                this.credits.$assignmentId,
-                this.credits.$exerciseId,
-                this.messages.$state,
-                this.translation.$isTranslated,
+          this.messages.setField({
+            entityId: assignmentId,
+            fieldId: exerciseId,
+            messages: [
+              ...currentMessages,
+              {
+                id: generateRandomId(),
+                message: message ?? '',
+                translatedMessage,
+                isOwnMessage: true,
+                threadId: payload.threadId ?? '',
+                options: [],
+              },
             ],
-            mapParams: (
-                params: SendHelpParams,
-                [assignmentId, exerciseId, messages, isTranslated],
-            ) => ({
-                ...params,
-                assignmentId,
-                exerciseId,
-                messages,
-                isTranslated,
-            }),
-            effect: createControllerEffect(
-                async ({
-                    payload,
-                    message = '',
-                    translationTargetLanguage,
-                    translatedMessage,
-                    assignmentId,
-                    exerciseId,
-                    messages,
-                    isTranslated,
-                }: SendHelpProps) => {
-                    this.setLastMessageIds({
-                        entityId: assignmentId,
-                        fieldId: exerciseId,
-                    })
-
-                    const currentMessages = messages[assignmentId]?.[exerciseId]?.messages || []
-
-                    this.messages.setField({
-                        entityId: assignmentId,
-                        fieldId: exerciseId,
-                        messages: [
-                            ...currentMessages,
-                            {
-                                id: generateRandomId(),
-                                message: message ?? '',
-                                translatedMessage,
-                                isOwnMessage: true,
-                                threadId: payload.threadId ?? '',
-                                options: [],
-                            },
-                        ],
-                    })
-
-                    const response = await this.api.postMessageFx(payload)
-                    if (isTranslated) {
-                        return await this.translation.translateMessageFx({
-                            targetLanguage: translationTargetLanguage as string,
-                            message: {
-                                ...response,
-                                id: generateRandomId(),
-                                isOwnMessage: false,
-                            },
-                        })
-                    }
-
-                    return response
-                },
-            ),
-        })
-    }
-
-    private initChatReplyWatcher() {
-        sample({
-            clock: this.sendHelpRequestFx.doneData,
-            source: [this.credits.$assignmentId, this.credits.$exerciseId, this.messages.$state] as const,
-            fn: ([assignmentId, exerciseId, messages], response) => ({
-                assignmentId,
-                exerciseId,
-                messages: messages[assignmentId]?.[exerciseId]?.messages || [],
-                response,
-            }),
-        }).watch(({ assignmentId, exerciseId, messages, response }) => {
-            this.messages.setField({
-                entityId: assignmentId,
-                fieldId: exerciseId,
-                messages: [
-                    ...messages,
-                    {
-                        id: generateRandomId(),
-                        message: response.message,
-                        isOwnMessage: false,
-                        threadId: response.threadId,
-                        options: response.options,
-                        translatedMessage:
-                            'translatedMessage' in response
-                                ? response.translatedMessage
-                                : undefined,
-                    },
-                ],
+          })
+
+          const response = await this.api.postMessageFx(payload)
+          if (isTranslated) {
+            return await this.translation.translateMessageFx({
+              targetLanguage: translationTargetLanguage as string,
+              message: {
+                ...response,
+                id: generateRandomId(),
+                isOwnMessage: false,
+              },
             })
+          }
 
-            this.threadIds.setField({
-                entityId: assignmentId,
-                fieldId: exerciseId,
-                threadId: response.threadId,
-            })
-        })
-
-        sample({
-            clock: this.sendHelpRequestFx.fail,
-            source: [this.credits.$assignmentId, this.credits.$exerciseId, this.messages.$state] as const,
-            fn: ([assignmentId, exerciseId, messages]) => ({
-                assignmentId,
-                exerciseId,
-                messages: messages[assignmentId]?.[exerciseId]?.messages || [],
-            }),
-            filter: (_, { error }) => error.message !== 'Request aborted manually',
-        }).watch(({ assignmentId, exerciseId, messages }) => {
-            this.messages.setField({
-                entityId: assignmentId,
-                fieldId: exerciseId,
-                messages: [...messages, this.defaultMessages.error],
-            })
-        })
-
-        sample({
-            clock: this.removeLastMessage,
-            source: [this.$lastMessageIds, this.messages.$state] as const,
-            fn: ([lastMessageIds, messages]) => ({
-                lastMessageIds,
-                messages,
-            }),
-        }).watch(({ lastMessageIds, messages }) => {
-            if (!lastMessageIds) return
-
-            const updatedMessages = [
-                ...messages[lastMessageIds.entityId][lastMessageIds.fieldId].messages,
-            ]
-            if (updatedMessages) updatedMessages.pop()
-
-            this.messages.setField({
-                entityId: lastMessageIds.entityId,
-                fieldId: lastMessageIds.fieldId,
-                messages: updatedMessages,
-            })
-        })
-    }
+          return response
+        },
+      ),
+    })
+  }
+
+  private initChatReplyWatcher() {
+    sample({
+      clock: this.sendHelpRequestFx.doneData,
+      source: [this.credits.$assignmentId, this.credits.$exerciseId, this.messages.$state] as const,
+      fn: ([assignmentId, exerciseId, messages], response) => ({
+        assignmentId,
+        exerciseId,
+        messages: messages[assignmentId]?.[exerciseId]?.messages || [],
+        response,
+      }),
+    }).watch(({ assignmentId, exerciseId, messages, response }) => {
+      this.messages.setField({
+        entityId: assignmentId,
+        fieldId: exerciseId,
+        messages: [
+          ...messages,
+          {
+            id: generateRandomId(),
+            message: response.message,
+            isOwnMessage: false,
+            threadId: response.threadId,
+            options: response.options,
+            translatedMessage:
+              'translatedMessage' in response ? response.translatedMessage : undefined,
+          },
+        ],
+      })
+
+      this.threadIds.setField({
+        entityId: assignmentId,
+        fieldId: exerciseId,
+        threadId: response.threadId,
+      })
+    })
 
-    public startConversation() {
-        const assignmentId = this.credits.$assignmentId.getState()
-        const exerciseId = this.credits.$exerciseId.getState()
-        const messages = this.messages.$state.getState()[assignmentId]?.[exerciseId]?.messages ?? []
+    sample({
+      clock: this.sendHelpRequestFx.fail,
+      source: [this.credits.$assignmentId, this.credits.$exerciseId, this.messages.$state] as const,
+      fn: ([assignmentId, exerciseId, messages]) => ({
+        assignmentId,
+        exerciseId,
+        messages: messages[assignmentId]?.[exerciseId]?.messages || [],
+      }),
+      filter: (_, { error }) => error.message !== 'Request aborted manually',
+    }).watch(({ assignmentId, exerciseId, messages }) => {
+      this.messages.setField({
+        entityId: assignmentId,
+        fieldId: exerciseId,
+        messages: [...messages, this.defaultMessages.error],
+      })
+    })
 
-        if (messages.length) return
+    sample({
+      clock: this.removeLastMessage,
+      source: [this.$lastMessageIds, this.messages.$state] as const,
+      fn: ([lastMessageIds, messages]) => ({
+        lastMessageIds,
+        messages,
+      }),
+    }).watch(({ lastMessageIds, messages }) => {
+      if (!lastMessageIds) return
+
+      const updatedMessages = [
+        ...messages[lastMessageIds.entityId][lastMessageIds.fieldId].messages,
+      ]
+      if (updatedMessages) updatedMessages.pop()
+
+      this.messages.setField({
+        entityId: lastMessageIds.entityId,
+        fieldId: lastMessageIds.fieldId,
+        messages: updatedMessages,
+      })
+    })
+  }
 
-        this.messages.setField({
-            entityId: assignmentId,
-            fieldId: exerciseId,
-            messages: [this.defaultMessages.start],
-        })
+  public startConversation() {
+    const assignmentId = this.credits.$assignmentId.getState()
+    const exerciseId = this.credits.$exerciseId.getState()
+    const messages = this.messages.$state.getState()[assignmentId]?.[exerciseId]?.messages ?? []
 
-        this.availableChats.setField({
-            entityId: assignmentId,
-            fieldId: exerciseId,
-            isAvailable: true,
-        });
-    }
+    if (messages.length) return
 
-    public setHintMessage({ entityId, fieldId, ...message }: SetHintMessageProps) {
-        const messages = this.messages.$state.getState()?.[entityId]?.[fieldId]?.messages
-        if (messages?.length > 1) return
+    this.messages.setField({
+      entityId: assignmentId,
+      fieldId: exerciseId,
+      messages: [this.defaultMessages.start],
+    })
 
-        this.messages.setField({
-            entityId,
-            fieldId,
-            messages: [
-                {
-                    ...message,
-                    options: this.$alternatives.getState()?.options.hints ?? message.options,
-                },
-            ],
-        })
-    }
+    this.availableChats.setField({
+      entityId: assignmentId,
+      fieldId: exerciseId,
+      isAvailable: true,
+    })
+  }
+
+  public setHintMessage({ entityId, fieldId, ...message }: SetHintMessageProps) {
+    const messages = this.messages.$state.getState()?.[entityId]?.[fieldId]?.messages
+    if (messages?.length > 1) return
+
+    this.messages.setField({
+      entityId,
+      fieldId,
+      messages: [
+        {
+          ...message,
+          options: this.$alternatives.getState()?.options.hints ?? message.options,
+        },
+      ],
+    })
+  }
 }