SimpleChatServer.h

Go to the documentation of this file.

// This may look like C code, but it's really -*- C++ -*-
/*
 * Copyright (C) 2008 Emweb bv, Herent, Belgium.
 *
 * See the LICENSE file for terms of use.
 */
#ifndef SIMPLECHATSERVER_H_
#define SIMPLECHATSERVER_H_
 
#include <Wt/WSignal.h>
#include <Wt/WString.h>
 
namespace Wt {
  class WServer;
}
 
#include <set>
#include <map>
#include <thread>
#include <mutex>
 
 
class ChatEvent
{
public:
  enum Type { Login, Logout, Rename, Message };
 
  Type type() const { return type_; }
 
  const Wt::WString& user() const { return user_; }
 
  const Wt::WString& message() const { return message_; }
 
  const Wt::WString& data() const { return data_; }
 
  const Wt::WString formattedHTML(const Wt::WString& user,
                                  Wt::TextFormat format) const;
 
private:
  Type    type_;
  Wt::WString user_;
  Wt::WString data_;
  Wt::WString message_;
 
  /*
   * Both user and html will be formatted as html
   */
  ChatEvent(const Wt::WString& user, const Wt::WString& message)
    : type_(Message), user_(user), message_(message)
  { }
 
  ChatEvent(Type type, const Wt::WString& user,
            const Wt::WString& data = Wt::WString::Empty)
    : type_(type), user_(user), data_(data)
  { }
 
  friend class SimpleChatServer;
};
 
typedef std::function<void (const ChatEvent&)> ChatEventCallback;
 
class SimpleChatServer
{
public:
  /*
   * A reference to a client.
   */
  class Client
  {
  };
 
  SimpleChatServer(Wt::WServer& server);
 
  SimpleChatServer(const SimpleChatServer &) = delete;
  SimpleChatServer &operator=(const SimpleChatServer &) = delete;
 
  bool connect(Client *client, const ChatEventCallback& handleEvent);
 
  bool disconnect(Client *client);
 
  bool login(const Wt::WString& user);
 
  void logout(const Wt::WString& user);
 
  bool changeName(const Wt::WString& user, const Wt::WString& newUser);
 
  Wt::WString suggestGuest();
 
  void sendMessage(const Wt::WString& user, const Wt::WString& message);
 
  typedef std::set<Wt::WString> UserSet;
 
  UserSet users();
 
private:
  struct ClientInfo {
    std::string       sessionId;
    ChatEventCallback eventCallback;
  };
 
  typedef std::map<Client *, ClientInfo> ClientMap;
 
  Wt::WServer&                server_;
  std::recursive_mutex    mutex_;
  ClientMap               clients_;
  UserSet                 users_;
 
  void postChatEvent(const ChatEvent& event);
};
 
#endif // SIMPLECHATSERVER_H_