cherry-studio/src/main/utils/systemInfo.ts

import { preferenceService } from '@data/PreferenceService'
import { app } from 'electron'
import macosRelease from 'macos-release'
import os from 'os'
import { v4 as uuidv4 } from 'uuid'

/**
 * System information interface
 */
export interface SystemInfo {
  platform: NodeJS.Platform
  arch: string
  osRelease: string
  appVersion: string
  osString: string
  archString: string
}

/**
 * Get basic system constants for quick access
 * @returns {Object} Basic system constants
 */
export function getSystemConstants() {
  return {
    platform: process.platform,
    arch: process.arch,
    osRelease: os.release(),
    appVersion: app.getVersion()
  }
}

/**
 * Get system information
 * @returns {SystemInfo} Complete system information object
 */
export function getSystemInfo(): SystemInfo {
  const platform = process.platform
  const arch = process.arch
  const osRelease = os.release()
  const appVersion = app.getVersion()

  let osString = ''

  switch (platform) {
    case 'win32': {
      // Get Windows version
      const parts = osRelease.split('.')
      const buildNumber = parseInt(parts[2], 10)
      osString = buildNumber >= 22000 ? 'Windows 11' : 'Windows 10'
      break
    }
    case 'darwin': {
      // macOS version handling using macos-release for better accuracy
      try {
        const macVersionInfo = macosRelease()
        const versionString = macVersionInfo.version.replace(/\./g, '_') // 15.0.0 -> 15_0_0
        osString = arch === 'arm64' ? `Mac OS X ${versionString}` : `Intel Mac OS X ${versionString}` // Mac OS X 15_0_0
      } catch (error) {
        // Fallback to original logic if macos-release fails
        const macVersion = osRelease.split('.').slice(0, 2).join('_')
        osString = arch === 'arm64' ? `Mac OS X ${macVersion}` : `Intel Mac OS X ${macVersion}`
      }
      break
    }
    case 'linux': {
      osString = `Linux ${arch}`
      break
    }
    default: {
      osString = `${platform} ${arch}`
    }
  }

  const archString = arch === 'x64' ? 'x86_64' : arch === 'arm64' ? 'arm64' : arch

  return {
    platform,
    arch,
    osRelease,
    appVersion,
    osString,
    archString
  }
}

/**
 * Generate User-Agent string based on user system data
 * @returns {string} Dynamically generated User-Agent string
 */
export function generateUserAgent(): string {
  const systemInfo = getSystemInfo()

  return `Mozilla/5.0 (${systemInfo.osString}; ${systemInfo.archString}) AppleWebKit/537.36 (KHTML, like Gecko) CherryStudio/${systemInfo.appVersion} Chrome/124.0.0.0 Safari/537.36`
}

/**
 * Get or generate a unique client ID
 * @returns {string} Client ID
 */
export function getClientId(): string {
  let clientId = preferenceService.get('app.user.id')

  // If it's the placeholder value, generate a new UUID
  if (!clientId || clientId.length === 0) {
    clientId = uuidv4()
    preferenceService.set('app.user.id', clientId)
  }

  return clientId
}