import Router from "koa-router";
import { Op } from "sequelize";
import { MAX_AVATAR_DISPLAY } from "@shared/constants";
import auth from "@server/middlewares/authentication";
import validate from "@server/middlewares/validate";
import { User, Event, Group, GroupUser } from "@server/models";
import { authorize } from "@server/policies";
import {
  presentGroup,
  presentPolicies,
  presentUser,
  presentGroupMembership,
} from "@server/presenters";
import { APIContext } from "@server/types";
import pagination from "../middlewares/pagination";
import * as T from "./schema";

const router = new Router();

router.post(
  "groups.list",
  auth(),
  pagination(),
  validate(T.GroupsListSchema),
  async (ctx: APIContext<T.GroupsListReq>) => {
    const { direction, sort } = ctx.input.body;
    const { user } = ctx.state.auth;

    const groups = await Group.findAll({
      where: {
        teamId: user.teamId,
      },
      order: [[sort, direction]],
      offset: ctx.state.pagination.offset,
      limit: ctx.state.pagination.limit,
    });

    ctx.body = {
      pagination: ctx.state.pagination,
      data: {
        groups: groups.map(presentGroup),
        groupMemberships: groups
          .map((g) =>
            g.groupMemberships
              .filter((membership) => !!membership.user)
              .slice(0, MAX_AVATAR_DISPLAY)
          )
          .flat()
          .map((membership) =>
            presentGroupMembership(membership, { includeUser: true })
          ),
      },
      policies: presentPolicies(user, groups),
    };
  }
);

router.post(
  "groups.info",
  auth(),
  validate(T.GroupsInfoSchema),
  async (ctx: APIContext<T.GroupsInfoReq>) => {
    const { id } = ctx.input.body;
    const { user } = ctx.state.auth;

    const group = await Group.findByPk(id);
    authorize(user, "read", group);

    ctx.body = {
      data: presentGroup(group),
      policies: presentPolicies(user, [group]),
    };
  }
);

router.post(
  "groups.create",
  auth(),
  validate(T.GroupsCreateSchema),
  async (ctx: APIContext<T.GroupsCreateReq>) => {
    const { name } = ctx.input.body;
    const { user } = ctx.state.auth;
    authorize(user, "createGroup", user.team);
    const g = await Group.create({
      name,
      teamId: user.teamId,
      createdById: user.id,
    });

    // reload to get default scope
    const group = await Group.findByPk(g.id, { rejectOnEmpty: true });

    await Event.create({
      name: "groups.create",
      actorId: user.id,
      teamId: user.teamId,
      modelId: group.id,
      data: {
        name: group.name,
      },
      ip: ctx.request.ip,
    });

    ctx.body = {
      data: presentGroup(group),
      policies: presentPolicies(user, [group]),
    };
  }
);

router.post(
  "groups.update",
  auth(),
  validate(T.GroupsUpdateSchema),
  async (ctx: APIContext<T.GroupsUpdateReq>) => {
    const { id, name } = ctx.input.body;
    const { user } = ctx.state.auth;

    const group = await Group.findByPk(id);
    authorize(user, "update", group);

    group.name = name;

    if (group.changed()) {
      await group.save();
      await Event.create({
        name: "groups.update",
        teamId: user.teamId,
        actorId: user.id,
        modelId: group.id,
        data: {
          name,
        },
        ip: ctx.request.ip,
      });
    }

    ctx.body = {
      data: presentGroup(group),
      policies: presentPolicies(user, [group]),
    };
  }
);

router.post(
  "groups.delete",
  auth(),
  validate(T.GroupsDeleteSchema),
  async (ctx: APIContext<T.GroupsDeleteReq>) => {
    const { id } = ctx.input.body;
    const { user } = ctx.state.auth;

    const group = await Group.findByPk(id);
    authorize(user, "delete", group);

    await group.destroy();
    await Event.create({
      name: "groups.delete",
      actorId: user.id,
      modelId: group.id,
      teamId: group.teamId,
      data: {
        name: group.name,
      },
      ip: ctx.request.ip,
    });

    ctx.body = {
      success: true,
    };
  }
);

router.post(
  "groups.memberships",
  auth(),
  pagination(),
  validate(T.GroupsMembershipsSchema),
  async (ctx: APIContext<T.GroupsMembershipsReq>) => {
    const { id, query } = ctx.input.body;
    const { user } = ctx.state.auth;

    const group = await Group.findByPk(id);
    authorize(user, "read", group);
    let userWhere;

    if (query) {
      userWhere = {
        name: {
          [Op.iLike]: `%${query}%`,
        },
      };
    }

    const memberships = await GroupUser.findAll({
      where: {
        groupId: id,
      },
      order: [["createdAt", "DESC"]],
      offset: ctx.state.pagination.offset,
      limit: ctx.state.pagination.limit,
      include: [
        {
          model: User,
          as: "user",
          where: userWhere,
          required: true,
        },
      ],
    });

    ctx.body = {
      pagination: ctx.state.pagination,
      data: {
        groupMemberships: memberships.map((membership) =>
          presentGroupMembership(membership, { includeUser: true })
        ),
        users: memberships.map((membership) => presentUser(membership.user)),
      },
    };
  }
);

router.post(
  "groups.add_user",
  auth(),
  validate(T.GroupsAddUserSchema),
  async (ctx: APIContext<T.GroupsAddUserReq>) => {
    const { id, userId } = ctx.input.body;
    const actor = ctx.state.auth.user;

    const user = await User.findByPk(userId);
    authorize(actor, "read", user);

    let group = await Group.findByPk(id);
    authorize(actor, "update", group);

    let membership = await GroupUser.findOne({
      where: {
        groupId: id,
        userId,
      },
    });

    if (!membership) {
      await group.$add("user", user, {
        through: {
          createdById: actor.id,
        },
      });
      // reload to get default scope
      membership = await GroupUser.findOne({
        where: {
          groupId: id,
          userId,
        },
        rejectOnEmpty: true,
      });

      // reload to get default scope
      group = await Group.findByPk(id, { rejectOnEmpty: true });

      await Event.create({
        name: "groups.add_user",
        userId,
        teamId: user.teamId,
        modelId: group.id,
        actorId: actor.id,
        data: {
          name: user.name,
        },
        ip: ctx.request.ip,
      });
    }

    ctx.body = {
      data: {
        users: [presentUser(user)],
        groupMemberships: [
          presentGroupMembership(membership, { includeUser: true }),
        ],
        groups: [presentGroup(group)],
      },
    };
  }
);

router.post(
  "groups.remove_user",
  auth(),
  validate(T.GroupsRemoveUserSchema),
  async (ctx: APIContext<T.GroupsRemoveUserReq>) => {
    const { id, userId } = ctx.input.body;
    const actor = ctx.state.auth.user;

    let group = await Group.findByPk(id);
    authorize(actor, "update", group);

    const user = await User.findByPk(userId);
    authorize(actor, "read", user);

    await group.$remove("user", user);
    await Event.create({
      name: "groups.remove_user",
      userId,
      modelId: group.id,
      teamId: user.teamId,
      actorId: actor.id,
      data: {
        name: user.name,
      },
      ip: ctx.request.ip,
    });

    // reload to get default scope
    group = await Group.findByPk(id, { rejectOnEmpty: true });

    ctx.body = {
      data: {
        groups: [presentGroup(group)],
      },
    };
  }
);

export default router;