loc-chain-backend/test_clients.js

const fs = require('fs')
const openpgp = require('openpgp')
const bcrypt = require('bcrypt')

class TestClient {
    constructor(keyName, id) {
        this.keyName = keyName
        this.id = id
        this.publicKey = fs.readFileSync(`./${this.keyName}.public.gpg.key`).toString('utf-8')
        this.privateKey = fs.readFileSync(`./${this.keyName}.private.gpg.key`).toString('utf-8')
        this.salt = fs.readFileSync(`./${this.keyName}.salt`).toString('utf-8')
    }

    getHash(otherClientId) {
        const [lesserId, greaterId] = [otherClientId, this.id].sort()
        return bcrypt.hashSync(`${lesserId}-${greaterId}`, this.salt)
    }

    buildExposure() {
        return {
            clientID: this.id,
            timestamp: (new Date).getTime(),
        }
    }

    async interactWith(otherClient) {
        const hash = otherClient.getHash(this.id)
        const message = openpgp.Message.fromText(hash)
        const signature = (await openpgp.sign({
            message,
            privateKeys: await openpgp.readKey({
                armoredKey: this.privateKey
            })
        }))

        return {
            combinedHash: hash,
            timestamp: (new Date).getTime(),
            encodedGPSLocation: JSON.stringify({ none: true }),
            partnerPublicKey: otherClient.publicKey,
            validationSignature: signature
        }
    }
}

;(async () => {
    const client_a = new TestClient('test_a', 'a893irwe')
    const client_b = new TestClient('test_b', 'bawesfdi')

    const a_transact = await client_a.interactWith(client_b)
    const b_transact = await client_b.interactWith(client_a)

    // Simulate an encounter between A and B
    console.log(a_transact, b_transact)
    postTo('/api/v1/encounter', a_transact)
    postTo('/api/v1/encounter', b_transact)

    // Simulate B being exposed
    const b_exposure = client_b.buildExposure()
    console.log(b_exposure)
    postTo('/api/v1/exposure', b_exposure)
})()

function postTo(endpoint, data) {
    data = JSON.stringify(data)

    const options = {
        hostname: 'localhost',
        port: 8000,
        path: endpoint,
        method: 'POST',
        headers: {
            'content-type': 'application/json',
            'content-length': data.length,
        }
    }

    const req = require('http').request(options)
    req.write(data)
    req.end()
}