import net.dv8tion.jda.api.events.guild.member.*;
import net.dv8tion.jda.api.Permission;
import net.dv8tion.jda.api.requests.restaction.MessageAction;
import net.dv8tion.jda.api.requests.RestAction;
import java.util.function.Consumer;

switchable S discordToken;

S discordBotName;
long discordBotID;

bool reactToBots = true;
transient bool sendToPostedLinesCRUD = true;
transient bool escapeAtEveryone;
transient bool membersIntent; // is members intent needed (reacting to user join/leave)

transient ShardManager discord;
transient Color discordImageEmbedMysteriousLineColor = colorFromHex("36393f");
transient L<IVF2<MessageChannel, S>> onPostedInChannel = syncList();
//transient bool debugPosting;

// when a discord action failed
transient Consumer<Throwable> onQueueError = error -> {
  temp enter();
  printStackTrace(error);
};

void startDiscord {
  if (!discordEnabled()) ret with print("Not enabled");
  
  vm_cleanPrints(); // avoid Swing Unicode problem
  logModuleOutput(); // good idea for chat bots

  // TODO: log JDA output going to System.out/err
  discord = shardedDiscordBot40(new ListenerAdapter {
    @Override
    public void onMessageUpdate(MessageUpdateEvent e) pcall {
      temp enter();
      ret if !discordEnabled() || !licensed();
      
      Message msg = e.getMessage();
      long msgID = msg.getIdLong();
      User user = e.getAuthor();
      long userID = user == null ? 0 : user.getIdLong();
      S content = e.getMessage().getContentRaw();
      MessageChannel channel = e.getChannel();
      long channelID = channel.getIdLong();
      S rendered = msgID + ": " + content;
      //print("Message edited: " + rendered);
      
      vmBus_send editedDiscordMessage(
        litmapparams(event := e, module := dm_me(), +msgID, +msg, +content,
          +channelID, +channel, +user, +userID));
          
      O lineConcept = dm_discord_lineForMsgID_unimported(msgID);
      if (lineConcept == null) ret; // logging module not enabled
      call(lineConcept, '_setField, editedText := content);
    }
    
    @Override
    public void onMessageReceived(MessageReceivedEvent e) pcall {
      temp enter();
      ret if !discordEnabled() || !licensed();
      
      // send out raw event
      vmBus_send discord_onMessageReceived(module(), e);
      
      User user = e.getAuthor();
      if (!reactToUser(user)) ret with print("Not reacting to user " + user);
      bool bot = user.isBot();

      long msgID = e.getMessage().getIdLong();
      long userID = user.getIdLong();
      Guild guild = e.isFromType(ChannelType.TEXT) ? e.getGuild() : null;
      long guildID = toLong(call(guild, 'getIdLong);
      Member member = e.getMember();
      S userName = member == null ? null : member.getNickname(); // the changeable nick name - null sometimes?
      if (userName == null && member != null) userName = member.getEffectiveName();
      if (userName == null) userName = e.getAuthor().getName();
      
      final Message msg = e.getMessage();
      
      MessageChannel channel = e.getChannel();
      long channelID = channel.getIdLong();
      //print("Channel type: " + e.getChannelType());
      bool isPrivate = e.getChannelType() == ChannelType.PRIVATE;
      S content = trim(msg.getContentRaw());
      //print("Msg from " + userName + ": " + content);

      vmBus_send incomingDiscordMessage(
        litmapparams(fromBot := bot, module := dm_me(), +msgID, +userID, +userName, event := e, +guildID, +guild, +member,
          +msg,
          +content, +isPrivate, +channelID, +channel));
    }
    
    @Override
    public void onMessageReactionAdd(MessageReactionAddEvent e) pcall {
      temp enter();
      ret if !discordEnabled() || !licensed();
      MessageReaction r = e.getReaction();
      
      User user = e.getUser();
      if (!reactToUser(user)) ret;
      bool bot = user.isBot();
      
      long msgID = r.getMessageIdLong();
      S emoji = r.getReactionEmote().getName();
      
      vmBus_send('incomingDiscordReaction, litmapparams(
        reaction := r,
        fromBot := bot, +user, userID := user.getIdLong(),
        module := dm_me(), +msgID, +emoji
      ));
    }
    
    @Override
    public void onMessageReactionRemove(MessageReactionRemoveEvent e) pcall {
      temp enter();
      ret if !discordEnabled() || !licensed();
      MessageReaction r = e.getReaction();
      
      User user = e.getUser();
      if (!reactToUser(user)) ret;
      bool bot = user.isBot();
      
      long msgID = r.getMessageIdLong();
      S emoji = r.getReactionEmote().getName();
      
      vmBus_send('discordReactionRemoved, litmapparams(
        reaction := r,
        fromBot := bot, +user, userID := user.getIdLong(),
        module := dm_me(), +msgID, +emoji
      ));
    }
    
    @Override
    public void onMessageReactionRemoveAll(MessageReactionRemoveAllEvent e) pcall {
      temp enter();
      print("TODO: onMessageReactionRemoveAll");
    }
    
    @Override
    public void onGuildMemberJoin(GuildMemberJoinEvent event) pcall {
      temp enter();
      print("Got guild join");
      ret if !discordEnabled() || !licensed();
      
      print("Join >> getting user ID");
      long userID = rcall_long getIdLong(rcall getUser(event.getMember()));
      print("Join >> sending");
      vmBus_send('discordGuildJoin, litmapparams(
        module := dm_me(), +event, +userID, guildID := discord_guildIDFromEvent_gen(event)
      ));
    }
    
    @Override
    public void onGuildMemberLeave(GuildMemberLeaveEvent event) pcall {
      temp enter();
      print("Got guild leave");
      ret if !discordEnabled() || !licensed();

      long userID = rcall_long getIdLong(rcall getUser(event.getMember()));
      vmBus_send('discordGuildLeave, litmapparams(
        module := dm_me(), +event, +userID
      ));
    } // end of JDA listener
  }, token := discordToken, +membersIntent);
  
  dm_registerAs('liveDiscordModule);
  dm_vmBus_answerToMessage activeDiscordTokens(func -> LS { ll(discordToken) });
  
  int safety = 0;
  int nShards = nShards();
  while (licensed() && ++safety < 60) {
    int connected = nShardsConnected();
    print(connected + " of " + n2(nShards, "shard") + " connected");
    if (connected >= nShards) break;
    sleepSeconds(1);
  }
  
  User selfUser = getSelfUser();
  if (selfUser == null) fail("No shards connected after 20 seconds");

  setField(discordBotName := selfUser.getName());
  setField(discordBotID := selfUser.getIdLong());
  print("Bot name: " + discordBotName);
}

int nShards() {
  ret l(discord.getShards());
}

public int nShardsConnected() {
  int n = 0;
  for (JDA shard : discord.getShards()) {
    if (shard.getStatus().equals(JDA.Status.CONNECTED))
      n++;
  }
  ret n;
}

public SelfUser getSelfUser() {
  for (JDA shard : discord.getShards()) {
    if (shard.getStatus().equals(JDA.Status.CONNECTED))
      ret shard.getSelfUser();
  }
  null;
}

void cleanMeUp {
  if (discord != null) pcall {
    print("Shutting down discord");
    discord.shutdown();
    print("Bot shut down");
  }
  discord = null;
}

O userConcept(User user) {
  S crud = dm_gazelle_linesCRUD();
  O userConcept = dm_call(crud, 'uniqUser, user.getIdLong());
  dm_call(crud, 'cset, userConcept, litobjectarray(name := user.getName()));
  ret userConcept;
}

// API

MessageChannel getChannel(long channelID) {
  ret discord.getTextChannelById(channelID);
}


void postInChannel(long channelID, S msg) {
  postInChannel(channelID, msg, null);
}

void postInChannel(long channelID, S msg, IVF1<Message> onPost) {
  if (channelID == 0) ret;
  postInChannel(getChannel(channelID), msg, onPost);
}

// returns null for your convenience
S uploadFileInChannel(long channelID, File file, S fileName, S msg, IVF1<Message> onPost) {
  if (channelID == 0) null;
  ret uploadFileInChannel(getChannel(channelID), file, fileName, msg, onPost);
}

void postInChannel(MessageChannel channel, S msg, IVF1<Message> onPost) {
  S originalMsg = msg;
  msg = ecapeAndShortenForDiscord(msg);
  if (empty(msg)) ret;
  
  S postID = !sendToPostedLinesCRUD ? null : (S) dm_pcall(gazelle_postedLinesCRUD(), 'postingLine, channel.getId(), msg);

  print("Posting in channel " + channel + ": " + msg);
  pcallFAll(onPostedInChannel, channel, msg);
  MessageAction a = channel.sendMessage(msg);
  if (msg != originalMsg)
    a.addFile(toUtf8(originalMsg), genericTextFileName());
  a.queue(m -> msgPosted(m, onPost, postID), onQueueError);
}

S genericTextFileName() {
  ret "full-output-" + ymd_minus_hms() + ".txt";
}

void msgPosted(Message m, IVF1<Message> onPost, S postID) enter {
  if (nempty(postID))
    dm_call(gazelle_postedLinesCRUD(), 'donePosting, postID, "discord msg " + m.getId());
  long msgId = m.getIdLong();
  print("I sent msg: " + msgId);
  pcallF(onPost, m);
}

void postInChannel(S channel, S msg) {
  long id = dm_discord_channelID(channel);
  if (id == 0) fail("Channel not found: " + channel);
  postInChannel(id, msg);
}

void postInChannel(MessageChannel channel, S msg) {
  postInChannel(channel, msg, null);
}

void postImage(Map msgMap, S url) {
  postImage(msgMap, url, "");
}

void postImage(Map msgMap, S url, S description) {
  postImage((MessageChannel) get channel(msgMap), url, description);
}

void postImage(MessageChannel channel, S url, S description) {
  print("Posting image: " + url);
  channel.sendMessage(
    new EmbedBuilder()
      .setImage(absoluteURL(url))
      //.setTitle(unnull(title))
      .setDescription(unnull(description))
      .setColor(discordImageEmbedMysteriousLineColor)
      .build()).queue(null, onQueueError); // TODO: posted lines
}

void editMessage(long channelID, long msgID, S text) {
  getChannel(channelID).editMessageById(str(msgID), text).queue(null, onQueueError);
}

void sendPM(long userID, S _text) {
  S text = ecapeAndShortenForDiscord(_text);
  if (empty(text)) ret;
  print("Sending PM to " + userID + ": " + text);
  discord.getUserById(userID).openPrivateChannel().queue(channel -> {
    channel.sendMessage(text).queue(null, onQueueError);
  }, onQueueError);
  print("PM queued");
}

void reply(Map msgMap, S text) {
  reply(msgMap, text, null);
}

void reply(Map msgMap, S text, O onPost) {
  if (empty(text = trim(text))) ret;
  postInChannel((MessageChannel) msgMap.get('channel), text, toIVF1(onPost));
}

void iAmTyping(Map msgMap) pcall {
  ((MessageChannel) msgMap.get('channel)).sendTyping().queue(null, onQueueError);
}

S startKeepAliveModule() enter {
  ret dm_discord_startKeepAliveModule(module());
}

bool reactToUser(User user) {
  if (user == null) false;
  
  bool bot = user.isBot();
  
  // don't react to other bots
  if (bot && !reactToBots) false;
  
  // don't read msgs from myself
  if (user.getIdLong() == discordBotID) false;
  
  true;
}

S ecapeAndShortenForDiscord(S s) {
  if (escapeAtEveryone) s = escapeAtEveryone(s);
  ret shortenForDiscord(s);
}

// returns null for your convenience
S uploadFileInChannel(MessageChannel channel, File file, S fileName, S msg, IVF1<Message> onPost) {
  msg = ecapeAndShortenForDiscord(msg);
  MessageAction a = empty(msg)
    ? channel.sendFile(file, fileName)
    : channel.sendMessage(msg).addFile(file, fileName);
  a.queue(m -> msgPosted(m, onPost, null), onQueueError);
  null;
}

// returns null... you know the drill
S uploadFileInChannel(MessageChannel channel, byte[] data, S fileName, S msg, IVF1<Message> onPost) {
  msg = ecapeAndShortenForDiscord(msg);
  MessageAction a = empty(msg)
    ? channel.sendFile(data, fileName)
    : channel.sendMessage(msg).addFile(data, fileName);
  a.queue(m -> msgPosted(m, onPost, null), onQueueError);
  null;
}

S uploadFileInChannel(long channelID, byte[] data, S fileName, S msg, IVF1<Message> onPost) {
  ret uploadFileInChannel(getChannel(channelID), data, fileName, msg, onPost);
}

void <A> queue(RestAction<A> action) {
  action.queue(null, onQueueError);
}

Began life as a copy of #1026292