import { Guild } from 'discord.js';
import { Database } from '../shared/Database';
import { userIsBanned } from './e621-utils';
import { config } from '../config';

type AltData = {
  type: 'e621' | 'discord'
  thisId: number | string
  banned: boolean
  alts: AltData[]
};


export async function getE621Alts(discordId: string, guild: Guild, depth = 1, ignore: number[] = []): Promise<string> {
  const e621UserIds = await Database.getE621Ids(discordId);

  const toIgnore = ignore.concat(e621UserIds);

  let content = '';

  for (const e621Id of e621UserIds) {
    if (ignore.includes(e621Id)) continue;

    const alts = await getDiscordAlts(e621Id, guild, depth + 1, toIgnore);

    const banned = await userIsBanned(e621Id);

    content += `${'- '.repeat(depth)}${config.E621_BASE_URL}/users/${e621Id}${banned ? ' [BANNED]' : ''}\n${alts}`;
  }

  return content;
}

export async function getDiscordAlts(e621Id: number, guild: Guild, depth = 1, ignore: number[] = []): Promise<string> {
  const discordIds = await Database.getDiscordIds(e621Id);

  let content = '';

  for (const discordId of discordIds) {
    const alts = await getE621Alts(discordId, guild, depth + 1, ignore);

    let banned = false;

    // It's either this or fetch all the bans and sift through them for every discord alt.
    try {
      banned = !!(await guild.bans.fetch(discordId));
    } catch (e) { }

    if (alts || banned) content += `${'- '.repeat(depth)}<@${discordId}>${banned ? ' [BANNED]' : ''}\n${alts}`;
  }

  return content;
}

export async function comprehensiveAltLookupFromDiscord(discordId: string, guild: Guild): Promise<AltData> {
  return getE621AltData(discordId, guild);
}

export async function comprehensiveAltLookupFromE621(e621Id: number, guild: Guild): Promise<AltData> {
  return getDiscordAltData(e621Id, guild);
}

async function getE621AltData(discordId: string, guild: Guild, depth = 1, ignore: number[] = []): Promise<AltData> {
  const e621UserIds = await Database.getE621Ids(discordId);

  const toIgnore = ignore.concat(e621UserIds);

  let banned = false;

  // It's either this or fetch all the bans and sift through them for every discord alt.
  try {
    banned = !!(await guild.bans.fetch(discordId));
  } catch (e) { }

  const data: AltData = { type: 'discord', thisId: discordId, banned, alts: [] };

  for (const e621Id of e621UserIds) {
    if (ignore.includes(e621Id)) continue;

    data.alts.push(await getDiscordAltData(e621Id, guild, depth + 1, toIgnore));
  }

  return data;
}

async function getDiscordAltData(e621Id: number, guild: Guild, depth = 1, ignore: number[] = []): Promise<AltData> {
  const discordIds = await Database.getDiscordIds(e621Id);

  const data: AltData = { type: 'e621', thisId: e621Id, banned: await userIsBanned(e621Id), alts: [] };

  for (const discordId of discordIds) {
    data.alts.push(await getE621AltData(discordId, guild, depth + 1, ignore));
  }

  return data;
}

export function e621IdsFromAltData(altData: AltData, data: number[] = []) {
  if (altData.type == 'e621' && !data.includes(altData.thisId as number)) data.push(altData.thisId as number);

  for (const alt of altData.alts) {
    e621IdsFromAltData(alt, data);
  }

  return data;
}