main.py

# Wessex - A log-reading bot for mordhau.
# Copyright (C) 2021  Morgan Chesser mfreelancedef@gmail.com

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

import logparse
import ftp
import aftp
import config
import fileio
import time
import asyncio
import discordio
import mydb
import sqlitedb
import serverstatus
import command
import logging
import sys

logging.basicConfig(level=logging.INFO)

DEBUG = False

async def main_loop(client):
	global DEBUG
	if DEBUG:
		logging.info('DEBUG is TRUE')
	else:
		if len(sys.argv) > 1:
			if sys.argv[1] == '-d':
				logging.info('DEBUG set TRUE from cli')
				DEBUG = True
	
	DB = mydb.db()
	SS = serverstatus.ServerStatus()
	#await client.wait_until_ready()
	await client._msg_queue_loaded.wait()
	#add commands
	client.add_cog(command.Status(client, SS))
	#client.add_cog(command.DB(client, DB))

	fname = config.config['fileio']['localcopy']
	while not client.is_closed():
		#initalize data
		data = bytearray()
		#create local copy file if not created
		fileio.create_if_not_created(fname)
		if not DEBUG:
			(host, port, usr, pwd, filepath) = config.get_ftp_config()
			try:
				data = await aftp.get_remote_file_binary(host, port, usr, pwd, filepath)
			except ConnectionResetError:
				logging.warning("Connection reset error on log download.")
				continue
			except:
				logging.error("Unexpected error on log download: " + str(sys.exc_info()[0]))
				continue
			#update local copy and return its previous contents
			old_data = fileio.update_binary(fname, data)
			#convert to text
			data = data.decode()
			old_data = old_data.decode()
		else:
			#for debug purposes, read Mordhau.log as new data
			data = fileio.read_binary(fname)
			old_data = ''


		#find the new part of the log
		print("Finding new log entries.")
		new_data = logparse.log_diff(old_data, data)
		print("New data length: {} bytes".format(len(new_data)))
		#init server status
		if old_data:
			#initialize from the old section of the log file.
			await SS.init_from_log(old_data, DB)
		else:
			#log file is new, completely reinitialize.
			await SS.force_init(new_data, DB)
		async def handle_msg(*args):
			"""Handle messages.

			logmsg 	- readable form of the message
			msgtype - message type
			match 	- regex match of the message"""
			# await SS.handle_msg(*args)
			# await client.handle_msg(*args)
			await asyncio.gather(
				SS.handle_msg(*args),
				client.handle_msg(*args),
				#DB.handle_msg(*args),
			)
		logging.info("Parsing lines.")
		await logparse.parse_lines(new_data, DB, handle_msg)
		#update status
		print("Updating status.")
		await client.update_status(SS)
		#wait
		interval = int(config.config['time']['interval'])
		print("Waiting {} seconds".format(interval))
		await asyncio.sleep(interval)

client = discordio.MyClient(intents=discordio.intents, command_prefix="$")

# @client.command()
# async def test(ctx, arg):
# 	print(arg)
# 	await ctx.send(arg)

client.set_loop(main_loop)
client.run(discordio.TOKEN)