Skip to content

Commit 88780b9

Browse files
committed
fix types errors
1 parent b4e666e commit 88780b9

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

playground/server/tasks/shield/clean.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,9 @@ export default defineTask({
88
const shieldStorage = useStorage("shield");
99

1010
const keys = await shieldStorage.getKeys();
11+
1112
keys.forEach(async (key) => {
12-
const rateLimit = await shieldStorage.getItem(key);
13+
const rateLimit = (await shieldStorage.getItem(key)) as RateLimit;
1314
if (isBanExpired(rateLimit)) {
1415
await shieldStorage.removeItem(key);
1516
}

src/runtime/server/middleware/shield.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ export default defineEventHandler(async (event) => {
1010
}
1111

1212
const shieldStorage = useStorage("shield");
13-
const requestIP = getRequestIP(event, { xForwardedFor: true });
13+
const requestIP = getRequestIP(event, { xForwardedFor: true }) || "unKnownIP";
1414

1515
if (!(await shieldStorage.hasItem(`ip:${requestIP}`))) {
1616
return await shieldStorage.setItem(`ip:${requestIP}`, {
@@ -19,7 +19,7 @@ export default defineEventHandler(async (event) => {
1919
});
2020
}
2121

22-
const req = await shieldStorage.getItem(`ip:${requestIP}`);
22+
const req = (await shieldStorage.getItem(`ip:${requestIP}`)) as RateLimit;
2323
req.count++;
2424

2525
shieldLog(req, requestIP, event.node.req.url);

0 commit comments

Comments
 (0)