f39f8ee830
- Fixed race condition on @olm-context.tsx when a session is created - Fixed an issue where the participants ids where being set to the current user only - Added safeguard for the dm event to try to either parse the message or use it as-is
75 lines
No EOL
1.9 KiB
TypeScript
75 lines
No EOL
1.9 KiB
TypeScript
import { v } from "convex/values";
|
|
import { Id } from "../../_generated/dataModel";
|
|
import { mutation, query } from "../_generated/server";
|
|
|
|
export const sendKeysToServer = mutation({
|
|
args: {
|
|
userId: v.string(),
|
|
identityKey: v.object({
|
|
curve25519: v.string(),
|
|
ed25519: v.string(),
|
|
}),
|
|
oneTimeKeys: v.array(v.object({
|
|
keyId: v.string(),
|
|
publicKey: v.string(),
|
|
})),
|
|
forceInsert: v.boolean(), // if true, insert even if user already has an olm account
|
|
},
|
|
handler: async (ctx, args) => {
|
|
|
|
// check if user already has an olm account
|
|
const olmAccount = await ctx.db.query("olmAccount").withIndex("userId", (q) => q.eq("userId", args.userId)).first();
|
|
|
|
if (olmAccount && !args.forceInsert) {
|
|
throw new Error("User already has an olm account");
|
|
}
|
|
|
|
const insert = await ctx.db.insert<"olmAccount">("olmAccount", {
|
|
userId: args.userId,
|
|
identityKey: args.identityKey,
|
|
oneTimeKeys: args.oneTimeKeys,
|
|
});
|
|
|
|
console.log("insert", insert);
|
|
return insert;
|
|
},
|
|
});
|
|
|
|
export const retrieveServerOlmAccount = query({
|
|
args: {
|
|
userId: v.string(),
|
|
},
|
|
handler: async (ctx, args) => {
|
|
const olmAccount = await ctx.db.get<"olmAccount">(args.userId as Id<"olmAccount">);
|
|
if (olmAccount) return olmAccount;
|
|
|
|
return null;
|
|
},
|
|
});
|
|
|
|
export const consumeOTK = mutation({
|
|
args: {
|
|
userId: v.string(),
|
|
keyId: v.string(),
|
|
},
|
|
handler: async (ctx, args) => {
|
|
const olmAccount = await ctx.db.query("olmAccount").withIndex("userId", (q) => q.eq("userId", args.userId)).first();
|
|
if (!olmAccount) throw new Error("User has no OLM account");
|
|
|
|
const oneTimeKeys = olmAccount.oneTimeKeys;
|
|
const keyIndex = oneTimeKeys.findIndex((key) => key.keyId === args.keyId);
|
|
|
|
if (keyIndex === -1) throw new Error("The key to be consumed was not found");
|
|
|
|
oneTimeKeys.splice(keyIndex, 1);
|
|
|
|
await ctx.db.patch(olmAccount._id, {
|
|
oneTimeKeys,
|
|
});
|
|
|
|
return {
|
|
consumed: true,
|
|
keysLeft: oneTimeKeys.length
|
|
}
|
|
},
|
|
}) |