diff --git a/striker-ui-api/src/lib/request_handlers/fence/createFence.ts b/striker-ui-api/src/lib/request_handlers/fence/createFence.ts new file mode 100644 index 00000000..fccf9bab --- /dev/null +++ b/striker-ui-api/src/lib/request_handlers/fence/createFence.ts @@ -0,0 +1,134 @@ +import assert from 'assert'; +import { RequestHandler } from 'express'; + +import { REP_PEACEFUL_STRING } from '../../consts'; + +import { getFenceSpec, timestamp, write } from '../../accessModule'; +import { sanitize } from '../../sanitize'; +import { stderr, stdoutVar, uuid } from '../../shell'; + +const MAP_TO_VAR_TYPE: Record< + AnvilDataFenceParameterType, + 'boolean' | 'number' | 'string' +> = { + boolean: 'boolean', + integer: 'number', + second: 'number', + select: 'string', + string: 'string', +}; + +export const createFence: RequestHandler< + { uuid?: string }, + undefined, + { + agent: string; + name: string; + parameters: { [parameterId: string]: boolean | number | string }; + } +> = async (request, response) => { + const { + body: { agent: rAgent, name: rName, parameters: rParameters }, + params: { uuid: rUuid }, + } = request; + + let fenceSpec: AnvilDataFenceHash; + + try { + fenceSpec = await getFenceSpec(); + } catch (error) { + stderr(`Failed to get fence devices specification; CAUSE: ${error}`); + + return response.status(500).send(); + } + + const agent = sanitize(rAgent, 'string'); + const name = sanitize(rName, 'string'); + const fenceUuid = sanitize(rUuid, 'string', { fallback: uuid() }); + + const { [agent]: agentSpec } = fenceSpec; + + try { + assert.ok(agentSpec, `Agent is unknown to spec; got [${agent}]`); + + assert( + REP_PEACEFUL_STRING.test(name), + `Name must be a peaceful string; got [${name}]`, + ); + + const rParamsType = typeof rParameters; + + assert( + rParamsType === 'object', + `Parameters must be an object; got type [${rParamsType}]`, + ); + } catch (error) { + assert( + `Failed to assert value when working with fence device; CAUSE: ${error}`, + ); + + return response.status(400).send(); + } + + const { parameters: agentSpecParams } = agentSpec; + + const args = Object.entries(agentSpecParams) + .reduce((previous, [paramId, paramSpec]) => { + const { content_type: paramType, default: paramDefault } = paramSpec; + const { [paramId]: rParamValue } = rParameters; + + if ( + [paramDefault, '', null, undefined].some((bad) => rParamValue === bad) + ) + return previous; + + // TODO: add SQL modifier after finding a way to escape single quotes + const paramValue = sanitize(rParamValue, MAP_TO_VAR_TYPE[paramType]); + + previous.push(`${paramId}="${paramValue}"`); + + return previous; + }, []) + .join(' '); + + stdoutVar( + { agent, args, name }, + `Proceed to record fence device (${fenceUuid}): `, + ); + + const modifiedDate = timestamp(); + + try { + const wcode = await write( + `INSERT INTO + fences ( + fence_uuid, + fence_name, + fence_agent, + fence_arguments, + modified_date + ) VALUES ( + '${fenceUuid}', + '${name}', + '${agent}', + '${args}', + '${modifiedDate}' + ) ON CONFLICT (fence_uuid) + DO UPDATE SET + fence_name = '${name}', + fence_agent = '${agent}', + fence_arguments = '${args}', + modified_date = '${modifiedDate}';`, + ); + + assert(wcode === 0, `Write exited with code ${wcode}`); + } catch (error) { + stderr(`Failed to write fence record; CAUSE: ${error}`); + + return response.status(500).send(); + } + + const scode = rUuid ? 201 : 200; + + return response.status(scode).send(); +}; diff --git a/striker-ui-api/src/lib/request_handlers/fence/index.ts b/striker-ui-api/src/lib/request_handlers/fence/index.ts index 8c5ca77f..dc13ad37 100644 --- a/striker-ui-api/src/lib/request_handlers/fence/index.ts +++ b/striker-ui-api/src/lib/request_handlers/fence/index.ts @@ -1,3 +1,5 @@ +export * from './createFence'; export * from './deleteFence'; export * from './getFence'; export * from './getFenceTemplate'; +export * from './updateFence'; diff --git a/striker-ui-api/src/lib/request_handlers/fence/updateFence.ts b/striker-ui-api/src/lib/request_handlers/fence/updateFence.ts new file mode 100644 index 00000000..037b407c --- /dev/null +++ b/striker-ui-api/src/lib/request_handlers/fence/updateFence.ts @@ -0,0 +1,3 @@ +import { createFence } from './createFence'; + +export const updateFence = createFence; diff --git a/striker-ui-api/src/routes/fence.ts b/striker-ui-api/src/routes/fence.ts index f502ce12..9c018ff2 100644 --- a/striker-ui-api/src/routes/fence.ts +++ b/striker-ui-api/src/routes/fence.ts @@ -1,9 +1,11 @@ import express from 'express'; import { + createFence, deleteFence, getFence, getFenceTemplate, + updateFence, } from '../lib/request_handlers/fence'; const router = express.Router(); @@ -11,6 +13,8 @@ const router = express.Router(); router .delete('/:uuid?', deleteFence) .get('/', getFence) - .get('/template', getFenceTemplate); + .get('/template', getFenceTemplate) + .post('/', createFence) + .put('/:uuid', updateFence); export default router;