hastic-server/server/src/routes/analytic_units_router.ts

import * as AnalyticUnit from '../models/analytic_unit_model';

import { createAnalyticUnitFromObject } from '../controllers/analytics_controller';

import * as Router from 'koa-router';


async function getStatus(ctx: Router.IRouterContext) {
  try {
    let analyticUnitId = ctx.request.query.id;
    if(analyticUnitId === undefined) {
      throw new Error('Cannot get status of undefined id');
    }

    let analyticUnit = await AnalyticUnit.findById(analyticUnitId);

    ctx.response.body = {
      status: analyticUnit.status
    };

    if(analyticUnit.status === AnalyticUnit.AnalyticUnitStatus.FAILED) {
      ctx.response.body.errorMessage = analyticUnit.error;
    }
  } catch(e) {
    console.error(e);
    ctx.response.status = 404;
    ctx.response.body = 'Can`t find anything';
  }
}

async function getUnit(ctx: Router.IRouterContext) {
  try {
    let analyticUnitId = ctx.request.query.id;

    if(analyticUnitId === undefined) {
      throw new Error('No id param in query');
    }

    let analyticUnit = await AnalyticUnit.findById(analyticUnitId);

    ctx.response.body = {
      name: analyticUnit.name,
      metric: analyticUnit.metric,
      status: analyticUnit.status
    };

  } catch(e) {
    console.error(e);
    // TODO: better send 404 when we know than isn`t found
    ctx.response.status = 500;
    ctx.response.body = 'Can`t find anything';
  }
}

async function createUnit(ctx: Router.IRouterContext) {
  try {
    let id = await createAnalyticUnitFromObject(ctx.request.body);
    ctx.response.body = { id };
  } catch(e) {
    ctx.response.status = 500;
    ctx.response.body = {
      code: 500,
      message: `Creation error: ${e.message}`
    };
  }

}

async function deleteUnit(ctx: Router.IRouterContext) {
  try {
    await AnalyticUnit.remove(ctx.request.query.id);
    ctx.response.body = {
      code: 200,
      message: 'Success'
    };
  } catch(e) {
    ctx.response.status = 500;
    ctx.response.body = {
      code: 500,
      message: `Deletion error: ${e.message}`
    };
  }
}


export var router = new Router();

router.get('/', getUnit);
router.get('/status', getStatus);
router.post('/', createUnit);
router.delete('/', deleteUnit);