Compare commits

...

3 Commits

Author SHA1 Message Date
196056099b
readme
Signed-off-by: Pratyush Desai <pratyush.desai@liberta.casa>
2023-10-04 13:24:11 +05:30
d701d1604e
runs in a container in a pod
Signed-off-by: Pratyush Desai <pratyush.desai@liberta.casa>
2023-10-04 13:18:16 +05:30
dd3a079d62
rm redundant code
Signed-off-by: Pratyush Desai <pratyush.desai@liberta.casa>
2023-10-04 13:11:10 +05:30
10 changed files with 31 additions and 785 deletions

6
DOCKERFILE Normal file
View File

@ -0,0 +1,6 @@
FROM golang
RUN mkdir /app
ADD . /app
WORKDIR /app
RUN go build /app
CMD [ "/app/watbot"]

View File

@ -1,5 +1,23 @@
# watbot
watbot
ported watbot from wuselfuzz
## Installation
### Prerequisites
* podman
* git
### Setup
* `git clone <repo>`
* `cd <repo>`
* `podman build -f DOCKERFILE`
* `podman run <container-id>
Note: You might wanna detach at the end there.

10
main.go
View File

@ -13,10 +13,10 @@ func main() {
flag.Parse()
fmt.Printf("PASS len %d\n", len(*pass))
config := irc.ClientConfig{
Nick: "watt",
Nick: "watbot",
Pass: *pass,
User: "wat",
Name: "wat",
User: "watbot",
Name: "watbot",
}
watConfig := wat.WatConfig{
PermittedChannels: []string{
@ -28,9 +28,9 @@ func main() {
},
}
tcpConf := &tls.Config{
InsecureSkipVerify: true,
InsecureSkipVerify: false,
}
conn, err := tls.Dial("tcp", "127.0.0.1:6697", tcpConf)
conn, err := tls.Dial("tcp", "irc.casa:6697", tcpConf)
if err != nil {
fmt.Println("err " + err.Error())
return

View File

@ -1,15 +0,0 @@
#!/usr/bin/python
# LICENSE: DO WAT YOU WANT WITH THIS!
from watbot_config import WatbotConfig
from watbot_db import WatbotDB
from watbot_game import WatbotGame
from watbot_irc import WatbotIRC
db = WatbotDB(WatbotConfig)
game = WatbotGame(WatbotConfig, db)
irc = WatbotIRC(WatbotConfig, game)
irc.main_loop()

View File

@ -1,19 +0,0 @@
#!/usr/bin/python
from watbot_config import WatbotConfig
from watbot_db import WatbotDB
from watbot_game import WatbotGame
from watbot_console import WatbotConsole
import sys
if len(sys.argv) > 1:
db = WatbotDB(WatbotConfig)
game = WatbotGame(WatbotConfig, db)
con = WatbotConsole(WatbotConfig, game, sys.argv[1])
con.main_loop()
else:
print "usage: " + sys.argv[0] + " <nickname>"

View File

@ -1,13 +0,0 @@
#!/usr/bin/python
config = {
"prefix": "abc",
"currency": "abccoin",
"bot_nick": "abcbot",
"channel": "#abc",
"server": "irc.abc.example.com",
"port": 6667,
"ssl": False,
"nickserv": False,
"password": ""
}

View File

@ -1,143 +0,0 @@
#!/usr/bin/python
import sqlite3
import time
class WatbotDB:
"""watbot database abstraction"""
def __init__(self, config):
self.config = config
self.currency = config["currency"]
self.conn = sqlite3.connect(self.currency + ".db")
self.c = self.conn.cursor()
self.c.execute("create table if not exists players ("
" nickname varchar(256),"
" watting integer,"
" anarchy integer, "
" trickery integer, "
" coins integer, "
" health integer, "
" last_mine integer, "
" last_rest integer, "
" primary key(nickname)"
")"
)
self.c.execute("create table if not exists ledger ("
" nickname varchar(256),"
" timestamp integer,"
" balance integer,"
" log text,"
" foreign key(nickname) references players(nickname)"
")"
)
self.c.execute("create table if not exists items ("
" itemname varchar(256),"
" nickname varchar(256),"
" price integer,"
" primary key(itemname),"
" foreign key(nickname) references players(nickname)"
")"
)
self.c.execute("create table if not exists inventory ("
" nickname varchar(256),"
" itemname varchar(256),"
" count integer,"
" foreign key(nickname) references players(nickname),"
" foreign key(itemname) references items(itemname)"
")"
)
self.conn.commit()
def get_account(self,nick):
self.c.execute("select watting, anarchy, trickery, coins, last_mine, health, last_rest from players where nickname=?", (nick.lower(),))
data = self.c.fetchone()
if data is None:
earlier = time.time() - ( 24 * 3600 + 1)
self.c.execute("insert into players(nickname, watting, anarchy, trickery, coins, last_mine, health, last_rest) values(?, 0, 0, 0, 0, ?, 100, ?)", (nick.lower(), earlier, earlier))
self.c.execute("insert into ledger(nickname, timestamp, balance, log) values(?, ?, ?, ?)", (nick.lower(), earlier, 0, "created"))
self.conn.commit()
data = (0, 0, 0, 0, earlier, 100, earlier)
return data
def update_account(self, nick, watting, anarchy, trickery, coins, last_mine, health, last_rest, log):
now = time.time()
self.c.execute("update players set watting=?, anarchy=?, trickery=?, coins=?, last_mine=?, health=?, last_rest=? where nickname=?", (watting, anarchy, trickery, coins, last_mine, health, last_rest, nick.lower()))
self.c.execute("insert into ledger(nickname, timestamp, balance, log) values(?, ?, ?, ?)", (nick.lower(), now, coins, log))
if not log is None:
print "log: " + log
self.conn.commit()
def close(self):
self.conn.close()
def topten(self):
out = ""
self.c.execute("select nickname, coins from players order by coins desc limit 10")
while True:
d = self.c.fetchone()
if d == None:
break
out += d[0] + "(" + str(d[1]) + ") "
return out
def items(self):
out = ""
self.c.execute("select itemname, nickname, price from items order by itemname")
while True:
d = self.c.fetchone()
if d == None:
break
out += d[0] + "(" + d[1] + ", " + str(d[2]) + ") "
return out
def invent_item(self, itemname, nickname, price):
try:
self.c.execute("insert into items(itemname, nickname, price) values(?, ?, ?)", (itemname.lower(), nickname.lower(), price))
self.conn.commit()
return True
except:
return False
def inventory(self, nickname):
out = ""
self.c.execute("select itemname, count from inventory where nickname = ? order by itemname", (nickname.lower(),))
while True:
d = self.c.fetchone()
if d == None:
break
out += d[0] + "(" + str(d[1]) + ") "
return out
def get_item(self, itemname):
self.c.execute("select itemname, nickname, price from items where itemname = ?", (itemname.lower(),))
return self.c.fetchone()
def get_item_count(self, nickname, itemname):
self.c.execute("select count from inventory where nickname = ? and itemname = ?", (nickname.lower(), itemname.lower()))
d = self.c.fetchone()
if d == None:
return 0
else:
return d[0]
def set_item_count(self, nickname, itemname, count):
if count == 0:
self.c.execute("delete from inventory where nickname = ? and itemname = ?", (nickname.lower(), itemname.lower()))
else:
if self.get_item_count(nickname, itemname) != 0:
self.c.execute("update inventory set count=? where nickname = ? and itemname = ?", (count, nickname.lower(), itemname.lower()))
else:
self.c.execute("insert into inventory(nickname, itemname, count) values(?, ?, ?)", (nickname.lower(), itemname.lower(), count))
self.conn.commit()

View File

@ -1,440 +0,0 @@
#!/usr/bin/python
import time
import random
import math
from watbot_player import WatbotPlayer
class WatbotGame:
"""Class containing the game logic"""
def __init__(self, config, db):
random.seed()
self.config = config
self.db = db
self.bot_nick = config["bot_nick"]
self.bot_player = WatbotPlayer(self.db, self.bot_nick)
self.quiet = False
self.help_text = (
"balance [<nickname>], "
"watch [<nickname>], "
"inventory [<nickname>], "
"topten, "
"items, "
"mine, "
"transfer <nickname> <amount>, "
"roll <amount>, "
"steal <nickname> <amount>, "
"frame <nickname> <amount>, "
"punch <nickname>, "
"give <nickname> <count> <itemname>, "
"invent <itemname> <price>, "
"create <itemname> <count>, "
)
self.rules_text = ("A new account is created with 5 hours time credit. "
"Mining exchanges time credit for coins: "
"1h - 10h: 1 coin/h; >10h: 10 coin; >1 day: 50 coin; >1 month: 1000 coin.")
def do_command(self, nick, command, args):
try:
player = WatbotPlayer(self.db, nick)
self.now = time.time()
if command == "wat":
out = self.mega_wat(player)
elif command == "speak" and (player.nick == "Burrito" or player.nick == "wuselfuzz"):
self.quiet = False
out = "wat!"
elif command == "shutup" and (player.nick == "Burrito" or player.nick == "wuselfuzz"):
self.quiet = True
out = "wat."
elif self.quiet:
out = None
elif command == "help":
out = self.help_text
elif command == "rules":
out = self.rules_text
elif command == "topten":
out = self.db.topten()
elif command == "items":
out = "temporarily disabled, because bug!"
#out = self.db.items()
elif command == "inventory":
if len(args) > 0:
out = self.db.inventory(args[0])
else:
out = self.db.inventory(player.nick)
elif command == "watch":
if len(args) > 0:
out = self.watch(self.get_target_player(player, args[0]))
else:
out = self.watch(player)
elif command == "balance":
if len(args) > 0:
out = self.balance(self.get_target_player(player, args[0]))
else:
out = self.balance(player)
elif command == "mine":
out = self.mine(player)
elif command == "transfer":
if len(args) < 2:
out = "transfer <target> <amount>"
else:
out = self.transfer(player, self.get_target_player(player, args[0]), int(args[1]))
elif player.health <= 0:
out = "You cannot do that while unconscious!"
# ----- commands that require consciousness below -----
elif command == "steal":
if len(args) < 2:
out = "steal <target> <amount> - rolls a d6. If <3, you steal <target> <amount> coins. Otherwise, you pay a <amount> * 2 fine to "+ self.bot_nick + "."
else:
out = self.steal(player, self.get_target_player(player, args[0]), int(args[1]))
elif command == "frame":
if len(args) < 2:
out = "frame <target> <amount> - rolls a d6. If <3, you make <target> pay a fine of <amount> coins to " + self.bot_nick + ". Otherwise, you pay a ceil(<amount>/2) to <target and floor(<amount>/2) to " + self.bot_nick + " as fines."
else:
out = self.frame(player, self.get_target_player(player, args[0]), int(args[1]))
elif command == "punch":
if len(args) < 1:
out = "punch <target>"
else:
out = self.punch(player, self.get_target_player(player, args[0]))
elif command == "roll":
if len(args) < 1:
out = "roll <amount> - rolls a d100 against watcoinbot. result<50 wins <amount>, result >=50 loses <amount>"
else:
out = self.roll(player, int(args[0]))
elif command == "invent":
if len(args) < 2:
out = "invent <itemname> <price> - invent an item called <itemname> which can be bought for <price>. An invention costs 100 coins."
else:
out = self.invent(player, args[0], int(args[1]))
elif command == "create":
if len(args) < 2:
out = "create <itemname> <count> - create <count> <itemname>s. You have to pay the price and must be the inventor of the item!"
else:
out = self.create(player, args[0], int(args[1]))
elif command == "give":
if len(args) < 3:
out = "give <target> <count> <itemname>"
else:
out = self.give(player, args[0], int(args[1]), args[2])
else:
out = None
return out
except:
return "wat?"
def get_target_player(self, player, target_nick):
if target_nick == player.nick:
return player
elif target_nick == self.bot_nick:
return self.bot_player
else:
return WatbotPlayer(self.db, target_nick)
def watch(self, player):
out = (
"Watting: " + str(player.watting) + "(" + str(player.watting_exp) + ") / " +
"Anarchy: " + str(player.anarchy) + "(" + str(player.anarchy_exp) + ") / " +
"Trickery: " + str(player.trickery) + "(" + str(player.trickery_exp) + ") " +
"Coins: " + str(player.coins) + " " +
"Health: " + str(player.health)
)
return out
def balance(self, player):
out = player.nick + "'s watcoin balance is " + str(player.coins) + ". Mining time credit: " + self.dhms(int(self.now - player.last_mine)) + " seconds."
return out
def mine(self, player):
delta = self.now - player.last_mine
if delta < 3600:
return "wat? not so soon again!"
if delta < 36000:
mined_coins = int(delta / 3600)
elif delta < 86400:
mined_coins = 10
elif delta < 2592000:
mined_coins = 50
else:
mined_coins = 1000
player.coins += mined_coins
player.last_mine = self.now
out = player.nick + " mined " + str(mined_coins) + " coins for " + self.dhms(int(delta)) + " seconds and now has " + str(player.coins) + " watcoins."
player.update(out)
return out
def transfer(self, player, target_player, amount):
if amount < 0:
return "wat? you thief!"
if player.coins < amount:
return "wat? you poor fuck don't have enough!"
player.coins -= amount
target_player.coins += amount
if amount != 1:
out = player.nick + " sent " + target_player.nick + " " +str(amount) + " watcoins."
else:
out = player.nick + " sent " + target_player.nick + " a watcoin."
player.update(out)
target_player.update(out)
return out
def mega_wat(self, player):
mega_number = random.randint(1,1000000)
kilo_number = random.randint(1,1000)
print "mega_wat(" + player.nick + ") mega_number == " + str(mega_number) + ", kilo_number == " + str(kilo_number)
out = None
if mega_number == 23:
player.coins += 1000000
out = "OMGWATWATWAT!!!! " + player.nick + " has won the MegaWat lottery and gains 1000000 watcoins!"
if kilo_number == 5:
player.coins += 1000
out = "OMGWAT! " + player.nick + " has won the KiloWat lottery and gains 1000 watcoins!"
player.watting_exp += 1
player.update(out)
return out
def roll(self, player, amount):
if amount < 0:
return "wat? nonono!"
if player.coins < amount:
return "wat? you broke, go away!"
if self.bot_player.coins < amount:
bot_mining_delta = self.now - self.bot_player.last_mine
if bot_mining_delta > 86400:
return self.bot_nick + " doesn't have enough coins for this, but " + self.bot_nick + " can mine! " + self.mine(self.bot_player) + self.roll(player, amount)
else:
return "wat? " + self.bot_nick + " only has " + str(bot_player.coins) + " wtc left. Try again later or beg someone to fund the bot. " + self.bot_nick + " will mine in " + str(self.dhms(int(86400 - bot_mining_delta))) + "."
number = random.randint(1, 100)
if number < 50:
player.coins += amount
player.trickery_exp += 1
self.bot_player.coins -= amount
out = player.nick + " rolls a d100 (<50 wins): " + str(number) + ". You win! Your new balance is " + str(player.coins) + "."
else:
player.coins -= amount
self.bot_player.coins += amount
out = player.nick + " rolls a d100 (<50 wins): " + str(number) + ". You lose. Your new balance is " + str(player.coins) + "."
player.update(out)
self.bot_player.update(out)
return out
def invent(self, player, itemname, price):
if price <= 0:
return "wat? nonono!"
invent_cost = 100 - player.watting
if player.coins < invent_cost:
return "wat? inventions cost you " + str(invent_cost) + " coins, but you're poor!"
if self.db.invent_item(itemname, player.nick, price):
player.coins -= invent_cost
self.bot_player.coins += invent_cost
out = player.nick + " invented " + itemname + " (" + str(price) + ")."
player.update(out)
self.bot_player.update(out)
else:
out = "wat?" + itemname + " already invented!"
return out
def create(self, player, itemname, count):
if count <= 0:
return "wat? nonono!"
(itemname, inventor_nick, price) = self.db.get_item(itemname)
if player.nick.lower() != inventor_nick:
return "wat? that's not your invention!"
if count * price > player.coins:
return "wat? you need more money for that!"
player.coins -= count * price
original_count = self.db.get_item_count(player.nick, itemname)
self.db.set_item_count(player.nick, itemname, original_count + count)
out = player.nick + " created " + str(count) + " " + itemname + "."
player.update(out)
return out
def give(self, player, target_nick, count, itemname):
player_item_count = self.db.get_item_count(player.nick, itemname)
if player_item_count < count:
return "wat? you don't have that many!"
self.db.set_item_count(player.nick, itemname, player_item_count - count)
target_item_count = self.db.get_item_count(target_nick, itemname)
self.db.set_item_count(target_nick, itemname, target_item_count + count)
return player.nick + " gave " + target_nick + " " + str(count) + " " + itemname
def steal(self, player, target_player, amount):
if amount < 0:
return "wat? nonono!"
if player.coins < amount * 2:
return "wat? you don't have enough to pay the possible fine!"
if target_player.coins < amount:
return "wat? " + target_player.nick + " is a poor fuck and doesn't have " + str(amount) + " coins."
number = random.randint(1,6)
if number <3:
out = player.nick + " rolls a d6 to steal " + str(amount) + " watcoins from " + target_player.nick + ": " + str(number) + " (<3 wins). You win! Sneaky bastard!"
player.coins += amount
player.anarchy_exp += 1
target_player.coins -= amount
player.update(out)
target_player.update(out)
else:
out = player.nick + " rolls a d6 to steal " + str(amount) + " watcoins from " + target_player.nick + ": " + str(number) + " (<3 wins). You were caught and pay " + str(2 * amount) + " coins to " + self.bot_nick + "."
player.coins -= 2 * amount
self.bot_player.coins += 2 * amount
player.update(out)
self.bot_player.update(out)
return out
def frame(self, player, target_player, amount):
if amount < 0:
return "wat? nonono!"
if player.coins < amount:
return "wat? you don't have enough to pay the possible fine!"
if target_player.coins < amount:
return "wat? " + target_player.nick + " is a poor fuck and doesn't have " + str(amount) + " coins."
number = random.randint(1,6)
if number <3:
out = player.nick + " rolls a d6 to frame " + str(amount) + " watcoins from " + target_player.nick + ": " + str(number) + " (<3 wins). You win! " + target_player.nick + " pays " + str(amount) + " to " + self.bot_nick + "."
player.anarchy_exp += 1
target_player.coins -= amount
self.bot_player.coins += amount
player.update(out)
target_player.update(out)
self.bot_player.update(out)
else:
target_amount = int(math.ceil(float(amount)/2))
bot_amount = int(math.floor(float(amount)/2))
out = player.nick + " rolls a d6 to frame " + str(amount) + " watcoins from " + target_player.nick + ": " + str(number) + " (<3 wins). You were caught and pay " + str(target_amount) + " coins to " + target_player.nick + " and " + str(bot_amount) + " coins to " + self.bot_nick + "."
player.coins -= amount
target_player.coins += target_amount
self.bot_player.coins += bot_amount
player.update(out)
target_player.update(out)
self.bot_player.update(out)
return out
def punch(self, player, target_player):
number = random.randint(1, 6)
dmg = random.randint(1, 6)
if number <3:
dmg += player.anarchy
out = player.nick + " rolls a d6 to punch " + target_player.nick + ": " + str(number) +". " + player.nick + " hits for " + str(dmg) + " points of damage."
target_player.health -= dmg
if target_player.health <= 0:
out += " " + target_player.nick + " falls unconscious."
target_player.update(out)
else:
dmg += target_player.anarchy
out = player.nick + " rolls a d6 to punch " + target_player.nick + ": " + str(number) +". " + player.nick + " misses. " + target_player.nick + " punches back for " + str(dmg) + " points of damage."
player.health -= dmg
if player.health <= 0:
out += " " + player.nick + " falls unconscious."
player.update(out)
return out
def dhms(self, seconds):
days = int(seconds / 86400)
seconds -= days * 86400
hours = int(seconds / 3600)
seconds -= hours * 3600
minutes = int(seconds / 60)
seconds -= minutes * 60
out = ""
show = False
if days > 0:
out += str(days) + "d "
show = True
if hours > 0 or show:
out += str(hours) + "h "
show = True
if minutes > 0 or show:
out += str(minutes) + "m "
show = True
out += str(seconds) + "s"
return out

View File

@ -1,102 +0,0 @@
#!/usr/bin/python
import irc.client
import ssl
class WatbotIRC:
"""irc frontend class"""
def __init__(self, config, game):
self.config = config
self.game = game
self.client = irc.client.IRC()
server = self.client.server()
if not config["ssl"]:
server.connect(
config["server"],
config["port"],
config["bot_nick"],
username = config["bot_nick"],
ircname = (config["prefix"] + " help"),
)
else:
ssl_factory = irc.connection.Factory(wrapper=ssl.wrap_socket)
server.connect(
config["server"],
config["port"],
config["bot_nick"],
username = config["bot_nick"],
ircname = (config["prefix"] + " help"),
connect_factory = ssl_factory
)
self.client.add_global_handler("welcome", self.on_connect)
self.client.add_global_handler("privmsg", self.on_msg)
self.client.add_global_handler("pubmsg", self.on_msg)
self.client.add_global_handler("disconnect", self.on_disconnect)
def main_loop(self):
self.client.process_forever()
def on_connect(self, connection, event):
if self.config["nickserv"]:
connection.privmsg("nickserv", "identify " + self.config["password"])
connection.join(self.config["channel"])
def on_msg(self, connection, event):
for a in event.arguments:
self.do_command(connection, event.source, event.target, a)
def do_command(self, connection, source, target, commandline):
cl_list = commandline.strip().split(" ", 2)
if len(cl_list) > 0:
if cl_list[0].lower() == "wat":
cl_list = [ self.config["prefix"], "wat" ]
if len(cl_list) < 2:
return
if cl_list[0].lower() != self.config["prefix"]:
return
command = cl_list[1].lower()
if len(cl_list) > 2:
args_list = cl_list[2].split(" ")
else:
args_list = []
if self.config["nickserv"]:
# The @ must be there, irc standard
host = source.split("@")[1]
host_split = host.split("/")
if len(host_split) < 3:
return
# Only allow nickserved users
if host_split[0] != "tripsit":
return
#nick = host_split[2]
nick = source.split("!", 1)[0]
output = self.game.do_command(nick, command, args_list)
if target == self.config["bot_nick"]:
target = nick
if not output is None:
connection.privmsg(target, output)
def on_disconnect(self, connection, event):
raise SystemExit()

View File

@ -1,46 +0,0 @@
#!/usr/bin/python
import time
class WatbotPlayer:
"""class representing a player account"""
def __init__(self, db, nick):
self.nick = nick
self.db = db
(
self.watting_exp,
self.anarchy_exp,
self.trickery_exp,
self.coins,
self.last_mine,
self.health,
self.last_rest
) = db.get_account(nick)
self.watting = self.get_level(self.watting_exp)
self.anarchy = self.get_level(self.anarchy_exp)
self.trickery = self.get_level(self.trickery_exp)
now = time.time()
delta = now - self.last_rest
if delta > 60:
self.health += int(delta/60)
if self.health > 10:
self.health = 10
self.last_rest += int(delta/60) * 60
def get_level(self, exp):
if exp < 100:
level = int(exp/10)
elif exp < 900:
level = 10 + int(exp/100)
else:
level = 99
return level
def update(self, log):
self.db.update_account(self.nick, self.watting_exp, self.anarchy_exp, self.trickery_exp, self.coins, self.last_mine, self.health, self.last_rest, log)