Agent.js

Go to the documentation of this file.

const db = require("../db");
 
module.exports = {
  createAgent,
  updateHeartbeat,
  getAgent,
};
 
/**
 *
 * @param root0
 * @param root0.tenantId
 * @param root0.version
 * @param root0.hostname
 * @param root0.platform
 * @param root0.machineId
 * @param root0.policyId
 */
async function createAgent({
  tenantId,
  version,
  hostname,
  platform,
  machineId,
  policyId
}) {
  const result = await db.query(
    `INSERT INTO agents
      (tenant_id, version, hostname, platform, machine_id, last_seen, policy_id, status)
     VALUES ($1,$2,$3,$4,$5,NOW(),$6,'enrolled')
     RETURNING agent_id, agent_uuid`,
    [tenantId, version, hostname, platform, machineId, policyId]
  );
 
  return {
    agentId: result.rows[0].agent_id,
    agentUuid: result.rows[0].agent_uuid
  };
}
 
/**
 *
 * @param agentUuid
 */
async function updateHeartbeat(agentUuid) {
  await db.query(
    `UPDATE agents SET last_seen = NOW() WHERE agent_uuid = $1`,
    [agentUuid]
  );
}
 
/**
 *
 * @param agentId
 */
async function getAgent(agentId) {
  const result = await db.query(
    `SELECT * FROM agents WHERE agent_id = $1`,
    [agentId]
  );
  return result.rows[0];
}