import {
  commit,
  rollback,
  select,
  startTransaction,
  update
} from '@evershop/postgres-query-builder';
import { getConnection, pool } from '../../../../lib/postgres/connection.js';
import {
  INTERNAL_SERVER_ERROR,
  INVALID_PAYLOAD,
  OK
} from '../../../../lib/util/httpStatus.js';
import { EvershopRequest } from '../../../../types/request.js';
import { EvershopResponse } from '../../../../types/response.js';
import addOrderActivityLog from '../../services/addOrderActivityLog.js';

export default async (
  request: EvershopRequest,
  response: EvershopResponse,
  next
) => {
  const connection = await getConnection();
  await startTransaction(connection);
  const { order_id, shipment_id } = request.params;
  const { carrier, tracking_number } = request.body;
  try {
    const order = await select()
      .from('order')
      .where('uuid', '=', order_id)
      .load(connection);

    if (!order) {
      response.status(INVALID_PAYLOAD);
      response.json({
        error: {
          status: INVALID_PAYLOAD,
          message: 'Invalid order id'
        }
      });
      return;
    }
    const shipment = await select()
      .from('shipment')
      .where('uuid', '=', shipment_id)
      .load(connection);

    if (!shipment) {
      response.status(INVALID_PAYLOAD);
      response.json({
        error: {
          status: INVALID_PAYLOAD,
          message: 'Invalid shipment id'
        }
      });
      return;
    }
    await update('shipment')
      .given({
        carrier,
        tracking_number
      })
      .where('uuid', '=', shipment_id)
      .execute(connection);
    /* Add an activity log message */
    await addOrderActivityLog(
      order.order_id,
      'Shipment information updated',
      false,
      connection
    );

    await commit(connection);

    // Load updated shipment
    const updatedShipment = await select()
      .from('shipment')
      .where('shipment_order_id', '=', order.order_id)
      .and('uuid', '=', shipment_id)
      .load(pool);

    response.status(OK);
    response.$body = {
      data: updatedShipment
    };
    next();
  } catch (e) {
    await rollback(connection);
    response.status(INTERNAL_SERVER_ERROR);
    response.json({
      error: {
        status: INTERNAL_SERVER_ERROR,
        message: e.message
      }
    });
  }
};