import { QueryResult } from '@aragon/connect-thegraph'

import { AgreementData } from '../../types'

export function parseAgreement(result: QueryResult): AgreementData {
  const agreement = result.data.agreement

  if (!agreement) {
    throw new Error('Unable to parse agreement.')
  }

  return {
    id: agreement.id,
    dao: agreement.dao,
    stakingFactory: agreement.stakingFactory,
    currentVersionId: agreement.currentVersion.id,
  }
}