Logon.cpp

#include "Logon.h"
#include "Game.h"
#include "PlayerDatabase.h"
#include "BasicLib.h"

using namespace SocketLib;

namespace SimpleMUD
{
    void Logon::Handle(string p_data)
    {
        if (m_errors == 5)
        {
            m_connection->Protocol().SendString(*m_connection, red + bold +
                "Too many incorrect responses, closing connection..." +
                newline);
            m_connection->Close();
            return;
        }
        if (m_state == NEWCONNECTION)  // has not entered name yet
        {
            if (BasicLib::LowerCase(p_data) == "new")
            {
                m_state = NEWUSER;
                m_connection->Protocol().SendString(*m_connection, yellow +
                    "Please enter your desired name: " + reset);
            }
            else
            {
                PlayerDatabase::iterator itr = PlayerDatabase::findfull(p_data);
                if (itr == PlayerDatabase::end())
                {
                    std::cout << "first player: " << std::endl;
                    PlayerDatabase::PrintPlayer(1);
                    std::cout << p_data << "No such player!" << "\n";
                    // name does not exist
                    m_errors++;
                    m_connection->Protocol().SendString(*m_connection,
                        red + bold + "Sorry, the user \"" + white + p_data + red +
                        "\" does not exist.\r\n" + 
                        "Please enter your name, or \"new\" if you are new: " +
                        reset);
                }
                else
                {
                    // name exists, go to password entrance.
                    m_state = ENTERPASS;
                    m_name = p_data;
                    m_password = itr->Password();

                    m_connection->Protocol().SendString(*m_connection,
                        green + bold + "Welcome, " + white + p_data + red +
                        newline + green + "Please enter your password: " +
                        reset);
                }
            }
            return;
        }
        if (m_state == NEWUSER)
        {
            // check if the name is taken:
            if (PlayerDatabase::hasfull(p_data))
            {
                m_errors++;
                m_connection->Protocol().SendString(*m_connection,
                    red + bold + "Sorry, the name \"" + white + p_data + red +
                    "\" has already been taken." + newline + yellow + 
                    "Please enter your desired name: " + reset);
            }
            else
            {
                if (!AcceptibleName(p_data))
                {
                    m_errors++;
                    m_connection->Protocol().SendString(*m_connection,
                        red + bold + "Sorry, the name \"" + white + p_data + red +
                        "\" is unacceptible." + newline + yellow + 
                        "Please enter your desired name: " + reset);
                }
                else
                {
                    m_state = ENTERNEWPASS;
                    m_name = p_data;
                    m_connection->Protocol().SendString(*m_connection,
                        green + "Please enter your desired password: " +
                        reset);
                }
            }
            return;
        }
        if (m_state == ENTERNEWPASS)
        {
            if (p_data.find_first_of(BasicLib::WHITESPACE) != string::npos)
            {
                m_errors++;
                m_connection->Protocol().SendString(*m_connection,
                    red + bold + "INVALID PASSWORD!" +
                    green + "Please enter your desired password: " +
                    reset);
                return;
            }
            m_connection->Protocol().SendString(*m_connection,
                green + "Thank you! You are now entering the realm..." +
                newline);
            Player p;
            p.Name() = m_name;
            std::cout << "New username: " << p.Name() << std::endl;
            p.Password() = p_data;
            // make the player the administrator if he's the first to log in.
            if (PlayerDatabase::size() == 0)
            {
                p.Rank() = ADMIN;
                p.ID() = 1;
            }
            else
            {
                p.ID() = PlayerDatabase::LastID() + 1;
            }
            // add the player
            PlayerDatabase::AddPlayer(p);
            // enter the game as a newbie.
            GotoGame(true);
            return;
        }
        if (m_state == ENTERPASS)
        {
            if (m_password == p_data)
            {
                m_connection->Protocol().SendString(*m_connection,
                    green + "Thank you! You are now entering the realm..." +
                    newline);

                // enter the game
                GotoGame();
            }
            else
            {
                m_errors++;
                m_connection->Protocol().SendString(*m_connection,
                    red + bold + "INVALID PASSWORD!" + newline +
                    yellow + "Please enter your password: " +
                    reset);
            }
            return;
        }
    }

	void Logon::Enter()
	{
		USERLOG.Log(
			GetIPString(m_connection->GetRemoteAddress()) +
			" - entered login state.");
		m_connection->Protocol().SendString(*m_connection,
			red + bold + "Welcome To SimpleMUD v.ckw\r\n" +
			"Please enter your name, or \"new\" if you are new: " + reset);
	}

	void Logon::GotoGame(bool p_newbie)
	{
        std::cout << m_name << "Welcome!" << std::endl;

		Player& p = *(PlayerDatabase::findfull(m_name));
        std::cout << p.Name() << "Welcome!" << std::endl;
		if (p.LoggedIn())
		{
			p.Conn()->Close();
			p.Conn()->Handler()->Hungup();
			p.Conn()->ClearHandlers();
		}
		p.Newbie() = p_newbie;
		p.Conn() = m_connection;
		p.Conn()->SwitchHandler(new Game(*p.Conn(), p.ID()));
	}

	bool Logon::AcceptibleName(const string& p_name)
	{
		static string inv = " \"'~!@#$%^&*+/\\[]{}<>()=.,?;:";
			// must not contain any invalid characters
			if (p_name.find_first_of(inv) != string::npos)
				return false;
		// must be less than 17 chars and more than 2
		if (p_name.size() > 16 || p_name.size() < 3)
			return false;
		// must start with an alphabetical character
		if (!isalpha(p_name[0]))
			return false;
		if (p_name == "new")
			return false;
		return true;
	}
}