nxtgauge-backend-rust/crates/db/migrations/20260706300000_ai_coupons_promotions.up.sql
Ashwin Kumar Sivakumar 3c0f45f1ee feat: Complete Ask Ash AI Credits implementation (Tasks 1-10)
- 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
2026-07-06 01:47:54 +05:30

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;