75 lines
1.8 KiB
Perl
75 lines
1.8 KiB
Perl
|
#!/usr/bin/perl
|
||
|
|
#
|
||
|
|
# DW::API::RateLimit
|
||
|
|
#
|
||
|
|
# API-specific rate limiting wrapper that uses DW::RateLimit
|
||
|
|
#
|
||
|
|
# Authors:
|
||
|
|
# Mark Smith <mark@dreamwidth.org>
|
||
|
|
#
|
||
|
|
# Copyright (c) 2025 by Dreamwidth Studios, LLC.
|
||
|
|
#
|
||
|
|
# This program is free software; you may redistribute it and/or modify it under
|
||
|
|
# the same terms as Perl itself. For a copy of the license, please reference
|
||
|
|
# 'perldoc perlartistic' or 'perldoc perlgpl'.
|
||
|
|
#
|
||
|
|
|
||
|
|
package DW::API::RateLimit;
|
||
|
|
|
||
|
|
use strict;
|
||
|
|
use warnings;
|
||
|
|
use JSON;
|
||
|
|
|
||
|
|
use DW::RateLimit;
|
||
|
|
use DW::Request;
|
||
|
|
|
||
|
|
# Wrap a function with rate limiting
|
||
|
|
sub wrap {
|
||
|
|
my ( $class, $code, %opts ) = @_;
|
||
|
|
|
||
|
|
# Validate required parameters
|
||
|
|
return $code unless $opts{rate};
|
||
|
|
|
||
|
|
# Create a rate limit object
|
||
|
|
my $limit = DW::RateLimit->get(
|
||
|
|
"api:" . ( $opts{name} || "unknown" ),
|
||
|
|
rate => $opts{rate},
|
||
|
|
mode => $opts{mode}
|
||
|
|
);
|
||
|
|
|
||
|
|
# Return a wrapped function that checks rate limits before executing
|
||
|
|
return sub {
|
||
|
|
my ( $self, $args ) = @_;
|
||
|
|
|
||
|
|
# Get the request object
|
||
|
|
my $r = DW::Request->get;
|
||
|
|
return $code->( $self, $args ) unless $r;
|
||
|
|
|
||
|
|
# Check rate limit
|
||
|
|
my $result = $limit->check(
|
||
|
|
userid => $args->{user} ? $args->{user}->userid : undef,
|
||
|
|
ip => $r->connection->remote_ip
|
||
|
|
);
|
||
|
|
|
||
|
|
if ( $result->{exceeded} ) {
|
||
|
|
$r->status(429);
|
||
|
|
$r->headers_out->{'Retry-After'} = $result->{time_remaining};
|
||
|
|
$r->print(
|
||
|
|
to_json(
|
||
|
|
{
|
||
|
|
success => 0,
|
||
|
|
error => "Rate limit exceeded",
|
||
|
|
retry_after => $result->{time_remaining}
|
||
|
|
}
|
||
|
|
)
|
||
|
|
);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
# Execute the original function
|
||
|
|
return $code->( $self, $args );
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
1;
|