/*
 * UserDBBackendMysql.h
 *
 * Copyright (C) 2008 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_MODULES_USERDBBACKENDMYSQL_USERDBBACKENDMYSQL_H_
#define MAD_MODULES_USERDBBACKENDMYSQL_USERDBBACKENDMYSQL_H_

#include <Common/UserDBBackend.h>
#include <Common/Application.h>

#include <Core/Configurable.h>
#include <Core/ConfigManager.h>

#include <mysql/mysql.h>

#include <boost/thread/mutex.hpp>
#include <boost/variant/variant.hpp>

namespace Mad {
namespace Modules {
namespace UserDBBackendMysql {

class UserDBBackendMysql : public Common::UserDBBackend, private Core::Configurable, private boost::noncopyable {
  private:
    typedef std::map<Core::String, boost::variant<Core::String, unsigned long> > ArgumentMap;

    class Result;
    Result query(const Core::String &query, const ArgumentMap &args = ArgumentMap()) throw(Core::Exception);

    class Result {
      private:
        friend Result UserDBBackendMysql::query(const Core::String &query, const ArgumentMap &args) throw(Core::Exception);

        boost::shared_ptr<MYSQL_RES> result;

        unsigned int mysqlErrno;
        const char *error;

        Result(MYSQL *mysql) {
          result = boost::shared_ptr<MYSQL_RES>(mysql_store_result(mysql), mysql_free_result);

          mysqlErrno = mysql_errno(mysql);
          error = mysql_error(mysql);
        }

      public:
        operator bool() const {
          return result;
        }

        unsigned int getErrno() const {
          return mysqlErrno;
        }

        const char* getError() const {
          return error;
        }

        unsigned int getFieldNumber() const {
          return mysql_num_fields(result.get());
        }

        my_ulonglong getRowNumber() const {
          return mysql_num_rows(result.get());
        }

        MYSQL_ROW getNextRow() {
          return mysql_fetch_row(result.get());
        }
    };

    static const Core::String name;

    Common::Application *application;

    Core::String host, username, passwd, db, unixSocket;
    uint16_t port;

    Core::String queryListUsers, queryListGroups;
    Core::String queryListUserGroups, queryListGroupUsers;
    Core::String queryUserById, queryUserByName;
    Core::String queryGroupById, queryGroupByName;
    Core::String queryUserGroupTable;
    Core::String queryAddUser, queryUpdateUser, queryDeleteUser;
    Core::String queryAddGroup, queryUpdateGroup, queryDeleteGroup;
    Core::String queryAddUserToGroup, queryDeleteUserFromGroup;

    MYSQL *mysql;

    boost::posix_time::ptime lastUpdate;

    boost::mutex mutex;

  protected:
    virtual bool handleConfigEntry(const Core::ConfigEntry &entry, bool handled);
    virtual void configFinished();

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

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

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

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

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

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

  public:
    UserDBBackendMysql(Common::Application *application0) : application(application0), port(0), mysql(0), lastUpdate(boost::posix_time::microsec_clock::universal_time()) {
      application->getConfigManager()->registerConfigurable(this);
    }

    virtual ~UserDBBackendMysql() {
      application->getConfigManager()->unregisterConfigurable(this);

      if(mysql) {
        mysql_close(mysql);
        mysql = 0;
      }
    }

    virtual const Core::String& getName() {
      return name;
    }
};

}
}
}

#endif /* MAD_MODULES_USERDBBACKENDMYSQL_USERBACKENDMYSQL_H_ */