forked from mirror/misskey
37 lines
874 B
TypeScript
37 lines
874 B
TypeScript
|
import IPCIDR from 'ip-cidr';
|
||
|
|
import { Redis } from 'ioredis';
|
||
|
|
|
||
|
|
/**
|
||
|
|
* This should be same as {@link file://./../src/misc/get-ip-hash.ts}.
|
||
|
|
*/
|
||
|
|
function getIpHash(ip: string) {
|
||
|
|
const prefix = IPCIDR.createAddress(ip).mask(64);
|
||
|
|
return `ip-${BigInt('0b' + prefix).toString(36)}`;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* This prevents hitting rate limit when login.
|
||
|
|
*/
|
||
|
|
export async function purgeLimit(host: string, client: Redis) {
|
||
|
|
const ipHash = getIpHash(process.env.TESTER_IP_ADDRESS!);
|
||
|
|
const key = `${host}:limit:${ipHash}:signin`;
|
||
|
|
const res = await client.zrange(key, 0, -1);
|
||
|
|
if (res.length !== 0) {
|
||
|
|
console.log(`${key} - ${JSON.stringify(res)}`);
|
||
|
|
await client.del(key);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
console.log('Daemon started running');
|
||
|
|
|
||
|
|
{
|
||
|
|
const redisClient = new Redis({
|
||
|
|
host: 'redis.local',
|
||
|
|
});
|
||
|
|
|
||
|
|
setInterval(() => {
|
||
|
|
purgeLimit('a.local', redisClient);
|
||
|
|
purgeLimit('b.local', redisClient);
|
||
|
|
}, 1000);
|
||
|
|
}
|