refactor: move client to sperate packages, rename cli to container

This commit is contained in:
wancat committed 2023-08-21 20:40:05 +08:00
1 parent 41a12c9aa8
commit 6ace18475b
15 files changed
+51 -28

No files matched your search

+84
View File
@@ -0,0 +1,84 @@
/* eslint-disable no-restricted-syntax */
const fs = require('fs').promises
const { exit } = require('process')
const { colorize, printResponses } = require('./print.js')
const discoverFiles = require('./discover.js')
async function readOrNone(file) {
try {
return await fs.readFile(file, 'utf8')
} catch (err) {
return ''
}
}
function parseCommand(command) {
const [name, ...args] = command.trim().split(' ')
return { name, args }
}
const commandListeners = {
ls: [discoverFiles],
}
async function handleCommand(command) {
const { name, args } = parseCommand(command)
const listeners = commandListeners[name]
if (!listeners) return {}
return listeners.reduce(async (result, listener) => {
const res = await listener(name, args)
return { ...res, ...result }
}, {})
}
async function main() {
const {
API_ENDPOINT,
TOKEN,
PWD,
CMD_NAME,
CMD_OUTPUT_FILE,
CMD_ERROR_FILE,
CMD_EXIT_CODE,
} = process.env
if (!CMD_NAME) exit(0)
const additionalData = await handleCommand(CMD_NAME)
const output = await readOrNone(CMD_OUTPUT_FILE)
const error = await readOrNone(CMD_ERROR_FILE)
const payload = {
command: CMD_NAME,
output,
error,
exit_code: CMD_EXIT_CODE,
token: TOKEN,
pwd: PWD,
...additionalData,
}
try {
const res = await fetch(`${API_ENDPOINT}/api/v1/commands`, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
},
body: JSON.stringify(payload),
})
const data = await res.json()
if (data.responses) {
await printResponses(data.responses, 60)
}
if (data.hints) {
for (const hint of data.hints) {
console.log(colorize(hint, 'yellow'))
}
}
if (data.end) {
console.log(colorize('Quest completed!', 'blue'))
}
} catch (err) {
console.error(err)
}
}
main().then(exit)
+38
View File
@@ -0,0 +1,38 @@
const fs = require('fs').promises
const path = require('path')
async function collectFilesInfo(directoryPath, level = 0) {
const entries = await fs.readdir(directoryPath, { withFileTypes: true })
const promises = entries.map(async (entry) => {
const filePath = path.join(directoryPath, entry.name)
const isDirectory = entry.isDirectory()
const discovered = level === 0 && !entry.name.startsWith('.')
const file = {
path: filePath,
name: entry.name,
type: isDirectory ? 'folder' : 'file',
discovered,
}
if (isDirectory && level < 2) {
// Get up to two levels deep
const subFilesInfo = await collectFilesInfo(filePath, level + 1)
return [file, ...subFilesInfo]
}
return file
})
const results = await Promise.all(promises)
return results.flat()
}
async function discoverFiles(name, args) {
let targetPath = args[0] || '.'
targetPath = path.resolve(process.cwd(), targetPath)
return { discover: await collectFilesInfo(targetPath) }
}
module.exports = discoverFiles
+9
View File
@@ -0,0 +1,9 @@
{
"name": "container",
"version": "1.0.0",
"main": "index.js",
"license": "MIT",
"dependencies": {
"chokidar": "^3.5.3"
}
}
+82
View File
@@ -0,0 +1,82 @@
/* eslint-disable no-restricted-syntax */
/* eslint-disable no-await-in-loop */
const readline = require('readline')
const RETURN_SYMBOL = '\u21B5'
// ANSI color map
const ANSI_COLORS = {
black: 30,
red: 31,
green: 32,
yellow: 33,
blue: 34,
magenta: 35,
cyan: 36,
white: 37,
}
function colorize(content, color) {
const colorCode = ANSI_COLORS[color]
if (!colorCode) return content
return `\x1b[${colorCode}m${content}\x1b[0m`
}
process.stdin.setRawMode(true)
const rl = readline.createInterface({
input: process.stdin,
output: process.stdout,
terminal: false,
})
function waitForEnter() {
return new Promise((resolve) => {
rl.once('line', resolve)
})
}
function printLine(content, delay, color) {
const printContent = `${colorize(content, color)}${RETURN_SYMBOL}`
return new Promise((resolve) => {
let i = 0
const timer = setInterval(() => {
process.stdout.write(printContent[i])
i += 1
if (i >= printContent.length) {
clearInterval(timer)
waitForEnter().then(resolve)
}
}, delay)
waitForEnter().then(() => {
clearInterval(timer)
process.stdout.write(printContent.slice(i))
process.stdout.write('\n')
resolve()
})
})
}
async function printResponses(responses, delay = 100) {
for (const response of responses) {
const { type, content, speaker, color } = response
if (type === 'narrative') {
for (const line of content) {
await printLine(line, delay, color || 'white')
}
}
if (type === 'dialogue') {
console.log(`${speaker}:`)
for (const line of content) {
await printLine(`\t${line}`, delay, color || 'green')
}
}
}
rl.close()
}
module.exports = {
printResponses,
colorize,
}
+6
View File
@@ -0,0 +1,6 @@
const chokidar = require('chokidar')
// One-liner for current directory
chokidar.watch('.').on('all', (event, path) => {
console.log(event, path)
})