/*
 * UserManager.h
 *
 * Copyright (C) 2009 Matthias Schiffer <matthias@gamezock.de>
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef MAD_SERVER_USERMANAGER_H_
#define MAD_SERVER_USERMANAGER_H_

#include "UserInfo.h"
#include "GroupInfo.h"

#include <Core/Exception.h>

#include <map>
#include <set>

#include <boost/thread/locks.hpp>
#include <boost/thread/shared_mutex.hpp>

namespace Mad {
namespace Common {

class Application;
class UserBackend;

class UserManager : boost::noncopyable {
  private:
    friend class Application;

    struct Compare {
      bool operator() (boost::shared_ptr<UserBackend> b1, boost::shared_ptr<UserBackend> b2);
    };

    std::set<boost::shared_ptr<UserBackend>, Compare> backends;

    boost::shared_mutex backendMutex;
    boost::shared_mutex mutex;

    bool caching;

    boost::shared_ptr<const std::map<unsigned long, UserInfo> > users;
    boost::shared_ptr<const std::map<std::string, unsigned long> > userNames;
    boost::shared_ptr<Core::Exception> userException;

    boost::shared_ptr<const std::map<unsigned long, GroupInfo> > groups;
    boost::shared_ptr<const std::map<std::string, unsigned long> > groupNames;
    boost::shared_ptr<Core::Exception> groupException;

    boost::shared_ptr<const std::multimap<unsigned long, unsigned long> > userGroups;
    boost::shared_ptr<const std::multimap<unsigned long, unsigned long> > groupUsers;
    boost::shared_ptr<Core::Exception> userGroupException;

    UserManager() : caching(false) {}

  public:
    void registerBackend(boost::shared_ptr<UserBackend> backend) {
      {
        boost::lock_guard<boost::shared_mutex> lock(backendMutex);
        backends.insert(backend);
      }

      update();
    }

    void unregisterBackend(boost::shared_ptr<UserBackend> backend) {
      {
        boost::lock_guard<boost::shared_mutex> lock(backendMutex);
        backends.erase(backend);
      }

      update();
    }

    bool isCaching() {
      boost::shared_lock<boost::shared_mutex> lock(mutex);
      return caching;
    }

    void setCaching(bool cache) {
      {
        boost::lock_guard<boost::shared_mutex> lock(mutex);
        caching = cache;
      }

      update();
    }

    void update(bool refresh = false) {
      {
        boost::lock_guard<boost::shared_mutex> lock(mutex);

        if(!caching)
          return;

        users.reset();
        userNames.reset();
        userException.reset();

        groups.reset();
        groupNames.reset();
        groupException.reset();

        userGroups.reset();
        groupUsers.reset();
        userGroupException.reset();
      }

      if(refresh) {
        getUserList();
        getGroupList();
      }
    }

    boost::shared_ptr<const std::map<unsigned long, UserInfo> > getUserList() throw(Core::Exception);
    boost::shared_ptr<const UserInfo> getUserInfo(unsigned long uid) throw(Core::Exception);
    boost::shared_ptr<const UserInfo> getUserInfoByName(const std::string &name) throw(Core::Exception);
    boost::shared_ptr<const std::set<unsigned long> > getUserGroupList(unsigned long uid) throw(Core::Exception);

    boost::shared_ptr<const std::map<unsigned long, GroupInfo> > getGroupList() throw(Core::Exception);
    boost::shared_ptr<const GroupInfo> getGroupInfo(unsigned long gid) throw(Core::Exception);
    boost::shared_ptr<const GroupInfo> getGroupInfoByName(const std::string &name) throw(Core::Exception);
    boost::shared_ptr<const std::set<unsigned long> > getGroupUserList(unsigned long gid) throw(Core::Exception);

    void setPassword(unsigned long uid, const std::string &password) throw(Core::Exception);

    void addUser(const UserInfo &userInfo) throw(Core::Exception);
};

}
}

#endif /* MAD_SERVER_USERMANAGER_H_ */