1
0
Fork 0
mirror of https://git.sr.ht/~ashkeel/strimertul synced 2024-09-18 01:50:50 +00:00

Add subscriber ACL level and make them work on bot commands

This commit is contained in:
Ash Keel 2021-11-30 17:37:33 +01:00
parent f1789a6d82
commit a72d52d685
No known key found for this signature in database
GPG key ID: BAD8D93E7314ED3E
5 changed files with 69 additions and 8 deletions

View file

@ -201,6 +201,7 @@
"description-help": "What does this command do?",
"access-level": "Access level",
"access-everyone": "Everyone",
"access-subscribers": "Subscribers",
"access-vips": "VIPs",
"access-moderators": "Moderators",
"access-streamer": "Streamer only",

View file

@ -35,7 +35,12 @@ interface TwitchModulesConfig {
enable_timers: boolean;
}
type AccessLevelType = 'everyone' | 'vip' | 'moderators' | 'streamer';
export type AccessLevelType =
| 'everyone'
| 'subscribers'
| 'vip'
| 'moderators'
| 'streamer';
export interface TwitchBotCustomCommand {
description: string;

View file

@ -5,7 +5,10 @@ import { useDispatch } from 'react-redux';
import { useModule } from '../../../lib/react-utils';
import { modules } from '../../../store/api/reducer';
import Modal from '../../components/Modal';
import { TwitchBotCustomCommand } from '../../../store/api/types';
import {
AccessLevelType,
TwitchBotCustomCommand,
} from '../../../store/api/types';
import Field from '../../components/Field';
interface CommandItemProps {
@ -92,6 +95,9 @@ function CommandModal({
const [description, setDescription] = useState(
initialData?.description ?? '',
);
const [accessLevel, setAccessLevel] = useState(
initialData?.access_level ?? 'everyone',
);
const [response, setResponse] = useState(initialData?.response ?? '');
const { t } = useTranslation();
@ -105,7 +111,7 @@ function CommandModal({
description,
response,
enabled: initialData?.enabled ?? false,
access_level: 'everyone',
access_level: accessLevel,
});
}
};
@ -171,10 +177,18 @@ function CommandModal({
<div className="field">
<p className="control">
<span className="select">
<select>
<select
value={accessLevel}
onChange={(ev) =>
setAccessLevel(ev.target.value as AccessLevelType)
}
>
<option value="everyone">
{t('twitch.commands.access-everyone')}
</option>
<option value="subscribers">
{t('twitch.commands.access-subscribers')}
</option>
<option value="vip">
{t('twitch.commands.access-vips')}
</option>

View file

@ -205,3 +205,27 @@ func (b *Bot) Connect() error {
func (b *Bot) WriteMessage(message string) {
b.Client.Say(b.config.Channel, message)
}
func getUserAccessLevel(user irc.User) AccessLevelType {
// Check broadcaster
if _, ok := user.Badges["broadcaster"]; ok {
return ALTStreamer
}
// Check mods
if _, ok := user.Badges["moderator"]; ok {
return ALTModerators
}
// Check VIP
if _, ok := user.Badges["vip"]; ok {
return ALTVIP
}
// Check subscribers
if _, ok := user.Badges["subscriber"]; ok {
return ALTSubscribers
}
return ALTEveryone
}

View file

@ -14,12 +14,21 @@ import (
type AccessLevelType string
const (
ALTEveryone AccessLevelType = "everyone"
ALTVIP AccessLevelType = "vip"
ALTModerators AccessLevelType = "moderators"
ALTStreamer AccessLevelType = "streamer"
ALTEveryone AccessLevelType = "everyone"
ALTSubscribers AccessLevelType = "subscriber"
ALTVIP AccessLevelType = "vip"
ALTModerators AccessLevelType = "moderators"
ALTStreamer AccessLevelType = "streamer"
)
var accessLevels = map[AccessLevelType]int{
ALTEveryone: 0,
ALTSubscribers: 1,
ALTVIP: 2,
ALTModerators: 3,
ALTStreamer: 999,
}
type BotCommandHandler func(bot *Bot, message irc.PrivateMessage)
type BotCommand struct {
@ -31,6 +40,14 @@ type BotCommand struct {
}
func cmdCustom(bot *Bot, cmd string, data BotCustomCommand, message irc.PrivateMessage) {
// Check access level
accessLevel := getUserAccessLevel(message.User)
// Ensure that access level is high enough
if accessLevels[accessLevel] < accessLevels[data.AccessLevel] {
return
}
// Add future logic (like counters etc.) here, for now it's just fixed messages
var buf bytes.Buffer
err := bot.customTemplates[cmd].Execute(&buf, message)