@feedoom/nls/lib/fileStats.js

The colorful ls command

const fs = require("fs").promises
const modeString = require("fs-mode-to-string")
const getIcon = require('./icons')
const color = require('./color')
const  groups = require('./groups');
const path = require('path')

function fileSize(size) {
  let carry = 0
  let unit = ['B', 'KB', 'MB', 'GB']
  while((size + '').length > 3) {
    size = Math.round(size * 10 / 1024) / 10
    carry++
  }
  let sizeString = ''
  if((size + '').length < 3) {
    sizeString = ' '.repeat(3 - (size + '').length) + size
  } else {
    sizeString = size
  }
  return color.size(`${sizeString} ${unit[carry] === 'B' ? 'B ' : unit[carry]}`)
}

function getRealpath(filePath) {
  return fs.readlink(filePath)
}


async function fileMessages(filePath, argvs, isFileInDir) {
  let fullPath = path.resolve(filePath)
  let stats = null

  try {
    stats = await fs.lstat(fullPath)
  } catch (err) {
    return Promise.reject(`nls: cannot access '${filePath}': No such file or directory`)
  }

  let fileMessage = {
    fullPathName: fullPath,
    pathName: isFileInDir ? filePath.split('/').pop() : filePath
  }

  if (stats.isDirectory()) {
    fileMessage.type = 'dir'
  } else if (stats.isSymbolicLink()) {
    fileMessage.type = 'link'
  } else {
    fileMessage.type = 'file'
  }

  // file 的具体类型
  if(fileMessage.type === 'dir') {
    fileMessage.file = 'dir'
  } else if (fileMessage.type === 'link') {
    fileMessage.file = 'link'
  } else if (fileMessage.type === 'file'){
    fileMessage.file = filePath.split(/[.\/]/).pop()
  } else {
    fileMessage.file = 'other'
  }

  // ishidden
  fileMessage.ishidden = fileMessage.pathName.split('/').pop()[0] === '.'

  if(argvs.list || argvs.execute) {
    // file mode
    fileMessage.mode = modeString(stats)
  }

  if(argvs.list || argvs.time) {
    // file time
    fileMessage.time = stats.mtimeMs
  }

  if(argvs.list) {
    // file uid
    fileMessage.uid = stats.uid
    // file username
    fileMessage.username = groups[stats.gid]
    // file gid
    fileMessage.gid = stats.gid
    // file groupname
    fileMessage.groupname = groups[stats.gid]
    // file size
    fileMessage.size= fileSize(stats.size)
  }

  if(argvs.icon) {
    // file icon
    fileMessage.icon = getIcon(fileMessage.file, fileMessage.type !== 'file' ? fileMessage.pathName : null)
  }

  if(argvs.indent) {
    // file icon
    fileMessage.indent = argvs.indent
  }

  // symlink realpath
  if(fileMessage.file === 'link') {
    let realPathName = await getRealpath(fullPath)
    fileMessage.realpath = realPathName
  }

  return fileMessage
}

/*
  fileStats 是一个 fileMessage 数组描述每个文件的信息
  fileMessage = {
    pathName: '文件名',
    ishidden: true,
    ...
  }
  */
async function fileStats (paths, argvs, isFileInDir) {
  const filesStats = []
  for(let i = 0; i < paths.length; i++) {
    try {
      let fileMessage = await fileMessages(paths[i], argvs, isFileInDir)
      filesStats.push(fileMessage)
    } catch (err) {
      return Promise.reject(err)
    }
  }
  return filesStats
}

module.exports = fileStats