/*
 * 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 Lesser 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef MAD_COMMON_USERMANAGER_H_
#define MAD_COMMON_USERMANAGER_H_

#include "export.h"

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

#include <Core/Configurable.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 UserConfigBackend;
class UserDBBackend;
class UserCache;

class MAD_COMMON_EXPORT UserManager : public Core::Configurable, private boost::noncopyable {
  private:
    friend class Application;

    typedef std::map<boost::shared_ptr<UserDBBackend>, std::pair<boost::shared_ptr<UserDBBackend>, Core::String> > BackendMap;

    Application *application;

    std::set<boost::shared_ptr<UserConfigBackend> > configBackends;

    BackendMap dbBackends;
    std::map<Core::String, boost::shared_ptr<UserDBBackend> > dbBackendNames;

    boost::shared_ptr<UserDBBackend> dbBackend;

    unsigned long minUid, maxUid, minGid, maxGid;

    boost::shared_mutex mutex;

    UserManager(Application *application0);
    ~UserManager();

  protected:
    virtual void configure();

  public:
    void registerBackend(boost::shared_ptr<UserDBBackend> backend);
    void registerBackendCached(boost::shared_ptr<UserDBBackend> backend);

    void unregisterBackend(boost::shared_ptr<UserDBBackend> backend) {
      boost::lock_guard<boost::shared_mutex> lock(mutex);

      BackendMap::iterator it = dbBackends.find(backend);
      if(it == dbBackends.end())
        return;

      boost::shared_ptr<UserDBBackend> oldBackend = it->second.first;

      std::map<Core::String, boost::shared_ptr<UserDBBackend> >::iterator it2 = dbBackendNames.find(it->second.second);
      if(it2->second == backend)
        dbBackendNames.erase(it2);

      dbBackends.erase(it);

      if(dbBackend == oldBackend) {
        if(dbBackends.empty())
          dbBackend.reset();
        else
          dbBackend = dbBackends.begin()->second.first;
      }
    }


    void registerBackend(boost::shared_ptr<UserConfigBackend> backend) {
      boost::lock_guard<boost::shared_mutex> lock(mutex);
      configBackends.insert(backend);
    }

    void unregisterBackend(boost::shared_ptr<UserConfigBackend> backend) {
      boost::lock_guard<boost::shared_mutex> lock(mutex);
      configBackends.erase(backend);
    }

    boost::shared_ptr<const std::map<unsigned long, UserInfo> > getUserList(boost::posix_time::ptime *timestamp = 0) throw(Core::Exception);
    boost::shared_ptr<const UserInfo> getUserInfo(unsigned long uid, boost::posix_time::ptime *timestamp = 0) throw(Core::Exception);
    boost::shared_ptr<const UserInfo> getUserInfoByName(const Core::String &name, boost::posix_time::ptime *timestamp = 0) throw(Core::Exception);
    boost::shared_ptr<const std::set<unsigned long> > getUserGroupList(unsigned long uid, boost::posix_time::ptime *timestamp = 0) throw(Core::Exception);

    boost::shared_ptr<const std::map<unsigned long, GroupInfo> > getGroupList(boost::posix_time::ptime *timestamp = 0) throw(Core::Exception);
    boost::shared_ptr<const GroupInfo> getGroupInfo(unsigned long gid, boost::posix_time::ptime *timestamp = 0) throw(Core::Exception);
    boost::shared_ptr<const GroupInfo> getGroupInfoByName(const Core::String &name, boost::posix_time::ptime *timestamp = 0) throw(Core::Exception);
    boost::shared_ptr<const std::set<unsigned long> > getGroupUserList(unsigned long gid, boost::posix_time::ptime *timestamp = 0) throw(Core::Exception);

    boost::shared_ptr<const std::multimap<unsigned long, unsigned long> > getFullUserGroupList(boost::posix_time::ptime *timestamp = 0) throw(Core::Exception);

    void checkUserInfo(const UserInfo &userInfo) throw(Core::Exception);
    void addUser(const UserInfo &userInfo) throw(Core::Exception);
    void updateUser(unsigned long uid, const UserInfo &userInfo) throw(Core::Exception);
    void deleteUser(unsigned long uid) throw(Core::Exception);

    void checkGroupInfo(const GroupInfo &groupInfo) throw(Core::Exception);
    void addGroup(const GroupInfo &groupInfo) throw(Core::Exception);
    void updateGroup(unsigned long gid, const GroupInfo &groupInfo) throw(Core::Exception);
    void deleteGroup(unsigned long gid) throw(Core::Exception);

    void addUserToGroup(unsigned long uid, unsigned long gid) throw(Core::Exception);
    void deleteUserFromGroup(unsigned long uid, unsigned long gid) throw(Core::Exception);

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

}
}

#endif /* MAD_COMMON_USERMANAGER_H_ */