Commands.js

Go to the documentation of this file.

const db = require("../db");
 
module.exports = {
  queueCommand,
  fetchPendingCommands,
  completeCommand
};
 
/**
 *
 * @param agentId
 * @param type
 * @param data
 */
async function queueCommand(agentId, type, data) {
  const result = await db.query(
    `INSERT INTO remote_commands (agent_id, type, data, status)
     VALUES ($1,$2,$3,'queued')
     RETURNING cmd_id`,
    [agentId, type, data]
  );
  return result.rows[0].cmd_id;
}
 
/**
 *
 * @param agentId
 */
async function fetchPendingCommands(agentId) {
  const result = await db.query(
    `SELECT * FROM remote_commands
     WHERE agent_id = $1 AND status = 'queued'`,
    [agentId]
  );
  return result.rows;
}
 
/**
 *
 * @param cmdId
 * @param result
 * @param success
 */
async function completeCommand(cmdId, result, success) {
  await db.query(
    `UPDATE remote_commands SET status='success', result=$2 WHERE cmd_id=$1`,
    [cmdId, result]
  );
}