//
// Copyright © 2023 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 DocumentId, type PlatformDocumentId } from '@hcengineering/collaborator-client'
import { WorkspaceId, generateId } from '@hcengineering/core'
import { decodeToken } from '@hcengineering/server-token'
import { onAuthenticatePayload } from '@hocuspocus/server'
import { ClientFactory, simpleClientFactory } from './platform'

export interface Context {
  connectionId: string
  workspaceId: WorkspaceId
  clientFactory: ClientFactory

  initialContentId?: DocumentId
  platformDocumentId?: PlatformDocumentId
}

interface WithContext {
  context: any
}

export type withContext<T extends WithContext> = Omit<T, 'context'> & {
  context: Context
}

export function buildContext (data: onAuthenticatePayload): Context {
  const context = data.context as Partial<Context>

  const connectionId = context.connectionId ?? generateId()
  const decodedToken = decodeToken(data.token)

  const initialContentId = (data.requestParameters.get('initialContentId') as DocumentId) ?? undefined
  const platformDocumentId = (data.requestParameters.get('platformDocumentId') as PlatformDocumentId) ?? undefined

  return {
    connectionId,
    workspaceId: decodedToken.workspace,
    clientFactory: simpleClientFactory(decodedToken),
    initialContentId,
    platformDocumentId
  }
}