//
// Copyright © 2022 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 { DOMAIN_MODEL_TX, DOMAIN_TX, withContext, type MeasureContext } from '@hcengineering/core'
import type {
  DbAdapter,
  DbConfiguration,
  Middleware,
  MiddlewareCreator,
  PipelineContext,
  TxAdapter
} from '@hcengineering/server-core'
import { BaseMiddleware, createServiceAdaptersManager, DbAdapterManagerImpl } from '@hcengineering/server-core'

/**
 * @public
 */
export class DBAdapterMiddleware extends BaseMiddleware implements Middleware {
  constructor (
    context: PipelineContext,
    next: Middleware | undefined,
    readonly conf: DbConfiguration
  ) {
    super(context, next)
  }

  static create (conf: DbConfiguration): MiddlewareCreator {
    return async (ctx, context, next): Promise<Middleware> => {
      const middleware = new DBAdapterMiddleware(context, next, conf)
      await middleware.init(ctx)
      return middleware
    }
  }

  @withContext('dbAdapter-middleware')
  async init (ctx: MeasureContext): Promise<void> {
    const adapters = new Map<string, DbAdapter>()

    await ctx.with('create-adapters', {}, async (ctx) => {
      for (const key in this.conf.adapters) {
        const adapterConf = this.conf.adapters[key]
        adapters.set(
          key,
          await adapterConf.factory(
            ctx,
            this.context.hierarchy,
            adapterConf.url,
            this.context.workspace,
            this.context.modelDb,
            this.context.storageAdapter
          )
        )
      }
    })

    const metrics = ctx.newChild('📔 adapters', {})

    const txAdapterName = this.conf.domains[DOMAIN_TX]
    const txAdapter = adapters.get(txAdapterName) as TxAdapter
    await txAdapter.init?.(metrics, this.context.contextVars, [DOMAIN_TX, DOMAIN_MODEL_TX])

    const defaultAdapter = adapters.get(this.conf.defaultAdapter)
    if (defaultAdapter === undefined) {
      throw new Error(`No default Adapter for ${this.conf.defaultAdapter}`)
    }

    this.context.serviceAdapterManager = await createServiceAdaptersManager(this.conf.serviceAdapters, metrics)

    // We need to init all next, since we will use model

    const adapterManager = new DbAdapterManagerImpl(metrics, this.conf, this.context, defaultAdapter, adapters)
    this.context.adapterManager = adapterManager
  }

  async close (): Promise<void> {
    await this.context.adapterManager?.close()
    await this.context.serviceAdapterManager?.close()
  }
}