linux-odyssey/server/src/game/commandHandler.js

import minimist from 'minimist'
import { FileGraph } from '@linux-odyssey/file-graph'
import { Quest } from '@linux-odyssey/models'
import { pushToSession } from '../api/socket.js'
import SessionHandler from './sessionHandler.js'
import { checkFile } from '../containers/cli.js'
import logger from '../utils/logger.js'

// 檢查 pattern 是否符合 input
const checkMatch = (pattern, input) => {
  if (!pattern || pattern.length === 0) return true
  return pattern.some((p) => {
    const regex = new RegExp(p)
    return regex.test(input)
  })
}

export default class CommandHandler extends SessionHandler {
  constructor(session, commandInput, params) {
    super(session)
    this.commandInput = commandInput
    this.argv = minimist(this.commandInput.command.split(' '))

    this.params = params
  }

  handleEvent() {
    if (this.params.discover) this.discoverHandler()
  }

  handleCommand() {
    const command = this.argv._[0]
    switch (command) {
      case 'cd':
        pushToSession(this.session.id, 'graph', {
          pwd: this.commandInput.pwd,
        })
        break

      case 'ls':
        pushToSession(this.session.id, 'graph', {
          pwd: this.commandInput.pwd,
          discover: this.params.discover,
        })
        break

      default:
        break
    }
  }

  async discoverHandler() {
    try {
      const graph = new FileGraph(this.session.graph)
      graph.discover(this.params.discover)
      this.session.graph = graph
    } catch (e) {
      logger.error('Update file graph failed', {
        error: e,
        session: this.session.id,
        graph: this.session.graph,
        discover: this.params.discover,
      })
    }
  }

  async isMatch(condition) {
    return (
      this.checkKeys(condition) &&
      (await this.checkFiles(condition.files)) &&
      (await this.checkNot(condition.not)) &&
      (await this.checkOr(condition.or)) === true
    )
  }

  checkKeys(condition) {
    const keys = ['command', 'output', 'error', 'pwd']
    return keys.every((k) => checkMatch(condition[k], this.commandInput[k]))
  }

  async checkFiles(files) {
    if (files.length === 0) {
      return true
    }
    try {
      const checks = await Promise.all(
        files.map((f) => checkFile(this.session.containerId, f))
      )
      return checks.every((c) => c === true)
    } catch (e) {
      return false
    }
  }

  async checkNot(condition) {
    if (!condition) {
      return true
    }
    return !(await this.isMatch(condition))
  }

  async checkOr(conditions) {
    if (conditions.length === 0) {
      return true
    }
    const matches = await Promise.all(conditions.map((c) => this.isMatch(c)))
    return matches.some((m) => m === true)
  }

  async checkException(stage) {
    const exceptions = stage.exceptions || []
    for (const exception of exceptions) {
      // eslint-disable-next-line no-await-in-loop
      if (exception.condition && (await this.isMatch(exception.condition))) {
        return exception
      }
      if (exception.catchAll) return exception
    }
    return null
  }

  async run() {
    try {
      this.quest = await Quest.findById(this.session.quest)
      const stages = this.getStages()
      if (stages.length === 0) {
        logger.warn('stage not found', this.session.tasks)
        return {}
      }

      this.handleEvent()
      this.handleCommand()

      const matches = await Promise.all(
        stages.map((stage) => this.isMatch(stage.condition))
      )

      const stage = stages.find((_, i) => matches[i])
      if (!stage) {
        for (const s of stages) {
          // eslint-disable-next-line no-await-in-loop
          const exception = await this.checkException(s)
          if (exception) {
            return this.executeException(exception)
          }
        }
        return {}
      }
      const response = this.execute(stage)

      return {
        stage: stage.id,
        end: this.session.status === 'finished',
        ...response,
      }
    } catch (err) {
      logger.error('CommandHandler error', {
        error: err,
        session: this.session.id,
        command: this.commandInput,
      })
      return {}
    }
  }
}