Momento/memento-note/lib/config.ts

import prisma from './prisma'

// Environment variable fallbacks for system config keys
const ENV_FALLBACKS: Record<string, string> = {
  // AI providers
  AI_PROVIDER_TAGS: process.env.AI_PROVIDER_TAGS || '',
  AI_MODEL_TAGS: process.env.AI_MODEL_TAGS || '',
  AI_PROVIDER_EMBEDDING: process.env.AI_PROVIDER_EMBEDDING || '',
  AI_MODEL_EMBEDDING: process.env.AI_MODEL_EMBEDDING || '',
  AI_PROVIDER_CHAT: process.env.AI_PROVIDER_CHAT || '',
  AI_MODEL_CHAT: process.env.AI_MODEL_CHAT || '',
  // Ollama
  OLLAMA_BASE_URL: process.env.OLLAMA_BASE_URL || '',
  // OpenAI
  OPENAI_API_KEY: process.env.OPENAI_API_KEY || '',
  // Custom OpenAI
  CUSTOM_OPENAI_API_KEY: process.env.CUSTOM_OPENAI_API_KEY || '',
  CUSTOM_OPENAI_BASE_URL: process.env.CUSTOM_OPENAI_BASE_URL || '',
  // DeepSeek
  DEEPSEEK_API_KEY: process.env.DEEPSEEK_API_KEY || '',
  // OpenRouter
  OPENROUTER_API_KEY: process.env.OPENROUTER_API_KEY || '',
  // Mistral
  MISTRAL_API_KEY: process.env.MISTRAL_API_KEY || '',
  // Z.AI
  ZAI_API_KEY: process.env.ZAI_API_KEY || '',
  // LM Studio
  LMSTUDIO_BASE_URL: process.env.LMSTUDIO_BASE_URL || '',
  LMSTUDIO_API_KEY: process.env.LMSTUDIO_API_KEY || '',
  // Email
  EMAIL_PROVIDER: process.env.EMAIL_PROVIDER || (process.env.RESEND_API_KEY ? 'resend' : 'smtp'),
  RESEND_API_KEY: process.env.RESEND_API_KEY || '',
  SMTP_HOST: process.env.SMTP_HOST || '',
  SMTP_PORT: process.env.SMTP_PORT || '587',
  SMTP_USER: process.env.SMTP_USER || '',
  SMTP_PASS: process.env.SMTP_PASS || '',
  SMTP_FROM: process.env.SMTP_FROM || '',
  SMTP_SECURE: process.env.SMTP_SECURE || 'false',
  SMTP_IGNORE_CERT: process.env.SMTP_IGNORE_CERT || 'false',
  // Tools
  WEB_SEARCH_PROVIDER: process.env.WEB_SEARCH_PROVIDER || 'searxng',
  SEARXNG_URL: process.env.SEARXNG_URL || '',
  BRAVE_SEARCH_API_KEY: process.env.BRAVE_SEARCH_API_KEY || '',
  JINA_API_KEY: process.env.JINA_API_KEY || '',
  // Auth / misc
  ALLOW_REGISTRATION: process.env.ALLOW_REGISTRATION || '',
  NEXTAUTH_URL: process.env.NEXTAUTH_URL || '',
}

export async function getSystemConfig() {
  let dbConfig: Record<string, string> = {}
  try {
    const configs = await prisma.systemConfig.findMany()
    dbConfig = configs.reduce((acc, conf) => {
      acc[conf.key] = conf.value
      return acc
    }, {} as Record<string, string>)
  } catch (e) {
    console.error('Failed to load system config from DB:', e)
  }

  // Merge: DB values take precedence, env vars as fallback
  const merged = { ...ENV_FALLBACKS, ...dbConfig }
  return merged
}

/**
 * Get a config value with a default fallback
 */
export async function getConfigValue(key: string, defaultValue: string = ''): Promise<string> {
  const config = await getSystemConfig()
  return config[key] || defaultValue
}

/**
 * Get a numeric config value with a default fallback
 */
export async function getConfigNumber(key: string, defaultValue: number): Promise<number> {
  const value = await getConfigValue(key, String(defaultValue))
  const num = parseFloat(value)
  return isNaN(num) ? defaultValue : num
}

/**
 * Get a boolean config value with a default fallback
 */
export async function getConfigBoolean(key: string, defaultValue: boolean): Promise<boolean> {
  const value = await getConfigValue(key, String(defaultValue))
  return value === 'true'
}

/**
 * Search configuration defaults
 */
export const SEARCH_DEFAULTS = {
  SEMANTIC_THRESHOLD: 0.65,
  RRF_K_BASE: 20,
  RRF_K_ADAPTIVE: true,
  KEYWORD_BOOST_EXACT: 2.0,
  KEYWORD_BOOST_CONCEPTUAL: 0.7,
  SEMANTIC_BOOST_EXACT: 0.7,
  SEMANTIC_BOOST_CONCEPTUAL: 1.5,
  QUERY_EXPANSION_ENABLED: false,
  QUERY_EXPANSION_MAX_SYNONYMS: 3,
  DEBUG_MODE: false,
} as const