mirror of
https://codeberg.org/redict/redict.git
synced 2025-01-22 08:08:53 -05:00
03d144a452
Replaces Redis => Redict API References: https://codeberg.org/redict/redict/issues/21 Signed-off-by: Drew DeVault <sir@cmpwn.com>
76 lines
2.6 KiB
C
76 lines
2.6 KiB
C
// SPDX-FileCopyrightText: 2024 Redict Contributors
|
|
// SPDX-FileCopyrightText: 2024 Salvatore Sanfilippo <antirez at gmail dot com>
|
|
//
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
// SPDX-License-Identifier: LGPL-3.0-only
|
|
|
|
#include "redictmodule.h"
|
|
#include <strings.h>
|
|
#include <assert.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
|
|
/* A sample with declarable channels, that are used to validate against ACLs */
|
|
int getChannels_subscribe(RedictModuleCtx *ctx, RedictModuleString **argv, int argc) {
|
|
if ((argc - 1) % 3 != 0) {
|
|
RedictModule_WrongArity(ctx);
|
|
return REDICTMODULE_OK;
|
|
}
|
|
char *err = NULL;
|
|
|
|
/* getchannels.command [[subscribe|unsubscribe|publish] [pattern|literal] <channel> ...]
|
|
* This command marks the given channel is accessed based on the
|
|
* provided modifiers. */
|
|
for (int i = 1; i < argc; i += 3) {
|
|
const char *operation = RedictModule_StringPtrLen(argv[i], NULL);
|
|
const char *type = RedictModule_StringPtrLen(argv[i+1], NULL);
|
|
int flags = 0;
|
|
|
|
if (!strcasecmp(operation, "subscribe")) {
|
|
flags |= REDICTMODULE_CMD_CHANNEL_SUBSCRIBE;
|
|
} else if (!strcasecmp(operation, "unsubscribe")) {
|
|
flags |= REDICTMODULE_CMD_CHANNEL_UNSUBSCRIBE;
|
|
} else if (!strcasecmp(operation, "publish")) {
|
|
flags |= REDICTMODULE_CMD_CHANNEL_PUBLISH;
|
|
} else {
|
|
err = "Invalid channel operation";
|
|
break;
|
|
}
|
|
|
|
if (!strcasecmp(type, "literal")) {
|
|
/* No op */
|
|
} else if (!strcasecmp(type, "pattern")) {
|
|
flags |= REDICTMODULE_CMD_CHANNEL_PATTERN;
|
|
} else {
|
|
err = "Invalid channel type";
|
|
break;
|
|
}
|
|
if (RedictModule_IsChannelsPositionRequest(ctx)) {
|
|
RedictModule_ChannelAtPosWithFlags(ctx, i+2, flags);
|
|
}
|
|
}
|
|
|
|
if (!RedictModule_IsChannelsPositionRequest(ctx)) {
|
|
if (err) {
|
|
RedictModule_ReplyWithError(ctx, err);
|
|
} else {
|
|
/* Normal implementation would go here, but for tests just return okay */
|
|
RedictModule_ReplyWithSimpleString(ctx, "OK");
|
|
}
|
|
}
|
|
|
|
return REDICTMODULE_OK;
|
|
}
|
|
|
|
int RedictModule_OnLoad(RedictModuleCtx *ctx, RedictModuleString **argv, int argc) {
|
|
REDICTMODULE_NOT_USED(argv);
|
|
REDICTMODULE_NOT_USED(argc);
|
|
if (RedictModule_Init(ctx, "getchannels", 1, REDICTMODULE_APIVER_1) == REDICTMODULE_ERR)
|
|
return REDICTMODULE_ERR;
|
|
|
|
if (RedictModule_CreateCommand(ctx, "getchannels.command", getChannels_subscribe, "getchannels-api", 0, 0, 0) == REDICTMODULE_ERR)
|
|
return REDICTMODULE_ERR;
|
|
|
|
return REDICTMODULE_OK;
|
|
}
|