Implement ads backend

Author: jahooma

cli/src/commands/ads.ts

@@ -1,5 +1,4 @@
 import { saveSettings, loadSettings } from '../utils/settings'
-import { getCliEnv } from '../utils/env'
 import { getSystemMessage } from '../utils/message-history'
 import { logger } from '../utils/logger'
 
@@ -8,24 +7,14 @@ import type { ChatMessage } from '../types/chat'
 export const handleAdsEnable = (): {
   postUserMessage: (messages: ChatMessage[]) => ChatMessage[]
 } => {
-  const apiKey = getCliEnv().GRAVITY_API_KEY
-  logger.info({ hasApiKey: !!apiKey }, '[gravity] Enabling ads')
+  logger.info('[gravity] Enabling ads')
 
   saveSettings({ adsEnabled: true })
 
-  if (!apiKey) {
-    return {
-      postUserMessage: (messages) => [
-        ...messages,
-        getSystemMessage('Ads enabled, but GRAVITY_API_KEY is not set. Set the environment variable to see ads.'),
-      ],
-    }
-  }
-
   return {
     postUserMessage: (messages) => [
       ...messages,
-      getSystemMessage('Ads enabled. You will see contextual ads above the input.'),
+      getSystemMessage('Ads enabled. You will see contextual ads above the input and earn credits from impressions.'),
     ],
   }
 }

cli/src/components/ad-banner.tsx

@@ -6,7 +6,7 @@ import { useTerminalDimensions } from '../hooks/use-terminal-dimensions'
 import { useTheme } from '../hooks/use-theme'
 import { logger } from '../utils/logger'
 
-import type { AdResponse } from '@gravity-ai/api'
+import type { AdResponse } from '../hooks/use-gravity-ad'
 
 interface AdBannerProps {
   ad: AdResponse
@@ -40,12 +40,10 @@ export const AdBanner: React.FC<AdBannerProps> = ({ ad }) => {
     }
   }, [ad.clickUrl])
 
-  // Use 'url' field for display domain (the actual destination), fallback to clickUrl
-  const displayUrl = (ad as { url?: string }).url || ad.clickUrl
-  const domain = displayUrl ? extractDomain(displayUrl) : ''
-  const title = (ad as { title?: string }).title
-  // Use title as CTA, or fallback to "Learn more" if there's a clickUrl
-  const ctaText = title || (ad.clickUrl ? 'Learn more' : '')
+  // Use 'url' field for display domain (the actual destination)
+  const domain = extractDomain(ad.url)
+  // Use title as CTA
+  const ctaText = ad.title
 
   // Calculate available width for ad text
   // Account for: padding (2), "Ad" label with space (3)

cli/src/components/usage-banner.tsx

@@ -43,7 +43,7 @@ export const UsageBanner = ({ showTime }: { showTime: number }) => {
     type: 'usage-response'
     usage: number
     remainingBalance: number | null
-    balanceBreakdown?: { free: number; paid: number }
+    balanceBreakdown?: { free: number; paid: number; ad?: number }
     next_quota_reset: string | null
   }>({
     queryKey: usageQueryKeys.current(),
@@ -83,6 +83,7 @@ export const UsageBanner = ({ showTime }: { showTime: number }) => {
         sessionCreditsUsed,
         remainingBalance: activeData.remainingBalance,
         next_quota_reset: activeData.next_quota_reset,
+        adCredits: activeData.balanceBreakdown?.ad,
       })
 
   return (

cli/src/hooks/use-gravity-ad.ts

@@ -1,12 +1,11 @@
-import { Client } from '@gravity-ai/api'
+import { WEBSITE_URL } from '@codebuff/sdk'
 import { useCallback, useEffect, useRef, useState } from 'react'
 
 import { getAdsEnabled } from '../commands/ads'
 import { useChatStore } from '../state/chat-store'
-import { getCliEnv } from '../utils/env'
+import { getAuthToken } from '../utils/auth'
 import { logger } from '../utils/logger'
 
-import type { AdResponse } from '@gravity-ai/api'
 import type { Message } from '@codebuff/common/types/messages/codebuff-message'
 
 const MAX_MESSAGES_FOR_AD = 100
@@ -16,6 +15,17 @@ const MAX_ADS_AFTER_ACTIVITY = 3 // Show up to 3 ads after last activity, then s
 
 type AdMessage = { role: 'user' | 'assistant'; content: string }
 
+// Ad response type (matches Gravity API response)
+export type AdResponse = {
+  adText: string
+  title: string
+  url: string
+  favicon: string
+  clickUrl: string
+  impUrl: string
+  payout: number
+}
+
 /**
  * Extract text content from a Message's content array
  */
@@ -71,7 +81,6 @@ export type GravityAdState = {
 export const useGravityAd = (): GravityAdState => {
   const [ad, setAd] = useState<AdResponse | null>(null)
   const [isLoading, setIsLoading] = useState(false)
-  const clientRef = useRef<Client | null>(null)
   const impressionFiredRef = useRef<Set<string>>(new Set())
 
   // Pre-fetched next ad ready to display
@@ -93,29 +102,42 @@ export const useGravityAd = (): GravityAdState => {
   // Get runState from chat store
   const runState = useChatStore((state) => state.runState)
 
-  // Initialize client on mount
-  useEffect(() => {
-    const apiKey = getCliEnv().GRAVITY_API_KEY
-    logger.info(
-      { hasApiKey: !!apiKey, adsEnabled: getAdsEnabled() },
-      '[gravity] Initializing Gravity ad client',
-    )
-    if (apiKey) {
-      clientRef.current = new Client(apiKey)
-      logger.info('[gravity] Gravity client initialized successfully')
-    } else {
-      logger.warn('[gravity] No GRAVITY_API_KEY found in environment')
-    }
-  }, [])
-
-  // Fire impression when ad changes
+  // Fire impression via web API when ad changes (grants credits)
   useEffect(() => {
     if (ad?.impUrl && !impressionFiredRef.current.has(ad.impUrl)) {
       impressionFiredRef.current.add(ad.impUrl)
-      logger.info({ impUrl: ad.impUrl }, '[gravity] Firing ad impression')
-      fetch(ad.impUrl).catch((err) => {
-        logger.debug({ err }, '[gravity] Failed to fire ad impression')
+      logger.info({ impUrl: ad.impUrl, payout: ad.payout }, '[gravity] Recording ad impression')
+      
+      const authToken = getAuthToken()
+      if (!authToken) {
+        logger.warn('[gravity] No auth token, skipping impression recording')
+        return
+      }
+      
+      // Call our web API to fire impression and grant credits
+      fetch(`${WEBSITE_URL}/api/v1/ads/impression`, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          'Authorization': `Bearer ${authToken}`,
+        },
+        body: JSON.stringify({
+          impUrl: ad.impUrl,
+          payout: ad.payout,
+        }),
       })
+        .then((res) => res.json())
+        .then((data) => {
+          if (data.creditsGranted > 0) {
+            logger.info(
+              { creditsGranted: data.creditsGranted },
+              '[gravity] Ad impression credits granted',
+            )
+          }
+        })
+        .catch((err) => {
+          logger.debug({ err }, '[gravity] Failed to record ad impression')
+        })
     }
   }, [ad])
 
@@ -131,10 +153,15 @@ export const useGravityAd = (): GravityAdState => {
     }
   }, [])
 
-  // Fetch an ad and return it (for pre-fetching)
+  // Fetch an ad via web API and return it (for pre-fetching)
   const fetchAdAsync = useCallback(async (): Promise<AdResponse | null> => {
-    const client = clientRef.current
-    if (!client || !getAdsEnabled()) return null
+    if (!getAdsEnabled()) return null
+    
+    const authToken = getAuthToken()
+    if (!authToken) {
+      logger.warn('[gravity] No auth token available')
+      return null
+    }
 
     const currentRunState = useChatStore.getState().runState
     const messageHistory =
@@ -143,22 +170,38 @@ export const useGravityAd = (): GravityAdState => {
 
     if (adMessages.length === 0) return null
 
-    logger.info('[gravity] Fetching ad from Gravity API')
+    logger.info('[gravity] Fetching ad from web API')
 
     try {
-      const response = await client.getAd({ messages: adMessages })
+      const response = await fetch(`${WEBSITE_URL}/api/v1/ads`, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          'Authorization': `Bearer ${authToken}`,
+        },
+        body: JSON.stringify({ messages: adMessages }),
+      })
+      
+      if (!response.ok) {
+        logger.warn({ status: response.status }, '[gravity] Web API returned error')
+        return null
+      }
+      
+      const data = await response.json()
+      const ad = data.ad as AdResponse | null
+      
       logger.info(
         {
-          hasAd: !!response,
-          adText: response?.adText,
-          title: (response as { title?: string })?.title,
-          clickUrl: response?.clickUrl,
-          impUrl: response?.impUrl,
-          payout: response?.payout,
+          hasAd: !!ad,
+          adText: ad?.adText,
+          title: ad?.title,
+          clickUrl: ad?.clickUrl,
+          impUrl: ad?.impUrl,
+          payout: ad?.payout,
         },
         '[gravity] Received ad response',
       )
-      return response
+      return ad
     } catch (err) {
       logger.error({ err }, '[gravity] Failed to fetch ad')
       return null
@@ -231,9 +274,9 @@ export const useGravityAd = (): GravityAdState => {
       runState?.sessionState?.mainAgentState?.messageHistory ?? []
     const hasMessages = messageHistory.length > 0
     const adsEnabled = getAdsEnabled()
-    const hasClient = !!clientRef.current
+    const hasAuth = !!getAuthToken()
 
-    if (hasMessages && adsEnabled && hasClient && !isStartedRef.current) {
+    if (hasMessages && adsEnabled && hasAuth && !isStartedRef.current) {
       logger.info('[gravity] Starting ad rotation')
       isStartedRef.current = true
       setIsLoading(true)

cli/src/hooks/use-usage-query.ts

@@ -20,6 +20,9 @@ interface UsageResponse {
   balanceBreakdown?: {
     free: number
     paid: number
+    ad?: number
+    referral?: number
+    admin?: number
   }
   next_quota_reset: string | null
   autoTopupEnabled?: boolean

cli/src/types/env.ts

@@ -52,9 +52,6 @@ export type CliEnv = BaseEnv & {
   OPEN_TUI_THEME?: string
   OPENTUI_THEME?: string
 
-  // Gravity AI Ads
-  GRAVITY_API_KEY?: string
-
   // Codebuff CLI-specific (set during binary build)
   CODEBUFF_IS_BINARY?: string
   CODEBUFF_CLI_VERSION?: string

cli/src/utils/env.ts

@@ -52,9 +52,6 @@ export const getCliEnv = (): CliEnv => ({
   OPEN_TUI_THEME: process.env.OPEN_TUI_THEME,
   OPENTUI_THEME: process.env.OPENTUI_THEME,
 
-  // Gravity AI Ads
-  GRAVITY_API_KEY: process.env.GRAVITY_API_KEY,
-
   // Binary build configuration
   CODEBUFF_IS_BINARY: process.env.CODEBUFF_IS_BINARY,
   CODEBUFF_CLI_VERSION: process.env.CODEBUFF_CLI_VERSION,

cli/src/utils/usage-banner-state.ts

@@ -66,6 +66,8 @@ export interface UsageBannerTextOptions {
   sessionCreditsUsed: number
   remainingBalance: number | null
   next_quota_reset: string | null
+  /** Ad impression credits earned */
+  adCredits?: number
   /** For testing purposes, allows overriding "today" */
   today?: Date
 }
@@ -87,6 +89,7 @@ export function generateUsageBannerText(
     sessionCreditsUsed,
     remainingBalance,
     next_quota_reset,
+    adCredits,
     today = new Date(),
   } = options
 
@@ -96,6 +99,11 @@ export function generateUsageBannerText(
     text += `. Credits remaining: ${remainingBalance.toLocaleString()}`
   }
 
+  // Show ad credits earned if any
+  if (adCredits && adCredits > 0) {
+    text += ` (${adCredits.toLocaleString()} from ads)`
+  }
+
   if (next_quota_reset) {
     const resetDate = new Date(next_quota_reset)
     const isToday = resetDate.toDateString() === today.toDateString()

common/src/constants/grant-priorities.ts

@@ -2,6 +2,7 @@ import type { GrantType } from '@codebuff/common/types/grant'
 
 export const GRANT_PRIORITIES: Record<GrantType, number> = {
   free: 20,
+  ad: 30, // Ad credits consumed after free, before referral
   referral: 40,
   admin: 60,
   organization: 70,

common/src/types/grant.ts

@@ -4,11 +4,13 @@ export type GrantType =
   | 'purchase'
   | 'admin'
   | 'organization'
+  | 'ad' // Credits earned from ads (impressions, clicks, acquisitions, etc.)
 
 export const GrantTypeValues = [
   'free',
   'referral',
   'purchase',
   'admin',
   'organization',
+  'ad',
 ] as const