mirror of
https://github.com/haiwen/ccnet-server.git
synced 2025-07-19 07:46:27 +00:00
786 lines
25 KiB
C
786 lines
25 KiB
C
|
/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
|
||
|
|
|
||
|
|
#include "common.h"
|
||
|
|
|
||
|
|
#include "server-session.h"
|
||
|
|
|
||
|
|
#include "ccnet-db.h"
|
||
|
|
#include "group-mgr.h"
|
||
|
|
#include "org-mgr.h"
|
||
|
|
|
||
|
|
#include "utils.h"
|
||
|
|
#include "log.h"
|
||
|
|
|
||
|
|
struct _CcnetGroupManagerPriv {
|
||
|
|
CcnetDB *db;
|
||
|
|
};
|
||
|
|
|
||
|
|
static int open_db (CcnetGroupManager *manager);
|
||
|
|
static int check_db_table (CcnetDB *db);
|
||
|
|
|
||
|
|
CcnetGroupManager* ccnet_group_manager_new (CcnetSession *session)
|
||
|
|
{
|
||
|
|
CcnetGroupManager *manager = g_new0 (CcnetGroupManager, 1);
|
||
|
|
|
||
|
|
manager->session = session;
|
||
|
|
manager->priv = g_new0 (CcnetGroupManagerPriv, 1);
|
||
|
|
|
||
|
|
return manager;
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
ccnet_group_manager_prepare (CcnetGroupManager *manager)
|
||
|
|
{
|
||
|
|
return open_db(manager);
|
||
|
|
}
|
||
|
|
|
||
|
|
void ccnet_group_manager_start (CcnetGroupManager *manager)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
static CcnetDB *
|
||
|
|
open_sqlite_db (CcnetGroupManager *manager)
|
||
|
|
{
|
||
|
|
CcnetDB *db = NULL;
|
||
|
|
char *db_dir;
|
||
|
|
char *db_path;
|
||
|
|
|
||
|
|
db_dir = g_build_filename (manager->session->config_dir, "GroupMgr", NULL);
|
||
|
|
if (checkdir_with_mkdir(db_dir) < 0) {
|
||
|
|
ccnet_error ("Cannot open db dir %s: %s\n", db_dir,
|
||
|
|
strerror(errno));
|
||
|
|
g_free (db_dir);
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
g_free (db_dir);
|
||
|
|
|
||
|
|
db_path = g_build_filename (manager->session->config_dir, "GroupMgr",
|
||
|
|
"groupmgr.db", NULL);
|
||
|
|
db = ccnet_db_new_sqlite (db_path);
|
||
|
|
|
||
|
|
g_free (db_path);
|
||
|
|
|
||
|
|
return db;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int
|
||
|
|
open_db (CcnetGroupManager *manager)
|
||
|
|
{
|
||
|
|
CcnetDB *db = NULL;
|
||
|
|
|
||
|
|
switch (ccnet_db_type(manager->session->db)) {
|
||
|
|
case CCNET_DB_TYPE_SQLITE:
|
||
|
|
db = open_sqlite_db (manager);
|
||
|
|
break;
|
||
|
|
case CCNET_DB_TYPE_PGSQL:
|
||
|
|
case CCNET_DB_TYPE_MYSQL:
|
||
|
|
db = manager->session->db;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!db)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
manager->priv->db = db;
|
||
|
|
return check_db_table (db);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* -------- Group Database Management ---------------- */
|
||
|
|
|
||
|
|
static int check_db_table (CcnetDB *db)
|
||
|
|
{
|
||
|
|
char *sql;
|
||
|
|
|
||
|
|
int db_type = ccnet_db_type (db);
|
||
|
|
if (db_type == CCNET_DB_TYPE_MYSQL) {
|
||
|
|
sql = "CREATE TABLE IF NOT EXISTS `Group` (`group_id` INTEGER"
|
||
|
|
" PRIMARY KEY AUTO_INCREMENT, `group_name` VARCHAR(255),"
|
||
|
|
" `creator_name` VARCHAR(255), `timestamp` BIGINT,"
|
||
|
|
" `type` VARCHAR(32))"
|
||
|
|
"ENGINE=INNODB";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
sql = "CREATE TABLE IF NOT EXISTS `GroupUser` (`group_id` INTEGER,"
|
||
|
|
" `user_name` VARCHAR(255), `is_staff` tinyint, PRIMARY KEY"
|
||
|
|
" (`group_id`, `user_name`), INDEX (`user_name`))"
|
||
|
|
"ENGINE=INNODB";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
sql = "CREATE TABLE IF NOT EXISTS GroupDNPair (group_id INTEGER,"
|
||
|
|
" dn VARCHAR(255))ENGINE=INNODB";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
} else if (db_type == CCNET_DB_TYPE_SQLITE) {
|
||
|
|
sql = "CREATE TABLE IF NOT EXISTS `Group` (`group_id` INTEGER"
|
||
|
|
" PRIMARY KEY AUTOINCREMENT, `group_name` VARCHAR(255),"
|
||
|
|
" `creator_name` VARCHAR(255), `timestamp` BIGINT,"
|
||
|
|
" `type` VARCHAR(32))";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
sql = "CREATE TABLE IF NOT EXISTS `GroupUser` (`group_id` INTEGER, "
|
||
|
|
"`user_name` VARCHAR(255), `is_staff` tinyint)";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
sql = "CREATE UNIQUE INDEX IF NOT EXISTS groupid_username_indx on "
|
||
|
|
"`GroupUser` (`group_id`, `user_name`)";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
sql = "CREATE INDEX IF NOT EXISTS username_indx on "
|
||
|
|
"`GroupUser` (`user_name`)";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
sql = "CREATE TABLE IF NOT EXISTS GroupDNPair (group_id INTEGER,"
|
||
|
|
" dn VARCHAR(255))";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
} else if (db_type == CCNET_DB_TYPE_PGSQL) {
|
||
|
|
sql = "CREATE TABLE IF NOT EXISTS \"Group\" (group_id SERIAL"
|
||
|
|
" PRIMARY KEY, group_name VARCHAR(255),"
|
||
|
|
" creator_name VARCHAR(255), timestamp BIGINT,"
|
||
|
|
" type VARCHAR(32))";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
sql = "CREATE TABLE IF NOT EXISTS GroupUser (group_id INTEGER,"
|
||
|
|
" user_name VARCHAR(255), is_staff smallint, UNIQUE "
|
||
|
|
" (group_id, user_name))";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
if (!pgsql_index_exists (db, "groupuser_username_idx")) {
|
||
|
|
sql = "CREATE INDEX groupuser_username_idx ON GroupUser (user_name)";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
sql = "CREATE TABLE IF NOT EXISTS GroupDNPair (group_id INTEGER,"
|
||
|
|
" dn VARCHAR(255))";
|
||
|
|
if (ccnet_db_query (db, sql) < 0)
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int
|
||
|
|
create_group_common (CcnetGroupManager *mgr,
|
||
|
|
const char *group_name,
|
||
|
|
const char *user_name,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
gint64 now = get_current_time();
|
||
|
|
char *sql;
|
||
|
|
int group_id = -1;
|
||
|
|
|
||
|
|
char *user_name_l = g_ascii_strdown (user_name, -1);
|
||
|
|
|
||
|
|
if (ccnet_db_type(db) == CCNET_DB_TYPE_PGSQL)
|
||
|
|
sql = "INSERT INTO \"Group\"(group_name, "
|
||
|
|
"creator_name, timestamp) VALUES(?, ?, ?)";
|
||
|
|
else
|
||
|
|
sql = "INSERT INTO `Group`(group_name, "
|
||
|
|
"creator_name, timestamp) VALUES(?, ?, ?)";
|
||
|
|
|
||
|
|
if (ccnet_db_statement_query (db, sql, 3,
|
||
|
|
"string", group_name, "string", user_name_l,
|
||
|
|
"int64", now) < 0) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0, "Failed to create group");
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (ccnet_db_type(db) == CCNET_DB_TYPE_PGSQL)
|
||
|
|
sql = "SELECT group_id FROM \"Group\" WHERE "
|
||
|
|
"group_name = ? AND creator_name = ? "
|
||
|
|
"AND timestamp = ?";
|
||
|
|
else
|
||
|
|
sql = "SELECT group_id FROM `Group` WHERE "
|
||
|
|
"group_name = ? AND creator_name = ? "
|
||
|
|
"AND timestamp = ?";
|
||
|
|
|
||
|
|
group_id = ccnet_db_statement_get_int (db, sql, 3,
|
||
|
|
"string", group_name, "string", user_name_l,
|
||
|
|
"int64", now);
|
||
|
|
if (group_id < 0) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0, "Failed to create group");
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
|
||
|
|
sql = "INSERT INTO GroupUser VALUES (?, ?, ?)";
|
||
|
|
|
||
|
|
if (ccnet_db_statement_query (db, sql, 3,
|
||
|
|
"int", group_id, "string", user_name_l,
|
||
|
|
"int", 1) < 0) {
|
||
|
|
if (ccnet_db_type(db) == CCNET_DB_TYPE_PGSQL)
|
||
|
|
sql = "DELETE FROM \"Group\" WHERE group_id=?";
|
||
|
|
else
|
||
|
|
sql = "DELETE FROM `Group` WHERE group_id=?";
|
||
|
|
ccnet_db_statement_query (db, sql, 1, "int", group_id);
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0, "Failed to create group");
|
||
|
|
group_id = -1;
|
||
|
|
goto out;
|
||
|
|
}
|
||
|
|
|
||
|
|
out:
|
||
|
|
g_free (user_name_l);
|
||
|
|
return group_id;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* static gboolean */
|
||
|
|
/* duplicate_group_name (CcnetGroupManager *mgr, */
|
||
|
|
/* const char *group_name, */
|
||
|
|
/* const char *user_name) */
|
||
|
|
/* { */
|
||
|
|
/* GList *groups = NULL, *ptr; */
|
||
|
|
/* CcnetOrgManager *org_mgr = NULL; */
|
||
|
|
/* gboolean ret = FALSE; */
|
||
|
|
|
||
|
|
/* groups = ccnet_group_manager_get_all_groups (mgr, -1, -1, NULL); */
|
||
|
|
/* if (!groups) { */
|
||
|
|
/* return FALSE; */
|
||
|
|
/* } */
|
||
|
|
|
||
|
|
/* for (ptr = groups; ptr; ptr = ptr->next) { */
|
||
|
|
/* CcnetGroup *group = (CcnetGroup *)ptr->data; */
|
||
|
|
/* org_mgr = ((CcnetServerSession *)(mgr->session))->org_mgr; */
|
||
|
|
/* if (ccnet_org_manager_is_org_group(org_mgr, ccnet_group_get_id (group), */
|
||
|
|
/* NULL)) { */
|
||
|
|
/* /\* Skip org groups. *\/ */
|
||
|
|
/* continue; */
|
||
|
|
/* } */
|
||
|
|
|
||
|
|
/* if (g_strcmp0 (group_name, ccnet_group_get_group_name (group)) == 0) { */
|
||
|
|
/* ret = TRUE; */
|
||
|
|
/* goto out; */
|
||
|
|
/* } */
|
||
|
|
/* } */
|
||
|
|
|
||
|
|
/* out: */
|
||
|
|
/* for (ptr = groups; ptr; ptr = ptr->next) */
|
||
|
|
/* g_object_unref ((GObject *)ptr->data); */
|
||
|
|
/* g_list_free (groups); */
|
||
|
|
/* return ret; */
|
||
|
|
/* } */
|
||
|
|
|
||
|
|
int ccnet_group_manager_create_group (CcnetGroupManager *mgr,
|
||
|
|
const char *group_name,
|
||
|
|
const char *user_name,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
|
||
|
|
/* if (duplicate_group_name (mgr, group_name, user_name)) { */
|
||
|
|
/* g_set_error (error, CCNET_DOMAIN, 0, "The group has already created"); */
|
||
|
|
/* return -1; */
|
||
|
|
/* } */
|
||
|
|
|
||
|
|
return create_group_common (mgr, group_name, user_name, error);
|
||
|
|
}
|
||
|
|
|
||
|
|
static gboolean
|
||
|
|
duplicate_org_group_name (CcnetGroupManager *mgr,
|
||
|
|
int org_id,
|
||
|
|
const char *group_name)
|
||
|
|
{
|
||
|
|
GList *org_groups = NULL, *ptr;
|
||
|
|
CcnetOrgManager *org_mgr = ((CcnetServerSession *)(mgr->session))->org_mgr;
|
||
|
|
|
||
|
|
org_groups = ccnet_org_manager_get_org_groups (org_mgr, org_id, -1, -1);
|
||
|
|
if (!org_groups)
|
||
|
|
return FALSE;
|
||
|
|
|
||
|
|
for (ptr = org_groups; ptr; ptr = ptr->next) {
|
||
|
|
int group_id = (int)(long)ptr->data;
|
||
|
|
CcnetGroup *group = ccnet_group_manager_get_group (mgr, group_id,
|
||
|
|
NULL);
|
||
|
|
if (!group)
|
||
|
|
continue;
|
||
|
|
|
||
|
|
if (g_strcmp0 (group_name, ccnet_group_get_group_name(group)) == 0) {
|
||
|
|
g_list_free (org_groups);
|
||
|
|
g_object_unref (group);
|
||
|
|
return TRUE;
|
||
|
|
} else {
|
||
|
|
g_object_unref (group);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
g_list_free (org_groups);
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
int ccnet_group_manager_create_org_group (CcnetGroupManager *mgr,
|
||
|
|
int org_id,
|
||
|
|
const char *group_name,
|
||
|
|
const char *user_name,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetOrgManager *org_mgr = ((CcnetServerSession *)(mgr->session))->org_mgr;
|
||
|
|
|
||
|
|
if (duplicate_org_group_name (mgr, org_id, group_name)) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0,
|
||
|
|
"The group has already created in this org.");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
int group_id = create_group_common (mgr, group_name, user_name, error);
|
||
|
|
if (group_id < 0) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0, "Failed to create org group.");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (ccnet_org_manager_add_org_group (org_mgr, org_id, group_id,
|
||
|
|
error) < 0) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0, "Failed to create org group.");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
return group_id;
|
||
|
|
}
|
||
|
|
|
||
|
|
static gboolean
|
||
|
|
check_group_staff (CcnetDB *db, int group_id, const char *user_name)
|
||
|
|
{
|
||
|
|
return ccnet_db_statement_exists (db, "SELECT group_id FROM GroupUser WHERE "
|
||
|
|
"group_id = ? AND user_name = ? AND "
|
||
|
|
"is_staff = 1",
|
||
|
|
2, "int", group_id, "string", user_name);
|
||
|
|
}
|
||
|
|
|
||
|
|
int ccnet_group_manager_remove_group (CcnetGroupManager *mgr,
|
||
|
|
int group_id,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
char *sql;
|
||
|
|
|
||
|
|
/* No permission check here, since both group staff and seahub staff
|
||
|
|
* can remove group.
|
||
|
|
*/
|
||
|
|
|
||
|
|
if (ccnet_db_type(db) == CCNET_DB_TYPE_PGSQL)
|
||
|
|
sql = "DELETE FROM \"Group\" WHERE group_id=?";
|
||
|
|
else
|
||
|
|
sql = "DELETE FROM `Group` WHERE group_id=?";
|
||
|
|
ccnet_db_statement_query (db, sql, 1, "int", group_id);
|
||
|
|
|
||
|
|
sql = "DELETE FROM GroupUser WHERE group_id=?";
|
||
|
|
ccnet_db_statement_query (db, sql, 1, "int", group_id);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static gboolean
|
||
|
|
check_group_exists (CcnetDB *db, int group_id)
|
||
|
|
{
|
||
|
|
if (ccnet_db_type(db) == CCNET_DB_TYPE_PGSQL)
|
||
|
|
return ccnet_db_statement_exists (db, "SELECT group_id FROM \"Group\" WHERE "
|
||
|
|
"group_id=?", 1, "int", group_id);
|
||
|
|
else
|
||
|
|
return ccnet_db_statement_exists (db, "SELECT group_id FROM `Group` WHERE "
|
||
|
|
"group_id=?", 1, "int", group_id);
|
||
|
|
}
|
||
|
|
|
||
|
|
int ccnet_group_manager_add_member (CcnetGroupManager *mgr,
|
||
|
|
int group_id,
|
||
|
|
const char *user_name,
|
||
|
|
const char *member_name,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
|
||
|
|
/* check whether user is the staff of the group */
|
||
|
|
if (!check_group_staff (db, group_id, user_name)) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0,
|
||
|
|
"Permission error: only group staff can add member");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* check whether group exists */
|
||
|
|
if (!check_group_exists (db, group_id)) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0, "Group not exists");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* check whether group is full */
|
||
|
|
/* snprintf (sql, sizeof(sql), "SELECT count(group_id) FROM `GroupUser` " */
|
||
|
|
/* "WHERE `group_id` = %d", group_id); */
|
||
|
|
/* int count = ccnet_db_get_int (db, sql); */
|
||
|
|
/* if (count >= MAX_GROUP_MEMBERS) { */
|
||
|
|
/* g_set_error (error, CCNET_DOMAIN, 0, "Group is full"); */
|
||
|
|
/* return -1; */
|
||
|
|
/* } */
|
||
|
|
|
||
|
|
char *member_name_l = g_ascii_strdown (member_name, -1);
|
||
|
|
int rc = ccnet_db_statement_query (db, "INSERT INTO GroupUser VALUES (?, ?, ?)",
|
||
|
|
3, "int", group_id, "string", member_name_l,
|
||
|
|
"int", 0);
|
||
|
|
g_free (member_name_l);
|
||
|
|
if (rc < 0) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0, "Failed to add member to group");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int ccnet_group_manager_remove_member (CcnetGroupManager *mgr,
|
||
|
|
int group_id,
|
||
|
|
const char *user_name,
|
||
|
|
const char *member_name,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
char *sql;
|
||
|
|
|
||
|
|
/* check whether user is the staff of the group */
|
||
|
|
if (!check_group_staff (db, group_id, user_name)) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0,
|
||
|
|
"Only group staff can remove member");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* check whether group exists */
|
||
|
|
if (!check_group_exists (db, group_id)) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0, "Group not exists");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* can not remove myself */
|
||
|
|
if (g_strcmp0 (user_name, member_name) == 0) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0, "Can not remove myself");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
sql = "DELETE FROM GroupUser WHERE group_id=? AND user_name=?";
|
||
|
|
ccnet_db_statement_query (db, sql, 2, "int", group_id, "string", member_name);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int ccnet_group_manager_set_admin (CcnetGroupManager *mgr,
|
||
|
|
int group_id,
|
||
|
|
const char *member_name,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
|
||
|
|
ccnet_db_statement_query (db,
|
||
|
|
"UPDATE GroupUser SET is_staff = 1 "
|
||
|
|
"WHERE group_id = ? and user_name = ?",
|
||
|
|
2, "int", group_id, "string", member_name);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int ccnet_group_manager_unset_admin (CcnetGroupManager *mgr,
|
||
|
|
int group_id,
|
||
|
|
const char *member_name,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
|
||
|
|
ccnet_db_statement_query (db,
|
||
|
|
"UPDATE GroupUser SET is_staff = 0 "
|
||
|
|
"WHERE group_id = ? and user_name = ?",
|
||
|
|
2, "int", group_id, "string", member_name);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int ccnet_group_manager_set_group_name (CcnetGroupManager *mgr,
|
||
|
|
int group_id,
|
||
|
|
const char *group_name,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
|
||
|
|
ccnet_db_statement_query (db,
|
||
|
|
"UPDATE `Group` SET group_name = ? "
|
||
|
|
"WHERE group_id = ?",
|
||
|
|
2, "string", group_name, "int", group_id);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int ccnet_group_manager_quit_group (CcnetGroupManager *mgr,
|
||
|
|
int group_id,
|
||
|
|
const char *user_name,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
|
||
|
|
/* check where user is the staff of the group */
|
||
|
|
if (check_group_staff (db, group_id, user_name)) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0,
|
||
|
|
"Group staff can not quit group");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* check whether group exists */
|
||
|
|
if (!check_group_exists (db, group_id)) {
|
||
|
|
g_set_error (error, CCNET_DOMAIN, 0, "Group not exists");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
ccnet_db_statement_query (db,
|
||
|
|
"DELETE FROM GroupUser WHERE group_id=? "
|
||
|
|
"AND user_name=?",
|
||
|
|
2, "int", group_id, "string", user_name);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static gboolean
|
||
|
|
get_group_ids_cb (CcnetDBRow *row, void *data)
|
||
|
|
{
|
||
|
|
GList **plist = data;
|
||
|
|
|
||
|
|
int group_id = ccnet_db_row_get_column_int (row, 0);
|
||
|
|
|
||
|
|
*plist = g_list_prepend (*plist, (gpointer)(long)group_id);
|
||
|
|
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
GList *
|
||
|
|
ccnet_group_manager_get_groupids_by_user (CcnetGroupManager *mgr,
|
||
|
|
const char *user_name,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
GList *group_ids = NULL;
|
||
|
|
|
||
|
|
if (ccnet_db_statement_foreach_row (db,
|
||
|
|
"SELECT group_id FROM GroupUser "
|
||
|
|
"WHERE user_name=?",
|
||
|
|
get_group_ids_cb, &group_ids,
|
||
|
|
1, "string", user_name) < 0) {
|
||
|
|
g_list_free (group_ids);
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
return g_list_reverse (group_ids);
|
||
|
|
}
|
||
|
|
|
||
|
|
static gboolean
|
||
|
|
get_ccnetgroup_cb (CcnetDBRow *row, void *data)
|
||
|
|
{
|
||
|
|
CcnetGroup **p_group = data;
|
||
|
|
int group_id;
|
||
|
|
const char *group_name;
|
||
|
|
const char *creator;
|
||
|
|
gint64 ts;
|
||
|
|
|
||
|
|
group_id = ccnet_db_row_get_column_int (row, 0);
|
||
|
|
group_name = (const char *)ccnet_db_row_get_column_text (row, 1);
|
||
|
|
creator = (const char *)ccnet_db_row_get_column_text (row, 2);
|
||
|
|
ts = ccnet_db_row_get_column_int64 (row, 3);
|
||
|
|
|
||
|
|
char *creator_l = g_ascii_strdown (creator, -1);
|
||
|
|
*p_group = g_object_new (CCNET_TYPE_GROUP,
|
||
|
|
"id", group_id,
|
||
|
|
"group_name", group_name,
|
||
|
|
"creator_name", creator_l,
|
||
|
|
"timestamp", ts,
|
||
|
|
"source", "DB",
|
||
|
|
NULL);
|
||
|
|
g_free (creator_l);
|
||
|
|
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
CcnetGroup *
|
||
|
|
ccnet_group_manager_get_group (CcnetGroupManager *mgr, int group_id,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
char *sql;
|
||
|
|
CcnetGroup *ccnetgroup = NULL;
|
||
|
|
|
||
|
|
if (ccnet_db_type(db) == CCNET_DB_TYPE_PGSQL)
|
||
|
|
sql = "SELECT * FROM \"Group\" WHERE group_id = ?";
|
||
|
|
else
|
||
|
|
sql = "SELECT * FROM `Group` WHERE group_id = ?";
|
||
|
|
if (ccnet_db_statement_foreach_row (db, sql,
|
||
|
|
get_ccnetgroup_cb, &ccnetgroup,
|
||
|
|
1, "int", group_id) < 0)
|
||
|
|
return NULL;
|
||
|
|
|
||
|
|
return ccnetgroup;
|
||
|
|
}
|
||
|
|
|
||
|
|
static gboolean
|
||
|
|
get_ccnet_groupuser_cb (CcnetDBRow *row, void *data)
|
||
|
|
{
|
||
|
|
GList **plist = data;
|
||
|
|
CcnetGroupUser *group_user;
|
||
|
|
|
||
|
|
int group_id = ccnet_db_row_get_column_int (row, 0);
|
||
|
|
const char *user = (const char *)ccnet_db_row_get_column_text (row, 1);
|
||
|
|
int is_staff = ccnet_db_row_get_column_int (row, 2);
|
||
|
|
|
||
|
|
char *user_l = g_ascii_strdown (user, -1);
|
||
|
|
group_user = g_object_new (CCNET_TYPE_GROUP_USER,
|
||
|
|
"group_id", group_id,
|
||
|
|
"user_name", user_l,
|
||
|
|
"is_staff", is_staff,
|
||
|
|
NULL);
|
||
|
|
g_free (user_l);
|
||
|
|
if (group_user != NULL) {
|
||
|
|
*plist = g_list_prepend (*plist, group_user);
|
||
|
|
}
|
||
|
|
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
GList *
|
||
|
|
ccnet_group_manager_get_group_members (CcnetGroupManager *mgr, int group_id,
|
||
|
|
GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
char *sql;
|
||
|
|
GList *group_users = NULL;
|
||
|
|
|
||
|
|
sql = "SELECT * FROM GroupUser WHERE group_id = ?";
|
||
|
|
if (ccnet_db_statement_foreach_row (db, sql,
|
||
|
|
get_ccnet_groupuser_cb, &group_users,
|
||
|
|
1, "int", group_id) < 0)
|
||
|
|
return NULL;
|
||
|
|
|
||
|
|
return g_list_reverse (group_users);
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
ccnet_group_manager_check_group_staff (CcnetGroupManager *mgr,
|
||
|
|
int group_id,
|
||
|
|
const char *user_name)
|
||
|
|
{
|
||
|
|
return check_group_staff (mgr->priv->db, group_id, user_name);
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
ccnet_group_manager_remove_group_user (CcnetGroupManager *mgr,
|
||
|
|
const char *user)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
|
||
|
|
ccnet_db_statement_query (db,
|
||
|
|
"DELETE FROM GroupUser "
|
||
|
|
"WHERE user_name = ?",
|
||
|
|
1, "string", user);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
ccnet_group_manager_is_group_user (CcnetGroupManager *mgr,
|
||
|
|
int group_id,
|
||
|
|
const char *user)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
|
||
|
|
return ccnet_db_statement_exists (db, "SELECT group_id FROM GroupUser "
|
||
|
|
"WHERE group_id=? AND user_name=?",
|
||
|
|
2, "int", group_id, "string", user);
|
||
|
|
}
|
||
|
|
|
||
|
|
static gboolean
|
||
|
|
get_all_ccnetgroups_cb (CcnetDBRow *row, void *data)
|
||
|
|
{
|
||
|
|
GList **plist = data;
|
||
|
|
int group_id;
|
||
|
|
const char *group_name;
|
||
|
|
const char *creator;
|
||
|
|
gint64 ts;
|
||
|
|
|
||
|
|
group_id = ccnet_db_row_get_column_int (row, 0);
|
||
|
|
group_name = (const char *)ccnet_db_row_get_column_text (row, 1);
|
||
|
|
creator = (const char *)ccnet_db_row_get_column_text (row, 2);
|
||
|
|
ts = ccnet_db_row_get_column_int64 (row, 3);
|
||
|
|
|
||
|
|
char *creator_l = g_ascii_strdown (creator, -1);
|
||
|
|
CcnetGroup *group = g_object_new (CCNET_TYPE_GROUP,
|
||
|
|
"id", group_id,
|
||
|
|
"group_name", group_name,
|
||
|
|
"creator_name", creator_l,
|
||
|
|
"timestamp", ts,
|
||
|
|
"source", "DB",
|
||
|
|
NULL);
|
||
|
|
g_free (creator_l);
|
||
|
|
|
||
|
|
*plist = g_list_prepend (*plist, group);
|
||
|
|
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
GList*
|
||
|
|
ccnet_group_manager_get_all_groups (CcnetGroupManager *mgr,
|
||
|
|
int start, int limit, GError **error)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
GList *ret = NULL;
|
||
|
|
int rc;
|
||
|
|
|
||
|
|
if (ccnet_db_type(mgr->priv->db) == CCNET_DB_TYPE_PGSQL) {
|
||
|
|
if (start == -1 && limit == -1) {
|
||
|
|
rc = ccnet_db_statement_foreach_row (db, "SELECT group_id, group_name, "
|
||
|
|
"creator_name, timestamp FROM \"Group\" "
|
||
|
|
"ORDER BY timestamp DESC",
|
||
|
|
get_all_ccnetgroups_cb, &ret,
|
||
|
|
0);
|
||
|
|
} else {
|
||
|
|
rc = ccnet_db_statement_foreach_row (db, "SELECT group_id, group_name, "
|
||
|
|
"creator_name, timestamp FROM \"Group\" "
|
||
|
|
"ORDER BY timestamp DESC LIMIT ? OFFSET ?",
|
||
|
|
get_all_ccnetgroups_cb, &ret,
|
||
|
|
2, "int", limit, "int", start);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if (start == -1 && limit == -1) {
|
||
|
|
rc = ccnet_db_statement_foreach_row (db, "SELECT `group_id`, `group_name`, "
|
||
|
|
"`creator_name`, `timestamp` FROM `Group` "
|
||
|
|
"ORDER BY timestamp DESC",
|
||
|
|
get_all_ccnetgroups_cb, &ret,
|
||
|
|
0);
|
||
|
|
} else {
|
||
|
|
rc = ccnet_db_statement_foreach_row (db, "SELECT `group_id`, `group_name`, "
|
||
|
|
"`creator_name`, `timestamp` FROM `Group` "
|
||
|
|
"ORDER BY timestamp DESC LIMIT ?, ?",
|
||
|
|
get_all_ccnetgroups_cb, &ret,
|
||
|
|
2, "int", start, "int", limit);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (rc < 0)
|
||
|
|
return NULL;
|
||
|
|
|
||
|
|
return g_list_reverse (ret);
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
ccnet_group_manager_set_group_creator (CcnetGroupManager *mgr,
|
||
|
|
int group_id,
|
||
|
|
const char *user_name)
|
||
|
|
{
|
||
|
|
CcnetDB *db = mgr->priv->db;
|
||
|
|
char *sql;
|
||
|
|
|
||
|
|
if (ccnet_db_type(db) == CCNET_DB_TYPE_PGSQL) {
|
||
|
|
sql = "UPDATE \"Group\" SET creator_name = ? WHERE group_id = ?";
|
||
|
|
} else {
|
||
|
|
sql = "UPDATE `Group` SET creator_name = ? WHERE group_id = ?";
|
||
|
|
}
|
||
|
|
|
||
|
|
ccnet_db_statement_query (db, sql, 2, "string", user_name, "int", group_id);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
|
||
|
|
}
|
||
|
|
|