import type { Client } from '../client/mod.ts'
import { BaseChildManager } from './baseChild.ts'
import type { RolePayload } from '../types/role.ts'
import { Role } from '../structures/role.ts'
import type { Member } from '../structures/member.ts'
import type { RolesManager } from './roles.ts'
import type { MemberPayload } from '../types/guild.ts'
import { GUILD_MEMBER_ROLE } from '../types/endpoint.ts'

export class MemberRolesManager extends BaseChildManager<RolePayload, Role> {
  member: Member

  constructor(client: Client, parent: RolesManager, member: Member) {
    super(client, parent as any)
    this.member = member
  }

  async get(id: string): Promise<Role | undefined> {
    const res = await this.parent.get(id)
    const mem = (await (this.parent as any).guild.members._get(
      this.member.id
    )) as MemberPayload
    if (
      res !== undefined &&
      (mem.roles.includes(res.id) === true || res.id === this.member.guild.id)
    )
      return res
    else return undefined
  }

  async array(): Promise<Role[]> {
    const arr = (await this.parent.array()) as Role[]
    const mem = (await (this.parent as any).guild.members._get(
      this.member.id
    )) as MemberPayload
    return arr.filter(
      (c: any) =>
        (mem.roles.includes(c.id) as boolean) || c.id === this.member.guild.id
    ) as any
  }

  async flush(): Promise<boolean> {
    const arr = await this.array()
    for (const elem of arr) {
      this.parent._delete(elem.id)
    }
    return true
  }

  async add(role: string | Role): Promise<boolean> {
    const res = await this.client.rest.put(
      GUILD_MEMBER_ROLE(
        this.member.guild.id,
        this.member.id,
        typeof role === 'string' ? role : role.id
      ),
      undefined,
      undefined,
      undefined,
      true
    )

    return res.response.status === 204
  }

  async remove(role: string | Role): Promise<boolean> {
    const res = await this.client.rest.delete(
      GUILD_MEMBER_ROLE(
        this.member.guild.id,
        this.member.id,
        typeof role === 'string' ? role : role.id
      ),
      undefined,
      undefined,
      undefined,
      true
    )

    return res.response.status === 204
  }
}