import path from 'path'
import { getString } from '@sasjs/utils/input'
import { createFolder, fileExists, folderExists, isWindows } from '@sasjs/utils'
import { RunTimeType } from './verifyEnvVariables'

export const getDesktopFields = async () => {
  const { SAS_PATH, NODE_PATH, PYTHON_PATH, R_PATH } = process.env

  let sasLoc, nodeLoc, pythonLoc, rLoc

  if (process.runTimes.includes(RunTimeType.SAS)) {
    sasLoc = SAS_PATH ?? (await getSASLocation())
  }

  if (process.runTimes.includes(RunTimeType.JS)) {
    nodeLoc = NODE_PATH ?? (await getNodeLocation())
  }

  if (process.runTimes.includes(RunTimeType.PY)) {
    pythonLoc = PYTHON_PATH ?? (await getPythonLocation())
  }

  if (process.runTimes.includes(RunTimeType.R)) {
    rLoc = R_PATH ?? (await getRLocation())
  }

  return { sasLoc, nodeLoc, pythonLoc, rLoc }
}

const getDriveLocation = async (): Promise<string> => {
  const validator = async (filePath: string) => {
    if (!filePath) return 'Path to files/drive is required.'

    const drivePath = path.join(process.cwd(), filePath)

    if (!(await folderExists(drivePath))) {
      await createFolder(drivePath)
      await createFolder(path.join(drivePath, 'files'))
    }

    return true
  }

  const defaultLocation = isWindows() ? '.\\tmp\\' : './tmp/'

  const targetName = await getString(
    'Please enter path to file/drive (relative to executable): ',
    validator,
    defaultLocation
  )

  return targetName
}

const getSASLocation = async (): Promise<string> => {
  const validator = async (filePath: string) => {
    if (!filePath) return 'Path to SAS executable is required.'

    if (!(await fileExists(filePath))) {
      return 'No file found at provided path.'
    }

    return true
  }

  const defaultLocation = isWindows()
    ? 'C:\\Program Files\\SASHome\\SASFoundation\\9.4\\sas.exe'
    : '/opt/sas/sas9/SASHome/SASFoundation/9.4/sasexe/sas'

  const targetName = await getString(
    'Please enter full path to a SAS executable with UTF-8 encoding: ',
    validator,
    defaultLocation
  )

  return targetName
}

const getNodeLocation = async (): Promise<string> => {
  const validator = async (filePath: string) => {
    if (!filePath) return 'Path to NodeJS executable is required.'

    if (!(await fileExists(filePath))) {
      return 'No file found at provided path.'
    }

    return true
  }

  const defaultLocation = isWindows()
    ? 'C:\\Program Files\\nodejs\\node.exe'
    : '/usr/local/nodejs/bin/node.sh'

  const targetName = await getString(
    'Please enter full path to a NodeJS executable: ',
    validator,
    defaultLocation
  )

  return targetName
}

const getPythonLocation = async (): Promise<string> => {
  const validator = async (filePath: string) => {
    if (!filePath) return 'Path to Python executable is required.'

    if (!(await fileExists(filePath))) {
      return 'No file found at provided path.'
    }

    return true
  }

  const defaultLocation = isWindows() ? 'C:\\Python' : '/usr/bin/python'

  const targetName = await getString(
    'Please enter full path to a Python executable: ',
    validator,
    defaultLocation
  )

  return targetName
}

const getRLocation = async (): Promise<string> => {
  const validator = async (filePath: string) => {
    if (!filePath) return 'Path to R executable is required.'

    if (!(await fileExists(filePath))) {
      return 'No file found at provided path.'
    }

    return true
  }

  const defaultLocation = isWindows() ? 'C:\\Rscript' : '/usr/bin/Rscript'

  const targetName = await getString(
    'Please enter full path to a R executable: ',
    validator,
    defaultLocation
  )

  return targetName
}