const express = require('express');
const moment = require('moment');
const _ = require('lodash');
const { query, body } = require('express-validator');
const { onSuccess, fill } = require('../../../utils');
const { limit } = require('../../config');
const check = require('../../../middleware/check');
const Sample = require('../../../db/models/sample');
const Area = require('../../../db/models/area');

const router = express.Router();

async function fillSingle(data) {
  if (data) {
    const { cityId, provinceId, regionId, volume, ratio, time } = data;
    const [cItem, pItem, rItem] = await Promise.all([
      Area.findOne({ where: { id: cityId } }),
      Area.findOne({ where: { id: provinceId } }),
      Area.findOne({ where: { id: regionId } }),
    ]);
    data.cityName = cItem.name;
    data.provinceName = pItem.name;
    data.regionName = rItem.name;
    let res = volume;
    const month = moment(time).month();
    const prediction = [{ date: month + 1, volume: res }];
    for (let i = month + 1; i < month + ratio.length; i++) {
      let j = i;
      j > 11 && (j -= 12);
      res *= 1 + ratio[j];
      prediction.push({ date: j + 1, volume: res.toFixed(2) });
    }
    data.prediction = prediction;
  }
  return data;
}
/**
 * showdoc
 * @catalog v1/商家/样本
 * @title 样本列表
 * @description 样本列表接口
 * @method GET
 * @url  /business/sample/
 * @param  page 0 Int 页码: 不传默认取第1页数据，每页20条数据
 * @param  cityId 0 integer 城市ID
 * @param  provinceId 0 integer 省ID
 * @param  regionId 0 integer 区ID
 * @return {"code":0, "count":1, "data": []}
 * @return_param  id integer id
 * @return_param  name string 样本名
 * @return_param  cityId integer 城市ID
 * @return_param  provinceId integer 省ID
 * @return_param  regionId integer 区域ID
 * @return_param  volume float 积蓄量
 * @return_param  reserves float 储蓄量
 * @return_param  plantCount integer 种植总量
 * @return_param  treeHeight float 树高
 * @return_param  treeDiameter float 树径
 * @return_param  img string 样本影像
 * @return_param  result string 样本结果
 * @return_param  report string 报告
 * @return_param  accumulate integer 基准累积量
 * @return_param  ratio string 成长系数
 * @return_param  cityName string 城市名称
 * @return_param  provinceName string 省名称
 * @return_param  regionName string 区名称
 * @return_param  prediction array 预测折线图
 * @return_param  createdAt date -
 * @return_param  updatedAt date -
 * @remark
 * @number
 */
router.get('/', async (req, res) => {
  const { page = 0 } = req.query;
  const where = _.omit(req.query, ['page']);
  const info = onSuccess();
  info.data = await Sample.findAll({ where, limit, offset: page * limit });

  info.data = await fill(info.data, fillSingle);
  if (!page) {
    info.count = await Sample.count({ where });
  }
  res.send(info);
});
/**
 * showdoc
 * @catalog v1/商家/样本
 * @title 样本详情
 * @description
 * @method GET
 * @url  /business/sample/:id
 * @param  :id 1 STRING 详情ID
 * @return {"code":0, "data": {}}
 * @return_param  id integer id
 * @return_param  name string 样本名
 * @return_param  cityId integer 城市ID
 * @return_param  provinceId integer 省ID
 * @return_param  regionId integer 区域ID
 * @return_param  volume float 积蓄量
 * @return_param  reserves float 储蓄量
 * @return_param  plantCount integer 种植总量
 * @return_param  treeHeight float 树高
 * @return_param  treeDiameter float 树径
 * @return_param  img string 样本影像
 * @return_param  result string 样本结果
 * @return_param  report string 报告
 * @return_param  accumulate integer 基准累积量
 * @return_param  ratio string 成长系数
 * @return_param  cityName string 城市名称
 * @return_param  provinceName string 省名称
 * @return_param  regionName string 区名称
 * @return_param  prediction array 预测折线图
 * @return_param  createdAt date -
 * @return_param  updatedAt date -
 * @remark
 * @number
 */
router.get('/:id', async (req, res) => {
  let data = await Sample.findOne({ where: req.params });
  data = await fill(data, fillSingle);
  res.send(onSuccess(data));
});
module.exports = router;