const nodeOsUtils = require('node-os-utils')
const getJanDataFolderPath = require('@janhq/core/node').getJanDataFolderPath
const path = require('path')
const { readFileSync } = require('fs')
const exec = require('child_process').exec

const NVIDIA_INFO_FILE = path.join(
  getJanDataFolderPath(),
  'settings',
  'settings.json'
)

const getResourcesInfo = () =>
  new Promise((resolve) => {
    nodeOsUtils.mem.used().then((ramUsedInfo) => {
      const totalMemory = ramUsedInfo.totalMemMb * 1024 * 1024
      const usedMemory = ramUsedInfo.usedMemMb * 1024 * 1024
      const response = {
        mem: {
          totalMemory,
          usedMemory,
        },
      }
      resolve(response)
    })
  })

const getCurrentLoad = () =>
  new Promise((resolve, reject) => {
    nodeOsUtils.cpu.usage().then((cpuPercentage) => {
      let data = {
        run_mode: 'cpu',
        gpus_in_use: [],
      }
      if (process.platform !== 'darwin') {
        data = JSON.parse(readFileSync(NVIDIA_INFO_FILE, 'utf-8'))
      }
      if (data.run_mode === 'gpu' && data.gpus_in_use.length > 0) {
        const gpuIds = data['gpus_in_use'].join(',')
        if (gpuIds !== '') {
          exec(
            `nvidia-smi --query-gpu=index,name,temperature.gpu,utilization.gpu,memory.total,memory.free,utilization.memory --format=csv,noheader,nounits --id=${gpuIds}`,
            (error, stdout, stderr) => {
              if (error) {
                console.error(`exec error: ${error}`)
                reject(error)
                return
              }
              const gpuInfo = stdout
                .trim()
                .split('\n')
                .map((line) => {
                  const [
                    id,
                    name,
                    temperature,
                    utilization,
                    memoryTotal,
                    memoryFree,
                    memoryUtilization,
                  ] = line.split(', ').map((item) => item.replace(/\r/g, ''))
                  return {
                    id,
                    name,
                    temperature,
                    utilization,
                    memoryTotal,
                    memoryFree,
                    memoryUtilization,
                  }
                })
              resolve({
                cpu: { usage: cpuPercentage },
                gpu: gpuInfo,
              })
            }
          )
        } else {
          // Handle the case where gpuIds is empty
          resolve({ cpu: { usage: cpuPercentage }, gpu: [] })
        }
      } else {
        // Handle the case where run_mode is not 'gpu' or no GPUs are in use
        resolve({ cpu: { usage: cpuPercentage }, gpu: [] })
      }
    })
  })

module.exports = {
  getResourcesInfo,
  getCurrentLoad,
}