You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
983 B
TypeScript
37 lines
983 B
TypeScript
import {
|
|
Context, db, ForbiddenError, UserModel,
|
|
} from 'hydrooj';
|
|
|
|
interface IpLoginInfo {
|
|
_id: string;
|
|
uid: number;
|
|
}
|
|
|
|
declare module 'hydrooj' {
|
|
interface Collections {
|
|
iplogin: IpLoginInfo;
|
|
}
|
|
}
|
|
const coll = db.collection('iplogin');
|
|
|
|
function normalizeIp(ip: string) {
|
|
if (ip.startsWith('::ffff:')) return ip.slice(7);
|
|
return ip;
|
|
}
|
|
|
|
export function apply(ctx: Context) {
|
|
ctx.on('handler/init', async (that) => {
|
|
const iplogin = await coll.findOne({ _id: normalizeIp(that.request.ip) });
|
|
if (iplogin) {
|
|
that.user = await UserModel.getById(that.domain._id, iplogin.uid);
|
|
if (!that.user) {
|
|
that.user = await UserModel.getById(that.domain._id, 0);
|
|
throw new ForbiddenError(`User ${iplogin.uid}not found`);
|
|
}
|
|
that.session.ipLoggedIn = true;
|
|
that.session.uid = iplogin.uid;
|
|
that.session.user = that.user;
|
|
}
|
|
});
|
|
}
|