DeFiCh/jellyfish

import { Model, ModelMapping } from '../module.database/model'
import { Injectable } from '@nestjs/common'
import { Database, SortOrder } from '../module.database/database'

const PoolSwapMapping: ModelMapping<PoolSwap> = {
  type: 'pool_swap',
  index: {
    key_sort: {
      name: 'pool_swap_sort',
      partition: {
        type: 'string',
        key: (b: PoolSwap) => b.poolPairId
      },
      sort: {
        type: 'string',
        key: (b: PoolSwap) => b.sort
      }
    }
  }
}

@Injectable()
export class PoolSwapMapper {
  public constructor (protected readonly database: Database) {
  }

  async query (key: string, limit: number, lt?: string, gt?: string): Promise<PoolSwap[]> {
    return await this.database.query(PoolSwapMapping.index.key_sort, {
      partitionKey: key,
      limit: limit,
      order: SortOrder.DESC,
      lt: lt,
      gt: gt
    })
  }

  async put (feed: PoolSwap): Promise<void> {
    return await this.database.put(PoolSwapMapping, feed)
  }

  async delete (id: string): Promise<void> {
    return await this.database.delete(PoolSwapMapping, id)
  }
}

export interface PoolSwap extends Model {
  id: string // ---------| poolpairId-txid
  txid: string
  txno: number

  poolPairId: string // ------| poolPairId (decimal encoded integer as string)
  sort: string // ------------| height-txnNo

  fromAmount: string // --------| bignumber
  fromTokenId: number // -------| number

  block: {
    hash: string
    height: number
    time: number
    medianTime: number
  }
}