import { DocumentNode, gql } from '@apollo/client/core'
import { getAddress } from 'ethers'

import {
	OptionPositionExtendedFragment,
	OptionPositionFragment,
} from '../fragments'
import { addFields } from '../../../../utils/voidnode'
import PremiaVoidnode from '../../index'

export class OptionPositionQuery {
	static optionPositionId(owner: string, poolAddress: string): string {
		return getAddress(owner) + ':' + getAddress(poolAddress)
	}

	@addFields
	static GetOptionPosition(
		voidnode: PremiaVoidnode,
		owner: string,
		poolAddress: string
	): DocumentNode {
		return gql`
        ${OptionPositionFragment}

        {
            optionPosition(id: "${this.optionPositionId(owner, poolAddress)}") {
                ...OptionPosition
            }
        }
    `
	}

	@addFields
	static GetOptionPositionExtended(
		voidnode: PremiaVoidnode,
		owner: string,
		poolAddress: string
	): DocumentNode {
		return gql`
        ${OptionPositionExtendedFragment}

        { 
            optionPosition(id: "${this.optionPositionId(owner, poolAddress)}") {
                ...OptionPositionExtended
            }
        }
    `
	}

	@addFields
	static GetOptionPositionsExtendedForUser(
		voidnode: PremiaVoidnode,
		owner: string,
		isOpen?: boolean
	): DocumentNode {
		let filter
		if (isOpen == undefined) {
			filter = ''
		} else {
			filter = isOpen
				? ', { closedAt: null }, { or: [{ pool_: { isExpiredOTM: false } }, { option_ne: null }, { isBuy: false }] }'
				: ', { or: [{ closedAt_ne: null }, { pool_: { isExpiredOTM: true }, isBuy: true }] }'
		}

		return gql`
			${OptionPositionExtendedFragment}

			{
				optionPositions(
					where: { 
						and: [
							{ owner: "${getAddress(owner)}" }
							${filter}
						]
					},
					limit: 1000, 
					orderBy: [{
						field: createdAt, 
						direction: DESC
					}]
				) {
					...OptionPositionExtended
				}
			}
		`
	}

	@addFields
	static GetRewardOptionPositionsExtendedForUser(
		voidnode: PremiaVoidnode,
		owner: string,
		timestamp?: number,
		isOpen?: boolean
	): DocumentNode {
		let filter = `owner: "${getAddress(owner)}", option_ne: null`
		if (isOpen !== undefined) {
			filter += isOpen ? ', closedAt: null' : ', closedAt_ne: null'
		}
		if (timestamp) {
			filter += `, maturity_lte: ${timestamp}`
		}
		return gql`
			${OptionPositionExtendedFragment}
			{
				optionPositions(
					where: { 
						 ${filter}
					},
					limit: 1000, 
					orderBy: [{
						field: createdAt, 
						direction: DESC
					}]
				) {
					...OptionPositionExtended
				}
			}
		`
	}
}