import { toHexString } from '@chainsafe/ssz'
import debug from 'debug'

import { PortalTrieDB, getDatabaseContent, getDatabaseKey, keyType, wrapDBContent } from './util.js'

import type { AbstractLevel } from 'abstract-level'
import type { Debugger } from 'debug'

export class StateDB {
  db: PortalTrieDB
  logger: Debugger
  blocks: Map<number, string>
  stateRoots: Map<string, string>
  constructor(db: AbstractLevel<string, string, string>, logger?: Debugger) {
    this.db = new PortalTrieDB(db)
    this.logger = logger
      ? logger.extend('StateNetwork').extend('StateDB')
      : debug('StateNetwork').extend('StateDB')
    this.stateRoots = new Map()
    this.blocks = new Map()
  }

  async keys() {
    const keys = await this.db.keys()
    this.logger(`keys: ${keys.length}`)
    return keys
  }

  /**
   * Store content by content key
   * @param contentKey serialized state network content key
   * @param content serialized state network content type
   * @returns true if content is stored successfully
   */
  async storeContent(contentKey: Uint8Array, content: Uint8Array) {
    const dbKey = getDatabaseKey(contentKey)
    const dbContent = getDatabaseContent(keyType(contentKey), content)
    await this.db.put(dbKey, dbContent)
    this.logger(
      `storeContent (${content.length}) bytes: \ncontentKey: ${toHexString(
        contentKey,
      )}\ndbKey: 0x${dbKey}\ndbSize: ${(await this.keys()).length}`,
    )
    return true
  }

  /**
   * Get database content by content key
   * @param contentKey serialized state network content key
   * @returns stored content (node or code) or undefined
   */
  async getContent(contentKey: Uint8Array): Promise<string | undefined> {
    const dbKey = getDatabaseKey(contentKey)
    this.logger!(`getContent: \ncontentKey: ${toHexString(contentKey)}\n: dbKey: ${dbKey}\n`)
    const allkeys = []
    for await (const key of this.db.db.keys()) {
      allkeys.push(key)
    }
    this.logger(`dbKeys: ${allkeys}`)
    const dbContent = await this.db.get(dbKey)
    if (dbContent === undefined) return dbContent
    const content = wrapDBContent(contentKey, dbContent)
    return content
  }
}