deep-stack/laconic-faucet
3
0
Fork 0
forked from cerc-io/laconic-faucet
Code Pull Requests Activity

Use a configurable period limit

Browse Source
This commit is contained in:
Prathamesh Musale 2024-07-24 18:33:33 +05:30
parent c5bf9793f7
commit e5a032cfa8
2 changed files with 34 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
environments/local.toml
View File

@ -8,6 +8,7 @@
[server] [server]
port = 3000 port = 3000
periodInSecs = 60
transferAmount = 1000000 transferAmount = 1000000
periodTransferLimit = 3000000 periodTransferLimit = 3000000
dbDir = "db" dbDir = "db"

40
src/index.ts
View File

@ -12,7 +12,7 @@ import KeyvSqlite from '@keyv/sqlite';
const CONFIG_PATH = 'environments/local.toml'; const CONFIG_PATH = 'environments/local.toml';
const FAUCET_DATA_FILE = 'faucet_data.sqlite'; const FAUCET_DATA_FILE = 'faucet_data.sqlite';
const FAUCET_DATA_TTL = 86400000; // 24 hrs const TTL_MULTIPLIER = 1.5; // 1.5 times the configured period
interface Config { interface Config {
registry: { registry: {
@ -25,12 +25,18 @@ interface Config {
}, },
server: { server: {
port: number port: number
periodInSecs: number
transferAmount: string transferAmount: string
periodTransferLimit: string periodTransferLimit: string
dbDir: string dbDir: string
} }
} }
interface FaucetStoreValue {
periodStart: string
amountSent: string
}
async function main (): Promise<void> { async function main (): Promise<void> {
// Read and parse the configuration // Read and parse the configuration
const configFile = fs.readFileSync(CONFIG_PATH, 'utf-8'); const configFile = fs.readFileSync(CONFIG_PATH, 'utf-8');
@ -54,22 +60,40 @@ async function main (): Promise<void> {
return res.status(400).json({ error: 'invalid address' }); return res.status(400).json({ error: 'invalid address' });
} }
// Check rate limit const currentTimeInSeconds = Math.floor(Date.now() / 1000);
const now = Date.now(); let amountSentToAccount = BigInt(0);
const today = new Date(now).toISOString().split('T')[0]; let accountPeriodStart = currentTimeInSeconds;
const faucetStoreKey = `${accountAddress}:${today}`;
const amountSentToAddress = await faucetDataStore.get(faucetStoreKey) || '0';
if (BigInt(amountSentToAddress) + BigInt(config.server.transferAmount) > BigInt(config.server.periodTransferLimit)) { // Get accounts data from faucet data store
const accountFaucetData: FaucetStoreValue = await faucetDataStore.get(accountAddress);
// Check rate limit
if (accountFaucetData !== undefined) {
const elapsedTime = currentTimeInSeconds - Number(accountFaucetData.periodStart);
// Check if saved entry is still within configured period
// Else, create a fresh entry
if (elapsedTime < config.server.periodInSecs) {
amountSentToAccount = BigInt(accountFaucetData.amountSent);
accountPeriodStart = Number(accountFaucetData.periodStart);
// Check if limit has been reached
if (amountSentToAccount + BigInt(config.server.transferAmount) > BigInt(config.server.periodTransferLimit)) {
return res.status(429).json({ error: 'Limit exceeded' }); return res.status(429).json({ error: 'Limit exceeded' });
} }
}
}
try { try {
const txHash = await sendTokens(config, accountAddress, String(config.server.transferAmount)); const txHash = await sendTokens(config, accountAddress, String(config.server.transferAmount));
console.log(`Sent tokens to address: ${accountAddress}, txHash: ${txHash}`); console.log(`Sent tokens to address: ${accountAddress}, txHash: ${txHash}`);
// Update rate limit // Update rate limit
await faucetDataStore.set(faucetStoreKey, (BigInt(amountSentToAddress) + BigInt(config.server.transferAmount)).toString(), FAUCET_DATA_TTL); const updatedAccountFaucetData: FaucetStoreValue = {
periodStart: accountPeriodStart.toString(),
amountSent: amountSentToAccount.toString()
};
await faucetDataStore.set(accountAddress, updatedAccountFaucetData, config.server.periodInSecs * TTL_MULTIPLIER * 1000);
res.json({ success: true, txHash }); res.json({ success: true, txHash });
} catch (error) { } catch (error) {