Huang Xin
67 lines
1.6 KiB
TypeScript
67 lines
1.6 KiB
TypeScript
import { createSupabaseAdminClient } from '@/utils/supabase';
|
|
|
|
export const USAGE_TYPES = {
|
|
TRANSLATION_CHARS: 'translation_chars',
|
|
} as const;
|
|
|
|
export const QUOTA_TYPES = {
|
|
DAILY: 'daily',
|
|
MONTHLY: 'monthly',
|
|
YEARLY: 'yearly',
|
|
} as const;
|
|
|
|
export class UsageStatsManager {
|
|
static async trackUsage(
|
|
userId: string,
|
|
usageType: string,
|
|
increment: number = 1,
|
|
metadata: Record<string, string | number> = {},
|
|
): Promise<number> {
|
|
try {
|
|
const supabase = createSupabaseAdminClient();
|
|
const { data, error } = await supabase.rpc('increment_daily_usage', {
|
|
p_user_id: userId,
|
|
p_usage_type: usageType,
|
|
p_usage_date: new Date().toISOString().split('T')[0],
|
|
p_increment: increment,
|
|
p_metadata: metadata,
|
|
});
|
|
|
|
if (error) {
|
|
console.error('Usage tracking error:', error);
|
|
return 0;
|
|
}
|
|
|
|
return data || 0;
|
|
} catch (error) {
|
|
console.error('Usage tracking failed:', error);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
static async getCurrentUsage(
|
|
userId: string,
|
|
usageType: string,
|
|
period: 'daily' | 'monthly' = 'daily',
|
|
): Promise<number> {
|
|
try {
|
|
const supabase = createSupabaseAdminClient();
|
|
const { data, error } = await supabase.rpc('get_current_usage', {
|
|
p_user_id: userId,
|
|
p_usage_type: usageType,
|
|
p_period: period,
|
|
});
|
|
|
|
if (error) {
|
|
console.error('Get current usage error:', error);
|
|
return 0;
|
|
}
|
|
|
|
return data || 0;
|
|
} catch (error) {
|
|
console.error('Get current usage failed:', error);
|
|
return 0;
|
|
}
|
|
}
|
|
}
|