/*
 * UserBackendMysql.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 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_MODULES_USERBACKENDMYSQL_USERBACKENDMYSQL_H_
#define MAD_MODULES_USERBACKENDMYSQL_USERBACKENDMYSQL_H_

#include <Common/UserBackend.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.hpp>

namespace Mad {
namespace Modules {
namespace UserBackendMysql {

class UserBackendMysql : public Common::UserBackend, private Core::Configurable, private boost::noncopyable {
  private:
    typedef std::map<std::string, boost::variant<std::string, unsigned long> > ArgumentMap;

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

    class Result {
      private:
        friend Result UserBackendMysql::query(const std::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());
        }
    };

    Common::Application *application;

    std::string host, username, passwd, db, unixSocket;
    uint16_t port;

    std::string queryListUsers, queryListGroups;
    std::string queryListUserGroups, queryListGroupUsers;
    std::string queryUserById, queryUserByName;
    std::string queryGroupById, queryGroupByName;
    std::string queryUserGroupTable;
    std::string queryAddUser;

    MYSQL *mysql;

    boost::posix_time::ptime lastUpdate;

    boost::mutex mutex;

  protected:
    virtual bool handleConfigEntry(const Core::ConfigEntry &entry, bool);
    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 std::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 std::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);

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

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

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

}
}
}

#endif /* MAD_MODULES_USERBACKENDMYSQL_USERBACKENDMYSQL_H_ */