package net.ercatec.hw2ircsvr;

import net.ercatec.hw.INetClient;

import net.ercatec.hw.ProtocolConnection;

import java.io.BufferedReader;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.io.OutputStream;

import java.net.InetSocketAddress;

import java.net.ServerSocket;

import java.net.Socket;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.HashMap;

import java.util.Hashtable;

import java.util.List;

import java.util.Map;

import java.util.Queue;

import java.util.concurrent.LinkedBlockingQueue;

import java.util.Collections;

import java.util.Vector;

import java.util.regex.Pattern;

import java.util.regex.PatternSyntaxException;

import java.lang.IllegalArgumentException;

// for auth files

import java.util.Properties;

import java.io.FileInputStream;

import java.io.IOException;

/* TODO

 * disconnect clients that are not irc clients

 * disconnect excess flooders

 * recognizes stuff after : as single arg

 * collect pre-irc-join messages and show on join

 * allow negating regexps

 * ban

 * banlist

 * commandquery // wth did I mean by that?

 * more room info

 * filter rooms

 * warnings

 * global notice

 */

/**

 * @author sheepluva

 * 

 * based on jircs by Alexander Boyd

 */

public class Connection implements INetClient, Runnable

{

    private static final String DESCRIPTION_SHORT

        = "connect to hedgewars via irc!";

    private static final String VERSION = "0.6.7-Alpha_2015-11-07";

    private static final String MAGIC_BYTES       = "[\1\2\3]";

    private static final char   MAGIC_BYTE_ACTION = ((char)1); // ^A

    private static final char   MAGIC_BYTE_BOLD   = ((char)2); // ^B

    private static final char   MAGIC_BYTE_COLOR  = ((char)3); // ^C

    private static final String[] DEFAULT_MOTD = {

        "                                         ",

        " "+MAGIC_BYTE_COLOR+"06"+MAGIC_BYTE_BOLD+"                            SUCH FLUFFY!",

        "                                         ",

        " "+MAGIC_BYTE_COLOR+"04 MUCH BAH     "+MAGIC_BYTE_COLOR+"00__  _                     ",

        " "+MAGIC_BYTE_COLOR+"00          .-.'  `; `-."+MAGIC_BYTE_COLOR+"00_  __  _          ",

        " "+MAGIC_BYTE_COLOR+"00         (_,         .-:'  `; `"+MAGIC_BYTE_COLOR+"00-._      ",

        " "+MAGIC_BYTE_COLOR+"14       ,'"+MAGIC_BYTE_COLOR+"02o "+MAGIC_BYTE_COLOR+"00(        (_,           )     ",

        " "+MAGIC_BYTE_COLOR+"14      (__"+MAGIC_BYTE_COLOR+"00,-'      "+MAGIC_BYTE_COLOR+"15,'"+MAGIC_BYTE_COLOR+"12o "+MAGIC_BYTE_COLOR+"00(            )>   ",

        " "+MAGIC_BYTE_COLOR+"00         (       "+MAGIC_BYTE_COLOR+"15(__"+MAGIC_BYTE_COLOR+"00,-'            )    ",

        " "+MAGIC_BYTE_COLOR+"00          `-'._.--._(             )     ",

        " "+MAGIC_BYTE_COLOR+"14             |||  |||"+MAGIC_BYTE_COLOR+"00`-'._.--._.-'      ",

        " "+MAGIC_BYTE_COLOR+"15                        |||  |||        ",

        " "+MAGIC_BYTE_COLOR+"07"+MAGIC_BYTE_BOLD+"  WOW!                                  ",

        " "+MAGIC_BYTE_COLOR+"09                   VERY SHEEP           ",

        "                                         ",

        "                                         ",

        "                                         ",

        " "+MAGIC_BYTE_COLOR+"4 Latest hw2irc crimes/changes:",

        "     ping: ping of hwserver will only get reply if irc client pingable",

        "     ping: pings of irc clients will only get reply if hwserver pingable",

        "     rooms: id-rotation, make channel names at least 2 digits wide",

        "     auth: support passhash being loaded local auth file and irc pass (sent as cleartext - DO NOT USE!)",

        "                                         ",

        "                                         ",

    };

    private static final String DEFAULT_QUIT_REASON = "User quit";

    // NOT final

    private static char CHAT_COMMAND_CHAR = '\\';

    private final class Room {

        public final int id;

        public final String chan;

        public String name;

        private String owner = "";

        public int nPlayers = 0;

        public int nTeams   = 0;

        public Room(final int id, final String name, final String owner) {

            this.id = id;

            this.chan = (id<10?"#0":"#") + id;

            this.name = name;

            this.setOwner(owner);

        }

        public String getOwner() { return this.owner; }

        public void setOwner(final String owner) {

            // don't to this for first owner

            if (!this.owner.isEmpty()) {

                // owner didn't change

                if (this.owner.equals(owner))

                    return;

                // update old room owner

                final Player oldOwner = allPlayers.get(this.owner);

                if (oldOwner != null)

                    oldOwner.isRoomAdm = false;

            }

            // update new room owner

            final Player newOwner = allPlayers.get(owner);

            if (newOwner != null)

                newOwner.isRoomAdm = true;

            this.owner = owner;

        }

    }

    private final class Player {

        public final String nick;

        public final String ircNick;

        private boolean isAdm;

        private boolean isCont;

        private boolean isReg;

        public boolean inRoom;

        public boolean isRoomAdm;

        private String ircId;

        private String ircHostname;

        private boolean announced;

        // server info

        private String version = "";

        private String ip = "";

        private String room = "";

        public Player(final String nick) {

            this.nick = nick;

            this.ircNick = hwToIrcNick(nick);

            this.announced = false;

            updateIrcHostname();

        }

        public String getIrcHostname() { return ircHostname; }

        public String getIrcId()       { return ircId; }

        public String getRoom()        {

            if (room.isEmpty())

                return room;

            return "[" + ((isAdm?"@":"") + (isRoomAdm?"+":"") + this.room);

        }

        public boolean needsAnnounce() {

            return !announced;

        }

        public void setAnnounced() {

            announced = true;

        }

        public void setInfo(final String ip, final String version, final String room) {

            if (this.version.isEmpty()) {

                this.version = version;

                this.ip = ip.replaceAll("^\\[|]$", "");

                updateIrcHostname();

            }

            if (room.isEmpty())

                this.room = room;

            else

                this.room = room.replaceAll("^\\[[@+]*", "");

        }

        public boolean isServerAdmin()  { return isAdm; }

        //public boolean isContributor()  { return isCont; }

        public boolean isRegistered()   { return isReg; }

        public void setServerAdmin(boolean isAdm) {

            this.isAdm = isAdm; updateIrcHostname(); }

        public void setContributor(boolean isCont) {

            this.isCont = isCont; updateIrcHostname(); }

        public void setRegistered(boolean isReg) {

            this.isReg = isReg; updateIrcHostname(); }

        private void updateIrcHostname() {

            ircHostname = ip.isEmpty()?"":(ip + '/');

            ircHostname += "hw/";

            if (!version.isEmpty())

                ircHostname += version;

            if (isAdm)

                ircHostname += "/admin";

            else if (isCont)

                ircHostname += "/contributor";

            else if (isReg)

                ircHostname += "/member";

            else

                ircHostname += "/player";

            updateIrcId();

        }

        private void updateIrcId() {

            ircId = ircNick + "!~" + ircNick + "@" + ircHostname;

        }

    }

    public String hw404NickToIrcId(String nick) {

        nick = hwToIrcNick(nick);

        return nick + "!~" + nick + "@hw/404";

    }

    // hash tables are thread-safe

    private final Map<String,  Player>  allPlayers = new Hashtable<String,  Player>();

    private final Map<String,  Player> roomPlayers = new Hashtable<String,  Player>();

    private final Map<Integer, Room>   roomsById   = new Hashtable<Integer, Room>();

    private final Map<String,  Room>   roomsByName = new Hashtable<String,  Room>();

    private final List<Room> roomsSorted = new Vector<Room>();

    private final List<String> ircPingQueue = new Vector<String>();

    private static final String DEFAULT_SERVER_HOST = "netserver.hedgewars.org";

    private static String SERVER_HOST = DEFAULT_SERVER_HOST;

    private static int IRC_PORT = 46667;

    private String hostname;

    private static final String LOBBY_CHANNEL_NAME = "#lobby";

    private static final String  ROOM_CHANNEL_NAME = "#room";

    // hack

    // TODO: ,

    private static final char MAGIC_SPACE   = ' ';

    private static final char MAGIC_ATSIGN  = '៙';

    private static final char MAGIC_PERCENT = '％';

    private static final char MAGIC_PLUS    = '＋';

    private static final char MAGIC_EXCLAM  = '❢';

    private static final char MAGIC_COMMA   = '，';

    private static final char MAGIC_COLON   = '：';

    private static String hwToIrcNick(final String nick) {

        return nick

            .replace(' ', MAGIC_SPACE)

            .replace('@', MAGIC_ATSIGN)

            .replace('%', MAGIC_PERCENT)

            .replace('+', MAGIC_PLUS)

            .replace('!', MAGIC_EXCLAM)

            .replace('，', MAGIC_COMMA)

            .replace('：', MAGIC_COLON)

            ;

    }

    private static String ircToHwNick(final String nick) {

        return nick

            .replace(MAGIC_COLON,   '：')

            .replace(MAGIC_COMMA,   '，')

            .replace(MAGIC_EXCLAM,  '!')

            .replace(MAGIC_PLUS,    '+')

            .replace(MAGIC_PERCENT, '%')

            .replace(MAGIC_ATSIGN,  '@')

            .replace(MAGIC_SPACE,   ' ')

            ;

    }

    private ProtocolConnection hwcon;

    private boolean joined = false;

    private boolean ircJoined = false;

    private void collectFurtherInfo() {

        hwcon.sendPing();

        hwcon.processNextClientFlagsMessages();

    }

    public void onPing() {

        send("PING :" + globalServerName);

    }

    public void onPong() {

        if (!ircPingQueue.isEmpty())

                send(":" + globalServerName + " PONG " + globalServerName

                        + " :" + ircPingQueue.remove(0));

    }

    public void onConnectionLoss() {

        quit("Connection Loss");

    }

    public void onDisconnect(final String reason) {

        quit(reason);

    }

    public String onPasswordHashNeededForAuth() {

        return passwordHash;

    }

    public void onMalformedMessage(String contents)

    {

        this.logError("MALFORMED MESSAGE: " + contents);

    }

    public void onChat(final String user, final String message) {

        String ircId;

        Player player = allPlayers.get(user);

        if (player == null) {

            // fake user - so probably a notice

            sendChannelNotice(message, hwToIrcNick(user));

            //logWarning("onChat(): Couldn't find player with specified nick! nick: " + user);

            //send(":" + hw404NickToIrcId(user) + " PRIVMSG "

                     //+ LOBBY_CHANNEL_NAME + " :" + hwActionToIrc(message));

        }

        else

            send(":" + player.getIrcId() + " PRIVMSG "

                     + LOBBY_CHANNEL_NAME + " :" + hwActionToIrc(message));

    }

    public void onWelcomeMessage(final String message) {

    }

    public void onNotice(int number) {

    }

    public void onBanListEntry(BanType type, String target, String duration, String reason) {

        // TODO

    }

    public void onBanListEnd() {

        // TODO

    }

    public String onNickCollision(final String nick) {

        return nick + "_";

    }

    public void onNickSet(final String nick) {

        final String newNick = hwToIrcNick(nick);

        // tell irc client

        send(":" + ownIrcNick + "!~" + username + "@"

                            + hostname + " NICK :" + nick);

        ownIrcNick = newNick;

        updateLogPrefix();

        logInfo("Nickname set to " + nick);

    }

    private void flagAsInLobby(final Player player) {

        if (!ircJoined)

            return;

        final String ircNick = player.ircNick;

        if (player.isServerAdmin())

            send(":room-part!~@~ MODE " + LOBBY_CHANNEL_NAME + " -h+o " + ircNick + " " + ircNick);

        //else

            //send(":room-part!~@~ MODE " + LOBBY_CHANNEL_NAME + " +v " + ircNick);

    }

    private void flagAsInRoom(final Player player) {

        if (!ircJoined)

            return;

        final String ircNick = player.ircNick;

        if (player.isServerAdmin())

            send(":room-join!~@~ MODE " + LOBBY_CHANNEL_NAME + " -o+h " + ircNick + " " + ircNick);

        //else

            //send(":room-join!~@~ MODE " + LOBBY_CHANNEL_NAME + " -v " + ircNick);

    }

// TODO somewhere: escape char for magic chars!

// TODO /join with playername => FOLLOW :D

    public void sendPlayerMode(final Player player) {

        char c;

        if (player.isServerAdmin())

            c = player.inRoom?'h':'o';

        else if (player.isRegistered())

            c = 'v';

        else

            // no mode

            return;

        send(":server-join!~@~ MODE " + LOBBY_CHANNEL_NAME + " +" + c + " " + player.ircNick);

    }

    private Player ownPlayer = null;

    public void onLobbyJoin(final String[] users) {

        final List<Player> newPlayers = new ArrayList<Player>(users.length);

        // process joins

        for (final String user : users) {

            final Player player = new Player(user);

            if (ownPlayer == null)

                ownPlayer = player;

            newPlayers.add(player);

            allPlayers.put(user, player);

        }

        // make sure we get the client flags before we announce anything

        collectFurtherInfo();

        // get player info

        // NOTE: if player is in room, then info was already retrieved

        for (final Player player : newPlayers) {

            if (!player.inRoom)

                hwcon.requestInfo(player.nick);

        }

        /* DISABLED - we'll announce later - when receiving info

        // announce joins

        if (ircJoined) {

            for (final Player player : newPlayers) {

                final String ircId = player.getIrcId();

                send(":" + ircId

                         + " JOIN "+ lobbyChannel.name);

                sendPlayerMode(player);

            }

        }

        */

        if (!ircJoined) {

            // don't announced players that were there before join already

            for (final Player player : newPlayers) {

                player.setAnnounced();

            }

        }

        if (!joined) {

            joined = true;

            // forget password hash, we don't need it anymore.

            passwordHash = "";

            logInfo("Hedgewars server/lobby joined.");

            sendSelfNotice("Hedgewars server was joined successfully");

            // do this after join so that rooms can be assigned to their owners

            hwcon.requestRoomsList();

        }

    }

    private void makeIrcJoinLobby() {

            sendGlobal("INVITE " + ownIrcNick + " " + LOBBY_CHANNEL_NAME);

            try{Thread.sleep(3000);}catch(Exception e){}

            join(lobbyChannel.name);

            sendSelfNotice("Joining lobby-channel: " + lobbyChannel.name);

    }

    private void announcePlayerJoinLobby(final Player player) {

            player.setAnnounced();

            send(":" + player.getIrcId()

                     + " JOIN "+ lobbyChannel.name);

            sendPlayerMode(player);

    }

    public void onLobbyLeave(final String user, final String reason) {

        final Player player = allPlayers.get(user);

        if (player == null) {

            logWarning("onLobbyLeave(): Couldn't find player with specified nick! nick: " + user);

            sendIfJoined(":" + hw404NickToIrcId(user)

                 + " PART " + lobbyChannel.name + " " + reason);

        }

        else {

            if (ircJoined && player.needsAnnounce())

                announcePlayerJoinLobby(player);

            sendIfJoined(":" + player.getIrcId()

                 + " PART " + lobbyChannel.name + " " + reason);

            allPlayers.remove(user);

        }

    }

    private int lastRoomId = 0;

    public void onRoomInfo(final String name, final String flags,

                           final String newName, final int nUsers,

                           final int nTeams, final String owner,

                           final String map, final String style,

                           final String scheme, final String weapons) {

        Room room = roomsByName.get(name);

        if (room == null) {

            // try to reuse old ids

            if (lastRoomId >= 90)

                lastRoomId = 9;

            // search for first free

            while(roomsById.containsKey(++lastRoomId)) { }

            room = new Room(lastRoomId, newName, owner);

            roomsById.put(lastRoomId, room);

            roomsByName.put(newName, room);

            roomsSorted.add(room);

        }

        else if (!room.name.equals(newName)) {

            room.name = newName;

            roomsByName.put(newName, roomsByName.remove(name));

        }

        // update data

        room.setOwner(owner);

        room.nPlayers = nUsers;

        room.nTeams = nTeams;

    }

    public void onRoomDel(final String name) {

        final Room room = roomsByName.remove(name);

        if (room != null) {

            roomsById.remove(room.id);

            roomsSorted.remove(room);

        }

    }