79 lines
2.3 KiB
JavaScript
79 lines
2.3 KiB
JavaScript
import { E_VAULT_KEY_EXISTS, GqlAuthenticationError, GqlInputError } from '@/lib/error'
|
|
|
|
export default {
|
|
Query: {
|
|
getVaultEntry: async (parent, { key }, { me, models }, info) => {
|
|
if (!me) throw new GqlAuthenticationError()
|
|
if (!key) throw new GqlInputError('must have key')
|
|
|
|
const k = await models.vault.findUnique({
|
|
where: {
|
|
userId_key_ownerId_ownerType: {
|
|
key,
|
|
userId: me.id
|
|
}
|
|
}
|
|
})
|
|
return k
|
|
},
|
|
getVaultEntries: async (parent, { keysFilter }, { me, models }, info) => {
|
|
if (!me) throw new GqlAuthenticationError()
|
|
|
|
const entries = await models.vault.findMany({
|
|
where: {
|
|
userId: me.id,
|
|
key: keysFilter?.length
|
|
? {
|
|
in: keysFilter
|
|
}
|
|
: undefined
|
|
}
|
|
})
|
|
return entries
|
|
}
|
|
},
|
|
Mutation: {
|
|
// atomic vault migration
|
|
updateVaultKey: async (parent, { entries, hash }, { me, models }) => {
|
|
if (!me) throw new GqlAuthenticationError()
|
|
if (!hash) throw new GqlInputError('hash required')
|
|
const txs = []
|
|
|
|
const { vaultKeyHash: oldKeyHash } = await models.user.findUnique({ where: { id: me.id } })
|
|
if (oldKeyHash) {
|
|
if (oldKeyHash !== hash) {
|
|
throw new GqlInputError('vault key already set', E_VAULT_KEY_EXISTS)
|
|
} else {
|
|
return true
|
|
}
|
|
} else {
|
|
txs.push(models.user.update({
|
|
where: { id: me.id },
|
|
data: { vaultKeyHash: hash }
|
|
}))
|
|
}
|
|
|
|
for (const entry of entries) {
|
|
txs.push(models.vaultEntry.upsert({
|
|
where: { userId: me.id, key: entry.key },
|
|
update: { key: entry.key, value: entry.value },
|
|
create: { key: entry.key, value: entry.value, userId: me.id, walletId: entry.walletId }
|
|
}))
|
|
}
|
|
await models.prisma.$transaction(txs)
|
|
return true
|
|
},
|
|
clearVault: async (parent, args, { me, models }) => {
|
|
if (!me) throw new GqlAuthenticationError()
|
|
const txs = []
|
|
txs.push(models.user.update({
|
|
where: { id: me.id },
|
|
data: { vaultKeyHash: '' }
|
|
}))
|
|
txs.push(models.vaultEntry.deleteMany({ where: { userId: me.id } }))
|
|
await models.prisma.$transaction(txs)
|
|
return true
|
|
}
|
|
}
|
|
}
|