Hydro/packages/hydrooj/src/model/oauth.ts

import { Collection } from 'mongodb';
import db from '../service/db';

interface OauthMap {
    _id: string, // source openId
    uid: number, // target uid
}

const coll: Collection<OauthMap> = db.collection('oauth');

class OauthModel {
    static async get(_id: string) {
        const doc = await coll.findOne({ _id });
        if (doc) return doc.uid;
        return null;
    }

    static async set(_id: string, value: number) {
        const res = await coll.findOneAndUpdate(
            { _id },
            { $set: { value } },
            { upsert: true, returnOriginal: false },
        );
        return res.value.uid;
    }
}

export = OauthModel;
global.Hydro.model.oauth = OauthModel;