import * as uint8arrays from 'uint8arrays'
import { EcdsaKeypair, parseDidKey, Secp256k1Keypair } from '@atproto/crypto'
import * as document from '../src/document'
import * as t from '../src/types'

describe('document', () => {
  it('formats a valid DID document', async () => {
    const atprotoKey = await Secp256k1Keypair.create()
    const otherKey = await EcdsaKeypair.create()
    const rotate1 = await Secp256k1Keypair.create()
    const rotate2 = await EcdsaKeypair.create()
    const alsoKnownAs = ['at://alice.test', 'https://bob.test']
    const atpPds = 'https://example.com'
    const otherService = 'https://other.com'
    const data: t.DocumentData = {
      did: 'did:example:alice',
      verificationMethods: {
        atproto: atprotoKey.did(),
        other: otherKey.did(),
      },
      rotationKeys: [rotate1.did(), rotate2.did()],
      alsoKnownAs,
      services: {
        atproto_pds: {
          type: 'AtprotoPersonalDataServer',
          endpoint: atpPds,
        },
        other: {
          type: 'SomeService',
          endpoint: otherService,
        },
      },
    }
    const doc = await document.formatDidDoc(data)
    // only expected keys
    expect(Object.keys(doc).sort()).toEqual(
      ['@context', 'id', 'alsoKnownAs', 'verificationMethod', 'service'].sort(),
    )
    expect(doc['@context']).toEqual([
      'https://www.w3.org/ns/did/v1',
      'https://w3id.org/security/suites/secp256k1-2019/v1',
      'https://w3id.org/security/suites/ecdsa-2019/v1',
    ])
    expect(doc.id).toEqual(data.did)
    expect(doc.alsoKnownAs).toEqual(alsoKnownAs)

    expect(doc.verificationMethod.length).toBe(2)

    expect(doc.verificationMethod[0].id).toEqual('#atproto')
    expect(doc.verificationMethod[0].type).toEqual(
      'EcdsaSecp256k1VerificationKey2019',
    )
    expect(doc.verificationMethod[0].controller).toEqual(data.did)
    const parsedAtprotoKey = parseDidKey(atprotoKey.did())
    const atprotoKeyMultibase =
      'z' + uint8arrays.toString(parsedAtprotoKey.keyBytes, 'base58btc')
    expect(doc.verificationMethod[0].publicKeyMultibase).toEqual(
      atprotoKeyMultibase,
    )

    expect(doc.verificationMethod[1].id).toEqual('#other')
    expect(doc.verificationMethod[1].type).toEqual(
      'EcdsaSecp256r1VerificationKey2019',
    )
    expect(doc.verificationMethod[1].controller).toEqual(data.did)
    const parsedOtherKey = parseDidKey(otherKey.did())
    const otherKeyMultibase =
      'z' + uint8arrays.toString(parsedOtherKey.keyBytes, 'base58btc')
    expect(doc.verificationMethod[1].publicKeyMultibase).toEqual(
      otherKeyMultibase,
    )

    expect(doc.service.length).toBe(2)
    expect(doc.service[0].id).toEqual('#atproto_pds')
    expect(doc.service[0].type).toEqual('AtprotoPersonalDataServer')
    expect(doc.service[0].serviceEndpoint).toEqual(atpPds)
    expect(doc.service[1].id).toEqual('#other')
    expect(doc.service[1].type).toEqual('SomeService')
    expect(doc.service[1].serviceEndpoint).toEqual(otherService)
  })
})