- Task 1: Admin endpoints for wallet management (balance, ledger, adjust, reconcile) - Task 2: AI Credits admin UI (pricing.tsx, credit.tsx) - Task 3: Ollama security (NetworkPolicy, prompt validation, audit logging) - Task 4: LiteLLM integration (litellm.rs, migrated AI feature handlers) - Task 5: Refund architecture (ai_refunds table, endpoints) - Task 6: Coupons, promotions, referrals (order creation with coupon validation) - Task 7: Subscription lifecycle (plan upgrades/downgrades/cancellations, cron jobs) - Task 8: Credit expiration enforcement (daily cron task) - Task 9: Token cost engine (ai_model_cost_config, margin view) - Task 10: Observability (metrics tables, aggregation function) New files: - apps/users/src/litellm.rs (LiteLLM client) - apps/users/src/ai_credits.rs (Charging primitives) - apps/users/src/ai_subscription.rs (Plan lifecycle) - apps/users/src/handlers/ai_credits.rs (Admin endpoints) - apps/payments/src/ai_credits.rs (Package purchase) - apps/payments/src/payu.rs (PayU integration) - apps/cron/src/tasks/ai_credits.rs (Cron jobs) - 9 SQL migrations for schema - crates/db/src/models/ai_credits.rs (Repository) - tests for ai_credits
160 lines
6.1 KiB
PL/PgSQL
160 lines
6.1 KiB
PL/PgSQL
-- Coupons, promotions, referral, and bonus credits (Task 6)
|
|
-- Implements coupon redemption, promotions, referral tracking
|
|
|
|
BEGIN;
|
|
|
|
-- Coupons table for discount codes
|
|
CREATE TABLE ai_coupons (
|
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
|
code VARCHAR(100) UNIQUE NOT NULL,
|
|
name VARCHAR(200) NOT NULL,
|
|
description TEXT,
|
|
|
|
-- Discount configuration
|
|
discount_type VARCHAR(20) NOT NULL CHECK (discount_type IN ('percentage', 'fixed_amount')),
|
|
discount_value NUMERIC(10, 2) NOT NULL CHECK (discount_value >= 0),
|
|
|
|
-- Applicability
|
|
applicable_package_ids UUID[] DEFAULT '{}',
|
|
applicable_plan_codes TEXT[] DEFAULT '{}',
|
|
min_purchase_amount NUMERIC(12, 2),
|
|
max_discount_amount NUMERIC(12, 2),
|
|
|
|
-- Limits
|
|
max_redemptions INT NOT NULL DEFAULT 1,
|
|
redemptions_used INT NOT NULL DEFAULT 0,
|
|
max_redemptions_per_user INT DEFAULT 1,
|
|
|
|
-- Validity
|
|
valid_from TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
|
valid_until TIMESTAMPTZ,
|
|
|
|
-- Status
|
|
is_active BOOLEAN NOT NULL DEFAULT TRUE,
|
|
|
|
-- Audit
|
|
created_by UUID REFERENCES users(id),
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
|
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
|
);
|
|
|
|
-- Coupon redemptions (tracks who used which coupon)
|
|
CREATE TABLE ai_coupon_redemptions (
|
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
|
coupon_id UUID NOT NULL REFERENCES ai_coupons(id) ON DELETE CASCADE,
|
|
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
|
order_id UUID, -- Reference to ai_credit_orders
|
|
credits_purchased INT NOT NULL,
|
|
discount_applied NUMERIC(12, 2) NOT NULL,
|
|
final_amount_paid NUMERIC(12, 2) NOT NULL,
|
|
redeemed_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
|
|
|
UNIQUE(coupon_id, user_id)
|
|
);
|
|
|
|
-- Promotions table for automatic bonuses
|
|
CREATE TABLE ai_promotions (
|
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
|
name VARCHAR(200) NOT NULL,
|
|
description TEXT,
|
|
|
|
-- Trigger conditions
|
|
trigger_type VARCHAR(50) NOT NULL CHECK (trigger_type IN ('signup', 'first_purchase', 'referral', 'milestone', 'manual')),
|
|
trigger_condition JSONB DEFAULT '{}', -- e.g., {"min_purchase": 100, "referrer_bonus": 50}
|
|
|
|
-- Reward
|
|
bonus_credits INT NOT NULL CHECK (bonus_credits > 0),
|
|
bonus_expires_after_days INT, -- NULL means no expiry
|
|
|
|
-- Applicability
|
|
applicable_plan_codes TEXT[] DEFAULT '{}',
|
|
applicable_user_types TEXT[] DEFAULT '{}', -- e.g., ['NEW_USER', 'REFERRER']
|
|
|
|
-- Limits
|
|
max_grants INT,
|
|
grants_used INT NOT NULL DEFAULT 0,
|
|
max_grants_per_user INT DEFAULT 1,
|
|
|
|
-- Validity
|
|
valid_from TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
|
valid_until TIMESTAMPTZ,
|
|
|
|
-- Status
|
|
is_active BOOLEAN NOT NULL DEFAULT TRUE,
|
|
|
|
-- Audit
|
|
created_by UUID REFERENCES users(id),
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
|
updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
|
);
|
|
|
|
-- Promotion grants (tracks who received which promotion)
|
|
CREATE TABLE ai_promotion_grants (
|
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
|
promotion_id UUID NOT NULL REFERENCES ai_promotions(id) ON DELETE CASCADE,
|
|
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
|
ledger_entry_id UUID REFERENCES ai_credit_ledger(id), -- Link to the bonus credit ledger entry
|
|
bonus_credits INT NOT NULL,
|
|
expires_at TIMESTAMPTZ,
|
|
granted_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
|
|
|
|
UNIQUE(promotion_id, user_id)
|
|
);
|
|
|
|
-- Bonus credits ledger reference
|
|
-- This extends ai_credit_ledger with source tracking for bonus credits
|
|
CREATE TABLE ai_bonus_credits (
|
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
|
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
|
source_type VARCHAR(30) NOT NULL CHECK (source_type IN ('promotion', 'coupon_bonus', 'referral', 'manual', 'signup_bonus')),
|
|
source_reference_id UUID, -- References ai_promotions, ai_coupon_redemptions, etc.
|
|
credits INT NOT NULL CHECK (credits > 0),
|
|
credits_used INT NOT NULL DEFAULT 0 CHECK (credits_used >= 0 AND credits_used <= credits),
|
|
expires_at TIMESTAMPTZ,
|
|
ledger_entry_id UUID REFERENCES ai_credit_ledger(id),
|
|
is_active BOOLEAN NOT NULL DEFAULT TRUE,
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
|
);
|
|
|
|
-- Referral tracking
|
|
CREATE TABLE ai_referrals (
|
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
|
referrer_user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
|
referred_user_id UUID REFERENCES users(id) ON DELETE SET NULL,
|
|
referral_code VARCHAR(100) NOT NULL UNIQUE,
|
|
|
|
-- Status
|
|
status VARCHAR(20) NOT NULL DEFAULT 'pending' CHECK (status IN ('pending', 'converted', 'expired')),
|
|
|
|
-- Rewards (granted when referred user completes first purchase)
|
|
referrer_bonus_credits INT,
|
|
referred_bonus_credits INT,
|
|
|
|
-- Conversion tracking
|
|
converted_at TIMESTAMPTZ,
|
|
first_purchase_order_id UUID REFERENCES ai_credit_orders(id),
|
|
|
|
-- Expiry
|
|
expires_at TIMESTAMPTZ NOT NULL DEFAULT (NOW() + INTERVAL '30 days'),
|
|
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
|
|
);
|
|
|
|
-- Indexes for performance
|
|
CREATE INDEX idx_ai_coupons_code ON ai_coupons(code) WHERE is_active = TRUE;
|
|
CREATE INDEX idx_ai_coupons_valid ON ai_coupons(valid_from, valid_until) WHERE is_active = TRUE;
|
|
CREATE INDEX idx_ai_coupon_redemptions_user ON ai_coupon_redemptions(user_id);
|
|
CREATE INDEX idx_ai_coupon_redemptions_coupon ON ai_coupon_redemptions(coupon_id);
|
|
|
|
CREATE INDEX idx_ai_promotions_trigger ON ai_promotions(trigger_type) WHERE is_active = TRUE;
|
|
CREATE INDEX idx_ai_promotions_valid ON ai_promotions(valid_from, valid_until) WHERE is_active = TRUE;
|
|
CREATE INDEX idx_ai_promotion_grants_user ON ai_promotion_grants(user_id);
|
|
|
|
CREATE INDEX idx_ai_bonus_credits_user ON ai_bonus_credits(user_id) WHERE is_active = TRUE;
|
|
CREATE INDEX idx_ai_bonus_credits_expires ON ai_bonus_credits(expires_at) WHERE expires_at IS NOT NULL AND is_active = TRUE;
|
|
|
|
CREATE INDEX idx_ai_referrals_referrer ON ai_referrals(referrer_user_id);
|
|
CREATE INDEX idx_ai_referrals_code ON ai_referrals(referral_code);
|
|
CREATE INDEX idx_ai_referrals_referred ON ai_referrals(referred_user_id) WHERE referred_user_id IS NOT NULL;
|
|
|
|
COMMIT;
|