Initial commit
commit
292747216b
|
|
@ -0,0 +1,2 @@
|
||||||
|
bots/*/bot
|
||||||
|
bots/*/.config
|
||||||
|
|
@ -0,0 +1,74 @@
|
||||||
|
[
|
||||||
|
{
|
||||||
|
"code": "help",
|
||||||
|
"grammar": "",
|
||||||
|
"handler": "hdlCharaHelp",
|
||||||
|
"allowOverwrite": false
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "learn",
|
||||||
|
"grammar": "",
|
||||||
|
"handler": "hdlGrammarLearn",
|
||||||
|
"allowOverwrite": false
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "unlearn",
|
||||||
|
"grammar": "",
|
||||||
|
"handler": "hdlGrammarUnlearn",
|
||||||
|
"allowOverwrite": false
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "forget",
|
||||||
|
"grammar": "",
|
||||||
|
"handler": "hdlGrammarUnlearn",
|
||||||
|
"allowOverwrite": false
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "show",
|
||||||
|
"grammar": "",
|
||||||
|
"handler": "hdlGrammarShow",
|
||||||
|
"allowOverwrite": false
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "ping",
|
||||||
|
"grammar": "ping.json",
|
||||||
|
"handler": "",
|
||||||
|
"allowOverwrite": false
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "pool",
|
||||||
|
"grammar": "",
|
||||||
|
"handler": "hdlPoolLook",
|
||||||
|
"allowOverwrite": false
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "pooladd",
|
||||||
|
"grammar": "",
|
||||||
|
"handler": "hdlPoolAdd",
|
||||||
|
"allowOverwrite": false
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "poolls",
|
||||||
|
"grammar": "",
|
||||||
|
"handler": "hdlPoolList",
|
||||||
|
"allowOverwrite": false
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "poolrm",
|
||||||
|
"grammar": "",
|
||||||
|
"handler": "hdlPoolRemove",
|
||||||
|
"allowOverwrite": false
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "8ball",
|
||||||
|
"grammar": "8ball.json",
|
||||||
|
"handler": "",
|
||||||
|
"allowOverwrite": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"code": "burger",
|
||||||
|
"grammar": "burger.json",
|
||||||
|
"handler": "",
|
||||||
|
"allowOverwrite": true
|
||||||
|
}
|
||||||
|
]
|
||||||
|
|
@ -0,0 +1,20 @@
|
||||||
|
{
|
||||||
|
"debug": false,
|
||||||
|
"servers": [
|
||||||
|
{
|
||||||
|
"host": "ircserver.tld",
|
||||||
|
"port": 6697,
|
||||||
|
"user": "botUser",
|
||||||
|
"pass": "botPassword",
|
||||||
|
"nick": "botNick",
|
||||||
|
"channels": ["#channel1", "#channel2"],
|
||||||
|
"codePrefix": "!",
|
||||||
|
"admins": [
|
||||||
|
{
|
||||||
|
"user": "adminUser",
|
||||||
|
"pass": "adminPass"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,24 @@
|
||||||
|
{
|
||||||
|
"origin": [
|
||||||
|
"it is certain",
|
||||||
|
"it is decidedly so",
|
||||||
|
"without a doubt",
|
||||||
|
"yes — definitely",
|
||||||
|
"you may rely on it",
|
||||||
|
"as I see it, yes",
|
||||||
|
"most likely",
|
||||||
|
"outlook good",
|
||||||
|
"yes",
|
||||||
|
"signs point to yes",
|
||||||
|
"reply hazy try again",
|
||||||
|
"ask again later",
|
||||||
|
"better not tell you now",
|
||||||
|
"cannot predict now",
|
||||||
|
"concentrate and ask again",
|
||||||
|
"don't count on it",
|
||||||
|
"my reply is no",
|
||||||
|
"my sources say no",
|
||||||
|
"outlook not so good",
|
||||||
|
"very doubtful"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,57 @@
|
||||||
|
{
|
||||||
|
"patty": [
|
||||||
|
"coconut",
|
||||||
|
"nutty",
|
||||||
|
"possible",
|
||||||
|
"soy",
|
||||||
|
"tofu",
|
||||||
|
"veggie delight"
|
||||||
|
],
|
||||||
|
"veggie": [
|
||||||
|
"kimchi",
|
||||||
|
"jalapeño peppers",
|
||||||
|
"lettuce",
|
||||||
|
"onions",
|
||||||
|
"pickles",
|
||||||
|
"tomatoes"
|
||||||
|
],
|
||||||
|
"cheese": [
|
||||||
|
"vegan mozzarella cheese"
|
||||||
|
],
|
||||||
|
"sauce": [
|
||||||
|
"barbecue sauce",
|
||||||
|
"curry sauce",
|
||||||
|
"Dijon mustard",
|
||||||
|
"ketchup"
|
||||||
|
],
|
||||||
|
"burger": [
|
||||||
|
"#patty# burger with #veggie#, #veggie# and #cheese# in #sauce#",
|
||||||
|
"#patty# burger with #veggie#, #veggie# and #veggie# in #sauce#"
|
||||||
|
],
|
||||||
|
"side": [
|
||||||
|
"avocado salad",
|
||||||
|
"BBQ baked beans",
|
||||||
|
"BBQ potato chips",
|
||||||
|
"caesar salad",
|
||||||
|
"coleslaw",
|
||||||
|
"corn on the cob",
|
||||||
|
"corn salad",
|
||||||
|
"french fries",
|
||||||
|
"french potato salad",
|
||||||
|
"fried pickles",
|
||||||
|
"fried shishito peppers",
|
||||||
|
"a fruit bowl",
|
||||||
|
"vegan mac and cheese",
|
||||||
|
"onion rings",
|
||||||
|
"potato wedges",
|
||||||
|
"roasted tomatoes",
|
||||||
|
"sweet potato fries",
|
||||||
|
"a three-bean salad",
|
||||||
|
"tempura green beans",
|
||||||
|
"zucchini chips"
|
||||||
|
],
|
||||||
|
"origin": [
|
||||||
|
"Here, have a #burger#!",
|
||||||
|
"Here, have a #burger#, served with #side#!"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1 @@
|
||||||
|
{ "origin": ["pong!"] }
|
||||||
|
|
@ -0,0 +1,12 @@
|
||||||
|
import ../../cirrus
|
||||||
|
import ../../modules/[grammar, javapool]
|
||||||
|
import chara
|
||||||
|
|
||||||
|
|
||||||
|
# Initialise cirrus
|
||||||
|
var
|
||||||
|
config = loadIrcConfig()
|
||||||
|
states = initServers(config.servers)
|
||||||
|
|
||||||
|
# Initialise grammar
|
||||||
|
listenGrammar(states)
|
||||||
|
|
@ -0,0 +1,18 @@
|
||||||
|
import std/[strutils, tables]
|
||||||
|
import ../../cirrus
|
||||||
|
from ../../modules/grammar import getGrammarNames
|
||||||
|
|
||||||
|
|
||||||
|
proc hdlCharaHelp*(st: ServerState, msg: Message): void =
|
||||||
|
var
|
||||||
|
names = getGrammarNames(st.codePrefix)
|
||||||
|
help = "Hi, I'm " & st.nick & ", a bot with random features. " &
|
||||||
|
"You can find my source code at https://git.tilde.town/mio/cirrus. " &
|
||||||
|
"I'll respond to these phrases: " & join(names, ", ")
|
||||||
|
message(st.sock, msg.replyTo, help)
|
||||||
|
|
||||||
|
|
||||||
|
when defined(nimHasUsed):
|
||||||
|
{.used.}
|
||||||
|
|
||||||
|
handlers["hdlCharaHelp"] = hdlCharaHelp
|
||||||
|
|
@ -0,0 +1,6 @@
|
||||||
|
#!/sbin/openrc-run
|
||||||
|
|
||||||
|
name="javapool"
|
||||||
|
command="$HOME/bin/cirrus/bots/javapool/bot"
|
||||||
|
pidfile="/var/run/javapool.pid"
|
||||||
|
start_stop_daemon_args="--background --make-pidfile"
|
||||||
|
|
@ -0,0 +1,11 @@
|
||||||
|
[Unit]
|
||||||
|
Description=javapool — an IRC bot
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
WorkingDirectory=%h/bin/cirrus/bots/javapool
|
||||||
|
ExecStart=./bot
|
||||||
|
Restart=always
|
||||||
|
RestartSec=300
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=default.target
|
||||||
|
|
@ -0,0 +1,340 @@
|
||||||
|
import std/[base64, json, jsonutils, net, os, sequtils, strutils, tables]
|
||||||
|
|
||||||
|
|
||||||
|
type
|
||||||
|
Admin* = tuple
|
||||||
|
user: string
|
||||||
|
pass: string
|
||||||
|
|
||||||
|
Server* = tuple
|
||||||
|
host: string
|
||||||
|
port: int
|
||||||
|
user: string
|
||||||
|
pass: string
|
||||||
|
nick: string
|
||||||
|
channels: seq[string]
|
||||||
|
codePrefix: string
|
||||||
|
admins: seq[Admin]
|
||||||
|
|
||||||
|
ServerState* = tuple
|
||||||
|
host: string
|
||||||
|
sock: Socket
|
||||||
|
isAuth: bool
|
||||||
|
channels: seq[string]
|
||||||
|
codePrefix: string
|
||||||
|
nick: string
|
||||||
|
admins: seq[Admin]
|
||||||
|
|
||||||
|
State* = seq[ServerState]
|
||||||
|
|
||||||
|
Message* = tuple
|
||||||
|
code: string
|
||||||
|
codeParams: seq[string]
|
||||||
|
sender: string
|
||||||
|
recipient: string
|
||||||
|
replyTo: string
|
||||||
|
body: string
|
||||||
|
|
||||||
|
IrcConfig* = tuple
|
||||||
|
debug: bool
|
||||||
|
servers: seq[Server]
|
||||||
|
|
||||||
|
SvcCode* = tuple
|
||||||
|
code: string
|
||||||
|
handler: string
|
||||||
|
|
||||||
|
SvcCodes* = seq[SvcCode]
|
||||||
|
|
||||||
|
|
||||||
|
# Look for .config directory in the bot executable's directory
|
||||||
|
# or in $HOME/.config/cirrus
|
||||||
|
var ircConfigDir*: string
|
||||||
|
if dirExists(unixToNativePath(getAppDir() & "/.config")):
|
||||||
|
ircConfigDir = getAppDir() & "/.config/"
|
||||||
|
else:
|
||||||
|
ircConfigDir = getConfigDir() & "cirrus/"
|
||||||
|
|
||||||
|
var
|
||||||
|
ircConfigFile* = ircConfigDir & "config.json"
|
||||||
|
ircCodeFile* = ircConfigDir & "codes.json"
|
||||||
|
ircDebug* = false
|
||||||
|
serviceCodes*: SvcCodes
|
||||||
|
handlers* = initTable[string, proc(st: ServerState, msg: Message): void]()
|
||||||
|
|
||||||
|
|
||||||
|
# Echo string to stdout.
|
||||||
|
proc debug*(tag: string, str: string): void =
|
||||||
|
if ircDebug:
|
||||||
|
echo join(["[debug]", "[", tag, "] ", str])
|
||||||
|
|
||||||
|
|
||||||
|
# Load JSON config file and return a config tuple of keys and values.
|
||||||
|
proc loadIrcConfig*(file = ircConfigFile): IrcConfig =
|
||||||
|
if fileExists(unixToNativePath(file)):
|
||||||
|
try:
|
||||||
|
var
|
||||||
|
nodes = parseFile(unixToNativePath(file))
|
||||||
|
conf = to(nodes, IrcConfig)
|
||||||
|
if conf.debug.type() is bool:
|
||||||
|
ircDebug = conf.debug
|
||||||
|
return conf
|
||||||
|
except IOError:
|
||||||
|
debug("loadIrcConfig", "error: could not read config file " & file)
|
||||||
|
except JsonParsingError:
|
||||||
|
debug("loadIrcConfig", "error: invalid JSON in config file " & file)
|
||||||
|
else:
|
||||||
|
debug("loadIrcConfig", "error: cannot find config file " & file)
|
||||||
|
|
||||||
|
|
||||||
|
# Connect to a server at the given host and port. Return a Socket object if
|
||||||
|
# successful, or output an error otherwise.
|
||||||
|
proc connectServer*(host: string, port: int, useSSL = true): Socket =
|
||||||
|
var sock = newSocket()
|
||||||
|
if useSSL:
|
||||||
|
var ctx = newContext(verifyMode = CVerifyNone)
|
||||||
|
wrapSocket(ctx, sock)
|
||||||
|
try:
|
||||||
|
connect(sock, host, Port(port))
|
||||||
|
except OSError:
|
||||||
|
debug("connectServer", "error: cannot connect to the server " & host &
|
||||||
|
"/" & intToStr(port))
|
||||||
|
return sock
|
||||||
|
|
||||||
|
|
||||||
|
# Wrap net's send() with carriage return and debug echo.
|
||||||
|
proc sendServer*(sock: Socket, str: string): void =
|
||||||
|
send(sock, str & "\r\n")
|
||||||
|
debug("sendServer", str)
|
||||||
|
|
||||||
|
|
||||||
|
# Wrap net's recvLine() with debug echo.
|
||||||
|
proc recvServer*(sock: Socket, debug = true): string =
|
||||||
|
var str = recvLine(sock)
|
||||||
|
debug("recvServer", str)
|
||||||
|
return str
|
||||||
|
|
||||||
|
|
||||||
|
# Send an IRC-formatted message to the server for a channel or user.
|
||||||
|
proc message*(sock: Socket, recipient: string, str: string): void =
|
||||||
|
sendServer(sock, join(["PRIVMSG ", recipient, " :", str]))
|
||||||
|
|
||||||
|
|
||||||
|
# Disconnect from a server with an optional quit message.
|
||||||
|
proc disconnectServer*(sock: Socket, quitMsg = "Bye."): void =
|
||||||
|
sendServer(sock, "QUIT :" & quitMsg)
|
||||||
|
close(sock)
|
||||||
|
|
||||||
|
|
||||||
|
# Authenticate to a server. Authentication modes: sasl-plain (default),
|
||||||
|
# nickserv.
|
||||||
|
proc authServer*(sock: Socket, user: string, pass: string, nick: string,
|
||||||
|
mode = "sasl-plain"): bool =
|
||||||
|
var resp: string
|
||||||
|
case mode
|
||||||
|
of "sasl-plain":
|
||||||
|
sendServer(sock, "CAP LS 302")
|
||||||
|
resp = recvServer(sock)
|
||||||
|
if find(resp, "sasl") != -1:
|
||||||
|
sendServer(sock, join(["USER", user, "0 *", user], " "))
|
||||||
|
sendServer(sock, "NICK " & nick)
|
||||||
|
sendServer(sock, "CAP REQ sasl")
|
||||||
|
resp = recvServer(sock)
|
||||||
|
if find(resp, "ACK") != 1:
|
||||||
|
sendServer(sock, "AUTHENTICATE PLAIN")
|
||||||
|
resp = recvServer(sock)
|
||||||
|
if find(resp, "AUTHENTICATE +") != 1:
|
||||||
|
# Format of the string to encode: "user\0user\0password"
|
||||||
|
sendServer(sock, "AUTHENTICATE " & encode(join([user, "\0", user, "\0",
|
||||||
|
pass])))
|
||||||
|
resp = recvServer(sock)
|
||||||
|
if find(resp, "successful") != 1:
|
||||||
|
sendServer(sock, "CAP END")
|
||||||
|
sendServer(sock, "NICK " & nick)
|
||||||
|
return true
|
||||||
|
return false
|
||||||
|
|
||||||
|
# PASS + Nickserv
|
||||||
|
else:
|
||||||
|
sendServer(sock, join(["PASS ", user, ":", pass]))
|
||||||
|
sendServer(sock, join(["USER", user, "0 *", user], " "))
|
||||||
|
sendServer(sock, "NICK " & nick)
|
||||||
|
sendServer(sock, join(["NickServ IDENTIFY", user, pass], " "))
|
||||||
|
resp = recvServer(sock)
|
||||||
|
if find(resp, "logged in") != -1 or find(resp, "identified") != -1:
|
||||||
|
return true
|
||||||
|
return false
|
||||||
|
|
||||||
|
|
||||||
|
# Respond to server pings to keep the socket connected.
|
||||||
|
proc sendPong*(sock: Socket, str: string): void =
|
||||||
|
if find(str, "PING") == 0:
|
||||||
|
sendServer(sock, "PONG " & split(str, " ")[1])
|
||||||
|
|
||||||
|
|
||||||
|
# Join a list of channels on a server.
|
||||||
|
proc joinChannels*(sock: Socket, chans: seq): void =
|
||||||
|
if chans.len() == 1:
|
||||||
|
sendServer(sock, "JOIN " & chans[0])
|
||||||
|
elif chans.len() > 1:
|
||||||
|
sendServer(sock, "JOIN " & join(chans, ","))
|
||||||
|
else:
|
||||||
|
debug("joinChannels", "warning: no channels joined, none found")
|
||||||
|
|
||||||
|
|
||||||
|
# Given a code prefix and message string from a server, parse the message
|
||||||
|
# string and return a message tuple of keys and values. The code prefix is set
|
||||||
|
# for a given server instance in the config file.
|
||||||
|
#
|
||||||
|
# Initial received string format:
|
||||||
|
# :[sender-nick]!~[sender-user]@domain.tld PRIVMSG [channel|recipient] :[body]
|
||||||
|
proc parseMessage*(prefix: string, str: string): Message =
|
||||||
|
var
|
||||||
|
msg: Message
|
||||||
|
strLen: int
|
||||||
|
if find(str, "PRIVMSG") != -1 and find(str, "!") != -1:
|
||||||
|
# code
|
||||||
|
strLen = split(str)[3].len() - 1
|
||||||
|
var cPLen = prefix.len()
|
||||||
|
msg.code = split(str)[3][(1 + cPLen)..strLen]
|
||||||
|
# codeParams
|
||||||
|
if split(split(str, ":")[2]).len() > 1:
|
||||||
|
strLen = split(str).len() - 1
|
||||||
|
msg.codeParams = split(str)[4..strLen]
|
||||||
|
else:
|
||||||
|
msg.codeParams = @[]
|
||||||
|
# sender
|
||||||
|
strLen = split(split(str, "!")[1], "@")[0].len() - 1
|
||||||
|
msg.sender = split(split(str, "!")[1], "@")[0][1..strLen]
|
||||||
|
# recipient and replyTo
|
||||||
|
msg.recipient = split(split(str, "PRIVMSG ")[1])[0]
|
||||||
|
msg.replyTo = msg.sender
|
||||||
|
if find(msg.recipient, "#") == 0:
|
||||||
|
msg.replyTo = msg.recipient
|
||||||
|
# body
|
||||||
|
strLen = split(str).len() - 1
|
||||||
|
var bodyLen = join(split(str)[3..strLen], " ").len() - 1
|
||||||
|
msg.body = join(split(str)[3..strLen], " ")[1..bodyLen]
|
||||||
|
return msg
|
||||||
|
|
||||||
|
|
||||||
|
# Load JSON service code file to global variable `serviceCodes`.
|
||||||
|
proc loadCodes*(file = ircCodeFile): void =
|
||||||
|
if fileExists(unixToNativePath(file)):
|
||||||
|
try:
|
||||||
|
var nodes = parseFile(unixToNativePath(file))
|
||||||
|
serviceCodes = to(nodes, SvcCodes)
|
||||||
|
except IOError:
|
||||||
|
debug("loadCodes", "error: could not JSON file " & file)
|
||||||
|
# Show error if the code file is missing fields
|
||||||
|
except KeyError:
|
||||||
|
debug("loadCodes", "error: invalid " & file)
|
||||||
|
# Warn if a code has no handler linked to it
|
||||||
|
for c in serviceCodes:
|
||||||
|
if c.handler == "":
|
||||||
|
debug("loadCodes", "warning: code " & c.code & " not added, no " &
|
||||||
|
"handler found")
|
||||||
|
else:
|
||||||
|
debug("loadCodes", "error: cannot find service codes file")
|
||||||
|
|
||||||
|
|
||||||
|
# Write service codes to a JSON file.
|
||||||
|
proc saveCodes*(codes: SvcCodes, file = ircCodeFile): void =
|
||||||
|
writeFile(unixToNativePath(file), pretty(toJson(codes)))
|
||||||
|
|
||||||
|
|
||||||
|
# Return a list of service code names given the code prefix.
|
||||||
|
proc getCodeNames*(prefix: string, codes = serviceCodes): seq[string] =
|
||||||
|
return map(serviceCodes, proc(c: SvcCode): string = prefix & c.code)
|
||||||
|
|
||||||
|
|
||||||
|
# Check whether a user and pass matches an admin credentials. Return true
|
||||||
|
# if there is a match or false otherwise.
|
||||||
|
proc isAdmin*(admins: seq[Admin], user: string, pass: string): bool =
|
||||||
|
for a in admins:
|
||||||
|
if user == a.user and pass == a.pass:
|
||||||
|
return true
|
||||||
|
return false
|
||||||
|
|
||||||
|
|
||||||
|
# Send a handler response to the server.
|
||||||
|
proc runHandler*(prc: string, st: ServerState, msg: Message, hdl = handlers):
|
||||||
|
void =
|
||||||
|
if hasKey(hdl, prc):
|
||||||
|
hdl[prc](st, msg)
|
||||||
|
else:
|
||||||
|
debug("runHandler", "warning: no corresponding handler found for " & prc)
|
||||||
|
|
||||||
|
|
||||||
|
# Given a sequence of servers, connect and authenticate to each server, and
|
||||||
|
# join pre-configured channels. Return a sequence of server states.
|
||||||
|
proc initServers*(servers: seq[Server]): State =
|
||||||
|
var states: State
|
||||||
|
for svr in servers:
|
||||||
|
var
|
||||||
|
sock = connectServer(svr.host, svr.port)
|
||||||
|
isAuth = authServer(sock, svr.user, svr.pass, svr.nick)
|
||||||
|
add(states, (host: svr.host, sock: sock, isAuth: isAuth, channels: @[],
|
||||||
|
codePrefix: svr.codePrefix, nick: svr.nick, admins: svr.admins))
|
||||||
|
if isAuth:
|
||||||
|
joinChannels(sock, svr.channels)
|
||||||
|
states[states.len() - 1].channels = svr.channels
|
||||||
|
return states
|
||||||
|
|
||||||
|
|
||||||
|
# Parse message strings received and pass them to handlers.
|
||||||
|
proc listenCodes*(st: ServerState, str: string): void =
|
||||||
|
var msg = parseMessage(st.codePrefix, str)
|
||||||
|
for c in serviceCodes:
|
||||||
|
if msg.code == c.code:
|
||||||
|
runHandler(c.handler, st, msg)
|
||||||
|
break
|
||||||
|
|
||||||
|
|
||||||
|
# Initialise a loop and add listeners to respond to service codes.
|
||||||
|
proc listen*(states: seq[ServerState]): void =
|
||||||
|
while true:
|
||||||
|
var
|
||||||
|
statesBuf = states
|
||||||
|
count = 0
|
||||||
|
for st in statesBuf:
|
||||||
|
try:
|
||||||
|
var recv = recvServer(st.sock)
|
||||||
|
sendPong(st.sock, recv)
|
||||||
|
listenCodes(st, recv)
|
||||||
|
except AssertionDefect:
|
||||||
|
# Remove server from the states sequence
|
||||||
|
delete(statesBuf, count)
|
||||||
|
# Exit if there was only one server connection
|
||||||
|
if states.len() == 1:
|
||||||
|
quit()
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
|
||||||
|
# Pre-configured handlers
|
||||||
|
# ----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
# Show a greeting message with a list of service codes.
|
||||||
|
proc hdlIrcHelp*(st: ServerState, msg: Message): void =
|
||||||
|
var
|
||||||
|
names = getCodeNames(st.codePrefix)
|
||||||
|
help = "Hello, I'm a bot. I respond to the following codes: " &
|
||||||
|
join(names, ", ")
|
||||||
|
message(st.sock, msg.replyTo, help)
|
||||||
|
|
||||||
|
|
||||||
|
# Disconnect from the current server.
|
||||||
|
proc hdlIrcQuit*(st: ServerState, msg: Message): void =
|
||||||
|
if msg.codeParams.len() == 1:
|
||||||
|
if isAdmin(st.admins, msg.sender, msg.codeParams[0]):
|
||||||
|
disconnectServer(st.sock)
|
||||||
|
else:
|
||||||
|
message(st.sock, msg.replyTo, "Unauthorised user or password.")
|
||||||
|
else:
|
||||||
|
message(st.sock, msg.replyTo, "Unauthorised user or password.")
|
||||||
|
|
||||||
|
|
||||||
|
# Init
|
||||||
|
# ----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
handlers["hdlIrcHelp"] = hdlIrcHelp
|
||||||
|
handlers["hdlIrcQuit"] = hdlIrcQuit
|
||||||
|
|
@ -0,0 +1,432 @@
|
||||||
|
import std/[httpclient, json, jsonutils, net, os, random, sequtils, strutils,
|
||||||
|
tables]
|
||||||
|
import ../cirrus
|
||||||
|
|
||||||
|
|
||||||
|
type
|
||||||
|
GrammarCode* = tuple
|
||||||
|
code: string
|
||||||
|
grammar: string
|
||||||
|
handler: string
|
||||||
|
allowOverwrite: bool
|
||||||
|
|
||||||
|
GrammarCodes* = seq[GrammarCode]
|
||||||
|
|
||||||
|
Grammar* = OrderedTable[string, JsonNode]
|
||||||
|
|
||||||
|
Grammars* = seq[tuple[code: string, grammar: Grammar]]
|
||||||
|
|
||||||
|
|
||||||
|
var
|
||||||
|
grammarDir = ircConfigDir & "grammar/"
|
||||||
|
codeFile = ircConfigDir & "codes.json"
|
||||||
|
grammars*: Grammars
|
||||||
|
grammarCodes*: GrammarCodes
|
||||||
|
reservedCodes* = @["help", "forget", "learn", "unlearn"]
|
||||||
|
pasteUrls* = @["https://ttm.sh"]
|
||||||
|
|
||||||
|
|
||||||
|
# Set the grammar and code file location.
|
||||||
|
proc setGrammarDir*(configDir = ircConfigDir, gramDir = grammarDir,
|
||||||
|
codes = codeFile): void =
|
||||||
|
ircConfigDir = configDir
|
||||||
|
grammarDir = gramDir
|
||||||
|
codeFile = codes
|
||||||
|
|
||||||
|
|
||||||
|
# Download a url and save to a file path. Return true if successful or false
|
||||||
|
# otherwise.
|
||||||
|
proc fetchUrl*(url: string, file: string): bool =
|
||||||
|
var
|
||||||
|
cl = newHttpClient(userAgent = defUserAgent,
|
||||||
|
sslContext = newContext(verifyMode = CVerifyPeer))
|
||||||
|
data: string
|
||||||
|
try:
|
||||||
|
data = cl.getContent(url)
|
||||||
|
except OSError:
|
||||||
|
debug("fetchUrl", "error: cannot fetch url " & url)
|
||||||
|
except ValueError:
|
||||||
|
debug("fetchUrl", "error: no uri scheme found for " & url)
|
||||||
|
if data != "":
|
||||||
|
try:
|
||||||
|
writeFile(unixToNativePath(file), data)
|
||||||
|
return true
|
||||||
|
except IOError:
|
||||||
|
debug("fetchUrl", "error: could not write to file " & file)
|
||||||
|
return false
|
||||||
|
else:
|
||||||
|
debug("fetchUrl", "error: file saved from " & url & " has no contents")
|
||||||
|
return false
|
||||||
|
|
||||||
|
|
||||||
|
# Upload a grammar JSON file to a pastebin and return the paste url.
|
||||||
|
proc postGrammar*(file: string): string =
|
||||||
|
var
|
||||||
|
cl = newHttpClient(userAgent = defUserAgent,
|
||||||
|
sslContext = newContext(verifyMode = CVerifyPeer))
|
||||||
|
data = newMultiPartData()
|
||||||
|
res: string
|
||||||
|
cl.headers = newHttpHeaders({"Content-Type": "application/json"})
|
||||||
|
randomize()
|
||||||
|
var host = sample(pasteUrls)
|
||||||
|
try:
|
||||||
|
data.addFiles({"file": file})
|
||||||
|
res = cl.postContent(host, multipart = data)
|
||||||
|
except OSError:
|
||||||
|
debug("postGrammar", "error: cannot post file " & file)
|
||||||
|
if res != "":
|
||||||
|
return strip(res)
|
||||||
|
else:
|
||||||
|
debug("postGrammar", "error: no response received from " & host)
|
||||||
|
return ""
|
||||||
|
|
||||||
|
|
||||||
|
# Load and return the grammar given a grammar JSON file path.
|
||||||
|
proc loadGrammar*(file: string): Grammar =
|
||||||
|
var gram: Grammar
|
||||||
|
try:
|
||||||
|
gram = getFields(parseFile(unixToNativePath(file)))
|
||||||
|
except IOError:
|
||||||
|
debug("loadGrammar", "error: could not read JSON file " & file)
|
||||||
|
except JsonParsingError:
|
||||||
|
debug("loadGrammar", "error: invalid JSON in file " & file)
|
||||||
|
return gram
|
||||||
|
|
||||||
|
|
||||||
|
# Check whether a code is writable. Return true if writable or false otherwise.
|
||||||
|
proc checkWritable*(name: string, codes = grammarCodes): bool =
|
||||||
|
# Check if the code is a reserved keyword
|
||||||
|
for c in reservedCodes:
|
||||||
|
if name == c:
|
||||||
|
return false
|
||||||
|
# Check if the code allows overwriting
|
||||||
|
for c in codes:
|
||||||
|
if name == c.code and c.allowOverwrite == false:
|
||||||
|
return false
|
||||||
|
return true
|
||||||
|
|
||||||
|
|
||||||
|
# Return the aliases of a given code name as a sequence and their indices, or
|
||||||
|
# the code name given an alias and its index. Return an empty sequence if none
|
||||||
|
# are found. Aliases are code names sharing the same grammar file or handler
|
||||||
|
# name.
|
||||||
|
proc lookupAliases*(name: string, ty = "aliases", codes = grammarCodes):
|
||||||
|
seq[(string, int)] =
|
||||||
|
var
|
||||||
|
grammar: string
|
||||||
|
handler: string
|
||||||
|
res: seq[(string, int)]
|
||||||
|
count = 0
|
||||||
|
tyCount = 0
|
||||||
|
for c in codes:
|
||||||
|
if name == c.code:
|
||||||
|
grammar = c.grammar
|
||||||
|
handler = c.handler
|
||||||
|
break
|
||||||
|
count += 1
|
||||||
|
case ty
|
||||||
|
of "aliases":
|
||||||
|
for c in codes:
|
||||||
|
if grammar != "" and grammar == c.grammar and c.code != name:
|
||||||
|
add(res, (c.code, tyCount))
|
||||||
|
elif handler != "" and handler == c.handler and c.code != name:
|
||||||
|
add(res, (c.code, tyCount))
|
||||||
|
tyCount += 1
|
||||||
|
else:
|
||||||
|
if grammar != "":
|
||||||
|
res = @[(split(grammar, ".")[0], count)]
|
||||||
|
else:
|
||||||
|
for c in codes:
|
||||||
|
# Take the first result found
|
||||||
|
if handler != "" and handler == c.handler and c.code != name:
|
||||||
|
add(res, (c.code, tyCount))
|
||||||
|
break
|
||||||
|
tyCount += 1
|
||||||
|
return res
|
||||||
|
|
||||||
|
|
||||||
|
# Return a list of service code names given the code prefix.
|
||||||
|
proc getGrammarNames*(prefix: string, codes = grammarCodes): seq[string] =
|
||||||
|
return map(grammarCodes, proc(c: GrammarCode): string = prefix & c.code)
|
||||||
|
|
||||||
|
|
||||||
|
# Given a service code name and grammars, return the associated grammar and the
|
||||||
|
# index of the grammar. If no grammar is found for the code, return an empty
|
||||||
|
# grammar and -1.
|
||||||
|
proc lookupGrammar*(code: string, grams = grammars): (Grammar, int) =
|
||||||
|
var
|
||||||
|
gram: Grammar
|
||||||
|
found = false
|
||||||
|
count = 0
|
||||||
|
for g in grams:
|
||||||
|
if g.code == code:
|
||||||
|
gram = g.grammar
|
||||||
|
found = true
|
||||||
|
break
|
||||||
|
count += 1
|
||||||
|
if found:
|
||||||
|
return (gram, count)
|
||||||
|
else:
|
||||||
|
return (gram, -1)
|
||||||
|
|
||||||
|
|
||||||
|
# Return a random origin string from a grammar. If the grammar origin array is
|
||||||
|
# empty, return an empty string.
|
||||||
|
proc getOrigin*(gram: Grammar): string =
|
||||||
|
if getOrDefault(gram, "origin") != nil:
|
||||||
|
var
|
||||||
|
line = getStr(sample(getElems(getOrDefault(gram, "origin"))))
|
||||||
|
spliced = false
|
||||||
|
by = "#"
|
||||||
|
check: seq[bool]
|
||||||
|
splice: string
|
||||||
|
word: string
|
||||||
|
exp: string
|
||||||
|
sub: seq[string]
|
||||||
|
# Convert splices to random selections from their respective blocks
|
||||||
|
while not spliced:
|
||||||
|
check = @[]
|
||||||
|
for w in split(line):
|
||||||
|
# Splices need to be at least 3 characters long, e.g. "#s#"
|
||||||
|
if w.len() >= 3:
|
||||||
|
# Allow for punctuation after a splice, e.g. "#block#,"
|
||||||
|
if rfind(w, by) == w.len() - 2:
|
||||||
|
splice = w[0..(w.len() - 2)]
|
||||||
|
word = w[1..(w.len() - 3)]
|
||||||
|
else:
|
||||||
|
splice = w[0..(w.len() - 1)]
|
||||||
|
word = w[1..(w.len() - 2)]
|
||||||
|
if find(w, by) == 0 and (rfind(w, by) == (w.len() - 1) or
|
||||||
|
rfind(w, by) == (w.len() - 2)):
|
||||||
|
if getOrDefault(gram, word) != nil:
|
||||||
|
randomize()
|
||||||
|
exp = getStr(sample(getElems(getOrDefault(gram, word))))
|
||||||
|
sub = split(line, splice, 1)
|
||||||
|
line = join([sub[0], exp, sub[1]])
|
||||||
|
# Confirm all splices have been expanded
|
||||||
|
for w in split(line):
|
||||||
|
if (find(w, by) != 0 and (rfind(w, by) != 0 or rfind(w, by) != 1)) or
|
||||||
|
w.len() <= 2:
|
||||||
|
add(check, true)
|
||||||
|
else:
|
||||||
|
add(check, false)
|
||||||
|
if find(check, false) == -1:
|
||||||
|
spliced = true
|
||||||
|
return line
|
||||||
|
else:
|
||||||
|
return ""
|
||||||
|
|
||||||
|
|
||||||
|
# Load JSON service code file to global variable `grammarCodes`.
|
||||||
|
proc loadGrammarCodes*(file = codeFile): void =
|
||||||
|
if fileExists(unixToNativePath(file)):
|
||||||
|
var
|
||||||
|
nodes: JsonNode
|
||||||
|
grammar: OrderedTable[string, JsonNode]
|
||||||
|
try:
|
||||||
|
nodes = parseFile(unixToNativePath(file))
|
||||||
|
except IOError:
|
||||||
|
debug("loadGrammar", "error: could not read file " & file)
|
||||||
|
try:
|
||||||
|
grammarCodes = to(nodes, GrammarCodes)
|
||||||
|
# Show error if the code index file is missing fields
|
||||||
|
except KeyError:
|
||||||
|
debug("loadGrammarCodes", "error: invalid code file " & file)
|
||||||
|
|
||||||
|
# Fetch the grammar for each code and add it to the grammars tuple. A code
|
||||||
|
# can have either grammar or handler, not both. If both are defined, the
|
||||||
|
# grammar will be selected (see `listenCodes()`).
|
||||||
|
for c in grammarCodes:
|
||||||
|
if fileExists(unixToNativePath(grammarDir & c.grammar)):
|
||||||
|
grammar = loadGrammar(grammarDir & c.grammar)
|
||||||
|
add(grammars, (code: c.code, grammar: grammar))
|
||||||
|
for a in lookupAliases(c.code):
|
||||||
|
add(grammars, (code: a[0], grammar: grammar))
|
||||||
|
else:
|
||||||
|
debug("loadGrammarCodes", "warning: code " & c.code &
|
||||||
|
" not added, no grammar file or handler found")
|
||||||
|
else:
|
||||||
|
debug("loadGrammarCodes", "error: cannot find service codes file")
|
||||||
|
|
||||||
|
|
||||||
|
# Write service codes to a JSON file.
|
||||||
|
proc saveGrammarCodes*(codes: GrammarCodes, file = codeFile): void =
|
||||||
|
try:
|
||||||
|
writeFile(unixToNativePath(file), pretty(toJson(codes)))
|
||||||
|
except IOError:
|
||||||
|
debug("saveGrammarCodes", "error: could not write to file " & file)
|
||||||
|
|
||||||
|
|
||||||
|
# Add new grammar.
|
||||||
|
proc addGrammar*(sock: Socket, msg: Message): void =
|
||||||
|
if msg.codeParams.len() >= 2:
|
||||||
|
var nameSplit = split(msg.codeParams[0], ",")
|
||||||
|
if not checkWritable(nameSplit[0]):
|
||||||
|
message(sock, msg.replyTo, "Sorry, " & nameSplit[0] &
|
||||||
|
" is not writable. Please use another code.")
|
||||||
|
return
|
||||||
|
|
||||||
|
var
|
||||||
|
aliases: seq[string]
|
||||||
|
file = grammarDir & nameSplit[0] & ".json"
|
||||||
|
urlFetched = fetchUrl(msg.codeParams[1], file)
|
||||||
|
grammar = loadGrammar(file)
|
||||||
|
origin = getOrigin(grammar)
|
||||||
|
if nameSplit.len() > 1:
|
||||||
|
var aliasesLen = nameSplit.len() - 1
|
||||||
|
aliases = nameSplit[1..aliasesLen]
|
||||||
|
|
||||||
|
# Add to the global codes and grammar tuples and save to the codes file
|
||||||
|
if origin != "" and urlFetched:
|
||||||
|
add(grammarCodes, (code: nameSplit[0], grammar: nameSplit[0] & ".json",
|
||||||
|
handler: "", allowOverwrite: true))
|
||||||
|
add(grammars, (code: nameSplit[0], grammar: grammar))
|
||||||
|
# Add aliases
|
||||||
|
for a in aliases:
|
||||||
|
if lookupGrammar(a)[1] == -1:
|
||||||
|
add(grammarCodes, (code: a, grammar: nameSplit[0] & ".json",
|
||||||
|
handler: "", allowOverwrite: true))
|
||||||
|
add(grammars, (code: a, grammar: grammar))
|
||||||
|
saveGrammarCodes(grammarCodes)
|
||||||
|
message(sock, msg.replyTo, "Learned.")
|
||||||
|
else:
|
||||||
|
message(sock, msg.replyTo, "Sorry, I didn't get that. " &
|
||||||
|
"The grammar file may be empty or invalid.")
|
||||||
|
# Remove the grammar file if the grammar is invalid
|
||||||
|
removeFile(unixToNativePath(file))
|
||||||
|
else:
|
||||||
|
message(sock, msg.replyTo, "Sorry, I didn't get that. " &
|
||||||
|
"Please provide a code name and JSON url.")
|
||||||
|
|
||||||
|
|
||||||
|
# Remove grammar. Only codes that are writable can be removed.
|
||||||
|
proc removeGrammar*(sock: Socket, msg: Message): void =
|
||||||
|
if msg.codeParams.len() >= 1:
|
||||||
|
# Alias entries will follow the writable status of the code name entry
|
||||||
|
# since the aliases depend on the same grammar file, which can be removed
|
||||||
|
# if the code name entry is writable.
|
||||||
|
if lookupAliases(msg.codeParams[0], "code").len() == 0:
|
||||||
|
message(sock, msg.replyTo, "Sorry, I don't remember learning that.")
|
||||||
|
return
|
||||||
|
var name = lookupAliases(msg.codeParams[0], "code")[0][0]
|
||||||
|
if not checkWritable(name) and name != msg.codeParams[0]:
|
||||||
|
message(sock, msg.replyTo, join(["Sorry, ", msg.codeParams[0],
|
||||||
|
" (alias of ", name, ") cannot be removed", "because ", name,
|
||||||
|
" is not writable."]))
|
||||||
|
return
|
||||||
|
# Remove from the global codes and grammars tuples, save to the codes file
|
||||||
|
# and remove the grammar json
|
||||||
|
var aliases = lookupAliases(name)
|
||||||
|
for a in aliases:
|
||||||
|
delete(grammars, lookupGrammar(a[0])[1])
|
||||||
|
delete(grammarCodes, a[1])
|
||||||
|
delete(grammars, lookupGrammar(name)[1])
|
||||||
|
delete(grammarCodes, lookupAliases(name, "code")[0][1])
|
||||||
|
saveGrammarCodes(grammarCodes)
|
||||||
|
removeFile(unixToNativePath(grammarDir & name & ".json"))
|
||||||
|
message(sock, msg.replyTo, msg.codeParams[0] & "? What's that?")
|
||||||
|
else:
|
||||||
|
message(sock, msg.replyTo, "Sorry, I didn't get that. " &
|
||||||
|
"Please provide a code.")
|
||||||
|
|
||||||
|
|
||||||
|
# Show the paste url of a grammar code.
|
||||||
|
proc showGrammar*(sock: Socket, msg: Message): void =
|
||||||
|
var reply: string
|
||||||
|
if msg.codeParams.len() > 0:
|
||||||
|
var
|
||||||
|
names = map(grammarCodes, proc(c: GrammarCode): string = c.code)
|
||||||
|
files = map(grammarCodes, proc(c: GrammarCode): string = c.grammar)
|
||||||
|
url: string
|
||||||
|
file: string
|
||||||
|
for p in msg.codeParams:
|
||||||
|
if find(names, p) != -1:
|
||||||
|
file = files[find(names, p)]
|
||||||
|
if file != "":
|
||||||
|
url = postGrammar(grammarDir & file)
|
||||||
|
if url != "":
|
||||||
|
message(sock, msg.replyTo, url)
|
||||||
|
else:
|
||||||
|
reply = "No grammar linked to " & p
|
||||||
|
message(sock, msg.replyTo, reply)
|
||||||
|
else:
|
||||||
|
reply = "No grammar found for " & p
|
||||||
|
message(sock, msg.replyTo, reply)
|
||||||
|
else:
|
||||||
|
reply = "Please provide a code."
|
||||||
|
message(sock, msg.replyTo, reply)
|
||||||
|
|
||||||
|
|
||||||
|
# Send a grammar response to the server.
|
||||||
|
proc runGrammar*(gram: Grammar, sock: Socket, msg: Message): void =
|
||||||
|
var origin = getOrigin(gram)
|
||||||
|
if origin != "":
|
||||||
|
message(sock, msg.replyTo, origin)
|
||||||
|
else:
|
||||||
|
debug("runGrammar", "warning: no origin found in grammar")
|
||||||
|
|
||||||
|
|
||||||
|
# Parse message strings received and pass them to handlers.
|
||||||
|
proc listenGrammarCodes*(st: ServerState, str: string): void =
|
||||||
|
var msg = parseMessage(st.codePrefix, str)
|
||||||
|
for c in grammarCodes:
|
||||||
|
if msg.code == c.code:
|
||||||
|
if lookupGrammar(c.code)[1] != -1:
|
||||||
|
runGrammar(lookupGrammar(c.code)[0], st.sock, msg)
|
||||||
|
else:
|
||||||
|
runHandler(c.handler, st, msg)
|
||||||
|
break
|
||||||
|
|
||||||
|
|
||||||
|
# Initialise a loop and add listeners to respond to service codes.
|
||||||
|
proc listenGrammar*(states: seq[ServerState]): void =
|
||||||
|
while true:
|
||||||
|
var
|
||||||
|
statesBuf = states
|
||||||
|
count = 0
|
||||||
|
for st in statesBuf:
|
||||||
|
try:
|
||||||
|
var recv = recvServer(st.sock)
|
||||||
|
sendPong(st.sock, recv)
|
||||||
|
listenGrammarCodes(st, recv)
|
||||||
|
except AssertionDefect:
|
||||||
|
# Remove server from the states sequence
|
||||||
|
delete(statesBuf, count)
|
||||||
|
# Exit if there was only one server connection
|
||||||
|
if states.len() == 1:
|
||||||
|
quit()
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
|
||||||
|
# Handlers
|
||||||
|
# ----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
proc hdlGrammarHelp*(st: ServerState, msg: Message): void =
|
||||||
|
var
|
||||||
|
names = getGrammarNames(st.codePrefix)
|
||||||
|
help = "Hello, I'm a tracery-inspired grammar bot. " &
|
||||||
|
"I respond to the following service codes: " & join(names, ", ")
|
||||||
|
message(st.sock, msg.replyTo, help)
|
||||||
|
|
||||||
|
|
||||||
|
proc hdlGrammarLearn*(st: ServerState, msg: Message): void =
|
||||||
|
addGrammar(st.sock, msg)
|
||||||
|
|
||||||
|
|
||||||
|
proc hdlGrammarShow*(st: ServerState, msg: Message): void =
|
||||||
|
showGrammar(st.sock, msg)
|
||||||
|
|
||||||
|
|
||||||
|
proc hdlGrammarUnlearn*(st: ServerState, msg: Message): void =
|
||||||
|
removeGrammar(st.sock, msg)
|
||||||
|
|
||||||
|
|
||||||
|
# Init
|
||||||
|
# ----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
setGrammarDir()
|
||||||
|
loadGrammarCodes()
|
||||||
|
handlers["hdlGrammarHelp"] = hdlGrammarHelp
|
||||||
|
handlers["hdlGrammarLearn"] = hdlGrammarLearn
|
||||||
|
handlers["hdlGrammarShow"] = hdlGrammarShow
|
||||||
|
handlers["hdlGrammarUnlearn"] = hdlGrammarUnlearn
|
||||||
|
|
@ -0,0 +1,123 @@
|
||||||
|
import std/[json, jsonutils, os, sequtils, strutils, tables]
|
||||||
|
import ../cirrus
|
||||||
|
|
||||||
|
|
||||||
|
type
|
||||||
|
PoolItem = tuple
|
||||||
|
name: string
|
||||||
|
desc: string
|
||||||
|
|
||||||
|
JavaPool = seq[PoolItem]
|
||||||
|
|
||||||
|
var
|
||||||
|
javaPool: JavaPool
|
||||||
|
poolFile* = ircConfigDir & "javapool.json"
|
||||||
|
|
||||||
|
|
||||||
|
# Save pool to pool config file.
|
||||||
|
proc savePool(file = poolFile, pool = javaPool): void =
|
||||||
|
try:
|
||||||
|
writeFile(unixToNativePath(file), pretty(toJson(pool)))
|
||||||
|
except IOError:
|
||||||
|
debug("savePool", "error: could not write to file " & file)
|
||||||
|
|
||||||
|
|
||||||
|
# Load pool config file.
|
||||||
|
proc loadPool(file = poolFile): JavaPool =
|
||||||
|
# If config file doesn't exist, create one
|
||||||
|
if not fileExists(unixToNativePath(file)):
|
||||||
|
savePool(file, @[])
|
||||||
|
try:
|
||||||
|
var
|
||||||
|
nodes = parseFile(unixToNativePath(file))
|
||||||
|
conf = to(nodes, JavaPool)
|
||||||
|
return conf
|
||||||
|
except IOError:
|
||||||
|
debug("loadPool", "error: could not read JSON file " & file)
|
||||||
|
except JsonParsingError:
|
||||||
|
debug("loadPool", "error: invalid JSON in file " & file)
|
||||||
|
|
||||||
|
|
||||||
|
# Given an item name, return its description and its index if the item is in
|
||||||
|
# the pool, or empty string and -1 otherwise.
|
||||||
|
proc lookupPoolItem(name: string, pool = javaPool): (string, int) =
|
||||||
|
var count = 0
|
||||||
|
for i in pool:
|
||||||
|
if name == i.name:
|
||||||
|
return (desc: i.desc, index: count)
|
||||||
|
count += 1
|
||||||
|
return (desc: "", index: -1)
|
||||||
|
|
||||||
|
|
||||||
|
# Handlers
|
||||||
|
# ----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
# Show a list of pool items.
|
||||||
|
proc hdlPoolList*(st: ServerState, msg: Message): void =
|
||||||
|
var reply: string
|
||||||
|
if javaPool.len() == 0:
|
||||||
|
reply = "There are no items in the pool yet."
|
||||||
|
else:
|
||||||
|
reply = "There are " & intToStr(javaPool.len()) & " items in the pool: " &
|
||||||
|
join(map(javaPool, proc(i: PoolItem): string = i.name), ", ")
|
||||||
|
message(st.sock, msg.replyTo, reply)
|
||||||
|
|
||||||
|
|
||||||
|
# Show an item's description if it exists in the pool, or list the pool items
|
||||||
|
# if no item name was given in the message.
|
||||||
|
proc hdlPoolLook*(st: ServerState, msg: Message): void =
|
||||||
|
if msg.codeParams.len() > 0:
|
||||||
|
var reply: string
|
||||||
|
for p in msg.codeParams:
|
||||||
|
if lookupPoolItem(p)[1] != -1:
|
||||||
|
reply = join(["The ", p, " is ", lookupPoolItem(p)[0], "."])
|
||||||
|
else:
|
||||||
|
reply = "There is no " & p & " in the pool."
|
||||||
|
message(st.sock, msg.replyTo, reply)
|
||||||
|
else:
|
||||||
|
hdlPoolList(st, msg)
|
||||||
|
|
||||||
|
|
||||||
|
# Add an item to the pool.
|
||||||
|
proc hdlPoolAdd*(st: ServerState, msg: Message): void =
|
||||||
|
var reply: string
|
||||||
|
if msg.codeParams.len() >= 2:
|
||||||
|
var desc = join(msg.codeParams[1..(msg.codeParams.len() - 1)], " ")
|
||||||
|
add(javaPool, (name: msg.codeParams[0], desc: desc))
|
||||||
|
savePool()
|
||||||
|
reply = "Item " & msg.codeParams[0] & " added to the pool."
|
||||||
|
message(st.sock, msg.replyTo, reply)
|
||||||
|
else:
|
||||||
|
reply = "Usage: " & st.codePrefix & msg.code & " [name] [description]"
|
||||||
|
message(st.sock, msg.replyTo, reply)
|
||||||
|
|
||||||
|
|
||||||
|
# Remove an item from the pool.
|
||||||
|
proc hdlPoolRemove*(st: ServerState, msg: Message): void =
|
||||||
|
var reply: string
|
||||||
|
if msg.codeParams.len() >= 1:
|
||||||
|
for p in msg.codeParams:
|
||||||
|
if lookupPoolItem(p)[1] != -1:
|
||||||
|
delete(javaPool, lookupPoolItem(p)[1])
|
||||||
|
reply = "Item " & p & " removed from the pool."
|
||||||
|
message(st.sock, msg.replyTo, reply)
|
||||||
|
else:
|
||||||
|
reply = "There is no " & p & " in the pool."
|
||||||
|
message(st.sock, msg.replyTo, reply)
|
||||||
|
savePool()
|
||||||
|
else:
|
||||||
|
reply = "Usage: " & st.codePrefix & msg.code & " [name]"
|
||||||
|
message(st.sock, msg.replyTo, reply)
|
||||||
|
|
||||||
|
|
||||||
|
# Init
|
||||||
|
# ----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
when defined(nimHasUsed):
|
||||||
|
{.used.}
|
||||||
|
|
||||||
|
javaPool = loadPool()
|
||||||
|
handlers["hdlPoolList"] = hdlPoolList
|
||||||
|
handlers["hdlPoolLook"] = hdlPoolLook
|
||||||
|
handlers["hdlPoolAdd"] = hdlPoolAdd
|
||||||
|
handlers["hdlPoolRemove"] = hdlPoolRemove
|
||||||
Reference in New Issue