File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed
playground/server/tasks/shield
src/runtime/server/middleware Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff 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 }
Original file line number Diff line number Diff 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 ) ;
You can’t perform that action at this time.
0 commit comments