/home/koen/project/wt/public-git/wt/examples/simplechat/SimpleChatServer.h

Go to the documentation of this file.

// This may look like C code, but it's really -*- C++ -*-
/*
 * Copyright (C) 2008 Emweb bvba, Heverlee, Belgium.
 *
 * See the LICENSE file for terms of use.
 */
#ifndef SIMPLECHATSERVER_H_
#define SIMPLECHATSERVER_H_

#include <boost/noncopyable.hpp>

#include <Wt/WSignal>
#include <Wt/WString>

namespace Wt {
  class WServer;
}

#include <set>
#include <map>
#include <boost/thread.hpp>


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) 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 boost::function<void (const ChatEvent&)> ChatEventCallback;

class SimpleChatServer : boost::noncopyable
{
public:
  /*
   * A reference to a client.
   */
  class Client
  {
  };

  SimpleChatServer(Wt::WServer& server);

  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_;
  boost::recursive_mutex mutex_;
  ClientMap clients_;
  UserSet users_;

  void postChatEvent(const ChatEvent& event);
};

#endif // SIMPLECHATSERVER_H_