platform/plugins/ai-bot-resources/src/requests.ts

//
// Copyright © 2024 Hardcore Engineering Inc.
//
// Licensed under the Eclipse Public License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License. You may
// obtain a copy of the License at https://www.eclipse.org/legal/epl-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//
// See the License for the specific language governing permissions and
// limitations under the License.
//
import {
  type ConnectMeetingRequest,
  type DisconnectMeetingRequest,
  type SummarizeMessagesRequest,
  type SummarizeMessagesResponse,
  type TranslateRequest,
  type TranslateResponse
} from '@hcengineering/ai-bot'
import { type Class, concatLink, type Doc, type Markup, type Ref } from '@hcengineering/core'
import { type Room, type RoomLanguage } from '@hcengineering/love'
import { getMetadata } from '@hcengineering/platform'
import presentation from '@hcengineering/presentation'

import aiBot from './plugin'

export async function translate (text: Markup, lang: string): Promise<TranslateResponse | undefined> {
  const url = getMetadata(aiBot.metadata.EndpointURL) ?? ''
  const token = getMetadata(presentation.metadata.Token) ?? ''

  if (url === '' || token === '') {
    return undefined
  }

  try {
    const req: TranslateRequest = { text, lang }
    const resp = await fetch(concatLink(url, '/translate'), {
      method: 'POST',
      headers: {
        Authorization: 'Bearer ' + token,
        'Content-Type': 'application/json'
      },
      body: JSON.stringify(req)
    })
    if (!resp.ok) {
      return undefined
    }

    return (await resp.json()) as TranslateResponse
  } catch (error) {
    console.error(error)
    return undefined
  }
}

export async function summarizeMessages (
  lang: string,
  target: Ref<Doc>,
  targetClass: Ref<Class<Doc>>
): Promise<SummarizeMessagesResponse | undefined> {
  const url = getMetadata(aiBot.metadata.EndpointURL) ?? ''
  const token = getMetadata(presentation.metadata.Token) ?? ''

  if (url === '' || token === '') {
    return undefined
  }

  try {
    const req: SummarizeMessagesRequest = {
      target,
      targetClass,
      lang
    }
    const resp = await fetch(concatLink(url, '/summarize'), {
      method: 'POST',
      headers: {
        Authorization: 'Bearer ' + token,
        'Content-Type': 'application/json'
      },
      body: JSON.stringify(req)
    })
    if (!resp.ok) {
      return undefined
    }

    return (await resp.json()) as SummarizeMessagesResponse
  } catch (error) {
    console.error(error)
    return undefined
  }
}

export async function connectMeeting (
  roomId: Ref<Room>,
  language: RoomLanguage,
  options: Partial<ConnectMeetingRequest>
): Promise<void> {
  const url = getMetadata(aiBot.metadata.EndpointURL) ?? ''
  const token = getMetadata(presentation.metadata.Token) ?? ''

  if (url === '' || token === '') {
    return undefined
  }

  try {
    const req: ConnectMeetingRequest = { roomId, transcription: options.transcription ?? false, language }
    await fetch(concatLink(url, 'love/connect'), {
      method: 'POST',
      headers: {
        Authorization: 'Bearer ' + token,
        'Content-Type': 'application/json'
      },
      body: JSON.stringify(req)
    })
  } catch (error) {
    console.error(error)
    return undefined
  }
}

export async function disconnectMeeting (roomId: Ref<Room>): Promise<void> {
  const url = getMetadata(aiBot.metadata.EndpointURL) ?? ''
  const token = getMetadata(presentation.metadata.Token) ?? ''

  if (url === '' || token === '') {
    return undefined
  }

  try {
    const req: DisconnectMeetingRequest = { roomId }
    await fetch(concatLink(url, 'love/disconnect'), {
      method: 'POST',
      headers: {
        Authorization: 'Bearer ' + token,
        'Content-Type': 'application/json'
      },
      body: JSON.stringify(req)
    })
  } catch (error) {
    console.error(error)
    return undefined
  }
}