|
1 | 1 | #![no_std] |
2 | 2 |
|
3 | | -use soroban_sdk::{contracttype, Address, String, Symbol, Vec}; |
| 3 | +use soroban_sdk::{contracttype, Address, BytesN, String, Symbol, Vec}; |
4 | 4 |
|
5 | 5 | /// Billing interval in seconds. |
6 | 6 | #[contracttype] |
@@ -745,3 +745,137 @@ pub struct PriceBounds { |
745 | 745 | /// Quote currency symbol used for price lookup (e.g. "USD"). |
746 | 746 | pub quote: Symbol, |
747 | 747 | } |
| 748 | + |
| 749 | +pub type ApiKeyId = u64; |
| 750 | + |
| 751 | +#[contracttype] |
| 752 | +#[derive(Clone, Debug, PartialEq)] |
| 753 | +pub enum ApiKeyStatus { |
| 754 | + Active, |
| 755 | + Revoked, |
| 756 | + Expired, |
| 757 | +} |
| 758 | + |
| 759 | +#[contracttype] |
| 760 | +#[derive(Clone, Debug, PartialEq)] |
| 761 | +pub enum UsageTier { |
| 762 | + Free, |
| 763 | + Basic, |
| 764 | + Pro, |
| 765 | + Enterprise, |
| 766 | +} |
| 767 | + |
| 768 | +impl UsageTier { |
| 769 | + pub fn default_rate_limit(&self) -> RateLimitConfig { |
| 770 | + match self { |
| 771 | + UsageTier::Free => RateLimitConfig { |
| 772 | + requests_per_minute: 100, |
| 773 | + requests_per_hour: 1_000, |
| 774 | + requests_per_day: 10_000, |
| 775 | + burst_limit: 10, |
| 776 | + }, |
| 777 | + UsageTier::Basic => RateLimitConfig { |
| 778 | + requests_per_minute: 1_000, |
| 779 | + requests_per_hour: 10_000, |
| 780 | + requests_per_day: 100_000, |
| 781 | + burst_limit: 50, |
| 782 | + }, |
| 783 | + UsageTier::Pro => RateLimitConfig { |
| 784 | + requests_per_minute: 10_000, |
| 785 | + requests_per_hour: 100_000, |
| 786 | + requests_per_day: 1_000_000, |
| 787 | + burst_limit: 200, |
| 788 | + }, |
| 789 | + UsageTier::Enterprise => RateLimitConfig { |
| 790 | + requests_per_minute: 100_000, |
| 791 | + requests_per_hour: 1_000_000, |
| 792 | + requests_per_day: 10_000_000, |
| 793 | + burst_limit: 1000, |
| 794 | + }, |
| 795 | + } |
| 796 | + } |
| 797 | + |
| 798 | + pub fn price_per_thousand(&self) -> i128 { |
| 799 | + match self { |
| 800 | + UsageTier::Free => 0, |
| 801 | + UsageTier::Basic => 1, // 0.001 per 1k requests (in stroops) |
| 802 | + UsageTier::Pro => 5, // 0.005 per 1k |
| 803 | + UsageTier::Enterprise => 10, // 0.01 per 1k |
| 804 | + } |
| 805 | + } |
| 806 | +} |
| 807 | + |
| 808 | +#[contracttype] |
| 809 | +#[derive(Clone, Debug, PartialEq)] |
| 810 | +pub struct RateLimitConfig { |
| 811 | + pub requests_per_minute: u32, |
| 812 | + pub requests_per_hour: u32, |
| 813 | + pub requests_per_day: u32, |
| 814 | + pub burst_limit: u32, |
| 815 | +} |
| 816 | + |
| 817 | +#[contracttype] |
| 818 | +#[derive(Clone, Debug, PartialEq)] |
| 819 | +pub struct ApiKeyConfig { |
| 820 | + pub name: String, |
| 821 | + pub rate_limit: RateLimitConfig, |
| 822 | + pub usage_tier: UsageTier, |
| 823 | + pub expires_at: u64, |
| 824 | +} |
| 825 | + |
| 826 | +#[contracttype] |
| 827 | +#[derive(Clone, Debug, PartialEq)] |
| 828 | +pub struct ApiKey { |
| 829 | + pub id: ApiKeyId, |
| 830 | + pub owner: Address, |
| 831 | + pub key_hash: BytesN<32>, |
| 832 | + pub name: String, |
| 833 | + pub rate_limit: RateLimitConfig, |
| 834 | + pub usage_tier: UsageTier, |
| 835 | + pub status: ApiKeyStatus, |
| 836 | + pub created_at: u64, |
| 837 | + pub expires_at: u64, |
| 838 | + pub last_used_at: u64, |
| 839 | + pub revoked_at: u64, |
| 840 | +} |
| 841 | + |
| 842 | +#[contracttype] |
| 843 | +#[derive(Clone, Debug, PartialEq)] |
| 844 | +pub struct RateLimitWindow { |
| 845 | + pub window_start: u64, |
| 846 | + pub count: u32, |
| 847 | +} |
| 848 | + |
| 849 | +#[contracttype] |
| 850 | +#[derive(Clone, Debug, PartialEq)] |
| 851 | +pub struct ApiUsageRecord { |
| 852 | + pub window_start: u64, |
| 853 | + pub count: u32, |
| 854 | +} |
| 855 | + |
| 856 | +#[contracttype] |
| 857 | +#[derive(Clone, Debug, PartialEq)] |
| 858 | +pub struct RateLimitStatus { |
| 859 | + pub is_allowed: bool, |
| 860 | + pub remaining: u32, |
| 861 | + pub reset_at: u64, |
| 862 | + pub retry_after: u64, |
| 863 | +} |
| 864 | + |
| 865 | +#[contracttype] |
| 866 | +#[derive(Clone, Debug, PartialEq)] |
| 867 | +pub struct UsageReport { |
| 868 | + pub key_id: ApiKeyId, |
| 869 | + pub period: TimeRange, |
| 870 | + pub total_requests: u32, |
| 871 | +} |
| 872 | + |
| 873 | +#[contracttype] |
| 874 | +#[derive(Clone, Debug, PartialEq)] |
| 875 | +pub struct ApiKeyAuditEntry { |
| 876 | + pub id: u64, |
| 877 | + pub key_id: ApiKeyId, |
| 878 | + pub action: String, |
| 879 | + pub changed_by: Address, |
| 880 | + pub timestamp: u64, |
| 881 | +} |
0 commit comments