import { AnalyticUnitId } from '../models/analytic_unit_model';
import * as Panel from '../models/panel_model';

import * as Router from 'koa-router';


async function getAnalyticUnits(ctx: Router.IRouterContext) {
  try {
    let panelUrl: string = ctx.request.query.panelUrl;
    if(panelUrl === undefined || panelUrl === '') {
      throw new Error('panelUrl is missing');
    }

    const analyticUnits = await Panel.findOne({ panelUrl });

    ctx.response.body = { analyticUnits };
  } catch(e) {
    ctx.response.status = 500;
    ctx.response.body = {
      code: 500,
      message: `GET /panel error: ${e.message}`
    };
  }
}

async function addAnalyticUnit(ctx: Router.IRouterContext) {
  try {
    let { panelUrl, analyticUnitId } = ctx.request.body as {
      panelUrl: string, analyticUnitId: AnalyticUnitId
    };

    await Panel.insertAnalyticUnit(panelUrl, analyticUnitId);

    ctx.response.body = {
      code: 200,
      message: 'Success'
    };
  } catch(e) {
    ctx.response.status = 500;
    ctx.response.body = {
      code: 500,
      message: `POST /panel error: ${e.message}`
    };
  }
}

async function deleteAnalyticUnit(ctx: Router.IRouterContext) {
  try {
    let { panelUrl, analyticUnitId } = ctx.request.body as {
      panelUrl: string, analyticUnitId: AnalyticUnitId
    };

    // TODO: stop task when analytic unit is removed
    await Panel.removeAnalyticUnit(panelUrl, analyticUnitId);

    ctx.response.body = {
      code: 200,
      message: 'Success'
    };
  } catch(e) {
    ctx.response.status = 500;
    ctx.response.body = {
      code: 500,
      message: `DELETE /panel error: ${e.message}`
    };
  }
}

export const router = new Router();

router.get('/', getAnalyticUnits);
router.post('/', addAnalyticUnit);
router.delete('/', deleteAnalyticUnit);